import os
from openai import OpenAI

# Inicializace klienta pro LiteLLM Proxy
client = OpenAI(
    base_url="http://127.0.0.1:4100",
    api_key=os.environ.get("LITELLM_MASTER_KEY", "sk-cb26-local")
)

try:
    response = client.chat.completions.create(
        model=os.environ.get("CB26_DEFAULT_MODEL", "gemini-3.6-flash"),
        messages=[{"role": "user", "content": "Ahoj, vygeneruj jedno krátké slovo na potvrzení funkčnosti."}],
        max_tokens=200
    )
    print("Odpověď z API:", response.choices[0].message.content)
except Exception as e:
    print("Chyba při volání API:", e)