Skip to content

Commit 6aea16b

Browse files
committed
feat(agentic): Support PROMPT_KEYWORD_EXTRACTOR env variable
Signed-off-by: Xin Liu <sam@secondstate.io>
1 parent e60c0a2 commit 6aea16b

File tree

1 file changed

+4
-2
lines changed
  • cardea-agentic-search/cardea-agentic-search-mcp-server/src

1 file changed

+4
-2
lines changed

cardea-agentic-search/cardea-agentic-search-mcp-server/src/search.rs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ use serde_json::{Value, json};
2020
use std::collections::HashSet;
2121
use tracing::{debug, error, info, warn};
2222

23-
const PROMPT_KEYWORD_EXTRACTOR: &str = r#"
23+
const DEFAULT_PROMPT_KEYWORD_EXTRACTOR: &str = r#"
2424
You are a multilingual keyword extractor. Your task is to extract the most relevant and concise keywords or key phrases from the given user query.
2525
2626
Follow these requirements strictly:
@@ -418,7 +418,9 @@ impl AgenticSearchServer {
418418
let config = self.config.chat_service.as_ref().unwrap();
419419

420420
let text = query.as_ref();
421-
let user_prompt = format!("{PROMPT_KEYWORD_EXTRACTOR}\n\n### Input Query\n{text:#?}");
421+
let prompt = std::env::var("PROMPT_KEYWORD_EXTRACTOR")
422+
.unwrap_or(DEFAULT_PROMPT_KEYWORD_EXTRACTOR.to_string());
423+
let user_prompt = format!("{prompt}\n\n### Input Query\n{text:#?}");
422424

423425
let user_message = ChatCompletionRequestMessage::new_user_message(
424426
ChatCompletionUserMessageContent::Text(user_prompt),

0 commit comments

Comments
 (0)