rerank_example.py
1 """ 2 Rerank Capabilities Example 3 4 Demonstrates document reranking using PraisonAI capabilities. 5 """ 6 7 from praisonai.capabilities import rerank 8 9 print("=== Document Reranking ===") 10 try: 11 documents = [ 12 "Python is a programming language", 13 "Machine learning uses algorithms", 14 "Python is great for data science", 15 "JavaScript runs in browsers" 16 ] 17 18 results = rerank( 19 query="Python programming", 20 documents=documents, 21 model="cohere/rerank-english-v3.0" 22 ) 23 24 print(f"Query: 'Python programming'") 25 print(f"Reranked results:") 26 for r in results: 27 print(f" Score {r.get('relevance_score', 0):.4f}: {r.get('document', {}).get('text', '')[:50]}") 28 except Exception as e: 29 print(f"Note: Reranking requires Cohere API key") 30 print(f"Error: {e}") 31 32 print("\nSee CLI: praisonai rerank <query> --documents 'doc1' 'doc2'")