fix: memory leak in OTEL traces

Author: michael-rubel

langfuse/_task_manager/media_manager.py

@@ -84,11 +84,9 @@ def _find_and_process_media(
         max_levels = 10
 
         def _process_data_recursively(data: Any, level: int) -> Any:
-            if id(data) in seen or level > max_levels:
+            if level > max_levels:
                 return data
 
-            seen.add(id(data))
-
             if isinstance(data, LangfuseMedia):
                 self._process_media(
                     media=data,
@@ -176,6 +174,24 @@ def _process_data_recursively(data: Any, level: int) -> Any:
                     for key, value in data.items()
                 }
 
+            if hasattr(data, "model_dump") and callable(data.model_dump):
+                # Pydantic v2 BaseModel
+                if id(data) in seen:
+                    return data
+
+                seen.add(id(data))
+
+                return _process_data_recursively(data.model_dump(), level + 1)
+
+            if hasattr(data, "dict") and callable(data.dict) and hasattr(data, "__fields__"):
+                # Pydantic v1 BaseModel
+                if id(data) in seen:
+                    return data
+
+                seen.add(id(data))
+
+                return _process_data_recursively(data.dict(), level + 1)
+
             return data
 
         return _process_data_recursively(data, 1)