freeplay 0.5.0a4__tar.gz → 0.5.0a6__tar.gz

{freeplay-0.5.0a4 → freeplay-0.5.0a6}/PKG-INFO

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

{freeplay-0.5.0a4 → freeplay-0.5.0a6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "freeplay"
-version = "0.5.0a4"
+version = "0.5.0a6"
 description = ""
 authors = ["FreePlay Engineering <engineering@freeplay.ai>"]
 license = "MIT"

{freeplay-0.5.0a4 → freeplay-0.5.0a6}/src/freeplay/resources/prompts.py

@@ -93,12 +93,17 @@ GenericProviderMessage = ProviderMessage
 
 
 # SDK-Exposed Classes
+
+@dataclass
+class PromptVersionInfo:
+    prompt_template_version_id: str
+    environment: Optional[str]
+
 @dataclass
-class PromptInfo:
+class PromptInfo(PromptVersionInfo):
     prompt_template_id: str
     prompt_template_version_id: str
     template_name: str
-    environment: Optional[str]
     model_parameters: LLMParameters
     provider_info: Optional[Dict[str, Any]]
     provider: str
@@ -488,6 +493,7 @@ class FilesystemTemplateResolver(TemplateResolver):
             'azure_openai_chat': 'azure',
             'anthropic_chat': 'anthropic',
             'openai_chat': 'openai',
+            "gemini_chat": "vertex",
         }
         provider = flavor_provider.get(flavor)
         if not provider:

{freeplay-0.5.0a4 → freeplay-0.5.0a6}/src/freeplay/resources/recordings.py

@@ -11,14 +11,13 @@ from freeplay.errors import FreeplayClientError, FreeplayError
 from freeplay.llm_parameters import LLMParameters
 from freeplay.model import (
     InputVariables,
-    MediaInput,
     MediaInputMap,
-    MediaInputUrl,
     OpenAIFunctionCall,
     TestRunInfo,
 )
 from freeplay.resources.prompts import (
     PromptInfo,
+    PromptVersionInfo,
 )
 from freeplay.resources.sessions import SessionInfo, TraceInfo
 from freeplay.support import CallSupport, media_inputs_to_json
@@ -85,7 +84,7 @@ class RecordPayload:
         default_factory=lambda: SessionInfo(session_id=str(uuid4()), custom_metadata=None)
     )
     inputs: Optional[InputVariables] = None
-    prompt_info: Optional[PromptInfo] = None
+    prompt_version_info: Optional[PromptVersionInfo] = None
     call_info: Optional[CallInfo] = None
     media_inputs: Optional[MediaInputMap] = None
     tool_schema: Optional[List[Dict[str, Any]]] = None
@@ -128,13 +127,12 @@ class Recordings:
             "inputs": record_payload.inputs,
             "tool_schema": record_payload.tool_schema,
             "session_info": {"custom_metadata": record_payload.session_info.custom_metadata},
-            
         }
 
-        if record_payload.prompt_info is not None:
+        if record_payload.prompt_version_info is not None:
             record_api_payload["prompt_info"] = {
-                "environment": record_payload.prompt_info.environment,
-                "prompt_template_version_id": record_payload.prompt_info.prompt_template_version_id,
+                "environment": record_payload.prompt_version_info.environment,
+                "prompt_template_version_id": record_payload.prompt_version_info.prompt_template_version_id,
             }
         
         if record_payload.call_info is not None: