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

freeplay/resources/prompts.py

@@ -8,7 +8,6 @@ from typing import (
     Any,
     Dict,
     List,
-    Literal,
     Optional,
     Protocol,
     Sequence,
@@ -56,7 +55,12 @@ logger = logging.getLogger(__name__)
 class UnsupportedToolSchemaError(FreeplayConfigurationError):
     def __init__(self) -> None:
         super().__init__(
-            f'Tool schema not supported for this model and provider.'
+            'Tool schema not supported for this model and provider.'
+        )
+class VertexAIToolSchemaError(FreeplayConfigurationError):
+    def __init__(self) -> None:
+        super().__init__(
+            'Vertex AI SDK not found. Install google-cloud-aiplatform to get proper Tool objects.'
         )
 
 
@@ -185,6 +189,21 @@ class BoundPrompt:
                     for tool_schema in tool_schema
                 ]
             }
+        elif flavor_name == "gemini_chat":
+            try:
+                from vertexai.generative_models import Tool, FunctionDeclaration  # type: ignore[import-untyped]
+                
+                function_declarations = [
+                    FunctionDeclaration(
+                        name=tool_schema.name,
+                        description=tool_schema.description,
+                        parameters=tool_schema.parameters
+                    )
+                    for tool_schema in tool_schema
+                ]
+                return [Tool(function_declarations=function_declarations)]
+            except ImportError:
+                raise VertexAIToolSchemaError()
 
         raise UnsupportedToolSchemaError()
 

freeplay/resources/recordings.py

@@ -22,6 +22,7 @@ from freeplay.resources.prompts import (
 )
 from freeplay.resources.sessions import SessionInfo, TraceInfo
 from freeplay.support import CallSupport
+from freeplay.utils import convert_provider_message_to_dict
 
 logger = logging.getLogger(__name__)
 
@@ -129,6 +130,9 @@ class Recordings:
         if len(record_payload.all_messages) < 1:
             raise FreeplayClientError("Messages list must have at least one message. "
                                       "The last message should be the current response.")
+        
+        if record_payload.tool_schema is not None:
+            record_payload.tool_schema = [convert_provider_message_to_dict(tool) for tool in record_payload.tool_schema]
 
         record_api_payload: Dict[str, Any] = {
             "messages": record_payload.all_messages,

freeplay/utils.py

@@ -75,14 +75,55 @@ 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:
-    if hasattr(obj, 'model_dump'):
+    """
+    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')):
+        # Regular to_dict (for Vertex AI wrappers without _raw_* attributes)
+        return obj.to_dict()
+    elif hasattr(obj, 'model_dump'):
         # Pydantic v2
         return obj.model_dump(mode='json')
     elif hasattr(obj, 'dict'):
         # 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.0a1.dist-info → freeplay-0.5.0a3.dist-info}/METADATA

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

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

@@ -9,15 +9,15 @@ freeplay/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 freeplay/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 freeplay/resources/adapters.py,sha256=6ZAPpoLeOkUkV1s9VNQNsYrnupV0-sy11zFfKfctM1Y,9296
 freeplay/resources/customer_feedback.py,sha256=6AUgHyOcXIpHvrxGAhsQgmDERvRHKutB6J-GkhkGH6s,928
-freeplay/resources/prompts.py,sha256=WNxQEXPFCSwp0SiK_qrqwPLgUTOAJXDLFvH52kbJXmg,23196
-freeplay/resources/recordings.py,sha256=gWFn4KavvSiPq43Atxy4qLuVKdJ0-OwnFkh7mEFHm9k,9570
+freeplay/resources/prompts.py,sha256=pkZa_b3HSrZoPqhcKCXsb26w2In-ETKydclbkXU_hus,24093
+freeplay/resources/recordings.py,sha256=OOSgSkQO7CVnOA2y7eSl5B2eMHZv5E3_-3J2XILwKUo,9811
 freeplay/resources/sessions.py,sha256=dZtd9nq2nH8pmXxQOJitBnN5Jl3kjggDItDcjC69TYo,3883
 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=Xvt4mNLXLL7E6MI2hTuDLV5cl5Y83DgdjCZSyDGMjR0,3187
-freeplay-0.5.0a1.dist-info/LICENSE,sha256=_jzIw45hB1XHGxiQ8leZ0GH_X7bR_a8qgxaqnHbCUOo,1064
-freeplay-0.5.0a1.dist-info/METADATA,sha256=-AHpXBmS7SOEoM9bRpCSj_VXxNf5e6HiNMY40dg6Vy4,1662
-freeplay-0.5.0a1.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-freeplay-0.5.0a1.dist-info/entry_points.txt,sha256=32s3rf2UUCqiJT4jnClEXZhdXlvl30uwpcxz-Gsy4UU,54
-freeplay-0.5.0a1.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,,