|
|
from openai import OpenAI |
|
|
from config import OPENAI_API_KEY |
|
|
|
|
|
|
|
|
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)}" |
|
|
|
