Merge pull request #1765 from codestoryai/features/add-support-for-de…

…epseek-r1

features/add support for deepseek r1

llm_client/src/clients/codestory.rs

@@ -168,6 +168,8 @@ impl CodeStoryClient {
             LLMType::O1 => Ok("o1".to_owned()),          // o1
             LLMType::O1Mini => Ok("o1-mini".to_owned()), // o1 mini
             LLMType::DeepSeekCoderV3 => Ok("deepseek/deepseek-chat".to_owned()),
+            LLMType::DeepSeekR1 => Ok("deepseek/deepseek-r1".to_owned()),
+            LLMType::Custom(custom) => Ok(custom.to_owned()),
             _ => Err(LLMClientError::UnSupportedModel),
         }
     }
@@ -336,7 +338,9 @@ impl LLMClient for CodeStoryClient {
         request: LLMClientCompletionRequest,
     ) -> Result<String, LLMClientError> {
         let (sender, _receiver) = tokio::sync::mpsc::unbounded_channel();
-        self.stream_completion(api_key, request, sender).await.map(|answer| answer.answer_up_until_now().to_owned())
+        self.stream_completion(api_key, request, sender)
+            .await
+            .map(|answer| answer.answer_up_until_now().to_owned())
     }
 
     async fn stream_completion(
@@ -386,7 +390,11 @@ impl LLMClient for CodeStoryClient {
                 }
             }
         }
-        Ok(LLMClientCompletionResponse::new(buffered_stream, None, model))
+        Ok(LLMClientCompletionResponse::new(
+            buffered_stream,
+            None,
+            model,
+        ))
     }
 
     async fn stream_prompt_completion(

llm_client/src/clients/types.rs

@@ -45,6 +45,8 @@ pub enum LLMType {
     DeepSeekCoderV2,
     /// DeepSeek Coder V3 model
     DeepSeekCoderV3,
+    /// DeepSeek r1
+    DeepSeekR1,
     /// CodeLLama 70B Instruct model
     CodeLLama70BInstruct,
     /// CodeLlama 13B Instruct model
@@ -135,6 +137,7 @@ impl<'de> Deserialize<'de> for LLMType {
                     "gemini-2.0-flash-thinking-exp-1219" => {
                         Ok(LLMType::Gemini2_0FlashThinkingExperimental)
                     }
+                    "deepseek/deepseek-r1" => Ok(LLMType::DeepSeekR1),
                     "Llama3_8bInstruct" => Ok(LLMType::Llama3_8bInstruct),
                     "Llama3_1_8bInstruct" => Ok(LLMType::Llama3_1_8bInstruct),
                     "Llama3_1_70bInstruct" => Ok(LLMType::Llama3_1_70bInstruct),
@@ -231,6 +234,7 @@ impl fmt::Display for LLMType {
             LLMType::Gemini2_0FlashThinkingExperimental => {
                 write!(f, "gemini-2.0-flash-thinking-exp-1219")
             }
+            LLMType::DeepSeekR1 => write!(f, "deepseek/deepseek-r1"),
             LLMType::Gpt4O => write!(f, "Gpt4O"),
             LLMType::GeminiProFlash => write!(f, "GeminiProFlash"),
             LLMType::DeepSeekCoderV2 => write!(f, "DeepSeekCoderV2"),

sidecar/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "sidecar"
-version = "0.1.18"
+version = "0.1.19"
 edition = "2021"
 build = "build.rs"