debug_api.py

#!/usr/bin/env python
"""
Debug script for testing API connectivity and question generation
"""

import requests
import json
import time
import sys

def test_api_connection():
    """Test basic connection to the qewertyy.dev API"""
    print("Testing connection to qewertyy.dev API...")
    
    try:
        api_url = "https://api.qewertyy.dev/models"
        payload = {
            "model_id": 5,  # GPT-3.5 Turbo
            "messages": [
                {
                    "role": "user",
                    "content": "Hello, this is a test message. Please respond with 'API connection successful'."
                }
            ]
        }
        headers = {"Content-Type": "application/json"}
        
        print("Sending test request...")
        start_time = time.time()
        response = requests.post(api_url, json=payload, headers=headers, timeout=10)
        end_time = time.time()
        
        print(f"Response received in {end_time - start_time:.2f} seconds")
        print(f"Status code: {response.status_code}")
        
        if response.status_code == 200:
            print("API connection successful!")
            try:
                data = response.json()
                content = data.get("content", "No content received")
                print(f"Response content: {content}")
                return True
            except json.JSONDecodeError:
                print("Error: Could not parse JSON response")
                print(f"Raw response: {response.text[:500]}")
                return False
        else:
            print(f"API request failed with status code: {response.status_code}")
            print(f"Response: {response.text[:500]}")
            return False
            
    except Exception as e:
        print(f"Error connecting to API: {str(e)}")
        return False

def test_question_generation():
    """Test question generation with a small sample text"""
    print("\nTesting question generation API...")
    
    sample_text = """
    The Python programming language was created by Guido van Rossum in the late 1980s. 
    It is named after the British comedy group Monty Python. 
    Python is known for its simple syntax and readability, making it a popular choice for beginners.
    Python supports multiple programming paradigms, including procedural, object-oriented, and functional programming.
    """
    
    try:
        api_url = "https://api.qewertyy.dev/models"
        
        # Create a simplified prompt for testing
        prompt = f"""Generate 2 questions and answers based on this text:

{sample_text}

Format each as "Question: [question]" and "Answer: [answer]".
"""
        
        payload = {
            "model_id": 5,  # GPT-3.5 Turbo
            "messages": [
                {
                    "role": "user",
                    "content": prompt
                }
            ]
        }
        headers = {"Content-Type": "application/json"}
        
        print("Sending question generation request...")
        start_time = time.time()
        response = requests.post(api_url, json=payload, headers=headers, timeout=20)
        end_time = time.time()
        
        print(f"Response received in {end_time - start_time:.2f} seconds")
        print(f"Status code: {response.status_code}")
        
        if response.status_code == 200:
            try:
                data = response.json()
                content = data.get("content", "No content received")
                
                if not content:
                    print("Error: Empty content received")
                    return False
                    
                print("\nGenerated questions and answers:")
                print("-" * 50)
                print(content)
                print("-" * 50)
                
                if "Question:" in content and "Answer:" in content:
                    print("\nSuccess! The API generated questions and answers correctly.")
                    return True
                else:
                    print("\nWarning: The API returned content but it doesn't contain the expected Question/Answer format.")
                    return False
                    
            except json.JSONDecodeError:
                print("Error: Could not parse JSON response")
                print(f"Raw response: {response.text[:500]}")
                return False
        else:
            print(f"API request failed with status code: {response.status_code}")
            print(f"Response: {response.text[:500]}")
            return False
            
    except Exception as e:
        print(f"Error testing question generation: {str(e)}")
        return False

if __name__ == "__main__":
    print("=" * 50)
    print("ASKPDF-AI API Connection Tester")
    print("=" * 50)
    
    connection_result = test_api_connection()
    
    if connection_result:
        print("\nBasic API connection successful, testing question generation...")
        question_result = test_question_generation()
        
        if question_result:
            print("\nAll tests passed! The API is working correctly for question generation.")
            sys.exit(0)
        else:
            print("\nQuestion generation test failed. The API may be working but not returning the expected format.")
            sys.exit(1)
    else:
        print("\nFailed to connect to the API. Please check your internet connection and try again.")
        sys.exit(1)