# test_models.py
import os
import sys
from src.llm_providers import test_model, get_available_openrouter_models

# Вивести всі доступні моделі з OpenRouter (опціонально)
print("Спроба отримати список моделей з OpenRouter API...")
models = get_available_openrouter_models()
if models:
    print("Доступні моделі OpenRouter:")
    for idx, model in enumerate(models, 1):
        print(f"{idx}. {model}")
else:
    print("Не вдалося отримати список моделей або список порожній")

# Тестування конкретних моделей
models_to_test = [
    ("openrouter", "google/gemini-2.5-pro-exp-03-25:free")
    # ("openrouter", "mistralai/mistral-large")
    # ("openrouter", "openrouter/quasar-alpha"),
    # ("openrouter", "deepseek/deepseek-r1-zero:free"),
    # ("openrouter", "google/gemini-2.0-flash-thinking-exp:free"),
    
    # ("anthropic", "claude-3-7-sonnet-20250219")
]

print("\nПочаток тестування моделей:")
print("="*50)

for provider, model in models_to_test:
    print(f"\nТестування {provider}/{model}...")
    success = test_model(provider, model)
    status = "✅ УСПІШНО" if success else "❌ ПОМИЛКА"
    print(f"{status} - {provider}/{model}")

print("="*50)
print("Тестування завершено!")