feat: add Ollama backend

Signed-off-by: Kay Yan <[email protected]>
k8sgpt-ai · Apr 14, 2024 · f82145d · f82145d
1 parent 9dfcce8
commit f82145d
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -307,6 +307,7 @@ Active:
 Unused: 
 > openai
 > localai
+> ollama
 > azureopenai
 > cohere
 > amazonbedrock

diff --git a/pkg/ai/iai.go b/pkg/ai/iai.go
@@ -22,6 +22,7 @@ var (
 		&OpenAIClient{},
 		&AzureAIClient{},
 		&LocalAIClient{},
+		&OllamaClient{},
 		&NoOpAIClient{},
 		&CohereClient{},
 		&AmazonBedRockClient{},
@@ -33,6 +34,7 @@ var (
 	Backends = []string{
 		openAIClientName,
 		localAIClientName,
+		ollamaClientName,
 		azureAIClientName,
 		cohereAIClientName,
 		amazonbedrockAIClientName,
@@ -150,7 +152,7 @@ func (p *AIProvider) GetProviderId() string {
 	return p.ProviderId
 }
 
-var passwordlessProviders = []string{"localai", "amazonsagemaker", "amazonbedrock", "googlevertexai"}
+var passwordlessProviders = []string{"localai", "ollama", "amazonsagemaker", "amazonbedrock", "googlevertexai"}
 
 func NeedPassword(backend string) bool {
 	for _, b := range passwordlessProviders {

diff --git a/pkg/ai/ollama.go b/pkg/ai/ollama.go
@@ -0,0 +1,11 @@
+package ai
+
+const ollamaClientName = "ollama"
+
+type OllamaClient struct {
+	OpenAIClient
+}
+
+func (a *OllamaClient) GetName() string {
+	return ollamaClientName
+}