Merge pull request #178 from skyline-GTRr32/feature/support-deepseek

Add DeepSeek Support to CodeBuddy

Author: olasunkanmi-SE

package.json

@@ -173,7 +173,8 @@
               "Gemini",
               "Groq",
               "Anthropic",
-              "XGrok"
+              "XGrok",
+              "Deepseek"
             ],
             "default": "Groq",
             "description": "Select the Generative AI to use with code buddy"
@@ -242,6 +243,22 @@
             "default": null,
             "markdownDescription": "Enter your [Grok API Key](https://console.x.ai/)"
           },
+          "deepseek.model": {
+            "type": [
+              "string",
+              "null"
+            ],
+            "default": "deepseek-chat",
+            "markdownDescription": "Provide the name of the Deepseek model you want to use."
+          },
+          "deepseek.apiKey": {
+            "type": [
+              "string",
+              "null"
+            ],
+            "default": null,
+            "markdownDescription": "Enter your Deepseek API Key"
+          },
           "font.family": {
             "type": "string",
             "enum": [
@@ -329,10 +346,11 @@
     "jsdom": "^26.0.0",
     "markdown-it": "^14.1.0",
     "node-fetch": "^3.3.2",
+    "openai": "^4.28.0",
     "prettier": "^3.2.5",
     "readable-stream": "^4.5.2",
     "simple-git": "^3.25.0",
     "sinon": "^17.0.1"
   },
   "license": "MIT"
-}
+}

src/application/constant.ts

@@ -14,24 +14,22 @@ export enum OLA_ACTIONS {
   generateCodeChart = "CodeBuddy.generateCodeChart",
   inlineChat = "CodeBuddy.inLineChat",
 }
-
 export enum COMMON {
   GROQ_CHAT_HISTORY = "groqChatHistory",
   GEMINI_CHAT_HISTORY = "geminiChatHistory",
   ANTHROPIC_CHAT_HISTORY = "anthropicChatHistory",
+  DEEPSEEK_CHAT_HISTORY = "deepseekChatHistory",
   USER_INPUT = "user-input",
   BOT = "bot",
   GEMINI_SNAPSHOT = "GeminiSnapshot",
 }
-
 export const GROQ_CONFIG = {
   temperature: 0.1,
   max_tokens: 5024,
   top_p: 1,
   stream: false,
   stop: null,
 };
-
 export const APP_CONFIG = {
   geminiKey: "google.gemini.apiKeys",
   geminiModel: "google.gemini.model",
@@ -42,32 +40,30 @@ export const APP_CONFIG = {
   anthropicApiKey: "anthropic.apiKey",
   grokApiKey: "grok.apiKey",
   grokModel: "grok.model",
+  deepseekApiKey: "deepseek.apiKey",
+  deepseekModel: "deepseek.model",
 };
-
 export enum generativeAiModels {
   GEMINI = "Gemini",
   GROQ = "Groq",
   ANTHROPIC = "Anthropic",
   GROK = "XGrok",
+  DEEPSEEK = "Deepseek",
 }
-
 export const MEMORY_CACHE_OPTIONS = {
   sessionTTL: 24 * 60 * 60 * 1000,
 };
-
 export type aIProviderConfig = {
   apiKey: string;
   model: string;
   providerName: string;
 };
-
 export enum FSPROPS {
   SRC_DIRECTORY = "src",
   TS_FILE_PATTERN = "**/*.ts",
   TSCONFIG_FILE = "tsconfig.json",
   NODE_MODULES_PATTERN = "**/node_modules/**",
 }
-
 export const EmbeddingsConfig = {
   batchSize: 5,
   maxRetries: 3,
@@ -126,24 +122,21 @@ export enum HTTP_STATUS_CODE {
   GATEWAY_TIMEOUT = 504,
   HTTP_VERSION_NOT_SUPPORTED = 505,
 }
-
 export enum RequestHeader {
   AUTHORIZATION = "authorization",
   CONTENT_TYPE = "Content-Type",
   ACCEPT = "accept",
   CONNECTION = "connection",
 }
-
 export enum HTTP_VERBS {
   GET = "GET",
   POST = "POST",
   PUT = "PUT",
 }
-
 export const HTTPS_DEFAULT_TIMEOUT = 10000;
 export const generateQuerySting = (query: string) =>
   `q=${encodeURIComponent(query)}`;
 export enum WEB_SEARCH_CONFIG {
   baseUrl = "https://www.startpage.com/sp/search?",
   userAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
-}
+}

src/extension.ts

@@ -26,6 +26,7 @@ import { AnthropicWebViewProvider } from "./providers/anthropic";
 import { CodeActionsProvider } from "./providers/code-actions";
 import { GeminiWebViewProvider } from "./providers/gemini";
 import { GroqWebViewProvider } from "./providers/groq";
+import { DeepseekWebViewProvider } from "./providers/deepseek";
 import { FileUploader } from "./services/file-uploader";
 import { initializeGenerativeAiEnvironment } from "./services/generative-ai-model-manager";
 import { Credentials } from "./services/github-authentication";
@@ -40,6 +41,8 @@ const {
   anthropicModel,
   grokApiKey,
   grokModel,
+  deepseekApiKey,
+  deepseekModel,
 } = APP_CONFIG;
 
 const logger = new Logger("extension");
@@ -214,6 +217,11 @@ export async function activate(context: vscode.ExtensionContext) {
         model: grokModel,
         webviewProviderClass: AnthropicWebViewProvider,
       },
+      [generativeAiModels.DEEPSEEK]: {
+        key: deepseekApiKey,
+        model: deepseekModel,
+        webviewProviderClass: DeepseekWebViewProvider,
+      },
     };
     if (selectedGenerativeAiModel in modelConfigurations) {
       const modelConfig = modelConfigurations[selectedGenerativeAiModel];
@@ -241,4 +249,4 @@ export function deactivate(context: vscode.ExtensionContext) {
   quickFixCodeAction.dispose();
   agentEventEmmitter.dispose();
   context.subscriptions.forEach((subscription) => subscription.dispose());
-}
+}