Merge pull request #2 from florianbuetow/feature-timeout-config

feat: add configurable request timeout for LLM calls

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,