Compare commits

...

1 Commits

Author SHA1 Message Date
Erick Friis
cb8e35df83 core: allow merging metadata in tracing 2025-01-24 16:23:37 -08:00
2 changed files with 24 additions and 6 deletions

View File

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

View File

@@ -154,7 +154,10 @@ class LangChainTracer(BaseTracer):
"""
start_time = datetime.now(timezone.utc)
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(
id=run_id,
parent_run_id=parent_run_id,