github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-07 05:52:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Core] Inherit tracing metadata & tags (#26838)

Browse Source
This commit is contained in:
William FH
2024-09-24 19:33:12 -07:00
committed by GitHub
parent 3796e143f8
commit 9b6ac41442
2 changed files with 23 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -390,7 +390,12 @@ def test_tree_is_constructed() -> None:
rid = uuid.uuid4()
with tracing_context(client=mock_client_, enabled=True):
with tracing_context(
client=mock_client_,
enabled=True,
metadata={"some_foo": "some_bar"},
tags=["afoo"],
):
assert parent(langsmith_extra={"on_end": collect_run, "run_id": rid}) == "foo"
assert collected
@@ -404,6 +409,10 @@ def test_tree_is_constructed() -> None:
grandchild_run = child_run.child_runs[0]
assert grandchild_run.name == "grandchild"
assert grandchild_run.child_runs
assert grandchild_run.metadata.get("some_foo") == "some_bar"
assert "afoo" in grandchild_run.tags # type: ignore
kitten_run = grandchild_run.child_runs[0]
assert kitten_run.name == "kitten"
assert not kitten_run.child_runs
assert kitten_run.metadata.get("some_foo") == "some_bar"
assert "afoo" in kitten_run.tags # type: ignore