Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-06 03:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Don't raise error if parent not found (#6538)

Browse Source 
Done so that you can pass in a run from the low level api
This commit is contained in:
Zander Chase 2023-06-26 22:57:52 -07:00 committed by GitHub
parent 49c864fa18
commit 07d802d088
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
langchain/callbacks/tracers/base.py
View File

@ -1,6 +1,7 @@
"""Base interfaces for tracing runs."""
from __future__ import annotations
import logging
from abc import ABC, abstractmethod
from datetime import datetime
from typing import Any, Dict, List, Optional, Union
@ -10,6 +11,8 @@ from langchain.callbacks.base import BaseCallbackHandler
from langchain.callbacks.tracers.schemas import Run, RunTypeEnum
from langchain.schema import LLMResult
logger = logging.getLogger(__name__)
class TracerException(Exception):
"""Base class for exceptions in tracers module."""
@ -41,9 +44,7 @@ class BaseTracer(BaseCallbackHandler, ABC):
if parent_run:
self._add_child_run(parent_run, run)
else:
raise TracerException(
f"Parent run with UUID {run.parent_run_id} not found."
)
logger.warning(f"Parent run with UUID {run.parent_run_id} not found.")
self.run_map[str(run.id)] = run
def _end_trace(self, run: Run) -> None:
@ -53,10 +54,8 @@ class BaseTracer(BaseCallbackHandler, ABC):
else:
parent_run = self.run_map.get(str(run.parent_run_id))
if parent_run is None:
raise TracerException(
f"Parent run with UUID {run.parent_run_id} not found."
)
if (
logger.warning(f"Parent run with UUID {run.parent_run_id} not found.")
elif (
run.child_execution_order is not None
and parent_run.child_execution_order is not None
and run.child_execution_order > parent_run.child_execution_order
@ -71,7 +70,8 @@ class BaseTracer(BaseCallbackHandler, ABC):
parent_run = self.run_map.get(parent_run_id)
if parent_run is None:
raise TracerException(f"Parent run with UUID {parent_run_id} not found.")
logger.warning(f"Parent run with UUID {parent_run_id} not found.")
return 1
if parent_run.child_execution_order is None:
raise TracerException(
f"Parent run with UUID {parent_run_id} has no child execution order."