|
from openai import OpenAI |
|
from config import OPENAI_API_KEY, OPENAI_DEFAULT_MODEL |
|
|
|
|
|
client = OpenAI(api_key=OPENAI_API_KEY) |
|
|
|
def chat_with_openai(system_message, user_message, model=OPENAI_DEFAULT_MODEL, temperature=0.3): |
|
""" |
|
Use OpenAI's Chat API to generate responses based on system and user messages. |
|
""" |
|
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.strip() |
|
|