53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
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}') |