using utils

Author: Akhilesh Negi

backend/app/api/routes/evaluation.py

@@ -8,6 +8,7 @@
 from app.utils import APIResponse
 from app.crud.credentials import get_provider_credential
 from app.api.routes.threads import threads_sync
+from app.core.util import configure_langfuse
 
 router = APIRouter(tags=["evaluation"])
 
@@ -47,17 +48,11 @@ async def evaluate_threads(
         )
 
     # Configure Langfuse
-    langfuse = Langfuse(
-        public_key=langfuse_credentials["public_key"],
-        secret_key=langfuse_credentials["secret_key"],
-        host=langfuse_credentials.get("host", "https://cloud.langfuse.com"),
-    )
-
-    langfuse_context.configure(
-        secret_key=langfuse_credentials["secret_key"],
-        public_key=langfuse_credentials["public_key"],
-        host=langfuse_credentials.get("host", "https://cloud.langfuse.com"),
-    )
+    langfuse, success = configure_langfuse(langfuse_credentials)
+    if not success:
+        return APIResponse.failure_response(
+            error="Failed to configure Langfuse client."
+        )
 
     try:
         # Get dataset

backend/app/api/routes/threads.py

@@ -14,6 +14,7 @@
 from app.utils import APIResponse
 from app.crud.credentials import get_provider_credential
 from app.core.security import decrypt_credentials
+from app.core.util import configure_langfuse
 
 logger = logging.getLogger(__name__)
 router = APIRouter(tags=["threads"])
@@ -245,11 +246,13 @@ async def threads(
             error="LANGFUSE keys not configured for this organization."
         )
 
-    langfuse_context.configure(
-        secret_key=langfuse_credentials["secret_key"],
-        public_key=langfuse_credentials["public_key"],
-        host=langfuse_credentials["host"],
-    )
+    # Configure Langfuse
+    _, success = configure_langfuse(langfuse_credentials)
+    if not success:
+        return APIResponse.failure_response(
+            error="Failed to configure Langfuse client."
+        )
+
     # Validate thread
     is_valid, error_message = validate_thread(client, request.get("thread_id"))
     if not is_valid:
@@ -310,11 +313,11 @@ async def threads_sync(
         )
 
     # Configure Langfuse
-    langfuse_context.configure(
-        secret_key=langfuse_credentials["secret_key"],
-        public_key=langfuse_credentials["public_key"],
-        host=langfuse_credentials.get("host", "https://cloud.langfuse.com"),
-    )
+    _, success = configure_langfuse(langfuse_credentials)
+    if not success:
+        return APIResponse.failure_response(
+            error="Failed to configure Langfuse client."
+        )
 
     # Validate thread
     is_valid, error_message = validate_thread(client, request.get("thread_id"))

backend/app/core/util.py

@@ -5,6 +5,8 @@
 from fastapi import HTTPException
 from requests import Session, RequestException
 from pydantic import BaseModel, HttpUrl
+from langfuse import Langfuse
+from langfuse.decorators import langfuse_context
 
 
 def now():
@@ -32,3 +34,37 @@ def post_callback(url: HttpUrl, payload: BaseModel):
             errno += 1
 
     return not errno
+
+
+def configure_langfuse(credentials: dict) -> tuple[Langfuse, bool]:
+    """
+    Configure Langfuse client and context with the provided credentials.
+
+    Args:
+        credentials: Dictionary containing Langfuse credentials (public_key, secret_key, host)
+
+    Returns:
+        Tuple of (Langfuse client instance, success boolean)
+    """
+    if not credentials:
+        return None, False
+
+    try:
+        # Configure Langfuse client
+        langfuse = Langfuse(
+            public_key=credentials["public_key"],
+            secret_key=credentials["secret_key"],
+            host=credentials.get("host", "https://cloud.langfuse.com"),
+        )
+
+        # Configure Langfuse context
+        langfuse_context.configure(
+            secret_key=credentials["secret_key"],
+            public_key=credentials["public_key"],
+            host=credentials.get("host", "https://cloud.langfuse.com"),
+        )
+
+        return langfuse, True
+    except Exception as e:
+        warnings.warn(f"Failed to configure Langfuse: {str(e)}")
+        return None, False