From 70fa94146a05ae57a4c09954b69c6717f245e15f Mon Sep 17 00:00:00 2001 From: Mubelotix Date: Thu, 17 Jul 2025 11:29:55 +0200 Subject: [PATCH] Add document template test --- .../tests/documents/render_documents.rs | 36 ++++++++++++++++++- 1 file changed, 35 insertions(+), 1 deletion(-) diff --git a/crates/meilisearch/tests/documents/render_documents.rs b/crates/meilisearch/tests/documents/render_documents.rs index 66ce0a269..2fca9b537 100644 --- a/crates/meilisearch/tests/documents/render_documents.rs +++ b/crates/meilisearch/tests/documents/render_documents.rs @@ -1,4 +1,4 @@ -use crate::common::shared_index_for_fragments; +use crate::common::{shared_index_for_fragments, Server}; use crate::json; use meili_snap::snapshot; @@ -296,3 +296,37 @@ async fn chat_completions_template_retrieval() { } "#); } + +#[actix_rt::test] +async fn retrieve_document_template() { + let server = Server::new_shared(); + let index = server.unique_index(); + + let (response, code) = index + .update_settings(json!( + { + "embedders": { + "doggo_embedder": { + "source": "huggingFace", + "model": "sentence-transformers/all-MiniLM-L6-v2", + "revision": "e4ce9877abf3edfe10b0d82785e83bdcb973e22e", + "documentTemplate": "This is a document template {{doc.doggo}}", + } + } + } + )) + .await; + snapshot!(code, @"202 Accepted"); + server.wait_task(response["taskUid"].as_u64().unwrap()).await; + + let (value, code) = index + .render(json! {{ "template": { "id": "embedders.doggo_embedder.documentTemplate" }}}) + .await; + snapshot!(code, @"200 OK"); + snapshot!(value, @r#" + { + "template": "This is a document template {{doc.doggo}}", + "rendered": null + } + "#); +}