fix: prevent langfuse initialization (#4057)
This commit is contained in:
parent
3ea7be12e9
commit
f7e58f96b0
1 changed files with 8 additions and 9 deletions
|
|
@ -30,22 +30,19 @@ class LangFuseTracer(BaseTracer):
|
||||||
self.flow_id = trace_name.split(" - ")[-1]
|
self.flow_id = trace_name.split(" - ")[-1]
|
||||||
self.last_span: StatefulSpanClient | None = None
|
self.last_span: StatefulSpanClient | None = None
|
||||||
self.spans: dict = {}
|
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
|
@property
|
||||||
def ready(self):
|
def ready(self):
|
||||||
return self._ready
|
return self._ready
|
||||||
|
|
||||||
def setup_langfuse(self) -> bool:
|
def setup_langfuse(self, config) -> bool:
|
||||||
try:
|
try:
|
||||||
from langfuse import Langfuse
|
from langfuse import Langfuse
|
||||||
from langfuse.callback.langchain import LangchainCallbackHandler
|
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._client = Langfuse(**config)
|
||||||
self.trace = self._client.trace(id=str(self.trace_id), name=self.flow_id)
|
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
|
||||||
return None # self._callback
|
return None # self._callback
|
||||||
|
|
||||||
def _get_config(self):
|
def _get_config(self) -> dict:
|
||||||
secret_key = os.getenv("LANGFUSE_SECRET_KEY", None)
|
secret_key = os.getenv("LANGFUSE_SECRET_KEY", None)
|
||||||
public_key = os.getenv("LANGFUSE_PUBLIC_KEY", None)
|
public_key = os.getenv("LANGFUSE_PUBLIC_KEY", None)
|
||||||
host = os.getenv("LANGFUSE_HOST", 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 {}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue