From c8f3a04f03a446d9af25c0f890507ef91f244a46 Mon Sep 17 00:00:00 2001
From: Your Name <your.email@example.com>
Date: Thu, 19 Feb 2026 03:07:35 +0800
Subject: [PATCH] perf(agent): remove slow llm optimize routing from auto graph

---
 lib/ai/auto-agent-graph.ts | 13 +++++--------
 1 file changed, 5 insertions(+), 8 deletions(-)

diff --git a/lib/ai/auto-agent-graph.ts b/lib/ai/auto-agent-graph.ts
index 470c412c9..553689fdb 100644
--- a/lib/ai/auto-agent-graph.ts
+++ b/lib/ai/auto-agent-graph.ts
@@ -1102,19 +1102,17 @@ const optimizeNode = async (state: AutoAgentStateType) => {
     };
   }
   if (state.analysis.intent === "image_edit" && !state.analysis.subIntent) {
-    const recommendedImageParams = await recommendImageParamsWithAgent(state.analysis, state);
+    const recommendedImageParams = heuristicImageParams(state.text, state.analysis);
     return { recommendedImageParams };
   }
   if (state.analysis.intent !== "image_edit" && state.analysis.intent !== "image_gen") {
     return {};
   }
-  const { optimizePromptWithLLM } = await import("./prompt-optimizer");
-  const optimized = await optimizePromptWithLLM({
+  const { optimizePrompt } = await import("./prompt-optimizer");
+  const optimized = optimizePrompt({
     originalPrompt: state.text,
     intent: state.analysis.intent,
     subIntent: state.analysis.subIntent,
-    historyMessages: state.historyMessages,
-    context: state.context ?? undefined,
   });
   const mergedParams = sanitizeRecommendedImageParams(
     {
@@ -1125,7 +1123,7 @@ const optimizeNode = async (state: AutoAgentStateType) => {
           : optimized.recommendedImageParams?.negativePrompt,
     },
     state.analysis.intent,
-  ) ?? (await recommendImageParamsWithAgent(state.analysis, state));
+  ) ?? heuristicImageParams(state.text, state.analysis);
 
   return {
     optimizedPrompt: optimized.optimizedPrompt,
@@ -1135,12 +1133,11 @@ const optimizeNode = async (state: AutoAgentStateType) => {
 
 const routeNode = async (state: AutoAgentStateType) => {
   const { routeToModel } = await import("./model-router");
-  const baseRoute = await routeToModel({
+  const route = await routeToModel({
     intent: state.analysis?.intent || "chat",
     subIntent: state.analysis?.subIntent,
     hasImage: state.hasImage,
   });
-  const route = await refineRouteWithAgent(state, baseRoute);
   return { route };
 };