import requests
import json

# Test query with enable_rerank=True
url = "http://localhost:3015/query"
headers = {
    "Content-Type": "application/json",
    "X-API-Key": "jleu1212"
}
data = {
    "query": "test query",
    "enable_rerank": True,
    "only_need_context": True  # Get only context to see what's retrieved
}

try:
    response = requests.post(url, headers=headers, json=data, timeout=10)
    print(f"Status Code: {response.status_code}")
    print(f"Response: {response.text}")
    
    if response.status_code == 200:
        result = response.json()
        print(f"\nQuery successful")
        print(f"Response length: {len(result.get('response', ''))}")
        
        # Try to parse if it's JSON
        try:
            parsed = json.loads(result.get('response', '{}'))
            print(f"Parsed response type: {type(parsed)}")
            if isinstance(parsed, dict):
                print(f"Has metadata: {'metadata' in parsed}")
                if 'metadata' in parsed:
                    print(f"Metadata keys: {list(parsed['metadata'].keys())}")
        except:
            print("Response is not JSON")
except Exception as e:
    print(f"Error: {e}")

# Also test without rerank for comparison
print("\n" + "="*50)
print("Testing without rerank:")
data_no_rerank = {
    "query": "test query",
    "enable_rerank": False,
    "only_need_context": True
}

try:
    response = requests.post(url, headers=headers, json=data_no_rerank, timeout=10)
    print(f"Status Code: {response.status_code}")
    print(f"Response length: {len(response.text)}")
except Exception as e:
    print(f"Error: {e}")