#!/usr/bin/env python3
"""
Simple test to validate ResetData API integration
"""

import os
import sys

# Add the local library path
sys.path.append('/home/tony/.local/lib/python3.12/site-packages')

try:
    from openai import OpenAI
    
    api_key = os.getenv('RESETDATA_API_KEY')
    if not api_key:
        print("❌ RESETDATA_API_KEY not set")
        sys.exit(1)
    
    print("🌐 Testing ResetData API directly...")
    print(f"📋 Using API key: {api_key[:20]}...")
    
    client = OpenAI(
        base_url="https://models.au-syd.resetdata.ai/v1",
        api_key=api_key
    )
    
    response = client.chat.completions.create(
        model="meta/llama-3.1-8b-instruct:ptu-9f3627a0-4909-4561-8996-272774e91fc8",
        messages=[
            {"role": "system", "content": "You are a helpful assistant."},
            {"role": "user", "content": "What is the capital of Australia? Answer briefly."}
        ],
        temperature=0.2,
        top_p=0.7,
        max_tokens=50
    )
    
    if response.choices and response.choices[0].message.content:
        print("✅ Direct ResetData API test successful!")
        print(f"📝 Response: {response.choices[0].message.content}")
        print("✨ ResetData integration working correctly!")
    else:
        print("❌ No valid response from ResetData API")
        sys.exit(1)
        
except Exception as e:
    print(f"❌ Direct ResetData API test failed: {e}")
    sys.exit(1)