edsl 0.1.55__tar.gz → 0.1.57__tar.gz

{edsl-0.1.55 → edsl-0.1.57}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: edsl
-Version: 0.1.55
+Version: 0.1.57
 Summary: Create and analyze LLM-based surveys
 Home-page: https://www.expectedparrot.com/
 License: MIT
@@ -23,7 +23,7 @@ Requires-Dist: azure-ai-inference (>=1.0.0b3,<2.0.0)
 Requires-Dist: black[jupyter] (>=24.4.2,<25.0.0)
 Requires-Dist: boto3 (>=1.34.161,<2.0.0)
 Requires-Dist: google-generativeai (>=0.8.2,<0.9.0)
-Requires-Dist: groq (>=0.9.0,<0.10.0)
+Requires-Dist: groq (==0.23.1)
 Requires-Dist: jinja2 (>=3.1.2,<4.0.0)
 Requires-Dist: json-repair (>=0.28.4,<0.29.0)
 Requires-Dist: jupyter (>=1.0.0,<2.0.0)

edsl-0.1.57/edsl/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.1.57"

{edsl-0.1.55 → edsl-0.1.57}/edsl/base/data_transfer_models.py

@@ -1,10 +1,11 @@
 from collections import UserDict
-from typing import NamedTuple, Dict, Optional, Any
+from typing import NamedTuple, Dict, Optional, Any, Union
 from dataclasses import dataclass, fields
 
 
 class ModelInputs(NamedTuple):
     "This is what was send by the agent to the model"
+
     user_prompt: str
     system_prompt: str
     encoded_image: Optional[str] = None
@@ -12,6 +13,7 @@ class ModelInputs(NamedTuple):
 
 class EDSLOutput(NamedTuple):
     "This is the edsl dictionary that is returned by the model"
+
     answer: Any
     generated_tokens: str
     comment: Optional[str] = None
@@ -19,11 +21,16 @@ class EDSLOutput(NamedTuple):
 
 class ModelResponse(NamedTuple):
     "This is the metadata that is returned by the model and includes info about the cache"
+
     response: dict
     cache_used: bool
     cache_key: str
     cached_response: Optional[Dict[str, Any]] = None
-    cost: Optional[float] = None
+    input_tokens: Optional[int] = None
+    output_tokens: Optional[int] = None
+    input_price_per_million_tokens: Optional[float] = None
+    output_price_per_million_tokens: Optional[float] = None
+    total_cost: Optional[Union[float, str]] = None
 
 
 class AgentResponseDict(NamedTuple):
@@ -44,7 +51,11 @@ class EDSLResultObjectInput(NamedTuple):
     comment: str
     validated: bool = False
     exception_occurred: Exception = None
-    cost: Optional[float] = None
+    input_tokens: Optional[int] = None
+    output_tokens: Optional[int] = None
+    input_price_per_million_tokens: Optional[float] = None
+    output_price_per_million_tokens: Optional[float] = None
+    total_cost: Optional[Union[float, str]] = None
 
 
 @dataclass
@@ -111,4 +122,4 @@ class Answers(UserDict):
 if __name__ == "__main__":
     import doctest
 
-    doctest.testmod()
+    doctest.testmod()

{edsl-0.1.55 → edsl-0.1.57}/edsl/coop/coop.py

@@ -1256,13 +1256,13 @@ class Coop(CoopFunctionsMixin):
         self, input: Union["Jobs", "Survey"], iterations: int = 1
     ) -> int:
         """
-        Get the cost of a remote inference job.
+        Get the estimated cost in credits of a remote inference job.
 
         :param input: The EDSL job to send to the server.
 
         >>> job = Jobs.example()
         >>> coop.remote_inference_cost(input=job)
-        {'credits': 0.77, 'usd': 0.0076950000000000005}
+        {'credits_hold': 0.77, 'usd': 0.0076950000000000005}
         """
         from ..jobs import Jobs
         from ..surveys import Survey
@@ -1290,7 +1290,7 @@ class Coop(CoopFunctionsMixin):
         self._resolve_server_response(response)
         response_json = response.json()
         return {
-            "credits": response_json.get("cost_in_credits"),
+            "credits_hold": response_json.get("cost_in_credits"),
             "usd": response_json.get("cost_in_usd"),
         }