File size: 737 Bytes
81a818a
6350527
ec83649
c3ff1f2
81a818a
ec83649
6350527
ec83649
6350527
ec83649
6350527
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from openai import OpenAI
from config import OPENAI_API_KEY

# Initialize OpenAI client
client = OpenAI(api_key=OPENAI_API_KEY)

def query_openai_text(system_message, user_message, model="gpt-4o", temperature=0.3):
    """
    Query OpenAI's Chat API for a response to medical questions or tasks.
    """
    try:
        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()
    except Exception as e:
        return f"Error querying OpenAI API: {str(e)}"