From f7e58f96b0d082e17f307a8b4884a5b7a1c670de Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=8Dtalo=20Johnny?= Date: Mon, 7 Oct 2024 17:38:28 -0300 Subject: [PATCH] fix: prevent langfuse initialization (#4057) --- .../base/langflow/services/tracing/langfuse.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/src/backend/base/langflow/services/tracing/langfuse.py b/src/backend/base/langflow/services/tracing/langfuse.py index f452de982..479b52494 100644 --- a/src/backend/base/langflow/services/tracing/langfuse.py +++ b/src/backend/base/langflow/services/tracing/langfuse.py @@ -30,22 +30,19 @@ class LangFuseTracer(BaseTracer): self.flow_id = trace_name.split(" - ")[-1] self.last_span: StatefulSpanClient | None = None self.spans: dict = {} - self._ready: bool = self.setup_langfuse() + + config = self._get_config() + self._ready: bool = self.setup_langfuse(config) if config else False @property def ready(self): return self._ready - def setup_langfuse(self) -> bool: + def setup_langfuse(self, config) -> bool: try: from langfuse import Langfuse from langfuse.callback.langchain import LangchainCallbackHandler - config = self._get_config() - if not all(config.values()): - msg = "Missing Langfuse configuration" - raise ValueError(msg) - self._client = Langfuse(**config) self.trace = self._client.trace(id=str(self.trace_id), name=self.flow_id) @@ -134,8 +131,10 @@ class LangFuseTracer(BaseTracer): return None return None # self._callback - def _get_config(self): + def _get_config(self) -> dict: secret_key = os.getenv("LANGFUSE_SECRET_KEY", None) public_key = os.getenv("LANGFUSE_PUBLIC_KEY", None) host = os.getenv("LANGFUSE_HOST", None) - return {"secret_key": secret_key, "public_key": public_key, "host": host} + if secret_key and public_key and host: + return {"secret_key": secret_key, "public_key": public_key, "host": host} + return {}