From 4ef4c93baccf30fc18936941ec9110b5f2fc2a59 Mon Sep 17 00:00:00 2001 From: tianzhipeng Date: Sat, 22 Feb 2025 02:48:04 +0800 Subject: [PATCH] fix: don't init langwatch if not configure (#6749) * fix: don't init langwatch if not configure; * feat: return langfuse callback for tracing * fix: move condition to setup_langwatch * fix: typo --------- Co-authored-by: tianzhipeng-jk Co-authored-by: Gabriel Luiz Freitas Almeida --- src/backend/base/langflow/services/tracing/langfuse.py | 2 +- src/backend/base/langflow/services/tracing/langwatch.py | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/services/tracing/langfuse.py b/src/backend/base/langflow/services/tracing/langfuse.py index 19a5f1fdb..5198b61b0 100644 --- a/src/backend/base/langflow/services/tracing/langfuse.py +++ b/src/backend/base/langflow/services/tracing/langfuse.py @@ -133,7 +133,7 @@ class LangFuseTracer(BaseTracer): def get_langchain_callback(self) -> BaseCallbackHandler | None: if not self._ready: return None - return None # self._callback + return self._callback @staticmethod def _get_config() -> dict: diff --git a/src/backend/base/langflow/services/tracing/langwatch.py b/src/backend/base/langflow/services/tracing/langwatch.py index 1c8376c78..01fd9633c 100644 --- a/src/backend/base/langflow/services/tracing/langwatch.py +++ b/src/backend/base/langflow/services/tracing/langwatch.py @@ -1,5 +1,6 @@ from __future__ import annotations +import os from typing import TYPE_CHECKING, Any, cast import nanoid @@ -31,7 +32,7 @@ class LangWatchTracer(BaseTracer): self.flow_id = trace_name.split(" - ")[-1] try: - self._ready = self.setup_langwatch() + self._ready: bool = self.setup_langwatch() if not self._ready: return @@ -58,6 +59,8 @@ class LangWatchTracer(BaseTracer): return self._ready def setup_langwatch(self) -> bool: + if "LANGWATCH_API_KEY" not in os.environ: + return False try: import langwatch