From 894dbdcd40bcd6020000b73c7afca913b4773abf Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 21 Aug 2023 22:07:35 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(base.py):=20improve=20erro?= =?UTF-8?q?r=20handling=20and=20error=20message=20when=20importing=20langf?= =?UTF-8?q?use=20callback=20module=20=F0=9F=9A=80=20feat(base.py):=20add?= =?UTF-8?q?=20langfuse=20callback=20import=20type=20hint=20to=20ignore=20t?= =?UTF-8?q?ype=20checking=20for=20better=20compatibility=20with=20dynamic?= =?UTF-8?q?=20imports?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/processing/base.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/processing/base.py b/src/backend/langflow/processing/base.py index ddf47d792..80ee3fef4 100644 --- a/src/backend/langflow/processing/base.py +++ b/src/backend/langflow/processing/base.py @@ -29,16 +29,21 @@ def get_langfuse_callback(): if settings.LANGFUSE_PUBLIC_KEY and settings.LANGFUSE_SECRET_KEY: logger.debug("Langfuse credentials found") try: - from langfuse.callback import CallbackHandler + from langfuse.callback import CallbackHandler # type: ignore return CallbackHandler( public_key=settings.LANGFUSE_PUBLIC_KEY, secret_key=settings.LANGFUSE_SECRET_KEY, host=settings.LANGFUSE_HOST, ) - + except ImportError as exc: + raise ImportError( + "Error importing langfuse callback. " + "Please install langfuse with `pip install langfuse`" + ) from exc except Exception as exc: logger.error(f"Error initializing langfuse callback: {exc}") + return None