File size: 653 Bytes
81a818a
 
ec83649
c3ff1f2
81a818a
ec83649
c3ff1f2
ec83649
c3ff1f2
ec83649
81a818a
 
 
c3ff1f2
 
81a818a
 
 
5090b92
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from openai import OpenAI
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):
    """
    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()