Fix type error in ApiManager

Pcyter · Aug 25, 2023 · c6e7d21 · c6e7d21
1 parent 6213558
commit c6e7d21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autogpt/llm/api_manager.py b/autogpt/llm/api_manager.py
@@ -45,7 +45,7 @@ def update_cost(self, prompt_tokens, completion_tokens, model):
         self.total_prompt_tokens += prompt_tokens
         self.total_completion_tokens += completion_tokens
         self.total_cost += prompt_tokens * model_info.prompt_token_cost / 1000
-        if issubclass(type(model_info), CompletionModelInfo):
+        if isinstance(model_info, CompletionModelInfo):
             self.total_cost += (
                 completion_tokens * model_info.completion_token_cost / 1000
             )