Skip to content

Commit c0fae36

Browse files
authored
Updated openai api
1 parent 0568239 commit c0fae36

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

bot-system/app/clients/llm/openai.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -48,13 +48,15 @@ async def generate_text(
4848
Generated text
4949
"""
5050
try:
51-
response = self.client.completions.create(
51+
response = self.client.chat.completions.create(
5252
model=self.model,
53-
prompt=prompt,
53+
messages=[
54+
{"role": "user", "content": prompt}
55+
],
5456
max_tokens=max_tokens,
5557
temperature=temperature,
5658
)
57-
raw_text = response.choices[0].text.strip()
59+
raw_text = response.choices[0].message.content.strip()
5860
return strip_think_tags(raw_text)
5961
except Exception as e:
60-
raise LLMError(f"OpenAI text generation error: {str(e)}")
62+
raise LLMError(f"OpenAI text generation error: {str(e)}")

0 commit comments

Comments
 (0)