1 files changed, 46 insertions, 30 deletions
diff --git a/internal/llm/provider.go b/internal/llm/provider.go
index c605081..ed9ca59 100644
--- a/internal/llm/provider.go
+++ b/internal/llm/provider.go
@@ -54,40 +54,56 @@ func WithStop(stop ...string) RequestOption {
 
 // Config defines provider configuration read from the Hexai config file.
 type Config struct {
-	Provider string
-	// OpenAI options
-	OpenAIBaseURL string
-	OpenAIModel   string
-	// Ollama options
-	OllamaBaseURL string
-	OllamaModel   string
-	// Copilot options
-	CopilotBaseURL string
-	CopilotModel   string
+    Provider string
+    // OpenAI options
+    OpenAIBaseURL string
+    OpenAIModel   string
+    OpenAITemperature *float64
+    // Ollama options
+    OllamaBaseURL string
+    OllamaModel   string
+    OllamaTemperature *float64
+    // Copilot options
+    CopilotBaseURL string
+    CopilotModel   string
+    CopilotTemperature *float64
 }
 
 // NewFromConfig creates an LLM client using only the supplied configuration.
 // The OpenAI API key is supplied separately and may be read from the environment
 // by the caller; other environment-based configuration is not used.
 func NewFromConfig(cfg Config, openAIAPIKey, copilotAPIKey string) (Client, error) {
-	p := strings.ToLower(strings.TrimSpace(cfg.Provider))
-	if p == "" {
-		p = "openai"
-	}
-	switch p {
-	case "openai":
-		if strings.TrimSpace(openAIAPIKey) == "" {
-			return nil, errors.New("missing OPENAI_API_KEY for provider openai")
-		}
-		return newOpenAI(cfg.OpenAIBaseURL, cfg.OpenAIModel, openAIAPIKey), nil
-	case "ollama":
-		return newOllama(cfg.OllamaBaseURL, cfg.OllamaModel), nil
-	case "copilot":
-		if strings.TrimSpace(copilotAPIKey) == "" {
-			return nil, errors.New("missing COPILOT_API_KEY for provider copilot")
-		}
-		return newCopilot(cfg.CopilotBaseURL, cfg.CopilotModel, copilotAPIKey), nil
-	default:
-		return nil, errors.New("unknown LLM provider: " + p)
-	}
+    p := strings.ToLower(strings.TrimSpace(cfg.Provider))
+    if p == "" {
+        p = "openai"
+    }
+    switch p {
+    case "openai":
+        if strings.TrimSpace(openAIAPIKey) == "" {
+            return nil, errors.New("missing OPENAI_API_KEY for provider openai")
+        }
+        // Set coding-friendly default temperature if none provided
+        if cfg.OpenAITemperature == nil {
+            t := 0.2
+            cfg.OpenAITemperature = &t
+        }
+        return newOpenAI(cfg.OpenAIBaseURL, cfg.OpenAIModel, openAIAPIKey, cfg.OpenAITemperature), nil
+    case "ollama":
+        if cfg.OllamaTemperature == nil {
+            t := 0.2
+            cfg.OllamaTemperature = &t
+        }
+        return newOllama(cfg.OllamaBaseURL, cfg.OllamaModel, cfg.OllamaTemperature), nil
+    case "copilot":
+        if strings.TrimSpace(copilotAPIKey) == "" {
+            return nil, errors.New("missing COPILOT_API_KEY for provider copilot")
+        }
+        if cfg.CopilotTemperature == nil {
+            t := 0.2
+            cfg.CopilotTemperature = &t
+        }
+        return newCopilot(cfg.CopilotBaseURL, cfg.CopilotModel, copilotAPIKey, cfg.CopilotTemperature), nil
+    default:
+        return nil, errors.New("unknown LLM provider: " + p)
+    }
 }