File size: 643 Bytes
81a818a ec83649 81a818a ec83649 81a818a ec83649 81a818a ec83649 81a818a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from openai import OpenAI
import os
from config import OPENAI_API_KEY, OPENAI_DEFAULT_MODEL
# Initialize OpenAI client
client = OpenAI(api_key=OPENAI_API_KEY)
def chat_with_openai(system_message, user_message, model=OPENAI_DEFAULT_MODEL, temperature=0.3):
"""
Chat with OpenAI's model using a system message and user message.
"""
response = client.chat.completions.create(
model=model,
messages=[
{"role": "system", "content": system_message},
{"role": "user", "content": user_message},
],
temperature=temperature,
)
return response.choices[0].message.content
|