"""
Test Ollama API with different parameters
"""
import requests
import json
import urllib3

# Disable SSL warnings
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)

API_URL = "https://ollama_pjapi.theaken.com"

print("=" * 60)
print("Testing Ollama Chat Completion - Variant Tests")
print("=" * 60)
print()

# Test 1: Using qwen2.5:72b (actual available model)
print("Test 1: Using qwen2.5:72b model...")
try:
    chat_request = {
        "model": "qwen2.5:72b",
        "messages": [
            {"role": "user", "content": "Say hello in Chinese."}
        ]
    }

    response = requests.post(
        f"{API_URL}/v1/chat/completions",
        json=chat_request,
        headers={'Content-Type': 'application/json'},
        timeout=60,
        verify=False
    )

    print(f"Status Code: {response.status_code}")

    if response.status_code == 200:
        result = response.json()
        text = result['choices'][0]['message']['content']
        print(f"Success! Response: {text}")
    else:
        print(f"Error: {response.text}")
except Exception as e:
    print(f"Error: {str(e)}")

print()

# Test 2: Try deepseek-chat model
print("Test 2: Using deepseek-chat model...")
try:
    chat_request = {
        "model": "deepseek-chat",
        "messages": [
            {"role": "user", "content": "Say hello in Chinese."}
        ]
    }

    response = requests.post(
        f"{API_URL}/v1/chat/completions",
        json=chat_request,
        headers={'Content-Type': 'application/json'},
        timeout=60,
        verify=False
    )

    print(f"Status Code: {response.status_code}")

    if response.status_code == 200:
        result = response.json()
        text = result['choices'][0]['message']['content']
        print(f"Success! Response: {text}")
    else:
        print(f"Error: {response.text}")
except Exception as e:
    print(f"Error: {str(e)}")

print()
print("=" * 60)