freeplay 0.5.0a4__py3-none-any.whl → 0.5.0a6__py3-none-any.whl

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/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:

{freeplay-0.5.0a4.dist-info → freeplay-0.5.0a6.dist-info}/METADATA

@@ -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.dist-info → freeplay-0.5.0a6.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=pkZa_b3HSrZoPqhcKCXsb26w2In-ETKydclbkXU_hus,24093
-freeplay/resources/recordings.py,sha256=f6h2b1qFQUJbjVjMaGdWuiVXD9TYbuAUTKe0ReEZ2dg,9452
+freeplay/resources/prompts.py,sha256=uSLMZrQBZ7LSKniQZj6xhl4xvuD8ue6ltY8g7VE7fOw,24223
+freeplay/resources/recordings.py,sha256=MK3xyPiAAF4z-Y0O8D0ixF-sfqtVbvCHDxQQ8IUXIpg,9466
 freeplay/resources/sessions.py,sha256=dZtd9nq2nH8pmXxQOJitBnN5Jl3kjggDItDcjC69TYo,3883
 freeplay/resources/test_cases.py,sha256=yJPtcAk1HznXSiJ8K5PtW_PIO_309LqObs4swBzzcNk,2378
 freeplay/resources/test_runs.py,sha256=ZQ7K2hjNRRiQOx8e4-mXTvPDb7ksTsIzBVlq2utrMNo,5117
 freeplay/support.py,sha256=smTwTTMRyK9NvhY0-uILWD_4Ri-Uw6_QZAagfhfOoJo,16372
 freeplay/utils.py,sha256=OtoSnlDrLEk3MWiXmKFJ4Sw42-1kQ94-d_2ekHT6eUo,5038
-freeplay-0.5.0a4.dist-info/LICENSE,sha256=_jzIw45hB1XHGxiQ8leZ0GH_X7bR_a8qgxaqnHbCUOo,1064
-freeplay-0.5.0a4.dist-info/METADATA,sha256=HLcOsZCmDOAftfWHKMj5YNJ_lNqkmlbnpYrM1eJR6Ig,1662
-freeplay-0.5.0a4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-freeplay-0.5.0a4.dist-info/entry_points.txt,sha256=32s3rf2UUCqiJT4jnClEXZhdXlvl30uwpcxz-Gsy4UU,54
-freeplay-0.5.0a4.dist-info/RECORD,,
+freeplay-0.5.0a6.dist-info/LICENSE,sha256=_jzIw45hB1XHGxiQ8leZ0GH_X7bR_a8qgxaqnHbCUOo,1064
+freeplay-0.5.0a6.dist-info/METADATA,sha256=5avT8WJXVKLKbWs8jBP0W2FFL1OeN8YG2vB4h8v2Hec,1662
+freeplay-0.5.0a6.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+freeplay-0.5.0a6.dist-info/entry_points.txt,sha256=32s3rf2UUCqiJT4jnClEXZhdXlvl30uwpcxz-Gsy4UU,54
+freeplay-0.5.0a6.dist-info/RECORD,,

{freeplay-0.5.0a4.dist-info → freeplay-0.5.0a6.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.3
+Generator: poetry-core 2.1.2
 Root-Is-Purelib: true
 Tag: py3-none-any