From 11d76b4535ee49d26999f1013e3189b883570f0d Mon Sep 17 00:00:00 2001 From: aakashb95 Date: Fri, 20 Feb 2026 17:16:09 +0530 Subject: [PATCH] chore(captain): support explicit cohere rerank endpoint config --- .../app/services/captain/documents/chunk_reranker_service.rb | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/enterprise/app/services/captain/documents/chunk_reranker_service.rb b/enterprise/app/services/captain/documents/chunk_reranker_service.rb index 3522c03a5..c1b6b0855 100644 --- a/enterprise/app/services/captain/documents/chunk_reranker_service.rb +++ b/enterprise/app/services/captain/documents/chunk_reranker_service.rb @@ -9,7 +9,10 @@ class Captain::Documents::ChunkRerankerService @account_id = account_id @model = resolve_model @api_key = InstallationConfig.find_by(name: 'CAPTAIN_COHERE_API_KEY')&.value.to_s - @endpoint = InstallationConfig.find_by(name: 'CAPTAIN_COHERE_API_BASE')&.value.to_s.presence || DEFAULT_ENDPOINT + @endpoint = + InstallationConfig.find_by(name: 'CAPTAIN_COHERE_RERANK_ENDPOINT')&.value.to_s.presence || + InstallationConfig.find_by(name: 'CAPTAIN_COHERE_API_BASE')&.value.to_s.presence || + DEFAULT_ENDPOINT end def rerank(query:, candidates:, limit:)