diff options
Diffstat (limited to 'internal/llm/openrouter.go')
| -rw-r--r-- | internal/llm/openrouter.go | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/internal/llm/openrouter.go b/internal/llm/openrouter.go index 21e3102..8aae6b8 100644 --- a/internal/llm/openrouter.go +++ b/internal/llm/openrouter.go @@ -22,6 +22,23 @@ type openRouterClient struct { defaultTemperature *float64 } +func init() { + RegisterProvider("openrouter", openRouterProviderFactory) +} + +func openRouterProviderFactory(cfg Config, keys ProviderKeys) (Client, error) { + if strings.TrimSpace(keys.OpenRouterAPIKey) == "" { + return nil, errors.New("missing OPENROUTER_API_KEY for provider openrouter") + } + return newOpenRouterWithTimeout( + cfg.OpenRouterBaseURL, + cfg.OpenRouterModel, + keys.OpenRouterAPIKey, + withDefaultTemperature(cfg.OpenRouterTemperature, 0.2), + cfg.RequestTimeout, + ), nil +} + func newOpenRouter(baseURL, model, apiKey string, defaultTemp *float64) Client { return newOpenRouterWithTimeout(baseURL, model, apiKey, defaultTemp, 0) } |
