github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-21 06:14:37 +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

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

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

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

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