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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: goose-py
-Version: 0.2.0
+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.0 → goose_py-0.2.2}/goose/agent.py

@@ -2,7 +2,7 @@ import base64
 import logging
 from datetime import datetime
 from enum import StrEnum
-from typing import Any, Callable, ClassVar, Literal, NotRequired, TypedDict
+from typing import Any, Awaitable, Callable, ClassVar, Literal, NotRequired, TypedDict
 
 from litellm import acompletion
 from pydantic import BaseModel, computed_field
@@ -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:
@@ -141,11 +147,11 @@ class Agent:
         *,
         flow_name: str,
         run_id: str,
-        logger: Callable[[AgentResponse[Any]], None] | None = None,
+        logger: Callable[[AgentResponse[Any]], Awaitable[None]] | None = None,
     ) -> None:
         self.flow_name = flow_name
         self.run_id = run_id
-        self.logger = logger or logging.info
+        self.logger = logger
 
     async def __call__[R: BaseModel](
         self,
@@ -192,5 +198,9 @@ class Agent:
             end_time=end_time,
         )
 
-        self.logger(agent_response)
+        if self.logger is not None:
+            await self.logger(agent_response)
+        else:
+            logging.info(agent_response.model_dump())
+
         return agent_response.response

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

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