github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-19 11:08:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

core: In LangChainTracer, send only the first token event (#31591)

Browse Source 
- only the first one is used for analytics
This commit is contained in:
Nuno Campos 2025-06-12 14:04:23 -07:00 committed by GitHub
parent 50f9354d31
commit ddc850ca72
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -95,6 +95,7 @@ class LangChainTracer(BaseTracer):
self.client = client or get_client()
self.tags = tags or []
self.latest_run: Optional[Run] = None
self.run_has_token_event_map: dict[str, bool] = {}
def _start_trace(self, run: Run) -> None:
if self.project_name:
@ -235,6 +236,11 @@ class LangChainTracer(BaseTracer):
parent_run_id: Optional[UUID] = None,
) -> Run:
"""Append token event to LLM run and return the run."""
run_id_str = str(run_id)
if run_id_str not in self.run_has_token_event_map:
self.run_has_token_event_map[run_id_str] = True
else:
return self._get_run(run_id, run_type={"llm", "chat_model"})
return super()._llm_run_with_token_event(
# Drop the chunk; we don't need to save it
token,