feat: add provider headers support for AI commit generation

Author: riccardogiorato

src/commands/aicommits.ts

@@ -126,6 +126,7 @@ export default async (
 		try {
 			const baseUrl = providerInstance.getBaseUrl();
 			const apiKey = providerInstance.getApiKey() || '';
+			const providerHeaders = providerInstance.getHeaders();
 
 			if (isChunking) {
 				// Split files into chunks
@@ -162,7 +163,8 @@ export default async (
 							config['max-length'],
 							config.type,
 							timeout,
-							customPrompt
+							customPrompt,
+							providerHeaders
 						);
 						chunkMessages.push(...result.messages);
 						if (result.usage) {
@@ -192,7 +194,8 @@ export default async (
 					config['max-length'],
 					config.type,
 					timeout,
-					customPrompt
+					customPrompt,
+					providerHeaders
 				);
 				messages = combineResult.messages;
 				if (combineResult.usage) {
@@ -229,7 +232,8 @@ export default async (
 					config['max-length'],
 					config.type,
 					timeout,
-					customPrompt
+					customPrompt,
+					providerHeaders
 				);
 				messages = result.messages;
 				usage = result.usage;

src/commands/prepare-commit-msg-hook.ts

@@ -51,6 +51,7 @@ export default () =>
 
 		const baseUrl = providerInstance.getBaseUrl();
 		const apiKey = providerInstance.getApiKey() || '';
+		const providerHeaders = providerInstance.getHeaders();
 
 		// Use config timeout, or default per provider
 		const timeout =
@@ -72,7 +73,9 @@ export default () =>
 				config.generate,
 				config['max-length'],
 				config.type,
-				timeout
+				timeout,
+				undefined,
+				providerHeaders
 			);
 			messages = result.messages;
 		} finally {

src/feature/providers/base.ts

@@ -9,6 +9,7 @@ export type ProviderDef = {
 	modelsFilter?: (models: any[]) => string[];
 	defaultModels: string[];
 	requiresApiKey: boolean;
+	headers?: Record<string, string>;
 };
 
 export class Provider {
@@ -111,6 +112,10 @@ export class Provider {
 		return this.def.defaultModels;
 	}
 
+	getHeaders(): Record<string, string> | undefined {
+		return this.def.headers;
+	}
+
 	validateConfig(): { valid: boolean; errors: string[] } {
 		const errors: string[] = [];
 		if (this.def.requiresApiKey && !this.getApiKey()) {

src/feature/providers/openrouter.ts

@@ -11,4 +11,8 @@ export const OpenRouterProvider: ProviderDef = {
 			.map((m: any) => m.id),
 	defaultModels: ['openai/gpt-oss-20b:free', 'z-ai/glm-4.5-air:free'],
 	requiresApiKey: true,
+	headers: {
+		'HTTP-Referer': 'https://github.com/nutlope/aicommits',
+		'X-Title': 'aicommits',
+	},
 };

src/utils/openai.ts

@@ -79,7 +79,8 @@ export const generateCommitMessage = async (
 	maxLength: number,
 	type: CommitType,
 	timeout: number,
-	customPrompt?: string
+	customPrompt?: string,
+	headers?: Record<string, string>
 ) => {
 	if (process.env.DEBUG) {
 		console.log('Diff being sent to AI:');
@@ -94,6 +95,7 @@ export const generateCommitMessage = async (
 						name: 'custom',
 						apiKey,
 						baseURL: baseUrl,
+						headers,
 				  });
 
 		const abortController = new AbortController();
@@ -208,7 +210,8 @@ export const combineCommitMessages = async (
 	maxLength: number,
 	type: CommitType,
 	timeout: number,
-	customPrompt?: string
+	customPrompt?: string,
+	headers?: Record<string, string>
 ) => {
 	try {
 		const provider =
@@ -218,6 +221,7 @@ export const combineCommitMessages = async (
 						name: 'custom',
 						apiKey,
 						baseURL: baseUrl,
+						headers,
 				  });
 
 		const abortController = new AbortController();