chore: update code

Signed-off-by: Xin Liu <[email protected]>
LlamaEdge · May 9, 2024 · ae82a99 · ae82a99
1 parent 670080e
commit ae82a99
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/backend/ggml.rs b/src/backend/ggml.rs
@@ -280,7 +280,7 @@ pub(crate) async fn rag_query_handler(
                     // create a embedding request
                     let embedding_request = EmbeddingRequest {
                         model: embedding_model_names[0].clone(),
-                        input: vec![query_text.clone()],
+                        input: query_text.into(),
                         encoding_format: None,
                         user: chat_request.user.clone(),
                     };
@@ -946,7 +946,7 @@ pub(crate) async fn doc_to_embeddings(
         // create an embedding request
         let embedding_request = EmbeddingRequest {
             model,
-            input: chunks,
+            input: chunks.into(),
             encoding_format: None,
             user: None,
         };
@@ -1098,7 +1098,7 @@ pub(crate) async fn retrieve_handler(
                     // create a embedding request
                     let embedding_request = EmbeddingRequest {
                         model: embedding_model_names[0].clone(),
-                        input: vec![query_text.clone()],
+                        input: query_text.into(),
                         encoding_format: None,
                         user: chat_request.user.clone(),
                     };