mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-07 05:30:39 +00:00
[Core] respect tracing project name cvar (#28792)
This commit is contained in:
parent
e6b41d081d
commit
50ea1c3ea3
@ -2379,6 +2379,7 @@ def _configure(
|
|||||||
if run_tree is not None
|
if run_tree is not None
|
||||||
else tracing_context["client"]
|
else tracing_context["client"]
|
||||||
),
|
),
|
||||||
|
tags=tracing_tags,
|
||||||
)
|
)
|
||||||
callback_manager.add_handler(handler, True)
|
callback_manager.add_handler(handler, True)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
@ -13,8 +13,8 @@ from typing import (
|
|||||||
)
|
)
|
||||||
from uuid import UUID
|
from uuid import UUID
|
||||||
|
|
||||||
|
from langsmith import run_helpers as ls_rh
|
||||||
from langsmith import utils as ls_utils
|
from langsmith import utils as ls_utils
|
||||||
from langsmith.run_helpers import get_run_tree_context
|
|
||||||
|
|
||||||
from langchain_core.tracers.langchain import LangChainTracer
|
from langchain_core.tracers.langchain import LangChainTracer
|
||||||
from langchain_core.tracers.run_collector import RunCollectorCallbackHandler
|
from langchain_core.tracers.run_collector import RunCollectorCallbackHandler
|
||||||
@ -149,7 +149,10 @@ def _tracing_v2_is_enabled() -> Union[bool, Literal["local"]]:
|
|||||||
|
|
||||||
|
|
||||||
def _get_tracer_project() -> str:
|
def _get_tracer_project() -> str:
|
||||||
run_tree = get_run_tree_context()
|
tracing_context = ls_rh.get_tracing_context()
|
||||||
|
run_tree = tracing_context["parent"]
|
||||||
|
if run_tree is None and tracing_context["project_name"] is not None:
|
||||||
|
return tracing_context["project_name"]
|
||||||
return getattr(
|
return getattr(
|
||||||
run_tree,
|
run_tree,
|
||||||
"session_name",
|
"session_name",
|
||||||
|
@ -35,6 +35,25 @@ def _get_posts(client: Client) -> list:
|
|||||||
return posts
|
return posts
|
||||||
|
|
||||||
|
|
||||||
|
def test_tracing_context() -> None:
|
||||||
|
mock_session = MagicMock()
|
||||||
|
mock_client_ = Client(
|
||||||
|
session=mock_session, api_key="test", auto_batch_tracing=False
|
||||||
|
)
|
||||||
|
|
||||||
|
@RunnableLambda
|
||||||
|
def my_function(a: int) -> int:
|
||||||
|
return a + 1
|
||||||
|
|
||||||
|
name = uuid.uuid4().hex
|
||||||
|
project_name = f"Some project {name}"
|
||||||
|
with tracing_context(project_name=project_name, client=mock_client_, enabled=True):
|
||||||
|
assert my_function.invoke(1) == 2
|
||||||
|
posts = _get_posts(mock_client_)
|
||||||
|
assert posts
|
||||||
|
assert all(post["session_name"] == project_name for post in posts)
|
||||||
|
|
||||||
|
|
||||||
def test_config_traceable_handoff() -> None:
|
def test_config_traceable_handoff() -> None:
|
||||||
get_env_var.cache_clear()
|
get_env_var.cache_clear()
|
||||||
mock_session = MagicMock()
|
mock_session = MagicMock()
|
||||||
|
Loading…
Reference in New Issue
Block a user