freeplay 0.5.0a5__py3-none-any.whl → 0.5.1__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

freeplay/resources/recordings.py

@@ -1,5 +1,6 @@
 import json
 import logging
+import warnings
 from dataclasses import dataclass, field
 from typing import Any, Dict, List, Literal, Optional, Union
 from uuid import UUID, uuid4
@@ -11,14 +12,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,16 +85,18 @@ 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
     response_info: Optional[ResponseInfo] = None
     test_run_info: Optional[TestRunInfo] = None
     eval_results: Optional[Dict[str, Union[bool, float]]] = None
-    trace_info: Optional[TraceInfo] = None
+    parent_id: Optional[UUID] = None
     completion_id: Optional[UUID] = None
-
+    # Deprecated field support for backward compatibility
+    # This field will be removed in v0.6.0
+    trace_info: Optional[TraceInfo] = None
 
 @dataclass
 class RecordUpdatePayload:
@@ -115,7 +117,7 @@ class Recordings:
     def __init__(self, call_support: CallSupport):
         self.call_support = call_support
 
-    def create(self, record_payload: RecordPayload) -> RecordResponse:  # type: ignore
+    def create(self, record_payload: RecordPayload) -> RecordResponse:
         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.")
@@ -128,13 +130,13 @@ class Recordings:
             "inputs": record_payload.inputs,
             "tool_schema": record_payload.tool_schema,
             "session_info": {"custom_metadata": record_payload.session_info.custom_metadata},
-            
+            "parent_id": str(record_payload.parent_id) if record_payload.parent_id is not None else None,
         }
 
-        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:
@@ -173,6 +175,7 @@ class Recordings:
             record_api_payload['eval_results'] = record_payload.eval_results
 
         if record_payload.trace_info is not None:
+            warnings.warn("trace_info in RecordPayload is deprecated and will be removed in v0.6.0. Use parent_id instead.", DeprecationWarning)
             record_api_payload['trace_info'] = {
                 "trace_id": record_payload.trace_info.trace_id
             }
@@ -213,6 +216,8 @@ class Recordings:
                       f'Status: {status_code}. {e.__class__}'
 
             raise FreeplayError(message) from e
+        
+        raise FreeplayError("Unexpected error occurred while recording to Freeplay.") 
 
     def update(self, record_update_payload: RecordUpdatePayload) -> RecordResponse:  # type: ignore
         record_update_api_payload: Dict[str, Any] = {

freeplay/resources/sessions.py

@@ -1,4 +1,4 @@
-import uuid
+from uuid import UUID, uuid4
 from dataclasses import dataclass
 from typing import Optional, Dict, Union
 
@@ -18,6 +18,7 @@ class TraceInfo:
     trace_id: str
     input: Optional[str] = None
     agent_name: Optional[str] = None
+    parent_id: Optional[UUID] = None
     custom_metadata: CustomMetadata = None
     _call_support: CallSupport
 
@@ -28,12 +29,14 @@ class TraceInfo:
             _call_support: CallSupport,
             input: Optional[str] = None,
             agent_name: Optional[str] = None,
+            parent_id: Optional[UUID] = None,
             custom_metadata: CustomMetadata = None,
     ):
         self.trace_id = trace_id
         self.session_id = session_id
         self.input = input
         self.agent_name = agent_name
+        self.parent_id = parent_id
         self.custom_metadata = custom_metadata
         self._call_support = _call_support
 
@@ -53,6 +56,7 @@ class TraceInfo:
             self.input,
             output,
             agent_name=self.agent_name,
+            parent_id=self.parent_id,
             custom_metadata=self.custom_metadata,
             eval_results=eval_results,
             test_run_info=test_run_info
@@ -78,11 +82,13 @@ class Session:
             self,
             input: str,
             agent_name: Optional[str] = None,
