chore(ci): Update CI workflow to include DEEPSEEK_API_KEY and deepseekAI

.env.example

@@ -3,6 +3,7 @@ IPS="420.69.247.365"
 NODE_ENV="development"
 OPENAI_API_KEY="DEEZ"
 CLAUDE_API_KEY="NUTZ"
+DEEPSEEK_API_KEY="BALLZ"
 
 # notify
 NOTIFY_URL="https://notify.jaw.dev"

.github/workflows/ci.yml

@@ -11,6 +11,7 @@ env:
   NODE_ENV: 'testing'
   OPENAI_API_KEY: 'DEEZ'
   CLAUDE_API_KEY: 'NUTZ'
+  DEEPSEEK_API_KEY: 'BALLZ'
   DISCORD_WEBHOOK_URL: 'x'
 
 jobs:

.vscode/settings.json

@@ -2,10 +2,14 @@
 	"cSpell.words": [
 		"adminjs",
 		"aicommit",
+		"BALLZ",
 		"claudeai",
+		"deepseek",
+		"DEEZ",
 		"fastq",
 		"healthz",
 		"nofollow",
+		"NUTZ",
 		"openai",
 		"pacman",
 		"Tahoma",

src/ai.ts

@@ -95,7 +95,7 @@
 				.filter((choice) => choice.message?.content)
 				.map((choice) => choice.message.content);
 			return getRandomElement(messages);
 		} catch (error: any) {
 			if (error.code === 'insufficient_quota') {
 				throw new UnauthorizedError(
 					'You exceeded your current quota, please check your plan and billing details',
@@ -133,7 +133,7 @@
 			});
 			// @ts-ignore - trust me bro
 			return getRandomElement(messages.content).text;
 		} catch (error: any) {
 			if (error?.error?.error?.type === 'authentication_error') {
 				throw new UnauthorizedError(error.error.error.message);
 			}
@@ -142,10 +142,48 @@
 	},
 };
 
