from typing import Any
import requests
API_URL: str = "https://openrouter.ai/api/v1/chat/completions"
def ask_model(prompt: str, token: str, model_name: str = "google/gemini-3.8-flash") -> str:
"""Отправляет запрос к модели и возвращает строковый ответ."""
headers: dict[str, str] = {
"Authorization": f"Bearer {token}",
"Content-Type": "application/json"
}
payload: dict[str, Any] = {
"model": model_name,
"messages": [{"role": "user", "content": prompt}]
}
response: requests.Response = requests.post(API_URL, headers=headers, json=payload)
data: dict[str, Any] = response.json()
result: str = data["choices"][0]["message"]["content"]
return result
# Вызов с контролем типов
user_token: str = "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"
answer: str = ask_model(prompt="Сколько планет в Солнечной системе?", token=user_token)
print(f"Ответ: {answer}")