RomanNum3ral
/
cai
mirror of https://github.com/aliasrobotics/cai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

undo changes in openai_responses.py and leave it as it was originally in v0.4.0

This commit is contained in:
Lidia 2025-04-11 09:53:51 +02:00
parent 23d2a0f22d
commit 3000cdfea0
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/cai/sdk/agents/models/openai_responses.py
View File

@ -28,7 +28,6 @@ from ..tracing import SpanError, response_span
from ..usage import Usage
from ..version import __version__
from .interface import Model, ModelTracing
from cai.util import calculate_model_cost
if TYPE_CHECKING:
from ..model_settings import ModelSettings
@ -154,8 +153,8 @@ class OpenAIResponsesModel(Model):
total_input_tokens=getattr(self, 'total_input_tokens', 0),
total_output_tokens=getattr(self, 'total_output_tokens', 0),
total_reasoning_tokens=getattr(self, 'total_reasoning_tokens', 0),
interaction_cost=calculate_model_cost(str(self.model), usage.input_tokens, usage.output_tokens),
total_cost=calculate_model_cost(str(self.model), getattr(self, 'total_input_tokens', 0), getattr(self, 'total_output_tokens', 0)),
interaction_cost=None,
total_cost=None,
)
# Update token totals
@ -255,8 +254,8 @@ class OpenAIResponsesModel(Model):
total_input_tokens=getattr(self, 'total_input_tokens', 0),
total_output_tokens=getattr(self, 'total_output_tokens', 0),
total_reasoning_tokens=getattr(self, 'total_reasoning_tokens', 0),
interaction_cost=calculate_model_cost(str(self.model), final_response.usage.input_tokens, final_response.usage.output_tokens),
total_cost=calculate_model_cost(str(self.model), getattr(self, 'total_input_tokens', 0), getattr(self, 'total_output_tokens', 0)),
interaction_cost=None,
total_cost=None,
)
# Update token totals