urf
This commit is contained in:
6
llm.py
6
llm.py
@@ -36,9 +36,9 @@ class LLMClient:
|
|||||||
)
|
)
|
||||||
output = response["response"]
|
output = response["response"]
|
||||||
elif self.PROVIDER == "openai":
|
elif self.PROVIDER == "openai":
|
||||||
response = self.openai_client.chat.completions.create(
|
response = self.openai_client.responses.create(
|
||||||
model="gpt-4o-mini",
|
model="gpt-4o-mini",
|
||||||
messages=[
|
input=[
|
||||||
{
|
{
|
||||||
"role": "system",
|
"role": "system",
|
||||||
"content": system_prompt,
|
"content": system_prompt,
|
||||||
@@ -49,7 +49,7 @@ class LLMClient:
|
|||||||
},
|
},
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
output = response.choices[0].message.content
|
output = response.output_text
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|||||||
Reference in New Issue
Block a user