refactor: adapt pr #70 to current code structure

Author: anjerodev

README.md

@@ -83,7 +83,7 @@ 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.
+- Custom 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
 

package.json

@@ -11,20 +11,10 @@
 	"engines": {
 		"vscode": "^1.98.0"
 	},
-	"categories": [
-		"Machine Learning",
-		"Programming Languages"
-	],
+	"categories": ["Machine Learning", "Programming Languages"],
 	"icon": "icon.jpg",
 	"license": "MIT",
-	"keywords": [
-		"code",
-		"assistant",
-		"ai",
-		"llm",
-		"commits",
-		"ollama"
-	],
+	"keywords": ["code", "assistant", "ai", "llm", "commits", "ollama"],
 	"main": "./out/extension.js",
 	"activationEvents": [],
 	"contributes": {
@@ -159,6 +149,11 @@
 						"type": "string",
 						"description": "Custom prompt to generate the commit description with.",
 						"order": 14
+					},
+					"commitollama.custom.requestHeaders": {
+						"type": "object",
+						"description": "Custom request headers to send with the Ollama request. Useful for authentication or custom headers.",
+						"order": 15
 					}
 				}
 			}
@@ -194,7 +189,5 @@
 	"dependencies": {
 		"ollama": "0.5.16"
 	},
-	"extensionDependencies": [
-		"vscode.git"
-	]
+	"extensionDependencies": ["vscode.git"]
 }

src/config.ts

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

src/generator.ts

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

src/types/config.d.ts

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