context-ai / configs /compute_rag_vector_index_openai_contextual_reranked.yaml
chinmayjha's picture
Enhanced conversation analysis UI with customer details and migrated to Keshav MongoDB
8223f74 unverified
parameters:
retriever_type: contextual_reranked # Enable re-ranking
embedding_model_id: text-embedding-3-small
embedding_model_type: openai
embedding_model_dim: 1536
device: mps # or cuda (for Nvidia GPUs) or mps (for Apple M1/M2/M3 chips)
# Re-ranking parameters
enable_reranking: true
rerank_model_name: "cross-encoder/ms-marco-MiniLM-L-2-v2"
stage1_limit: 50 # Retrieve 50 candidates in stage 1
final_k: 10 # Return top 10 after re-ranking