test_api.py

import asyncio
import httpx
import json

async def test_suggest():
    """Test the suggestion endpoint"""
    url = "http://localhost:8001/api/suggest"  # Note the port change
    
    payload = {
        "current_text": "The JavaScript event loop is fundamental to understanding asynchronous programming.",
        "context": ["Modern web applications rely heavily on non-blocking operations."]
    }
    
    # Increased timeout to 60 seconds
    async with httpx.AsyncClient(timeout=60.0) as client:
        print("Sending request to API...")
        try:
            response = await client.post(url, json=payload)
            
            print(f"Status: {response.status_code}")
            if response.status_code == 200:
                data = response.json()
                print("\nSuggestions:")
                for i, suggestion in enumerate(data["suggestions"], 1):
                    print(f"{i}. {suggestion}")
                
                print("\nLatencies:")
                for key, value in data["latencies"].items():
                    print(f"{key}: {value:.2f}ms")
                
                print("\nSource chunks used (first 2):")
                for i, chunk in enumerate(data["source_chunks"][:2], 1):
                    print(f"{i}. {chunk['text'][:100]}...")
            else:
                print(f"Error: {response.text}")
        except Exception as e:
            print(f"Request failed: {str(e)}")

if __name__ == "__main__":
    asyncio.run(test_suggest())