cura/chatbot.py

from openai import OpenAI

def ask_question(query: str):
    client = OpenAI()
    try:
        response = client.chat.completions.create(
            model="gpt-4o-mini",
            messages=[
                {"role": "system", "content": "You are a code assistant."},
                {"role": "user", "content": query},
            ],
            max_tokens=1024,
        )
        return response.choices[0].message.content
    except Exception as e:
        return f"Failed to ask question: {e}"