feat: add Google Gemini provider support

Author: m7medVision

bun.lockb

@@ -38,6 +38,7 @@
   },
   "type": "module",
   "dependencies": {
+    "@ai-sdk/google": "^1.0.11",
     "@ai-sdk/openai": "^1.0.10",
     "@clack/core": "^0.3.4",
     "@clack/prompts": "^0.7.0",

package.json

@@ -62,7 +62,7 @@ function hasOwn<T extends object, K extends PropertyKey>(
 export const configPath = path.join(os.homedir(), ".lazycommit");
 
 export interface Config {
-	provider: "openai"; // Removed "google"
+	provider: "openai" | "google";
 	API_KEY: string;
 	model: string;
 	templates: Record<string, string>;
@@ -71,7 +71,7 @@ export interface Config {
 const DEFAULT_CONFIG: Config = {
 	provider: "openai",
 	API_KEY: "",
-	model: "gpt-4o",
+	model: "gpt-4",
 	templates: {
 		default: path.join(os.homedir(), ".lazycommit-template"),
 	},
@@ -233,10 +233,8 @@ export async function showConfigUI() {
 			const provider = await p.select({
 				message: "Provider",
 				options: [
-					{
-						label: "OpenAI",
-						value: "openai",
-						},
+					{ label: "OpenAI", value: "openai" },
+					{ label: "Google", value: "google" },
 				],
 				initialValue: config.provider,
 			});
@@ -268,7 +266,14 @@ async function getModels() {
 		});
 		const models = await oai.models.list();
 		return models.data.map((model) => model.id);
-	} else {
-		throw new Error("Invalid provider");
+	} else if (provider === "google") {
+		return [
+			"gemini-2.0-flash-exp",
+			"gemini-1.5-pro-latest",
+			"gemini-1.5-pro",
+			"gemini-1.5-flash-latest",
+			"gemini-1.5-flash",
+		];
 	}
+	throw new Error("Invalid provider");
 }

src/config.ts

@@ -1,5 +1,6 @@
 import { $ } from "bun";
 import { createOpenAI } from "@ai-sdk/openai";
+import { createGoogleGenerativeAI } from "@ai-sdk/google";
 import { generateText } from "ai";
 import { readConfigFile } from "./config";
 import simpleGit from "simple-git";
@@ -93,14 +94,23 @@ export async function run(options: RunOptions, templateName?: string) {
   const system_message =
     "You are a commit message generator. I will provide you with a git diff, and I would like you to generate an appropriate commit message using the conventional commit format. Do not write any explanations or other words, just reply with the commit message.";
 
-  const aiProvider = createOpenAI({
-    compatibility: 'strict',
-    apiKey: config.API_KEY,
-  });
-
   try {
     if (options.verbose) {
-      console.debug("Sending request to OpenAI service...");
+      console.debug("Sending request to AI service...");
+    }
+
+    let aiProvider;
+    if (config.provider === "openai") {
+      aiProvider = createOpenAI({
+        compatibility: 'strict',
+        apiKey: config.API_KEY,
+      });
+    } else if (config.provider === "google") {
+      aiProvider = createGoogleGenerativeAI({
+        apiKey: config.API_KEY,
+      });
+    } else {
+      throw new Error("Invalid provider");
     }
 
     const { text } = await generateText({
@@ -109,7 +119,7 @@ export async function run(options: RunOptions, templateName?: string) {
     });
 
     if (options.verbose) {
-      console.debug("Response received from OpenAI service.");
+      console.debug("Response received from AI service.");
       console.debug(text);
     }
 
@@ -120,7 +130,7 @@ export async function run(options: RunOptions, templateName?: string) {
     }
 
   } catch (error) {
-    console.error(`Failed to fetch from OpenAI service: ${error}`);
+    console.error(`Failed to fetch from AI service: ${error}`);
     process.exit(1);
   }
 }