Skip to content

Commit 21d32e4

Browse files
authored
Update models.py
1 parent b22b397 commit 21d32e4

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

sweagent/agent/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -746,7 +746,7 @@ def _single_query(self, messages: list[dict[str, str]], n: int | None = None, te
746746
if fallback_cost == 0:
747747
input_tokens = litellm.utils.token_counter(messages=messages_no_cache_control, model=self.config.name)
748748
output_tokens = sum(litellm.utils.token_counter(text=choice.message.content or "", model=self.config.name) for choice in response.choices)
749-
fallback_cost = (input_tokens * 1.0 + output_tokens * 3.0) / 1000000
749+
fallback_cost = (input_tokens * 0.28 + output_tokens * 0.42) / 1_000_000
750750
self.logger.info(f"Using estimated DeepSeek pricing for cost calculation: ${fallback_cost:.6f}")
751751
except Exception as fallback_error:
752752
self.logger.debug(f"Fallback cost calculation also failed: {fallback_error}")

0 commit comments

Comments
 (0)