feat: support rag in remove_url_handler func

Signed-off-by: Xin Liu <sam@secondstate.io>
LlamaEdge · Jan 13, 2025 · 6fa7579 · 6fa7579
1 parent 3da6821
commit 6fa7579
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/handler.rs b/src/handler.rs
@@ -179,6 +179,7 @@ pub(crate) async fn remove_url_handler(
         "chat" => UrlType::Chat,
         "whisper" => UrlType::AudioWhisper,
         "image" => UrlType::Image,
+        "rag" => UrlType::Rag,
         _ => {
             let err_msg = format!("invalid url type: {}", url_type);
             error!(target: "stdout", "{}", &err_msg);