Semantic Search Pack

from semantic_search_pack.tool import run

# Index a collection of support articles
result = run(
    action="index",
    documents=[
        {"id": "doc-1", "text": "To reset your password, click 'Forgot Password' on the login page and follow the email instructions.", "metadata": {"category": "account"}},
        {"id": "doc-2", "text": "Billing cycles run from the 1st to the last day of each month. Invoices are generated on the 2nd.", "metadata": {"category": "billing"}},
        {"id": "doc-3", "text": "Our API rate limit is 1000 requests per minute per API key. Contact support to request a limit increase.", "metadata": {"category": "api"}},
        {"id": "doc-4", "text": "Two-factor authentication can be enabled under Settings > Security. We support TOTP and SMS.", "metadata": {"category": "account"}},
        {"id": "doc-5", "text": "To cancel your subscription, go to Settings > Billing > Cancel Plan. Cancellation takes effect at period end.", "metadata": {"category": "billing"}}
    ],
    index_name="support-kb"
)
print(f"Indexed {result['document_count']} documents")

# Search by meaning
search_result = run(
    action="semantic_search",
    query="How do I change my password?",
    index_name="support-kb",
    top_k=3
)

for match in search_result["results"]:
    print(f"  [{match['score']:.3f}] {match['id']}: {match['text'][:80]}...")