ykl7 commited on
Commit
da4d7e6
·
1 Parent(s): 0993480

gpt4o debug

Browse files
Files changed (1) hide show
  1. llm_reasoner.py +2 -1
llm_reasoner.py CHANGED
@@ -28,7 +28,8 @@ class LLMReasoner():
28
  self.frequency_penalty = 0.0 if "frequency_penalty" not in options else options["frequency_penalty"]
29
  self.presence_penalty = 0.0 if "presence_penalty" not in options else options["presence_penalty"]
30
 
31
- def make_openai_chat_completions_api_call(self, prompt):
 
32
  try:
33
  if "gpt-4o" in self.model_name:
34
  response = self.client.chat.completions.create(
 
28
  self.frequency_penalty = 0.0 if "frequency_penalty" not in options else options["frequency_penalty"]
29
  self.presence_penalty = 0.0 if "presence_penalty" not in options else options["presence_penalty"]
30
 
31
+ def make_openai_chat_completions_api_call(self, message):
32
+ prompt = [{"role": "user", "content": message}]
33
  try:
34
  if "gpt-4o" in self.model_name:
35
  response = self.client.chat.completions.create(