freeplay 0.3.11__tar.gz → 0.3.12__tar.gz

{freeplay-0.3.11 → freeplay-0.3.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: freeplay
-Version: 0.3.11
+Version: 0.3.12
 Summary: 
 License: MIT
 Author: FreePlay Engineering

{freeplay-0.3.11 → freeplay-0.3.12}/pyproject.toml

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

{freeplay-0.3.11 → freeplay-0.3.12}/src/freeplay/resources/recordings.py

@@ -2,6 +2,7 @@ import json
 import logging
 from dataclasses import dataclass
 from typing import Any, Dict, List, Optional, Union
+from uuid import UUID
 
 from requests import HTTPError
 
@@ -64,6 +65,7 @@ class RecordPayload:
     test_run_info: Optional[TestRunInfo] = None
     eval_results: Optional[Dict[str, Union[bool, float]]] = None
     trace_info: Optional[TraceInfo] = None
+    completion_id: Optional[UUID] = None
 
 
 @dataclass
@@ -80,7 +82,7 @@ class Recordings:
             raise FreeplayClientError("Messages list must have at least one message. "
                                       "The last message should be the current response.")
 
-        record_api_payload = {
+        record_api_payload: Dict[str, Any] = {
             "messages": record_payload.all_messages,
             "inputs": record_payload.inputs,
             "tool_schema": record_payload.tool_schema,
@@ -99,6 +101,9 @@ class Recordings:
             }
         }
 
+        if record_payload.completion_id is not None:
+            record_api_payload['completion_id'] = str(record_payload.completion_id)
+
         if record_payload.session_info.custom_metadata is not None:
             record_api_payload['custom_metadata'] = record_payload.session_info.custom_metadata