diff options
Diffstat (limited to 'internal/llm/openrouter.go')
| -rw-r--r-- | internal/llm/openrouter.go | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/internal/llm/openrouter.go b/internal/llm/openrouter.go index 4aae398..21e3102 100644 --- a/internal/llm/openrouter.go +++ b/internal/llm/openrouter.go @@ -23,14 +23,21 @@ type openRouterClient struct { } func newOpenRouter(baseURL, model, apiKey string, defaultTemp *float64) Client { + return newOpenRouterWithTimeout(baseURL, model, apiKey, defaultTemp, 0) +} + +func newOpenRouterWithTimeout(baseURL, model, apiKey string, defaultTemp *float64, timeoutSec int) Client { if strings.TrimSpace(baseURL) == "" { baseURL = "https://openrouter.ai/api/v1" } if strings.TrimSpace(model) == "" { model = "openrouter/auto" } + if timeoutSec <= 0 { + timeoutSec = 30 + } return openRouterClient{ - httpClient: &http.Client{Timeout: 30 * time.Second}, + httpClient: &http.Client{Timeout: time.Duration(timeoutSec) * time.Second}, apiKey: apiKey, baseURL: baseURL, defaultModel: model, |