pr cleanups

rminchev1 · Oct 26, 2023 · 59b68a2 · 59b68a2
1 parent ba3e6df
commit 59b68a2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/gpt_engineer/core/ai.py b/gpt_engineer/core/ai.py
@@ -107,7 +107,7 @@ def __init__(self, model_name="gpt-4", temperature=0.1, azure_endpoint=""):
         self.llm = self._create_chat_model()
         self.token_usage_log = TokenUsageLog(model_name)
 
-        logger.debug(f"Using model {self.model_name} with llm {self.llm}")
+        logger.debug(f"Using model {self.model_name}")
 
     def start(self, system: str, user: str, step_name: str) -> List[Message]:
         """

diff --git a/tests/test_ai.py b/tests/test_ai.py
@@ -58,7 +58,10 @@ def test_token_logging(monkeypatch):
 
     # act
     response_messages = ai.start("system prompt", "user prompt", "step name")
+    usageCostAfterStart = ai.token_usage_log.usage_cost()
     ai.next(response_messages, "next user prompt", step_name="step name")
+    usageCostAfterNext = ai.token_usage_log.usage_cost()
 
     # assert
-    assert ai.token_usage_log.usage_cost() > 0
+    assert usageCostAfterStart > 0
+    assert usageCostAfterNext > usageCostAfterStart