fix(galileo): handle negative TZ offsets, 2xx success, and Pydantic ImageObject serialization

cursoragent · yassin-berriai · cursoragent · commit 2a0436419f5e · 2026-05-25T07:05:40.000Z
Co-authored-by: Yassin Kortam &lt;yassin@berri.ai&gt;
diff --git a/litellm/integrations/galileo.py b/litellm/integrations/galileo.py
@@ -1,5 +1,6 @@
 import json
 import os
+import re
 from typing import Any, Dict, List, Optional, Tuple
 
 from pydantic import BaseModel, Field
@@ -142,7 +143,7 @@ def _galileo_input_messages(
     @staticmethod
     def _record_to_v2_span(record: Dict[str, Any]) -> Dict[str, Any]:
         created_at = record.get("created_at", "")
-        if created_at and "Z" not in created_at and "+" not in created_at:
+        if created_at and not re.search(r"(Z|[+-]\d{2}:?\d{2})$", created_at):
             created_at = f"{created_at}Z"
 
         span: Dict[str, Any] = {
@@ -203,7 +204,7 @@ def get_output_str_from_response(
         if isinstance(response_obj, litellm.TextCompletionResponse):
             return response_obj.choices[0].text
         if isinstance(response_obj, litellm.ImageResponse):
-            return json.dumps(response_obj["data"])
+            return json.dumps(response_obj["data"], default=str)
         if isinstance(response_obj, (litellm.ModelResponse, dict)):
             return get_content_from_model_response(response_obj)
         return None
@@ -289,7 +290,7 @@ async def flush_in_memory_records(self):
             )
             return
 
-        if response.status_code == 200:
+        if 200 <= response.status_code < 300:
             verbose_logger.debug(
                 "Galileo Logger: successfully flushed in memory records"
             )