+            parent_id: Optional[UUID] = None,
             custom_metadata: CustomMetadata = None
     ) -> TraceInfo:
         return TraceInfo(
-            trace_id=str(uuid.uuid4()),
+            trace_id=str(uuid4()),
             session_id=self.session_id,
+            parent_id=parent_id,
             input=input,
             agent_name=agent_name,
             custom_metadata=custom_metadata,
@@ -91,9 +97,10 @@ class Session:
 
     def restore_trace(
             self,
-            trace_id: uuid.UUID,
+            trace_id: UUID,
             input: Optional[str],
             agent_name: Optional[str] = None,
+            parent_id: Optional[UUID] = None,
             custom_metadata: CustomMetadata = None
     ) -> TraceInfo:
         return TraceInfo(
@@ -101,6 +108,7 @@ class Session:
             session_id=self.session_id,
             input=input,
             agent_name=agent_name,
+            parent_id=parent_id,
             custom_metadata=custom_metadata,
             _call_support=self._call_support
         )
@@ -112,7 +120,7 @@ class Sessions:
 
     def create(self, custom_metadata: CustomMetadata = None) -> Session:
         return Session(
-            session_id=str(uuid.uuid4()),
+            session_id=str(uuid4()),
             custom_metadata=custom_metadata,
             _call_support=self.call_support,
         )

freeplay/support.py

@@ -400,6 +400,7 @@ class CallSupport:
             trace_id: str,
             input: str,
             output: str,
+            parent_id: Optional[UUID] = None,
             agent_name: Optional[str] = None,
             custom_metadata: CustomMetadata = None,
             eval_results: Optional[Dict[str, Union[bool, float]]] = None,
@@ -409,6 +410,7 @@ class CallSupport:
             'agent_name': agent_name,
             'input': input,
             'output': output,
+            'parent_id': str(parent_id) if parent_id else None,
             'custom_metadata': custom_metadata,
             'eval_results': eval_results,
             'test_run_info': asdict(test_run_info) if test_run_info else None

{freeplay-0.5.0a5.dist-info → freeplay-0.5.1.dist-info}/METADATA

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

{freeplay-0.5.0a5.dist-info → freeplay-0.5.1.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=co79tarmUpDNaIRDRJZzTNRnnC1V945XxOK8UXcMKkM,24130
-freeplay/resources/recordings.py,sha256=f6h2b1qFQUJbjVjMaGdWuiVXD9TYbuAUTKe0ReEZ2dg,9452
-freeplay/resources/sessions.py,sha256=dZtd9nq2nH8pmXxQOJitBnN5Jl3kjggDItDcjC69TYo,3883
+freeplay/resources/prompts.py,sha256=uSLMZrQBZ7LSKniQZj6xhl4xvuD8ue6ltY8g7VE7fOw,24223
+freeplay/resources/recordings.py,sha256=a_W4B2erhb258xNLOuqC1JrTpaZ87gV9F5gjrF6d8fs,9950
+freeplay/resources/sessions.py,sha256=D8rI_qGRBWnaOrcNxYhILbgG9Mgiz1ksS2aMmo4uCmc,4199
 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/support.py,sha256=B1JwMBejBUAm5sKBuE9y6mv_MyfOful4Si2kgAv2W-k,16482
 freeplay/utils.py,sha256=OtoSnlDrLEk3MWiXmKFJ4Sw42-1kQ94-d_2ekHT6eUo,5038
-freeplay-0.5.0a5.dist-info/LICENSE,sha256=_jzIw45hB1XHGxiQ8leZ0GH_X7bR_a8qgxaqnHbCUOo,1064
-freeplay-0.5.0a5.dist-info/METADATA,sha256=zpSSQ-a2yhsIH4rUNyXYgj6OcchxA_gO8dde928KUOI,1662
-freeplay-0.5.0a5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-freeplay-0.5.0a5.dist-info/entry_points.txt,sha256=32s3rf2UUCqiJT4jnClEXZhdXlvl30uwpcxz-Gsy4UU,54
-freeplay-0.5.0a5.dist-info/RECORD,,
+freeplay-0.5.1.dist-info/LICENSE,sha256=_jzIw45hB1XHGxiQ8leZ0GH_X7bR_a8qgxaqnHbCUOo,1064
+freeplay-0.5.1.dist-info/METADATA,sha256=fTYx_xgff2bP7UGUW5oZo0XG-p_inqsDpsoqr9S8wns,1660
+freeplay-0.5.1.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+freeplay-0.5.1.dist-info/entry_points.txt,sha256=32s3rf2UUCqiJT4jnClEXZhdXlvl30uwpcxz-Gsy4UU,54
+freeplay-0.5.1.dist-info/RECORD,,

{freeplay-0.5.0a5.dist-info → freeplay-0.5.1.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