Setup langfuse #716
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -4,6 +4,7 @@ | |
| from typing import Annotated, List, Optional | ||
|
|
||
| from fastapi import Depends | ||
| from pydantic import SecretStr | ||
| from pydantic_settings import BaseSettings, SettingsConfigDict | ||
|
|
||
|
|
||
|
|
@@ -75,6 +76,12 @@ class Settings(BaseSettings): | |
| task_tracker_redis_url: str = "redis://redis/0" | ||
| task_tracker_prefix: str = "task_metadata:" | ||
|
|
||
| # Langfuse | ||
| langfuse_is_enabled: bool = True | ||
| langfuse_host: Optional[str] = None | ||
| langfuse_public_key: Optional[str] = None | ||
| langfuse_secret_key: Optional[SecretStr] = None | ||
|
|
||
|
|
||
| @lru_cache | ||
| def get_settings(): | ||
|
|
||
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,101 @@ | ||
| """Wrapper around Langfuse observability.""" | ||
|
|
||
| from __future__ import annotations | ||
|
|
||
| import logging | ||
| from contextlib import nullcontext | ||
| from typing import Any, Callable, ContextManager | ||
|
|
||
| logger = logging.getLogger(__name__) | ||
|
|
||
| try: | ||
| from langfuse import Langfuse as _Langfuse | ||
| from langfuse import observe as _lf_observe | ||
| except Exception as e: | ||
| logger.debug("Langfuse import failed: %s", e) | ||
| _Langfuse = None | ||
| _lf_observe = None | ||
|
|
||
|
|
||
| class Observability: | ||
| """Wrapper around Langfuse observability.""" | ||
|
|
||
| def __init__( | ||
| self, is_enabled, langfuse_host, langfuse_public_key, langfuse_secret_key | ||
| ) -> None: | ||
| """Initialize the Observability instance.""" | ||
| self._client = None | ||
| if hasattr(langfuse_secret_key, "get_secret_value"): | ||
| langfuse_secret_key = langfuse_secret_key.get_secret_value() | ||
|
|
||
| self._enabled = bool( | ||
| is_enabled and langfuse_host and langfuse_public_key and langfuse_secret_key | ||
| ) | ||
|
|
||
| if not self._enabled or _Langfuse is None: | ||
| self._enabled = False | ||
| return | ||
|
|
||
| try: | ||
| self._client = _Langfuse( | ||
| public_key=langfuse_public_key, | ||
| secret_key=langfuse_secret_key, | ||
| host=langfuse_host, | ||
| ) | ||
| except Exception as e: | ||
| logger.warning("Langfuse init failed: %s", e) | ||
| self._enabled = False | ||
| self._client = None | ||
|
|
||
| def observe( | ||
| self, **decorator_kwargs | ||
| ) -> Callable[[Callable[..., Any]], Callable[..., Any]]: | ||
| """Decorator to observe a function with Langfuse. If disabled, returns a no-op decorator.""" # noqa: E501 | ||
| if self._enabled and self._client and _lf_observe is not None: | ||
| return _lf_observe(**decorator_kwargs) | ||
|
|
||
| def _noop(fn): | ||
| return fn | ||
|
|
||
| return _noop | ||
|
|
||
| def span(self, name: str, **kwargs) -> ContextManager[Any]: | ||
| """Context manager to create a span with Langfuse.""" | ||
| if self._enabled and self._client: | ||
| start_span = getattr(self._client, "start_as_current_span", None) | ||
| if callable(start_span): | ||
| return start_span(name=name, **kwargs) | ||
| return nullcontext() | ||
|
|
||
| def generation(self, **kwargs) -> ContextManager[Any]: | ||
| """Context manager to create a generation with Langfuse.""" | ||
| if self._enabled and self._client: | ||
| start_gen = getattr(self._client, "start_as_current_generation", None) | ||
| if callable(start_gen): | ||
| return start_gen(**kwargs) | ||
| return nullcontext() | ||
|
|
||
| def update_current_trace(self, **kwargs) -> None: | ||
| """Update the current trace with additional metadata.""" | ||
| if not (self._enabled and self._client): | ||
| return | ||
| try: | ||
| self._client.update_current_trace(**kwargs) | ||
| except Exception as e: | ||
| logger.warning("Langfuse update_current_trace failed: %s", e) | ||
| pass | ||
|
|
||
| def flush(self) -> None: | ||
| """Flush any buffered data to Langfuse.""" | ||
| if not (self._enabled and self._client): | ||
| return | ||
| try: | ||
| self._client.flush() | ||
| except Exception as e: | ||
| logger.warning("Langfuse flush failed: %s", e) | ||
| pass | ||
|
|
||
| @property | ||
| def is_enabled(self) -> bool: | ||
| """Check if observability is enabled.""" | ||
| return bool(self._enabled and self._client) |
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🧹 Nitpick | 🔵 Trivial
🧩 Analysis chain
Verify the Langfuse version is current and secure.
The pinned version
langfuse==3.4.0should be validated to ensure it's up-to-date and free from known vulnerabilities.Run the following script to check the latest version and security advisories:
🏁 Script executed:
Length of output: 137
Update langfuse to latest version (3.5.2)
In
src/summary/pyproject.toml(line 18), changelangfuse==3.4.0tolangfuse==3.5.2. No known vulnerabilities.🤖 Prompt for AI Agents