freeplay 0.5.0a5__tar.gz → 0.5.1__tar.gz

{freeplay-0.5.0a5 → freeplay-0.5.1}/PKG-INFO

@@ -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 → freeplay-0.5.1}/pyproject.toml

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

{freeplay-0.5.0a5 → freeplay-0.5.1}/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

{freeplay-0.5.0a5 → freeplay-0.5.1}/src/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-0.5.0a5 → freeplay-0.5.1}/src/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-0.5.0a5 → freeplay-0.5.1}/src/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