From 05c73bbbdf341f9e8625b0f57027676a6a45228e Mon Sep 17 00:00:00 2001
From: Carlos Alexandro Becker <caarlos0@users.noreply.github.com>
Date: Tue, 22 Jul 2025 12:16:11 -0300
Subject: [PATCH] fix: improv diff

---
 internal/llm/agent/agent.go | 36 ++++++++++++++++++------------------
 1 file changed, 18 insertions(+), 18 deletions(-)

diff --git a/internal/llm/agent/agent.go b/internal/llm/agent/agent.go
index 84ef27a86e1a91557a7af88c868fdd4ba8ce9362..501cf3aad0d1f6fa57dbda1f055e7f23a855b862 100644
--- a/internal/llm/agent/agent.go
+++ b/internal/llm/agent/agent.go
@@ -98,6 +98,24 @@ func NewAgent(
 	ctx := context.Background()
 	cfg := config.Get()
 
+	var agentTool tools.BaseTool
+	if agentCfg.ID == "coder" {
+		taskAgentCfg := config.Get().Agents["task"]
+		if taskAgentCfg.ID == "" {
+			return nil, fmt.Errorf("task agent not found in config")
+		}
+		taskAgent, err := NewAgent(taskAgentCfg, permissions, sessions, messages, history, lspClients)
+		if err != nil {
+			return nil, fmt.Errorf("failed to create task agent: %w", err)
+		}
+
+		agentTool = NewAgentTool(
+			taskAgent,
+			sessions,
+			messages,
+		)
+	}
+
 	providerCfg := config.Get().GetProviderForModel(agentCfg.Model)
 	if providerCfg == nil {
 		return nil, fmt.Errorf("provider for agent %s not found in config", agentCfg.Name)
@@ -154,24 +172,6 @@ func NewAgent(
 		return nil, err
 	}
 
-	var agentTool tools.BaseTool
-	if agentCfg.ID == "coder" {
-		taskAgentCfg := config.Get().Agents["task"]
-		if taskAgentCfg.ID == "" {
-			return nil, fmt.Errorf("task agent not found in config")
-		}
-		taskAgent, err := NewAgent(taskAgentCfg, permissions, sessions, messages, history, lspClients)
-		if err != nil {
-			return nil, fmt.Errorf("failed to create task agent: %w", err)
-		}
-
-		agentTool = NewAgentTool(
-			taskAgent,
-			sessions,
-			messages,
-		)
-	}
-
 	agent := &agent{
 		Broker:              pubsub.NewBroker[AgentEvent](),
 		agentCfg:            agentCfg,