test_api.py

import requests
import json
import time

def test_rag_api():
    url = "http://localhost:8080/api/ask"
    
    payload = {
        "query": "TO BE FILLED"
    }
    
    headers = {
        "Content-Type": "application/json"
    }
    
    print(f"[{time.strftime('%X')}] Sending questions to server...")
    print(f"Query: {payload['query']}")
    print("-" * 50)
    
    try:
        start_time = time.time()
        response = requests.post(url, json=payload, headers=headers)
        end_time = time.time()
        
        if response.status_code == 200:
            result = response.json()
            print(f"[{time.strftime('%X')}] Answer came in! (Time: {end_time - start_time:.2f} sec)\n")
            print("Answer: ")
            print(result.get("answer", "No answer."))
        else:
            print(f"Error : {response.status_code}")
            print(response.text)
            
    except requests.exceptions.ConnectionError:
        print("Cannot connect to server")

if __name__ == "__main__":
    test_rag_api()