fix: return in case langwatch is not ready (#2773)
This commit is contained in:
parent
bf16cc158e
commit
1573d6126e
1 changed files with 11 additions and 0 deletions
|
|
@ -1,3 +1,4 @@
|
|||
import os
|
||||
from typing import TYPE_CHECKING, Any, Dict, Optional, cast
|
||||
from uuid import UUID
|
||||
|
||||
|
|
@ -28,6 +29,8 @@ class LangWatchTracer(BaseTracer):
|
|||
|
||||
try:
|
||||
self._ready = self.setup_langwatch()
|
||||
if not self._ready:
|
||||
return
|
||||
|
||||
# import after setting up langwatch so we are sure to be available
|
||||
import nanoid # type: ignore
|
||||
|
|
@ -52,6 +55,8 @@ class LangWatchTracer(BaseTracer):
|
|||
return self._ready
|
||||
|
||||
def setup_langwatch(self):
|
||||
if os.getenv("LANGWATCH_API_KEY") is None:
|
||||
return False
|
||||
try:
|
||||
import langwatch
|
||||
|
||||
|
|
@ -78,6 +83,8 @@ class LangWatchTracer(BaseTracer):
|
|||
metadata: Dict[str, Any] | None = None,
|
||||
vertex: Optional["Vertex"] = None,
|
||||
):
|
||||
if not self._ready:
|
||||
return
|
||||
# If user is not using session_id, then it becomes the same as flow_id, but
|
||||
# we don't want to have an infinite thread with all the flow messages
|
||||
if "session_id" in inputs and inputs["session_id"] != self.flow_id:
|
||||
|
|
@ -111,6 +118,8 @@ class LangWatchTracer(BaseTracer):
|
|||
error: Exception | None = None,
|
||||
logs: list[Log | dict] = [],
|
||||
):
|
||||
if not self._ready:
|
||||
return
|
||||
if self.spans.get(trace_id):
|
||||
# Workaround for when model is used just as a component not actually called as an LLM,
|
||||
# to prevent LangWatch from calculating the cost based on it when it was in fact never called
|
||||
|
|
@ -131,6 +140,8 @@ class LangWatchTracer(BaseTracer):
|
|||
error: Exception | None = None,
|
||||
metadata: dict[str, Any] | None = None,
|
||||
):
|
||||
if not self._ready:
|
||||
return
|
||||
self.trace.root_span.end(
|
||||
input=self._convert_to_langwatch_types(inputs),
|
||||
output=self._convert_to_langwatch_types(outputs),
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue