freeplay 0.3.14__py3-none-any.whl → 0.3.16__py3-none-any.whl

freeplay/__init__.py

@@ -1,6 +1,6 @@
 from .freeplay import Freeplay
 from .resources.prompts import PromptInfo
-from .resources.recordings import CallInfo, ResponseInfo, RecordPayload, TestRunInfo
+from .resources.recordings import CallInfo, ResponseInfo, RecordPayload, TestRunInfo, UsageTokens
 from .resources.sessions import SessionInfo, TraceInfo
 
 __all__ = [
@@ -12,4 +12,5 @@ __all__ = [
     'SessionInfo',
     'TestRunInfo',
     'TraceInfo',
+    'UsageTokens',
 ]

freeplay/resources/recordings.py

@@ -1,7 +1,7 @@
 import json
 import logging
 from dataclasses import dataclass
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional, Union, Literal
 from uuid import UUID
 
 from requests import HTTPError
@@ -17,6 +17,15 @@ from freeplay.support import CallSupport
 logger = logging.getLogger(__name__)
 
 
+@dataclass
+class UsageTokens:
+    prompt_tokens: int
+    completion_tokens: int
+
+
+ApiStyle = Union[Literal['batch'], Literal['default']]
+
+
 @dataclass
 class CallInfo:
     provider: str
@@ -25,9 +34,17 @@ class CallInfo:
     end_time: float
     model_parameters: LLMParameters
     provider_info: Optional[Dict[str, Any]] = None
+    usage: Optional[UsageTokens] = None
+    api_style: Optional[ApiStyle] = None
 
     @staticmethod
-    def from_prompt_info(prompt_info: PromptInfo, start_time: float, end_time: float) -> 'CallInfo':
+    def from_prompt_info(
+            prompt_info: PromptInfo,
+            start_time: float,
+            end_time: float,
+            usage: Optional[UsageTokens] = None,
+            api_style: Optional[ApiStyle] = None
+    ) -> 'CallInfo':
         return CallInfo(
             provider=prompt_info.provider,
             model=prompt_info.model,
@@ -35,6 +52,8 @@ class CallInfo:
             end_time=end_time,
             model_parameters=prompt_info.model_parameters,
             provider_info=prompt_info.provider_info,
+            usage=usage,
+            api_style=api_style
         )
 
 
@@ -138,6 +157,15 @@ class Recordings:
                 "trace_id": record_payload.trace_info.trace_id
             }
 
+        if record_payload.call_info.usage is not None:
+            record_api_payload['call_info']['usage'] = {
+                "prompt_tokens": record_payload.call_info.usage.prompt_tokens,
+                "completion_tokens": record_payload.call_info.usage.completion_tokens,
+            }
+
+        if record_payload.call_info.api_style is not None:
+            record_api_payload['call_info']['api_style'] = record_payload.call_info.api_style
+
         try:
             recorded_response = api_support.post_raw(
                 api_key=self.call_support.freeplay_api_key,
@@ -163,7 +191,6 @@ class Recordings:
 
             raise FreeplayError(message) from e
 
-
     def update(self, record_update_payload: RecordUpdatePayload) -> RecordResponse:  # type: ignore
         record_update_api_payload: Dict[str, Any] = {
             "new_messages": record_update_payload.new_messages,

{freeplay-0.3.14.dist-info → freeplay-0.3.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: freeplay
-Version: 0.3.14
+Version: 0.3.16
 Summary: 
 License: MIT
 Author: FreePlay Engineering

{freeplay-0.3.14.dist-info → freeplay-0.3.16.dist-info}/RECORD

@@ -1,4 +1,4 @@
-freeplay/__init__.py,sha256=oseuUqIVAi-2a_ns4ZbbFqkZez6KGGwI6fPkA0AKt6I,374
+freeplay/__init__.py,sha256=fipXoYuxUtsy6mxshut2ujWJh4lVbQ5v_xQ-33TPC-g,406
 freeplay/api_support.py,sha256=Kn2x3g6yloHQl3NwFRjbZE9BnIh7d1sgwGwC0mHuvw4,2483
 freeplay/errors.py,sha256=vwotUBldxDzREZOmLUeoiDoZjcvDwgH1AMwKBLhLooE,807
 freeplay/freeplay.py,sha256=J04-erDD6rI2SAje_Nsf3x5Qx-Z6p8gQvGrMRHFWoD4,1602
@@ -9,14 +9,14 @@ freeplay/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 freeplay/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 freeplay/resources/customer_feedback.py,sha256=bw8MfEOKbGgn4FOyvcADrcs9GhcpNXNTgxKjBjIzywE,899
 freeplay/resources/prompts.py,sha256=NZi4K6oGnbSgw_i0NFssSqRNonl6Ov8eGPFFbZ6O5aI,22185
-freeplay/resources/recordings.py,sha256=k_ZQ-9YYeIcaSkdRFaELJF1dAkomdtNEcxbqDpeLXZU,7615
+freeplay/resources/recordings.py,sha256=uslhoKWsXq51sAbF1lBs2uGL14qvBGc3dM8ZL6-H_04,8488
 freeplay/resources/sessions.py,sha256=Qz5v7VOf1DmQTd1wCOFXnrizlW5WFJT5V8-pq22Ifvg,2793
 freeplay/resources/test_cases.py,sha256=nXL_976RwSJDT6OWDM4GEzbcOzcGkJ9ulvb0XOzCRDM,2240
 freeplay/resources/test_runs.py,sha256=Tp2N-odInT5XEEWrEsVhdgfnsclOE8n92_C8gTwO2MI,2623
 freeplay/support.py,sha256=we_FEtxcqc-8R0uOWy8p0nX0pHUbs-ulw7TC5NarlX4,11091
 freeplay/utils.py,sha256=Xvt4mNLXLL7E6MI2hTuDLV5cl5Y83DgdjCZSyDGMjR0,3187
-freeplay-0.3.14.dist-info/LICENSE,sha256=_jzIw45hB1XHGxiQ8leZ0GH_X7bR_a8qgxaqnHbCUOo,1064
-freeplay-0.3.14.dist-info/METADATA,sha256=attHahy983-M4iFtaU-00Uzf_dsVVTSOAhgny3lpxk4,1654
-freeplay-0.3.14.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-freeplay-0.3.14.dist-info/entry_points.txt,sha256=32s3rf2UUCqiJT4jnClEXZhdXlvl30uwpcxz-Gsy4UU,54
-freeplay-0.3.14.dist-info/RECORD,,
+freeplay-0.3.16.dist-info/LICENSE,sha256=_jzIw45hB1XHGxiQ8leZ0GH_X7bR_a8qgxaqnHbCUOo,1064
+freeplay-0.3.16.dist-info/METADATA,sha256=-HV1aJ7HGVIDEvjVoa7zP92wujEBuMiCK0vcNdhZNCY,1654
+freeplay-0.3.16.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+freeplay-0.3.16.dist-info/entry_points.txt,sha256=32s3rf2UUCqiJT4jnClEXZhdXlvl30uwpcxz-Gsy4UU,54
+freeplay-0.3.16.dist-info/RECORD,,

{freeplay-0.3.14.dist-info → freeplay-0.3.16.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.1
+Generator: poetry-core 2.1.2
 Root-Is-Purelib: true
 Tag: py3-none-any