freeplay 0.5.0a2__py3-none-any.whl → 0.5.0a3__py3-none-any.whl

freeplay/utils.py

@@ -75,8 +75,42 @@ def get_user_agent() -> str:
 # Recursively convert Pydantic models, lists, and dicts to dict compatible format -- used to allow us to accept
 # provider message shapes (usually generated types) or the default {'content': ..., 'role': ...} shape.
 def convert_provider_message_to_dict(obj: Any) -> Any:
+    """
+    Convert provider message objects to dictionaries.
+    For Vertex AI objects, automatically converts to camelCase.
+    """
+    # List of possible raw attribute names in Vertex AI objects
+    vertex_raw_attrs = [
+        '_raw_content',      # For Content objects
+        '_raw_tool',         # For Tool objects  
+        '_raw_message',      # For message objects
+        '_raw_candidate',    # For Candidate objects
+        '_raw_response',     # For response objects
+        '_raw_function_declaration',  # For FunctionDeclaration
+        '_raw_generation_config',     # For GenerationConfig
+        '_pb',              # Generic protobuf attribute
+    ]
+    
+    # Check for Vertex AI objects with raw protobuf attributes
+    for attr_name in vertex_raw_attrs:
+        if hasattr(obj, attr_name):
+            raw_obj = getattr(obj, attr_name)
+            if raw_obj is not None:
+                try:
+                    # Use the metaclass to_dict with camelCase conversion
+                    return type(raw_obj).to_dict(
+                        raw_obj,
+                        preserving_proto_field_name=False,  # camelCase
+                        use_integers_for_enums=False,  # Keep as strings (we'll lowercase them)
+                        including_default_value_fields=False  # Exclude defaults
+                    )
+                except:  # noqa: E722
+                    # If we can't convert, continue to the next attribute
+                    pass
+    
+    # For non-Vertex AI objects, use their standard to_dict methods
     if hasattr(obj, 'to_dict') and callable(getattr(obj, 'to_dict')):
-        # Vertex AI has a to_dict method
+        # Regular to_dict (for Vertex AI wrappers without _raw_* attributes)
         return obj.to_dict()
     elif hasattr(obj, 'model_dump'):
         # Pydantic v2
@@ -85,7 +119,11 @@ def convert_provider_message_to_dict(obj: Any) -> Any:
         # Pydantic v1
         return obj.dict(encode_json=True)
     elif isinstance(obj, dict):
+        # Handle dictionaries recursively
         return {k: convert_provider_message_to_dict(v) for k, v in obj.items()}
     elif isinstance(obj, list):
+        # Handle lists recursively
         return [convert_provider_message_to_dict(item) for item in obj]
+    
+    # Return as-is for primitive types
     return obj

{freeplay-0.5.0a2.dist-info → freeplay-0.5.0a3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: freeplay
-Version: 0.5.0a2
+Version: 0.5.0a3
 Summary: 
 License: MIT
 Author: FreePlay Engineering

{freeplay-0.5.0a2.dist-info → freeplay-0.5.0a3.dist-info}/RECORD

@@ -15,9 +15,9 @@ freeplay/resources/sessions.py,sha256=dZtd9nq2nH8pmXxQOJitBnN5Jl3kjggDItDcjC69TY
 freeplay/resources/test_cases.py,sha256=nXL_976RwSJDT6OWDM4GEzbcOzcGkJ9ulvb0XOzCRDM,2240
 freeplay/resources/test_runs.py,sha256=IkqW2LrVrLzFcGtalwP4FV-DeQKKb6Nekvy02FfoH8k,5007
 freeplay/support.py,sha256=9X-utKyM9BkhCLJbPT184ud26Dc69f3DiHjyN8qq0vQ,15266
-freeplay/utils.py,sha256=5mNC3NTHDlzltyUDF8q2wNddgGy6-MhI8ldnRvtAl7U,3329
-freeplay-0.5.0a2.dist-info/LICENSE,sha256=_jzIw45hB1XHGxiQ8leZ0GH_X7bR_a8qgxaqnHbCUOo,1064
-freeplay-0.5.0a2.dist-info/METADATA,sha256=pzFZ1UFoEM-rCaL11UHjH_pMjYg20tRGog0zptc5cnQ,1662
-freeplay-0.5.0a2.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-freeplay-0.5.0a2.dist-info/entry_points.txt,sha256=32s3rf2UUCqiJT4jnClEXZhdXlvl30uwpcxz-Gsy4UU,54
-freeplay-0.5.0a2.dist-info/RECORD,,
+freeplay/utils.py,sha256=OtoSnlDrLEk3MWiXmKFJ4Sw42-1kQ94-d_2ekHT6eUo,5038
+freeplay-0.5.0a3.dist-info/LICENSE,sha256=_jzIw45hB1XHGxiQ8leZ0GH_X7bR_a8qgxaqnHbCUOo,1064
+freeplay-0.5.0a3.dist-info/METADATA,sha256=wJZwNOsRV5Oq1xPG8Wbx1H4UpbC4oH6RbuVP_erRPqM,1662
+freeplay-0.5.0a3.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+freeplay-0.5.0a3.dist-info/entry_points.txt,sha256=32s3rf2UUCqiJT4jnClEXZhdXlvl30uwpcxz-Gsy4UU,54
+freeplay-0.5.0a3.dist-info/RECORD,,