Update openai_models.py

kyegomez · web-flow · commit 7694c5a94035 · 2024-02-23T10:34:10.000-08:00
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]

Original file line number	Diff line number	Diff line change
`@@ -169,7 +169,7 @@ def evaluate_states(self, states, initial_prompt):`
`169`	`169`	`f" state of reasoning achieveing {initial_prompt} and become"`
`170`	`170`	`" very pessimistic very NOTHING ELSE"`
`171`	`171`	`)`
`172`		`- response = self.openai_api_call_handler(prompt, 50, 1)`
	`172`	`+ response = self.model(prompt)`
`173`	`173`	`print(f"state response: {response}")`
`174`	`174`	`best_state_text = self.openai_choice2text_handler(`
`175`	`175`	`response.choices[0]`