github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-03 11:47:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Dereference run tree (#30377)

Browse Source
This commit is contained in:
William FH 2025-03-19 12:05:06 -07:00 committed by GitHub
parent 8265be4d3e
commit ce84f8ba7e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/core/langchain_core/tracers/langchain.py
View File

@ -173,7 +173,13 @@ class LangChainTracer(BaseTracer):
return chat_model_run return chat_model_run
def _persist_run(self, run: Run) -> None: def _persist_run(self, run: Run) -> None:
self.latest_run = run # We want to free up more memory by avoiding keeping a reference to the
# whole nested run tree.
self.latest_run = Run.construct(
**run.dict(exclude={"child_runs", "inputs", "outputs"}),
inputs=run.inputs,
outputs=run.outputs,
)
def get_run_url(self) -> str: def get_run_url(self) -> str:
"""Get the LangSmith root run URL. """Get the LangSmith root run URL.

14
libs/core/tests/unit_tests/runnables/test_tracing_interops.py
View File

@ -445,11 +445,12 @@ def test_tree_is_constructed(parent_type: Literal["ls", "lc"]) -> None:
metadata={"some_foo": "some_bar"}, metadata={"some_foo": "some_bar"},
tags=["afoo"], tags=["afoo"],
): ):
if parent_type == "ls": collected: dict[str, RunTree] = {} # noqa
collected: dict[str, RunTree] = {} # noqa
def collect_run(run: RunTree) -> None: def collect_run(run: RunTree) -> None:
collected[str(run.id)] = run collected[str(run.id)] = run
if parent_type == "ls":
@traceable @traceable
def parent() -> str: def parent() -> str:
@ -459,7 +460,6 @@ def test_tree_is_constructed(parent_type: Literal["ls", "lc"]) -> None:
parent(langsmith_extra={"on_end": collect_run, "run_id": rid}) == "foo" parent(langsmith_extra={"on_end": collect_run, "run_id": rid}) == "foo"
) )
assert collected assert collected
run = collected.get(str(rid))
else: else:
@ -468,8 +468,10 @@ def test_tree_is_constructed(parent_type: Literal["ls", "lc"]) -> None:
return child.invoke("foo") return child.invoke("foo")
tracer = LangChainTracer() tracer = LangChainTracer()
tracer._persist_run = collect_run # type: ignore
assert parent.invoke(..., {"run_id": rid, "callbacks": [tracer]}) == "foo" # type: ignore assert parent.invoke(..., {"run_id": rid, "callbacks": [tracer]}) == "foo" # type: ignore
run = tracer.latest_run run = collected.get(str(rid))
assert run is not None assert run is not None
assert run.name == "parent" assert run.name == "parent"