Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions langfuse/_task_manager/media_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,6 +126,7 @@ def _process_data_recursively(data: Any, level: int) -> Any:
and data["type"] == "base64"
and "media_type" in data
and "data" in data
and isinstance(data["data"], str)
):
media = LangfuseMedia(
base64_data_uri=f"data:{data['media_type']};base64," + data["data"],
Expand All @@ -150,6 +151,7 @@ def _process_data_recursively(data: Any, level: int) -> Any:
and data["type"] == "media"
and "mime_type" in data
and "data" in data
and isinstance(data["data"], str)
):
media = LangfuseMedia(
base64_data_uri=f"data:{data['mime_type']};base64," + data["data"],
Expand Down
22 changes: 22 additions & 0 deletions tests/unit/test_media_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -194,3 +194,25 @@ def test_find_and_process_media_sse_in_nested_structure_passes_through():

assert result == data
assert queue.empty()


def test_find_and_process_media_base64_block_with_bytes_data_passes_through():
queue = Queue()
manager = MediaManager(
api_client=SimpleNamespace(media=Mock()),
httpx_client=Mock(),
media_upload_queue=queue,
)

# Anthropic/Vertex media blocks whose "data" is raw bytes rather than a
# base64 string must not crash media processing.
data = [
{"type": "base64", "media_type": "image/png", "data": b"\x89PNG raw"},
{"type": "media", "mime_type": "image/png", "data": b"\x89PNG raw"},
]
result = manager._find_and_process_media(
data=data, trace_id="trace-id", observation_id=None, field="input"
)

assert result == data
assert queue.empty()