push

Author: hassiebp

langfuse/_client/client.py

@@ -5,9 +5,9 @@
 
 import logging
 import os
-import warnings
 import re
 import urllib.parse
+import warnings
 from datetime import datetime
 from hashlib import sha256
 from time import time_ns
@@ -17,8 +17,8 @@
     List,
     Literal,
     Optional,
-    Union,
     Type,
+    Union,
     cast,
     overload,
 )
@@ -36,6 +36,13 @@
 from packaging.version import Version
 
 from langfuse._client.attributes import LangfuseOtelSpanAttributes
+from langfuse._client.constants import (
+    ObservationTypeGenerationLike,
+    ObservationTypeLiteral,
+    ObservationTypeLiteralNoEvent,
+    ObservationTypeSpanLike,
+    get_observation_types_list,
+)
 from langfuse._client.datasets import DatasetClient, DatasetItemClient
 from langfuse._client.environment_variables import (
     LANGFUSE_DEBUG,
@@ -47,25 +54,18 @@
     LANGFUSE_TRACING_ENABLED,
     LANGFUSE_TRACING_ENVIRONMENT,
 )
-from langfuse._client.constants import (
-    ObservationTypeLiteral,
-    ObservationTypeLiteralNoEvent,
-    ObservationTypeGenerationLike,
-    ObservationTypeSpanLike,
-    get_observation_types_list,
-)
 from langfuse._client.resource_manager import LangfuseResourceManager
 from langfuse._client.span import (
-    LangfuseEvent,
-    LangfuseGeneration,
-    LangfuseSpan,
     LangfuseAgent,
-    LangfuseTool,
     LangfuseChain,
-    LangfuseRetriever,
-    LangfuseEvaluator,
     LangfuseEmbedding,
+    LangfuseEvaluator,
+    LangfuseEvent,
+    LangfuseGeneration,
     LangfuseGuardrail,
+    LangfuseRetriever,
+    LangfuseSpan,
+    LangfuseTool,
 )
 from langfuse._utils import _get_timestamp
 from langfuse._utils.parse_error import handle_fern_exception
@@ -2996,11 +2996,10 @@ def _url_encode(self, url: str, *, is_url_param: Optional[bool] = False) -> str:
         # we need add safe="" to force escaping of slashes
         # This is necessary for prompts in prompt folders
         return urllib.parse.quote(url, safe="")
-    
-    def clear_prompt_cache(self):
-        """
-        Clear the entire prompt cache, removing all cached prompts.
-        
+
+    def clear_prompt_cache(self) -> None:
+        """Clear the entire prompt cache, removing all cached prompts.
+
         This method is useful when you want to force a complete refresh of all
         cached prompts, for example after major updates or when you need to
         ensure the latest versions are fetched from the server.

langfuse/_utils/prompt_cache.py

@@ -2,18 +2,20 @@
 
 import atexit
 import logging
+import os
 from datetime import datetime
 from queue import Empty, Queue
 from threading import Thread
 from typing import Callable, Dict, List, Optional, Set
-import os
 
-from langfuse.model import PromptClient
 from langfuse._client.environment_variables import (
-    LANGFUSE_PROMPT_CACHE_DEFAULT_TTL_SECONDS
+    LANGFUSE_PROMPT_CACHE_DEFAULT_TTL_SECONDS,
 )
+from langfuse.model import PromptClient
 
-DEFAULT_PROMPT_CACHE_TTL_SECONDS = int(os.getenv(LANGFUSE_PROMPT_CACHE_DEFAULT_TTL_SECONDS, 60))
+DEFAULT_PROMPT_CACHE_TTL_SECONDS = int(
+    os.getenv(LANGFUSE_PROMPT_CACHE_DEFAULT_TTL_SECONDS, 60)
+)
 
 DEFAULT_PROMPT_CACHE_REFRESH_WORKERS = 1
 

langfuse/langchain/CallbackHandler.py

@@ -939,7 +939,7 @@ def __join_tags_and_metadata(
     def _convert_message_to_dict(self, message: BaseMessage) -> Dict[str, Any]:
         # assistant message
         if isinstance(message, HumanMessage):
-            message_dict = {"role": "user", "content": message.content}
+            message_dict: Dict[str, Any] = {"role": "user", "content": message.content}
         elif isinstance(message, AIMessage):
             message_dict = {"role": "assistant", "content": message.content}
 
@@ -949,6 +949,7 @@ def _convert_message_to_dict(self, message: BaseMessage) -> Dict[str, Any]:
                 and len(message.tool_calls) > 0
             ):
                 message_dict["tool_calls"] = message.tool_calls
+
         elif isinstance(message, SystemMessage):
             message_dict = {"role": "system", "content": message.content}
         elif isinstance(message, ToolMessage):