41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
import requests
|
|
import json
|
|
|
|
def test_bee_search():
|
|
base_url = 'http://localhost:3015'
|
|
headers = {'X-API-Key': 'jleu1212', 'Content-Type': 'application/json'}
|
|
|
|
# Test search for bee-related content
|
|
queries = ['bee', 'photo of a bee', 'insect', 'image classification']
|
|
|
|
for query in queries:
|
|
print(f"\n🔍 Testing search for: '{query}'")
|
|
try:
|
|
response = requests.post(f'{base_url}/api/search', headers=headers, json={'query': query, 'top_k': 10}, timeout=20)
|
|
print(f'Status: {response.status_code}')
|
|
|
|
if response.status_code == 200:
|
|
results = response.json()
|
|
chunks = results.get('chunks', [])
|
|
entities = results.get('entities', [])
|
|
|
|
print(f'Chunks found: {len(chunks)}')
|
|
print(f'Entities found: {len(entities)}')
|
|
|
|
if entities:
|
|
print('Entities:')
|
|
for entity in entities:
|
|
print(f' - {entity.get("entity_name")}: {entity.get("entity_type")}')
|
|
|
|
if chunks:
|
|
print('First chunk content preview:')
|
|
if chunks[0].get('content'):
|
|
print(f' {chunks[0]["content"][:200]}...')
|
|
else:
|
|
print(f'Error: {response.text}')
|
|
|
|
except Exception as e:
|
|
print(f'Request failed: {e}')
|
|
|
|
if __name__ == '__main__':
|
|
test_bee_search() |