"""Take a question plus relevant documents and write the final answer."""
from openai import OpenAI

import config
from retrieve import retrieve

client = OpenAI(api_key=config.OPENAI_API_KEY)


def answer(question):
    """Find relevant docs, then ask the model to answer using them."""
    context = "\n\n".join(retrieve(question))
    prompt = f"Use this context to answer.\n\nContext:\n{context}\n\nQuestion: {question}"
    resp = client.chat.completions.create(
        model=config.CHAT_MODEL,
        messages=[{"role": "user", "content": prompt}],
    )
    return resp.choices[0].message.content