[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

GiladShapira94 · 2026-01-19T08:22:42Z

modules/src/langchain_mlrun/item.yaml

@@ -0,0 +1,23 @@
+apiVersion: v1
+categories:
+- langchain


@Eyal-Danieli, which categories do we support?
Should it be on general for now?

@guy1992l do we have an option to deploy the agent code into the user cluster?

GiladShapira94 · 2026-01-19T09:30:13Z

modules/src/langchain_mlrun/requirements.txt

@@ -0,0 +1,3 @@
+pytest
+langchain


Worth pointing out to a specific version, to make sure the function do not change between package releases

GiladShapira94 · 2026-01-19T09:30:50Z

modules/src/langchain_mlrun/langchain_mlrun.ipynb

@@ -0,0 +1,899 @@
+{


I think we can generate the stream path by using the project name of the user or this env MLRUN_ACTIVE_PROJECT in addition, model monitoring always uses the project container so we can just set these values as default.
In this example, the user does not create any serving function, so we can just set the value under the hood if the user does not provide one.
I think we can remove this line - This is a temporary workaround until custom endpoint creation support is added to MLRun.
Once MLRun will add this feature, we will update the notebook

Reply via ReviewNB

GiladShapira94 · 2026-01-19T09:30:50Z

modules/src/langchain_mlrun/langchain_mlrun.ipynb

@@ -0,0 +1,899 @@
+{


We should use the same requirements.txt file. If it should be a different file I think it would be better to create an application_requirements.txt
Maybe worth to write another step that invoke the chain after the application deployed

Reply via ReviewNB

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 14 commits October 8, 2024 15:26

[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

GiladShapira94 reviewed Jan 19, 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!

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!

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

Uh oh!

Choose a reason for hiding this comment

Uh oh!

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

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

7 participants

guy1992l commented Jan 8, 2026 •

edited

Loading

GiladShapira94 Jan 19, 2026 •

edited

Loading

GiladShapira94 Jan 19, 2026 •

edited

Loading