Merge pull request #7 from m7medVision/add-custom-endpoint

Add custom endpoint option for OpenAI-like APIs

Author: m7medVision

README.md

@@ -72,3 +72,7 @@ customCommands:
   check out these other projects that inspired this one:
 
 - https://github.com/BuilderIO/ai-shell
+
+## Custom Endpoint
+
+You can now specify a custom endpoint for using OpenAI-like APIs. To set a custom endpoint, use the `bunx @m7medvision/lazycommit@latest config` command and select the "Custom Endpoint" option. This allows you to use different API endpoints that are compatible with OpenAI.

src/config.ts

@@ -11,6 +11,10 @@ async function editFile(filePath: string, onExit: () => void) {
 		(await p.select({
 			message: "Select an editor",
 			options: [
+				{
+					label: "neovim",
+					value: "nvim",
+				},
 				{
 					label: "vim",
 					value: "vim",
@@ -19,6 +23,10 @@ async function editFile(filePath: string, onExit: () => void) {
 					label: "nano",
 					value: "nano",
 				},
+				{
+					label: "code",
+					value: "code",
+				},
 				{
 					label: "cancel",
 					value: "cancel",
@@ -62,16 +70,18 @@ function hasOwn<T extends object, K extends PropertyKey>(
 export const configPath = path.join(os.homedir(), ".lazycommit");
 
 export interface Config {
-	provider: "openai" | "google";
+	provider: "openai" | "google" | "custom";
 	API_KEY: string;
 	model: string;
 	templates: Record<string, string>;
+	customEndpoint?: string;
 }
 
 const DEFAULT_CONFIG: Config = {
 	provider: "openai",
 	API_KEY: "",
-	model: "gpt-4",
+	customEndpoint: "",
+	model: "gpt-4o",
 	templates: {
 		default: path.join(os.homedir(), ".lazycommit-template"),
 	},
@@ -157,6 +167,11 @@ export async function showConfigUI() {
 					value: "template",
 					hint: "edit the prompt template",
 				},
+				{
+					label: "Custom Endpoint",
+					value: "customEndpoint",
+					hint: config.customEndpoint || "not set",
+				},
 				{
 					label: "Done", // Changed from "Cancel" to "Done"
 					value: "done",
@@ -235,11 +250,19 @@ export async function showConfigUI() {
 				options: [
 					{ label: "OpenAI", value: "openai" },
 					{ label: "Google", value: "google" },
+					{ label: "Custom", value: "custom" },
 				],
 				initialValue: config.provider,
 			});
 
 			await setConfigs([["provider", provider as string]]);
+		} else if (choice === "customEndpoint") {
+			const customEndpoint = await p.text({
+				message: "Custom Endpoint",
+				initialValue: config.customEndpoint,
+			});
+
+			await setConfigs([["customEndpoint", customEndpoint as string]]);
 		}
 
 		if (p.isCancel(choice)) {
@@ -266,6 +289,16 @@ async function getModels() {
 		});
 		const models = await oai.models.list();
 		return models.data.map((model) => model.id);
+	} else if (provider === "custom") {
+		if (!config.customEndpoint) {
+			throw new Error("Custom endpoint is not set");
+		}
+		const oai = new OpenAI({
+			apiKey,
+			baseURL: config.customEndpoint,
+		});
+		const models = await oai.models.list();
+		return models.data.map((model) => model.id);
 	} else if (provider === "google") {
 		return [
 			"gemini-2.0-flash-exp",

src/run.ts

@@ -109,6 +109,12 @@ export async function run(options: RunOptions, templateName?: string) {
       aiProvider = createGoogleGenerativeAI({
         apiKey: config.API_KEY,
       });
+    } else if (config.provider === "custom") {
+      aiProvider = createOpenAI({
+        compatibility: 'strict',
+        apiKey: config.API_KEY,
+        baseURL: config.customEndpoint
+      })
     } else {
       throw new Error("Invalid provider");
     }
@@ -133,4 +139,4 @@ export async function run(options: RunOptions, templateName?: string) {
     console.error(`Failed to fetch from AI service: ${error}`);
     process.exit(1);
   }
-}
+}