[module][langchain_mlrun] First version #963

guy1992l · 2026-01-08T22:50:00Z

Created the module langchain_mlrun - MLRun's integration with LangChain, allowing to auto-trace and monitor LangChain and LangGraph based code.

Added the module itself - fully documented and sphinx ready.
Added about 76 tests for 100% of the current functionality.
Added an end-to-end notebook with brief explanations and tutorial.
Made preparations for MLRun-CE support (using Kafka stream).
Wrote TODOs for future work.

[Build] Fix html links, add <function>.html as source in documentation

[XGB-Custom] Fix test artifact key name

[XGB-Serving][XGB-Test][XGB-Trainer] Fix tests - artifact key

* [Build] Install python 3.9 when testing * [Build] Update python version in CI * .

This reverts commit 0cd1f15.

This reverts commit 3301415.

* [Build] Build with python 3.9 * .

… development

* [Noise-reduction] Add new function to hub * fix test * added multiprocessing and silence removal to function

This reverts commit 8872798.

* delete EOS functions * bring back validate_great_expectations * bring back load_dataset

* adjust batch infer v2 * update docs in NB

…#836)

* [text to audio generator] Replaced bark with openai tts models * [text to audio generator] Fix base url env var * fix version * Add speech engine * after review

* [Build] Fix html links, Add <function>.html as source in documentation * Update CI temporarily and update index * [XGB-Custom] Fix test artifact key name * [XGB-Serving][XGB-Test][XGB-Trainer] Fix tests - artifact key * [Build] Install python 3.9 when testing (mlrun#618) * [Build] Update python version in CI (mlrun#620) * [Build] Install python 3.9 when testing * [Build] Update python version in CI * . * Revert "[Build] Update python version in CI (mlrun#620)" (mlrun#621) This reverts commit 0cd1f15. * Revert "[Build] Install python 3.9 when testing (mlrun#618)" (mlrun#619) This reverts commit 3301415. * [Build] Build with python 3.9 (mlrun#622) * [Build] Build with python 3.9 * . * Update requirements.txt

* fix feature_selection * fix feature_selection * fix feature_selection nb * update yaml name * fix test * fix test

* fix arbitrary file vulnerability * fix arbitrary file vulnerability * fix test

* add traversal test * add traversal test * add traversal test

… development

review-notebook-app · 2026-01-08T22:50:05Z

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

Eyal-Danieli

very nice, had few comments

modules/src/langchain_mlrun/item.yaml

Eyal-Danieli · 2026-01-11T13:17:33Z

modules/src/langchain_mlrun/langchain_mlrun.py

+        :returns: An MLRun model endpoint monitoring client.
+        """
+        if mlrun.mlconf.is_ce_mode():
+            return _KafkaMLRunEndPointClient(


missing monitoring_broker and monitoring_topic params

its a "hachana lemazgan"... there is a TODO basically anywhere related to Kafka usage.

Eyal-Danieli · 2026-01-11T13:18:56Z

modules/src/langchain_mlrun/langchain_mlrun.py

+            ) from value_error
+        except ImportError as import_error:
+            raise ImportError(
+                f"Could not import '{module_path}'. Tried to import '{module_name}' and failed with the following "


module_name might be referenced before assignment (also object_name in the next error)

its not, Import Error cannot be raised during split so the module name is there, same for object name.

modules/src/langchain_mlrun/langchain_mlrun.py

modules/src/langchain_mlrun/requirements.txt

Eyal-Danieli · 2026-01-11T13:30:12Z

modules/src/langchain_mlrun/test_langchain_mlrun.py

+        self,
+        tools: Sequence[
+            dict[str, Any] | type | Callable | BaseTool  # noqa: UP006
+        ],
+        *,
+        tool_choice: str | None = None,
+        **kwargs: Any,


didn't find any usage for that func

Suggested change

self,

tools: Sequence[

dict[str, Any] | type | Callable | BaseTool # noqa: UP006

],

*,

tool_choice: str | None = None,

**kwargs: Any,

*

It is used under the hood by create_agent in LangChain.

Eyal-Danieli · 2026-01-11T13:31:31Z

modules/src/langchain_mlrun/test_langchain_mlrun.py

+        response = model.invoke(state["messages"])
+        return {"messages": [response], "attempts": state["attempts"] + 1}
+
+    def reflect_node(state: AgentState):


not in use

Suggested change

def reflect_node(state: AgentState):

def reflect_node():

That's the nodes pattern in LangGraph. I want to mimic real flow.

Eyal-Danieli · 2026-01-11T13:31:53Z

modules/src/langchain_mlrun/test_langchain_mlrun.py

+    """
+    run_func, expected_events = run_suites
+
+    with mlrun_monitoring(settings=manual_mode_settings) as tracer:


Suggested change

with mlrun_monitoring(settings=manual_mode_settings) as tracer:

with mlrun_monitoring(settings=manual_mode_settings):

it is used like 2 rows ahead in 451:

assert len(tracer.settings.monitor.debug_target_list) == expected_events

Eyal-Danieli

plus add copyright

modules/src/langchain_mlrun/langchain_mlrun.py

modules/src/langchain_mlrun/test_langchain_mlrun.py

omermaim

looks good to me

yonishelach and others added 30 commits March 8, 2023 11:01

[Build] Fix html links, Add <function>.html as source in documentation

356264f

Update CI temporarily and update index

2202caf

Merge pull request mlrun#615 from yonishelach/fix-links-html

b0f2b5d

[Build] Fix html links, add <function>.html as source in documentation

[XGB-Custom] Fix test artifact key name

8b17c8f

Merge pull request mlrun#616 from yonishelach/fix-xgb-custom-test

f42f3e3

[XGB-Custom] Fix test artifact key name

[XGB-Serving][XGB-Test][XGB-Trainer] Fix tests - artifact key

a469dca

Merge pull request mlrun#617 from yonishelach/fix-xgb-custom-test

80c7de2

[XGB-Serving][XGB-Test][XGB-Trainer] Fix tests - artifact key

[Build] Install python 3.9 when testing (mlrun#618)

3301415

[Build] Update python version in CI (mlrun#620)

0cd1f15

* [Build] Install python 3.9 when testing * [Build] Update python version in CI * .

Revert "[Build] Update python version in CI (mlrun#620)" (mlrun#621)

33e7ab8

This reverts commit 0cd1f15.

Revert "[Build] Install python 3.9 when testing (mlrun#618)" (mlrun#619)

7a7473b

This reverts commit 3301415.

[Build] Build with python 3.9 (mlrun#622)

81437da

* [Build] Build with python 3.9 * .

Merge branch 'development' of https://github.com/mlrun/functions into…

c0a1bb8

… development

Merge branch 'development' of https://github.com/mlrun/functions into…

10cf170

… development

Merge branch 'development' of https://github.com/mlrun/functions into…

81b45ae

… development

Merge branch 'development' of https://github.com/mlrun/functions into…

f1aa62a

… development

Merge branch 'development' of https://github.com/mlrun/functions into…

5bb578c

… development

Merge branch 'development' of https://github.com/mlrun/functions into…

58b0d3e

… development

Merge branch 'development' of https://github.com/mlrun/functions into…

1d2f407

… development

[onnx utils] update onnx utils packages

8872798

[Noise-reduction] Add new function to hub (mlrun#765)

3b34fef

* [Noise-reduction] Add new function to hub * fix test * added multiprocessing and silence removal to function

Merge remote-tracking branch 'upstream/development' into development

55a1907

delete load_dask (mlrun#822)

1e0cb66

Merge remote-tracking branch 'upstream/development' into development

1ffaede

[feature selection] update function yaml

6fdc1b1

[feature selection] update function yaml

64907af

Revert "[onnx utils] update onnx utils packages"

d1d61a0

This reverts commit 8872798.

[feature selection] update function yaml

98d0942

[feature selection] update function yaml

2fb2080

Delete unsupported functions from the hub (mlrun#824)

55e1dc0

* delete EOS functions * bring back validate_great_expectations * bring back load_dataset

Eyal-Danieli and others added 16 commits September 25, 2024 15:53

set navigation_with_keys to False (mlrun#829)

6ded2af

remove xgb and churn functions (mlrun#830)

60c5349

[Batch Infer V2] Adjust function to 1.7 (mlrun#832)

6dd4ded

* adjust batch infer v2 * update docs in NB

bring back deprecated params and add warn (mlrun#834)

c4c558e

fix PyYAML loading (mlrun#837)

e978719

[text to audio generator] Replaced bark with openai tts models (mlrun…

7d29949

…#836)

[Text to audio generator] Add speech engine (mlrun#838)

b3eb31a

* [text to audio generator] Replaced bark with openai tts models * [text to audio generator] Fix base url env var * fix version * Add speech engine * after review

[Feature Selection] Fix deprecated get_offline_features (mlrun#844)

fe39028

* fix feature_selection * fix feature_selection * fix feature_selection nb * update yaml name * fix test * fix test

limit torchaudio for unit test (mlrun#845)

c146acd

Update requirements.txt (mlrun#843)

ec61826

[Open Archive] Fix arbitrary file vulnerability (mlrun#847)

b23f095

* fix arbitrary file vulnerability * fix arbitrary file vulnerability * fix test

[open_archive] Add traversal attack test (mlrun#849)

9798aed

* add traversal test * add traversal test * add traversal test

Merge branch 'development' of https://github.com/mlrun/functions into…

34ec86a

… development

Merge branch 'development' of https://github.com/mlrun/functions into…

6d065eb

… development

first version

8fc3f35

guy1992l requested a review from Copilot January 8, 2026 22:50

Copilot started reviewing on behalf of guy1992l January 8, 2026 22:50 View session

guy1992l requested review from Eyal-Danieli and omermaim and removed request for Copilot January 8, 2026 22:50

run upg to pydantic v2

daf0db8

guy1992l requested a review from GiladShapira94 January 11, 2026 08:21

added kafka and mlrun-ce code preparation

3f417fd

Eyal-Danieli reviewed Jan 11, 2026

View reviewed changes

modules/src/langchain_mlrun/langchain_mlrun.py Show resolved Hide resolved

modules/src/langchain_mlrun/test_langchain_mlrun.py Show resolved Hide resolved

Eyal review

3ad0984

guy1992l requested a review from Eyal-Danieli January 11, 2026 21:30

omermaim approved these changes Jan 13, 2026

View reviewed changes

	with mlrun_monitoring(settings=manual_mode_settings) as tracer:
	with mlrun_monitoring(settings=manual_mode_settings):

[module][langchain_mlrun] First version #963

Are you sure you want to change the base?

[module][langchain_mlrun] First version #963

Conversation

guy1992l commented Jan 8, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

review-notebook-app bot commented Jan 8, 2026

Uh oh!

Eyal-Danieli left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Eyal-Danieli left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

omermaim left a comment

Choose a reason for hiding this comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

6 participants

guy1992l commented Jan 8, 2026 •

edited

Loading