Medapp / models.py
mgbam's picture
Update models.py
5090b92 verified
raw
history blame
682 Bytes
from openai import OpenAI
import os
from config import OPENAI_API_KEY, OPENAI_DEFAULT_MODEL
# Initialize OpenAI
client = OpenAI(api_key=OPENAI_API_KEY)
def chat_with_openai(system_prompt, user_prompt, model=OPENAI_DEFAULT_MODEL, temperature=0.3):
"""
Orchestrates a conversation with OpenAI's Chat API,
combining system + user messages for a final answer.
"""
response = client.chat.completions.create(
model=model,
messages=[
{"role": "system", "content": system_prompt},
{"role": "user", "content": user_prompt}
],
temperature=temperature,
)
return response.choices[0].message.content.strip()