"""Prosta rozmowa wieloturowa z ograniczoną historią."""
import os

from openai import OpenAI

client = OpenAI(
    api_key=os.environ["JANUSZ_API_KEY"],
    base_url=os.getenv("JANUSZ_BASE_URL", "https://janusz.fabryka.ai/v1"),
    timeout=30.0,
)
messages = [{"role": "system", "content": "Jesteś pomocnym doradcą. Odpowiadaj po polsku."}]


def ask(prompt: str) -> str:
    messages.append({"role": "user", "content": prompt})
    response = client.chat.completions.create(
        model="bielik-11b-v3",
        messages=messages[-12:],
        temperature=0.2,
        max_tokens=300,
    )
    answer = response.choices[0].message.content or ""
    messages.append({"role": "assistant", "content": answer})
    return answer


if __name__ == "__main__":
    print(ask("Zapamiętaj, że mój projekt nazywa się Atlas."))
    print(ask("Jak nazywa się mój projekt?"))