lunanting/ai-chatbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf(agent): remove slow llm optimize routing from auto graph
Some checks failed
Lint / build (20) (push) Has been cancelled
Details
Playwright Tests / test (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Your Name 2026-02-19 03:07:35 +08:00
parent ee2026a973
commit c8f3a04f03
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
lib/ai/auto-agent-graph.ts
View File

@ -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 };
};