freeplay 0.3.2__tar.gz → 0.3.3__tar.gz

{freeplay-0.3.2 → freeplay-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: freeplay
-Version: 0.3.2
+Version: 0.3.3
 Summary: 
 License: MIT
 Author: FreePlay Engineering
@@ -12,6 +12,7 @@ Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: click (==8.1.7)
 Requires-Dist: dacite (>=1.8.0,<2.0.0)
 Requires-Dist: pystache (>=0.6.5,<0.7.0)

{freeplay-0.3.2 → freeplay-0.3.3}/pyproject.toml

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

{freeplay-0.3.2 → freeplay-0.3.3}/src/freeplay/model.py

@@ -4,6 +4,7 @@ from typing import List, Union, Any, Dict, Mapping, TypedDict
 InputValue = Union[str, int, bool, Dict[str, Any], List[Any]]
 InputVariables = Mapping[str, InputValue]
 TestRunInput = Mapping[str, InputValue]
+FeedbackValue = Union[bool, str, int, float]
 
 
 @dataclass

freeplay-0.3.3/src/freeplay/resources/customer_feedback.py

@@ -0,0 +1,33 @@
+from dataclasses import dataclass
+from typing import Dict, Union
+
+from freeplay.model import FeedbackValue
+from freeplay.support import CallSupport
+
+
+@dataclass
+class CustomerFeedbackResponse:
+    pass
+
+
+@dataclass
+class TraceFeedbackResponse:
+    pass
+
+
+class CustomerFeedback:
+    def __init__(self, call_support: CallSupport) -> None:
+        self.call_support = call_support
+
+    def update(self, completion_id: str, feedback: Dict[str, FeedbackValue]) -> CustomerFeedbackResponse:
+        self.call_support.update_customer_feedback(completion_id, feedback)
+        return CustomerFeedbackResponse()
+
+    def update_trace(
+            self,
+            project_id: str,
+            trace_id: str,
+            feedback: Dict[str, FeedbackValue]
+    ) -> TraceFeedbackResponse:
+        self.call_support.update_trace_feedback(project_id, trace_id, feedback)
+        return TraceFeedbackResponse()

{freeplay-0.3.2 → freeplay-0.3.3}/src/freeplay/resources/prompts.py

@@ -303,8 +303,11 @@ class FilesystemTemplateResolver(TemplateResolver):
     def __normalize_roles(messages: List[Dict[str, str]]) -> List[Dict[str, str]]:
         normalized = []
         for message in messages:
-            role = FilesystemTemplateResolver.__role_translations.get(message['role']) or message['role']
-            normalized.append({'role': role, 'content': message['content']})
+            if 'kind' in message:
+                normalized.append(message)
+            else:
+                role = FilesystemTemplateResolver.__role_translations.get(message['role']) or message['role']
+                normalized.append({'role': role, 'content': message['content']})
         return normalized
 
     @staticmethod

{freeplay-0.3.2 → freeplay-0.3.3}/src/freeplay/resources/test_runs.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import List, Optional
+from typing import List, Optional, Dict
 
 from freeplay.model import InputVariables
 from freeplay.resources.recordings import TestRunInfo
@@ -13,10 +13,12 @@ class TestCase:
             test_case_id: str,
             variables: InputVariables,
             output: Optional[str],
+            history: Optional[List[Dict[str, str]]]
     ):
         self.id = test_case_id
         self.variables = variables
         self.output = output
+        self.history = history
 
 
 @dataclass
@@ -64,7 +66,10 @@ class TestRuns:
     ) -> TestRun:
         test_run = self.call_support.create_test_run(project_id, testlist, include_outputs, name, description)
         test_cases = [
-            TestCase(test_case_id=test_case.id, variables=test_case.variables, output=test_case.output)
+            TestCase(test_case_id=test_case.id,
+                     variables=test_case.variables,
+                     output=test_case.output,
+                     history=test_case.history)
             for test_case in test_run.test_cases
         ]
 

{freeplay-0.3.2 → freeplay-0.3.3}/src/freeplay/support.py

@@ -5,7 +5,7 @@ from typing import Optional, Dict, Any, List, Union
 from freeplay import api_support
 from freeplay.api_support import try_decode
 from freeplay.errors import freeplay_response_error, FreeplayServerError
-from freeplay.model import InputVariables
+from freeplay.model import InputVariables, FeedbackValue
 
 
 @dataclass
@@ -49,6 +49,7 @@ class TestCaseTestRunResponse:
         self.variables: InputVariables = test_case['variables']
         self.id: str = test_case['test_case_id']
         self.output: Optional[str] = test_case.get('output')
+        self.history: Optional[List[Dict[str, str]]] = test_case.get('history')
 
 
 class TestRunResponse:
@@ -163,6 +164,20 @@ class CallSupport:
         if response.status_code != 201:
             raise freeplay_response_error("Error updating customer feedback", response)
 
+    def update_trace_feedback(
+            self, project_id: str, trace_id: str, feedback: Dict[str, FeedbackValue]
+    ) -> None:
+        response = api_support.post_raw(
+            self.freeplay_api_key,
+            f'{self.api_base}/v2/projects/{project_id}/trace-feedback/id/{trace_id}',
+            feedback
+        )
+        if response.status_code != 201:
+            raise freeplay_response_error(
+                f'Error updating trace feedback for {trace_id} in project {project_id}',
+                response
+            )
+
     def create_test_run(
             self,
             project_id: str,

freeplay-0.3.2/src/freeplay/resources/customer_feedback.py

@@ -1,18 +0,0 @@
-from dataclasses import dataclass
-from typing import Dict, Union
-
-from freeplay.support import CallSupport
-
-
-@dataclass
-class CustomerFeedbackResponse:
-    pass
-
-
-class CustomerFeedback:
-    def __init__(self, call_support: CallSupport) -> None:
-        self.call_support = call_support
-
-    def update(self, completion_id: str, feedback: Dict[str, Union[bool, str, int, float]]) -> CustomerFeedbackResponse:
-        self.call_support.update_customer_feedback(completion_id, feedback)
-        return CustomerFeedbackResponse()