freeplay 0.5.0a6__tar.gz → 0.5.2a1__tar.gz

{freeplay-0.5.0a6 → freeplay-0.5.2a1}/PKG-INFO

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

{freeplay-0.5.0a6 → freeplay-0.5.2a1}/pyproject.toml

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

{freeplay-0.5.0a6 → freeplay-0.5.2a1}/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
@@ -91,9 +92,11 @@ class RecordPayload:
     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:
@@ -114,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.")
@@ -127,6 +130,7 @@ 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_version_info is not None:
@@ -171,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
             }
@@ -211,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.0a6 → freeplay-0.5.2a1}/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.0a6 → freeplay-0.5.2a1}/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