Support overwriten prompts of the search query

This commit is contained in:
Clément Renault 2025-05-13 16:33:58 +02:00
parent 951be67060
commit 82fa70da83
No known key found for this signature in database
GPG key ID: F250A4C4E3AE5F5F
2 changed files with 36 additions and 4 deletions

View file

@ -17,7 +17,7 @@ use meilisearch_types::milli::index::IndexEmbeddingConfig;
use meilisearch_types::milli::prompt::PromptData;
use meilisearch_types::milli::vector::EmbeddingConfig;
use meilisearch_types::{Document, Index};
use serde::Deserialize;
use serde::{Deserialize, Serialize};
use serde_json::json;
use crate::extractors::authentication::policies::ActionPolicy;
@ -68,6 +68,24 @@ async fn chat(
"Meilisearch /chat only support one completion at a time (n = 1, n = null)"
);
let rtxn = index_scheduler.read_txn().unwrap();
let search_in_index_description = index_scheduler
.chat_prompts(&rtxn, "searchInIndex-description")
.unwrap()
.unwrap_or(DEFAULT_SEARCH_IN_INDEX_TOOL_DESCRIPTION)
.to_string();
let search_in_index_q_param_description = index_scheduler
.chat_prompts(&rtxn, "searchInIndex-q-param-description")
.unwrap()
.unwrap_or(DEFAULT_SEARCH_IN_INDEX_Q_PARAMETER_TOOL_DESCRIPTION)
.to_string();
let search_in_index_index_description = index_scheduler
.chat_prompts(&rtxn, "searchInIndex-index-param-description")
.unwrap()
.unwrap_or(DEFAULT_SEARCH_IN_INDEX_INDEX_PARAMETER_TOOL_DESCRIPTION)
.to_string();
drop(rtxn);
let mut response;
loop {
let mut tools = chat_completion.tools.get_or_insert_default();