rename to LangfuseMedia

Author: hassiebp

langfuse/_task_manager/media_manager.py

@@ -6,7 +6,7 @@
 
 from langfuse.api import GetMediaUploadUrlRequest, PatchMediaBody
 from langfuse.api.client import FernLangfuse
-from langfuse.langfuse_media import MediaWrapper
+from langfuse.media import LangfuseMedia
 from langfuse.utils import _get_timestamp
 
 from .media_upload_queue import MediaUploadQueue, UploadMediaJob
@@ -87,7 +87,7 @@ def _process_data_recursively(data: Any):
 
             seen.add(id(data))
 
-            if isinstance(data, MediaWrapper):
+            if isinstance(data, LangfuseMedia):
                 self._process_media(
                     media=data,
                     trace_id=trace_id,
@@ -98,7 +98,7 @@ def _process_data_recursively(data: Any):
                 return data
 
             if isinstance(data, str) and data.startswith("data:"):
-                media = MediaWrapper(
+                media = LangfuseMedia(
                     obj=data,
                     base64_data_uri=data,
                 )
@@ -127,7 +127,7 @@ def _process_data_recursively(data: Any):
     def _process_media(
         self,
         *,
-        media: MediaWrapper,
+        media: LangfuseMedia,
         trace_id: str,
         observation_id: Optional[str],
         field: str,

langfuse/langfuse_media.py langfuse/media.pylangfuse/langfuse_media.py renamed to langfuse/media.py

@@ -4,7 +4,7 @@
 from typing import Optional
 
 
-class MediaWrapper:
+class LangfuseMedia:
     """A class for wrapping media objects for upload to Langfuse.
 
     This class handles the preparation and formatting of media content for Langfuse,

langfuse/openai.py

@@ -33,7 +33,7 @@
 from langfuse import Langfuse
 from langfuse.client import StatefulGenerationClient
 from langfuse.decorators import langfuse_context
-from langfuse.langfuse_media import MediaWrapper
+from langfuse.media import LangfuseMedia
 from langfuse.utils import _get_timestamp
 from langfuse.utils.langfuse_singleton import LangfuseSingleton
 
@@ -234,7 +234,9 @@ def _process_message(message):
                     {
                         "type": "input_audio",
                         "input_audio": {
-                            "data": MediaWrapper(None, base64_data_uri=base64_data_uri),
+                            "data": LangfuseMedia(
+                                None, base64_data_uri=base64_data_uri
+                            ),
                             "format": format,
                         },
                     }
@@ -273,7 +275,7 @@ def _extract_chat_response(kwargs: any):
             {
                 "type": "output_audio",
                 "output_audio": {
-                    "data": MediaWrapper(
+                    "data": LangfuseMedia(
                         None,
                         base64_data_uri=base64_data_uri,
                     ),

langfuse/serializer.py

@@ -14,7 +14,7 @@
 from pydantic import BaseModel
 
 from langfuse.api.core import pydantic_utilities, serialize_datetime
-from langfuse.langfuse_media import MediaWrapper
+from langfuse.media import LangfuseMedia
 
 # Attempt to import Serializable
 try:
@@ -43,7 +43,7 @@ def default(self, obj: Any):
                 # Timezone-awareness check
                 return serialize_datetime(obj)
 
-            if isinstance(obj, MediaWrapper):
+            if isinstance(obj, LangfuseMedia):
                 return (
                     obj._reference_string
                     or "<Failed to generate reference string for LangfuseMediaWrapper>"

tests/test_decorators.py

@@ -9,7 +9,7 @@
 from langchain_community.chat_models import ChatOpenAI
 
 from langfuse.decorators import langfuse_context, observe
-from langfuse.langfuse_media import MediaWrapper
+from langfuse.media import LangfuseMedia
 from langfuse.openai import AsyncOpenAI
 from tests.utils import create_uuid, get_api, get_llama_index_index
 
@@ -1472,7 +1472,7 @@ def test_pdf_in_metadata():
     def main():
         langfuse_context.update_current_trace(
             metadata={
-                "context": MediaWrapper(
+                "context": LangfuseMedia(
                     pdf_bytes, content_bytes=pdf_bytes, content_type="application/pdf"
                 )
             },