opencode/picoclaw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: add proper Ollama provider support without requiring API key

Browse source
This commit is contained in:
Ashish Sharma 2026-02-17 21:59:42 +05:30
parent 0d16525fab
commit 032f107935
1 changed files with 13 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
pkg/providers/http_provider.go
View file

@ -324,6 +324,13 @@ func CreateProvider(cfg *config.Config) (LLMProvider, error) {
model = "deepseek-chat" model = "deepseek-chat"
} }
} }
case "ollama":
apiKey = ""
apiBase = cfg.Providers.Ollama.APIBase
if apiBase == "" {
apiBase = "http://localhost:11434/v1"
}
return NewHTTPProvider(apiKey, apiBase, proxy), nil
case "github_copilot", "copilot": case "github_copilot", "copilot":
if cfg.Providers.GitHubCopilot.APIBase != "" { if cfg.Providers.GitHubCopilot.APIBase != "" {
apiBase = cfg.Providers.GitHubCopilot.APIBase apiBase = cfg.Providers.GitHubCopilot.APIBase
@ -331,9 +338,7 @@ func CreateProvider(cfg *config.Config) (LLMProvider, error) {
apiBase = "localhost:4321" apiBase = "localhost:4321"
} }
return NewGitHubCopilotProvider(apiBase, cfg.Providers.GitHubCopilot.ConnectMode, model) return NewGitHubCopilotProvider(apiBase, cfg.Providers.GitHubCopilot.ConnectMode, model)
} }
} }
// Fallback: detect provider from model name // Fallback: detect provider from model name
@ -409,15 +414,15 @@ func CreateProvider(cfg *config.Config) (LLMProvider, error) {
if apiBase == "" { if apiBase == "" {
apiBase = "https://integrate.api.nvidia.com/v1" apiBase = "https://integrate.api.nvidia.com/v1"
} }
case (strings.Contains(lowerModel, "ollama") || strings.HasPrefix(model, "ollama/")) && cfg.Providers.Ollama.APIKey != "":
fmt.Println("Ollama provider selected based on model name prefix") case cfg.Providers.Ollama.APIBase != "":
apiKey = cfg.Providers.Ollama.APIKey apiKey = ""
apiBase = cfg.Providers.Ollama.APIBase apiBase = cfg.Providers.Ollama.APIBase
proxy = cfg.Providers.Ollama.Proxy proxy = cfg.Providers.Ollama.Proxy
if apiBase == "" { if apiBase == "" {
apiBase = "http://localhost:11434/v1" apiBase = "http://localhost:11434/v1"
} }
fmt.Println("Ollama apiBase:", apiBase)
case cfg.Providers.VLLM.APIBase != "": case cfg.Providers.VLLM.APIBase != "":
apiKey = cfg.Providers.VLLM.APIKey apiKey = cfg.Providers.VLLM.APIKey
apiBase = cfg.Providers.VLLM.APIBase apiBase = cfg.Providers.VLLM.APIBase
@ -438,7 +443,7 @@ func CreateProvider(cfg *config.Config) (LLMProvider, error) {
} }
} }
if apiKey == "" && !strings.HasPrefix(model, "bedrock/") { if apiKey == "" && !strings.HasPrefix(model, "bedrock/") && !strings.Contains(apiBase, "11434") {
return nil, fmt.Errorf("no API key configured for provider (model: %s)", model) return nil, fmt.Errorf("no API key configured for provider (model: %s)", model)
} }
@ -447,4 +452,4 @@ func CreateProvider(cfg *config.Config) (LLMProvider, error) {
} }
return NewHTTPProvider(apiKey, apiBase, proxy), nil return NewHTTPProvider(apiKey, apiBase, proxy), nil
} }