/ examples / capabilities / rerank_example.py
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'")