Spaces:
Sleeping
Sleeping
File size: 1,003 Bytes
05ebfc6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
beeRoot - Exemplo de uso da API
"""
import requests
# URL da sua abelha
BEEROOT_URL = "https://seu-usuario-sua-abelha.hf.space"
def search(query, top_k=10):
"""Busca documentos"""
response = requests.post(
f"{BEEROOT_URL}/search/embedding",
json={"query": query, "top_k": top_k}
)
return response.json()
# Exemplo 1: Busca simples
print("🔍 Busca: FGTS")
results = search("FGTS", top_k=5)
for doc in results['results']:
print(f"\nID: {doc['id']} | Score: {doc['score']:.4f}")
print(f"Ementa: {doc['ementa'][:200]}...")
# Exemplo 2: Múltiplas queries
queries = [
"FGTS correção monetária",
"aposentadoria servidor público",
"execução fiscal prescrição"
]
for query in queries:
print(f"\n{'='*60}")
print(f"Query: {query}")
print('='*60)
results = search(query, top_k=3)
for doc in results['results']:
print(f" • ID {doc['id']}: {doc['score']:.2f}")
|