Add support for custom request headers in commit generation

Author: Awilen-Bernkastel

README.md

@@ -83,6 +83,8 @@ Default value: `{{type}} {{emoji}}: {{message}}`
 
 - Custom Description Prompt: A custom prompt to generate the commit description.
 
+- Request headers: custom request headers that will be sent with each request to Ollama or a custom endpoint. This is useful for authentication and other purposes.
+
 ## Known Issues
 
 Sometimes, depending on the model used, it can generate quite long commit messages. However, it provides a good starting point for what the commit should be and can be manually edited to achieve the desired length.

src/config.ts

@@ -16,6 +16,7 @@ export const defaultConfig = {
 	language: Languages.English,
 	commitTemplate: '{{type}} {{emoji}}: {{message}}',
 	promptTemperature: 0.2,
+    requestHeaders: {},
 	emojis: {
 		feat: '✨',
 		fix: '🐛',
@@ -83,6 +84,9 @@ class Config {
 		const customCommitMessageRules = getConfig('custom.commitMessageRules')
 		const customDescriptionPrompt = getConfig('custom.descriptionPrompt')
 
+        // Load request headers
+        const requestHeaders = getConfig('request.headers') || defaultConfig.requestHeaders
+
 		return {
 			commitEmojis,
 			promptTemperature,
@@ -97,6 +101,7 @@ class Config {
 			useDescription,
 			useEmojis,
 			useLowerCase,
+            requestHeaders,
 		}
 	}
 }

src/generator.ts

@@ -23,8 +23,9 @@ export async function generateStructuredCommit(
 		customTypeRules,
 		customCommitMessageRules,
 		customDescriptionPrompt,
+        requestHeaders,
 	} = config.inference
-	const ollama = new Ollama({ host: endpoint })
+	const ollama = new Ollama({ host: endpoint, headers: requestHeaders })
 
 	const typeRules =
 		customTypeRules ||

src/types/config.d.ts

@@ -16,4 +16,5 @@ export type ExtensionConfig = {
 	'custom.typeRules'?: string
 	'custom.commitMessageRules'?: string
 	'custom.descriptionPrompt'?: string
+    'request.headers': Record<string, string>
 }