update Openai API call method (#27628)

Co-authored-by: 张兴言 <SENSETIME\[email protected]>
huggingface · Nov 22, 2023 · 8aca43b · 8aca43b
1 parent 7f6a804
commit 8aca43b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/tools/agents.py b/src/transformers/tools/agents.py
@@ -440,13 +440,13 @@ def generate_one(self, prompt, stop):
             return self._completion_generate([prompt], stop)[0]
 
     def _chat_generate(self, prompt, stop):
-        result = openai.ChatCompletion.create(
+        result = openai.chat.completions.create(
             model=self.model,
             messages=[{"role": "user", "content": prompt}],
             temperature=0,
             stop=stop,
         )
-        return result["choices"][0]["message"]["content"]
+        return result.choices[0].message.content
 
     def _completion_generate(self, prompts, stop):
         result = openai.Completion.create(