Skip to content

Commit 8aca43b

Browse files
Strive-for-excellence张兴言
andauthored
update Openai API call method (#27628)
Co-authored-by: 张兴言 <SENSETIME\[email protected]>
1 parent 7f6a804 commit 8aca43b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/transformers/tools/agents.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -440,13 +440,13 @@ def generate_one(self, prompt, stop):
440440
return self._completion_generate([prompt], stop)[0]
441441

442442
def _chat_generate(self, prompt, stop):
443-
result = openai.ChatCompletion.create(
443+
result = openai.chat.completions.create(
444444
model=self.model,
445445
messages=[{"role": "user", "content": prompt}],
446446
temperature=0,
447447
stop=stop,
448448
)
449-
return result["choices"][0]["message"]["content"]
449+
return result.choices[0].message.content
450450

451451
def _completion_generate(self, prompts, stop):
452452
result = openai.Completion.create(

0 commit comments

Comments
 (0)