import requests
import json

base_url = 'http://localhost:3015'
headers = {'X-API-Key': 'jleu1212'}

print('=== TESTING SEARCH WITHOUT LLM (NAIVE MODE) ===')
search_payload = {
    'query': 'Safety Precautions Minimum Safe Distance',
    'top_k': 5,
    'mode': 'naive'
}
response = requests.post(f'{base_url}/api/search', headers=headers, json=search_payload)
print(f'Search status: {response.status_code}')
if response.status_code == 200:
    results = response.json()
    print(f'Found {len(results.get("chunks", []))} chunks')
    for i, chunk in enumerate(results.get('chunks', [])[:3]):
        print(f'  Chunk {i+1}: {chunk.get("content", "")[:100]}...')
else:
    print(f'Search error: {response.text}')

print('\n=== TESTING DATA ENDPOINT (NO LLM) ===')
data_payload = {
    'query': 'Safety Precautions',
    'mode': 'naive',
    'only_need_context': True
}
response = requests.post(f'{base_url}/query/data', headers=headers, json=data_payload)
print(f'Data query status: {response.status_code}')
if response.status_code == 200:
    results = response.json()
    print(f'Found {len(results.get("chunks", []))} chunks')
    for i, chunk in enumerate(results.get('chunks', [])[:3]):
        print(f'  Chunk {i+1}: {chunk.get("content", "")[:100]}...')
else:
    print(f'Data query error: {response.text}')

print('\n=== TESTING SIMPLE SEARCH ENDPOINT ===')
simple_search_payload = {
    'query': 'Minimum Safe Distance',
    'top_k': 3,
    'mode': 'naive'
}
response = requests.post(f'{base_url}/search', headers=headers, json=simple_search_payload)
print(f'Simple search status: {response.status_code}')
if response.status_code == 200:
    results = response.json()
    print(f'Found {len(results.get("results", []))} results')
    for i, result in enumerate(results.get('results', [])[:3]):
        print(f'  Result {i+1}: {result.get("content", "")[:100]}...')
else:
    print(f'Simple search error: {response.text}')