summaryrefslogtreecommitdiff
path: root/internal/llm/openrouter.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/llm/openrouter.go')
-rw-r--r--internal/llm/openrouter.go17
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)
}