+export const deepseekAI: AIService = {
+	generate: async (diff: string, apiKey?: string) => {
+		try {
+			const API_KEY = apiKey ? apiKey : appConfig.DEEPSEEK_API_KEY;
+			const chatCompletion = await new OpenAI({
+				baseURL: 'https://api.deepseek.com',
+				apiKey: API_KEY,
+			}).chat.completions.create({
+				model: 'deepseek-chat',
+				temperature: 0.7,
+				max_tokens: 200,
+				stream: false,
+				messages: [
+					{
+						role: 'system',
+						content: prompt,
+					},
+					{
+						role: 'user',
+						content: diff,
+					},
+				],
+			});
+			const messages = chatCompletion.choices
+				.filter((choice) => choice.message?.content)
+				.map((choice) => choice.message.content);
+			return getRandomElement(messages);
+		} catch (error: any) {
+			if (error?.error?.type === 'invalid_api_key') {
+				throw new UnauthorizedError(error.message);
+			}
+			throw error;
+		}
+	},
+};
+
 export function ai(type?: Provider): AIService {
 	switch (type) {
 		case 'claudeai':
 			return claudeAI;
+		case 'deepseek':
+			return deepseekAI;
 		case 'openai':
 		default:
 			return openAI;

src/commit.sh

@@ -32,7 +32,7 @@ show_help() {
     printf "${YELLOW}Options:${NC}\n"
     printf "  ${GREEN}-dr, --dry-run${NC}        Run the script without making any changes\n"
     printf "  ${GREEN}-nv, --no-verify${NC}      Skip message selection\n"
-    printf "  ${GREEN}-ai, --ai-provider${NC}    Specify AI provider (openai or claude, default: openai)\n"
+    printf "  ${GREEN}-ai, --ai-provider${NC}    Specify AI provider (openai, claude or deepseek, default: openai)\n"
     printf "  ${GREEN}-k, --api-key${NC}         Specify the API key for the AI provider\n"
     printf "  ${GREEN}-v, --verbose${NC}         Enable verbose logging\n"
     printf "  ${GREEN}-h, --help${NC}            Display this help message\n"
@@ -71,7 +71,7 @@ parse_arguments() {
             -ai|--ai-provider)
                 AI_PROVIDER=$2
                 log_verbose "AI provider set to: " "$AI_PROVIDER"
-                if [[ "$AI_PROVIDER" != "openai" && "$AI_PROVIDER" != "claudeai" ]]; then
+                if [[ "$AI_PROVIDER" != "openai" && "$AI_PROVIDER" != "claudeai" && "$AI_PROVIDER" != "deepseek" ]]; then
                     log_verbose "Invalid AI provider specified"
                     echo -e "${RED}Invalid AI provider. Please use 'openai' or 'claudeai'.${NC}\n"
                     exit 1

src/config.ts

@@ -8,33 +8,33 @@
 	NOTIFY_URL: {
 		value: process.env.NOTIFY_URL,
 		required: true,
 		type: (value: any) => String(value),
 	},
 	NOTIFY_X_API_KEY: {
 		value: process.env.NOTIFY_X_API_KEY,
 		required: true,
 		type: (value: any) => String(value),
 	},
 	IPS: {
 		value: process.env.IPS,
 		required: true,
 		type: (value: any) => String(value),
 	},
 	PORT: {
 		value: process.env.PORT,
 		default: 80,
 		type: (value: any) => Number(value),
 		required: false,
 	},
 	OPENAI_API_KEY: {
 		value: process.env.OPENAI_API_KEY,
 		required: true,
 		type: (value: any) => String(value),
 	},
 	CLAUDE_API_KEY: {
 		value: process.env.CLAUDE_API_KEY,
 		required: true,
 		type: (value: any) => String(value),
 	},
 	NODE_ENV: {
 		value: process.env.NODE_ENV,
@@ -42,4 +42,9 @@
 		required: false,
 		type: (value: any) => value as 'development' | 'production',
 	},
+	DEEPSEEK_API_KEY: {
+		value: process.env.DEEPSEEK_API_KEY,
+		required: true,
+		type: (value: any) => String(value),
+	},
 });

src/handler.ts

@@ -120,7 +120,7 @@ export function postGenerateCommitMessageHandler(ai: (type?: Provider) => AIServ
 			throw new ValidationError('Diff must not be empty!');
 		}
 
-		if (provider && provider !== 'openai' && provider !== 'claudeai') {
+		if (provider && provider !== 'openai' && provider !== 'claudeai' && provider !== 'deepseek') {
 			throw new ValidationError('Invalid provider specified!');
 		}
 

src/types.ts

@@ -1,6 +1,6 @@
 import { Request } from 'express';
 
-export type Provider = 'openai' | 'claudeai';
+export type Provider = 'openai' | 'claudeai' | 'deepseek';
 
 export interface GenerateCommitMessageRequest extends Request {
 	body: {
@@ -27,7 +27,7 @@ export interface ConfigItem<T> {
 	readonly required: boolean;
 }
 
-export interface Logger {
+export interface LoggerType {
 	debug: (...value: any) => void;
 	error: (...value: any) => void;
 	info: (...value: any) => void;

src/util.ts

@@ -2,7 +2,7 @@ import fastq from 'fastq';
 import { Request } from 'express';
 import { styleText } from 'node:util';
 import { appConfig } from './config';
-import { CacheType, ConfigItem, Logger } from './types';
+import { CacheType, ConfigItem, LoggerType } from './types';
 
 export const sendNotificationQueue = fastq.promise(sendNotification, 1);
 
@@ -81,7 +81,7 @@ function Cache(): CacheType {
 
 export const cache = Cache();
 
-export const logger: Logger = {
+export const logger: LoggerType = {
 	debug: (...value: any) => {
 		const timestamp = new Date().toLocaleString();
 		console.debug(styleText('red', `🐛 ${timestamp} ${value}`));