added ollama integration with the CLI

Author: vinyas-bharadwaj

cmd/cli/createMsg.go

@@ -12,6 +12,7 @@ import (
 	"github.com/dfanso/commit-msg/internal/gemini"
 	"github.com/dfanso/commit-msg/internal/git"
 	"github.com/dfanso/commit-msg/internal/grok"
+	"github.com/dfanso/commit-msg/internal/ollama"
 	"github.com/dfanso/commit-msg/internal/stats"
 	"github.com/dfanso/commit-msg/pkg/types"
 	"github.com/pterm/pterm"
@@ -106,6 +107,17 @@ func CreateCommitMsg () {
 
 		case "Claude":
 			commitMsg, err = claude.GenerateCommitMessage(config, changes, apiKey)
+		case "Ollama":
+			url := os.Getenv("OLLAMA_URL")
+			if url == "" {
+				url = "http://localhost:11434/api/generate"
+			}
+			model := os.Getenv("OLLAMA_MODEL")
+			if model == "" {
+				model = "llama3:latest"
+			}
+			commitMsg, err = ollama.GenerateCommitMessage(config, changes, url, model)
+
 
 		default:
 			commitMsg, err = grok.GenerateCommitMessage(config, changes, apiKey)

cmd/cli/llmSetup.go

@@ -11,7 +11,7 @@ import (
 
 func SetupLLM() error {
 
-	providers := []string{"OpenAI", "Claude", "Gemini", "Grok"}
+	providers := []string{"OpenAI", "Claude", "Gemini", "Grok", "Ollama"}
 	prompt := promptui.Select{
 		Label: "Select LLM",
 		Items: providers,
@@ -22,15 +22,21 @@ func SetupLLM() error {
 		return fmt.Errorf("prompt failed")
 	}
 
-	apiKeyPrompt := promptui.Prompt{
-		Label: "Enter API Key",
-		Mask: '*',
-		
-	}
-
-	apiKey, err := apiKeyPrompt.Run()
-	if err != nil {
-		return fmt.Errorf("failed to read API Key: %w", err)
+	var apiKey string
+	
+	// Skip API key prompt for Ollama (local LLM)
+	if model != "Ollama" {
+		apiKeyPrompt := promptui.Prompt{
+			Label: "Enter API Key",
+			Mask: '*',
+		}
+
+		apiKey, err = apiKeyPrompt.Run()
+		if err != nil {
+			return fmt.Errorf("failed to read API Key: %w", err)
+		}
+	} else {
+		apiKey = "" // No API key needed for Ollama
 	}
 
 	LLMConfig := store.LLMProvider{