goose-py 0.2.1__tar.gz → 0.2.2__tar.gz

{goose_py-0.2.1 → goose_py-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: goose-py
-Version: 0.2.1
+Version: 0.2.2
 Summary: A tool for AI workflows based on human-computer collaboration and structured output.
 Home-page: https://github.com/chelle-ai/goose
 Keywords: ai,yaml,configuration,llm

{goose_py-0.2.1 → goose_py-0.2.2}/goose/agent.py

@@ -125,14 +125,20 @@ class AgentResponse[R: BaseModel](BaseModel):
     def duration_ms(self) -> int:
         return int((self.end_time - self.start_time).total_seconds() * 1000)
 
+    @computed_field
+    @property
+    def input_cost(self) -> float:
+        return self.INPUT_CENTS_PER_MILLION_TOKENS[self.model] * self.input_tokens
+
+    @computed_field
+    @property
+    def output_cost(self) -> float:
+        return self.OUTPUT_CENTS_PER_MILLION_TOKENS[self.model] * self.output_tokens
+
     @computed_field
     @property
     def total_cost(self) -> float:
-        input_cost = self.INPUT_CENTS_PER_MILLION_TOKENS[self.model] * self.input_tokens
-        output_cost = (
-            self.OUTPUT_CENTS_PER_MILLION_TOKENS[self.model] * self.output_tokens
-        )
-        return input_cost + output_cost
+        return self.input_cost + self.output_cost
 
 
 class Agent:

{goose_py-0.2.1 → goose_py-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "goose-py"
-version = "0.2.1"
+version = "0.2.2"
 description = "A tool for AI workflows based on human-computer collaboration and structured output."
 authors = [
     "Nash Taylor <nash@chelle.ai>",