refactor: add/modify rag-related config (#599)

1 files changed, 9 insertions, 2 deletions

diff --git a/src/client/model.rs b/src/client/model.rs
index d555232..56421bf 100644
--- a/src/client/model.rs
+++ b/src/client/model.rs
@@ -1,5 +1,5 @@
 use super::{
-    list_chat_models,
+    list_chat_models, list_embedding_models,
     message::{Message, MessageContent},
     EmbeddingsData,
 };
@@ -43,13 +43,20 @@ impl Model {
             .collect()
     }
 
-    pub fn retrieve(config: &Config, model_id: &str) -> Result<Self> {
+    pub fn retrieve_chat(config: &Config, model_id: &str) -> Result<Self> {
         match Self::find(&list_chat_models(config), model_id) {
             Some(v) => Ok(v),
             None => bail!("Invalid model '{model_id}'"),
         }
     }
 
+    pub fn retrieve_embedding(config: &Config, model_id: &str) -> Result<Self> {
+        match Self::find(&list_embedding_models(config), model_id) {
+            Some(v) => Ok(v),
+            None => bail!("Invalid model '{model_id}'"),
+        }
+    }
+
     pub fn find(models: &[&Self], model_id: &str) -> Option<Self> {
         let mut model = None;
         let (client_name, model_name) = match model_id.split_once(':') {