github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-21 22:29:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

core: allow merging metadata in tracing

Browse Source
This commit is contained in:
Erick Friis 2025-01-24 16:23:37 -08:00
parent dbb6b7b103
commit cb8e35df83
2 changed files with 24 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
libs/core/langchain_core/tracers/core.py
View File

@ -178,7 +178,10 @@ class _TracerCore(ABC):
raise NotImplementedError(msg) raise NotImplementedError(msg)
start_time = datetime.now(timezone.utc) start_time = datetime.now(timezone.utc)
if metadata: if metadata:
kwargs.update({"metadata": metadata}) if isinstance(kwargs.get("metadata"), dict):
kwargs["metadata"].update(metadata)
else:
kwargs.update({"metadata": metadata})
return Run( return Run(
id=run_id, id=run_id,
parent_run_id=parent_run_id, parent_run_id=parent_run_id,
@ -209,7 +212,10 @@ class _TracerCore(ABC):
"""Create a llm run.""" """Create a llm run."""
start_time = datetime.now(timezone.utc) start_time = datetime.now(timezone.utc)
if metadata: if metadata:
kwargs.update({"metadata": metadata}) if isinstance(kwargs.get("metadata"), dict):
kwargs["metadata"].update(metadata)
else:
kwargs.update({"metadata": metadata})
return Run( return Run(
id=run_id, id=run_id,
parent_run_id=parent_run_id, parent_run_id=parent_run_id,
@ -314,7 +320,10 @@ class _TracerCore(ABC):
"""Create a chain Run.""" """Create a chain Run."""
start_time = datetime.now(timezone.utc) start_time = datetime.now(timezone.utc)
if metadata: if metadata:
kwargs.update({"metadata": metadata}) if isinstance(kwargs.get("metadata"), dict):
kwargs["metadata"].update(metadata)
else:
kwargs.update({"metadata": metadata})
return Run( return Run(
id=run_id, id=run_id,
parent_run_id=parent_run_id, parent_run_id=parent_run_id,
@ -399,7 +408,10 @@ class _TracerCore(ABC):
"""Create a tool run.""" """Create a tool run."""
start_time = datetime.now(timezone.utc) start_time = datetime.now(timezone.utc)
if metadata: if metadata:
kwargs.update({"metadata": metadata}) if isinstance(kwargs.get("metadata"), dict):
kwargs["metadata"].update(metadata)
else:
kwargs.update({"metadata": metadata})
if self._schema_format in ("original", "original+chat"): if self._schema_format in ("original", "original+chat"):
inputs = {"input": input_str} inputs = {"input": input_str}
@ -464,7 +476,10 @@ class _TracerCore(ABC):
"""Create a retrieval run.""" """Create a retrieval run."""
start_time = datetime.now(timezone.utc) start_time = datetime.now(timezone.utc)
if metadata: if metadata:
kwargs.update({"metadata": metadata}) if isinstance(kwargs.get("metadata"), dict):
kwargs["metadata"].update(metadata)
else:
kwargs.update({"metadata": metadata})
return Run( return Run(
id=run_id, id=run_id,
name=name or "Retriever", name=name or "Retriever",

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

@ -154,7 +154,10 @@ class LangChainTracer(BaseTracer):
""" """
start_time = datetime.now(timezone.utc) start_time = datetime.now(timezone.utc)
if metadata: if metadata:
kwargs.update({"metadata": metadata}) if isinstance(kwargs.get("metadata"), dict):
kwargs["metadata"].update(metadata)
else:
kwargs.update({"metadata": metadata})
chat_model_run = Run( chat_model_run = Run(
id=run_id, id=run_id,
parent_run_id=parent_run_id, parent_run_id=parent_run_id,