Update openai_models.py

kyegomez · Feb 23, 2024 · 7694c5a · 7694c5a
1 parent 9be8b0f
commit 7694c5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tree_of_thoughts/openai_models.py b/tree_of_thoughts/openai_models.py
@@ -169,7 +169,7 @@ def evaluate_states(self, states, initial_prompt):
                 f" state of reasoning achieveing {initial_prompt} and become"
                 " very pessimistic very NOTHING ELSE"
             )
-            response = self.openai_api_call_handler(prompt, 50, 1)
+            response = self.model(prompt)
             print(f"state response: {response}")
             best_state_text = self.openai_choice2text_handler(
                 response.choices[0]