feat: refactor agent prompts and add utility functions for system prompt parsing

Author: francismiko

packages/service/core/workflow/dispatch/ai/agent/constants.ts

@@ -7,32 +7,13 @@ export enum SubAppIds {
   fileRead = 'file_read'
 }
 
-export const getTopAgentDefaultPrompt = () => {
-  return `你是一位Supervisor Agent，具备以下核心能力：
-
-## 核心能力
-1. **计划制定与管理**：根据用户需求制定详细的执行计划，并实时跟踪和调整计划进度
-2. **工具调用编排**：可以调用各种工具来完成特定任务，支持并行和串行工具调用
-3. **上下文理解**：能够理解对话历史、文档内容和当前状态
-4. **自主决策**：根据当前情况和计划进度做出最优决策
-
-## 工作流程
-1. **需求分析**：深入理解用户需求，识别关键目标和约束条件
-2. **计划制定**：使用 plan_agent 工具制定详细的执行计划
-3. **工具编排**：根据计划选择和调用合适的工具
-4. **结果处理**：分析工具返回结果，判断是否满足预期
-5. **计划调整**：根据执行结果动态调整计划, 再次使用 plan_agent 工具去更新计划进度
-6. **更新粒度**：将任务拆分为多个 TODO 去做, 确保每次得到结果后去更新一个或多个 TODO 的状态
-7. **最终输出**：给出完整、准确的回答
-
-## 特殊指令
+export const getTopAgentConstantPrompt = () => {
+  return `## 特殊指令
 - 对于复杂任务，必须先使用 plan_agent 制定计划
 - 在执行过程中如需调整计划，再次调用 plan_agent
 - 始终保持计划的可见性和可追踪性
 - 每次有新的进度完成时，都要调用 plan_agent 更新计划
-- 遇到错误时要有容错和重试机制
-
-请始终保持专业、准确、有条理的回答风格，确保用户能够清楚了解执行进度和结果。`;
+- 遇到错误时要有容错和重试机制`;
 };
 
 export const StopAgentTool: ChatCompletionTool = {

packages/service/core/workflow/dispatch/ai/agent/index.ts

@@ -34,7 +34,7 @@ import {
   initToolNodes,
   parseToolArgs
 } from '../utils';
-import { getFileReadTool, getTopAgentDefaultPrompt, StopAgentTool, SubAppIds } from './constants';
+import { getFileReadTool, getTopAgentConstantPrompt, StopAgentTool, SubAppIds } from './constants';
 import { runWorkflow } from '../..';
 import type { ChatCompletionTool } from '@fastgpt/global/core/ai/type';
 import type { ToolNodeItemType } from './type';
@@ -44,6 +44,8 @@ import { transferPlanAgent } from '../sub/plan';
 import { transferModelAgent } from '../sub/model';
 import { PlanAgentTool } from '../sub/plan/constants';
 import { ModelAgentTool } from '../sub/model/constants';
+import { getSubIdsByAgentSystem, parseAgentSystem } from './utils';
+import { getChildAppPreviewNode } from '../../../../../core/app/plugin/controller';
 
 export type DispatchAgentModuleProps = ModuleDispatchProps<{
   [NodeInputKeyEnum.history]?: ChatItemType[];
@@ -101,7 +103,8 @@ export const dispatchRunAgent = async (props: DispatchAgentModuleProps): Promise
     }
 
     // Init tool params
-    const toolNodeIds = filterToolNodeIdByEdges({ nodeId, edges: runtimeEdges });
+    // const toolNodeIds = filterToolNodeIdByEdges({ nodeId, edges: runtimeEdges });
+    const toolNodeIds = getSubIdsByAgentSystem(systemPrompt);
     const toolNodes = getToolNodesByIds({ toolNodeIds, runtimeNodes });
     // TODO: 补充系统 agent
     const toolNodesMap = new Map<string, ToolNodeItemType>();
@@ -118,10 +121,12 @@ export const dispatchRunAgent = async (props: DispatchAgentModuleProps): Promise
 
     const subApps = getSubApps({ toolNodes, urls: fileLinks });
 
+    const combinedSystemPrompt = `${parseAgentSystem({ systemPrompt, toolNodesMap })}\n\n${getTopAgentConstantPrompt()}`;
+
     // TODO: 把 files 加入 query 中。
     const messages: ChatItemType[] = (() => {
       const value: ChatItemType[] = [
-        ...getSystemPrompt_ChatItemType(systemPrompt || getTopAgentDefaultPrompt()),
+        ...getSystemPrompt_ChatItemType(combinedSystemPrompt),
         // Add file input prompt to histories
         ...chatHistories,
         {
@@ -436,7 +441,7 @@ const getSubApps = ({
         type: 'function',
         function: {
           name: item.nodeId,
-          description: item.intro || item.name,
+          description: `调用${item.flowNodeType}:${item.name || item.intro}完成任务`,
           parameters: item.jsonSchema
         }
       };
@@ -459,7 +464,7 @@ const getSubApps = ({
       type: 'function',
       function: {
         name: item.nodeId,
-        description: item.toolDescription || item.intro || item.name,
+        description: `调用${item.flowNodeType}:${item.name || item.toolDescription || item.intro}完成任务`,
         parameters: {
           type: 'object',
           properties,
@@ -468,6 +473,7 @@ const getSubApps = ({
       }
     };
   });
+  console.dir(nodeTools, { depth: null });
 
   return [...systemTools, ...nodeTools];
 };

packages/service/core/workflow/dispatch/ai/agent/utils.ts

@@ -0,0 +1,61 @@
+import type { ToolNodeItemType } from './type';
+
+const namespaceMap = new Map<string, string>([
+  ['a', '子应用'],
+  ['t', '工具'],
+  ['d', '知识库'],
+  ['m', '模型']
+]);
+
+// e.g: {{@a.appId@}} -> a.appId
+const buildPattern = (options?: { prefix?: string }): RegExp => {
+  const config = {
+    prefix: '@',
+    ...options
+  };
+
+  const escapedPrefix = config.prefix.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+  return new RegExp(`\\{\\{${escapedPrefix}([^${escapedPrefix}]+)${escapedPrefix}\\}\\}`, 'g');
+};
+
+export const getSubIdsByAgentSystem = (
+  systemPrompt: string,
+  options?: { prefix?: string }
+): string[] => {
+  const pattern = buildPattern(options);
+  const ids: string[] = [];
+  let match;
+
+  while ((match = pattern.exec(systemPrompt)) !== null) {
+    const fullName = match[1];
+    const [, id] = fullName.split('.');
+    if (id) {
+      ids.push(id);
+    }
+  }
+
+  return ids;
+};
+
+export const parseAgentSystem = ({
+  systemPrompt,
+  toolNodesMap,
+  options
+}: {
+  systemPrompt: string;
+  toolNodesMap: Map<string, ToolNodeItemType>;
+  options?: { prefix?: string };
+}): string => {
+  const pattern = buildPattern(options);
+
+  const processedPrompt = systemPrompt.replace(pattern, (_, toolName) => {
+    const [namespace, id] = toolName.split('.');
+    const toolNode = toolNodesMap.get(id);
+    const name = toolNode?.name || toolNode?.toolDescription || toolNode?.intro || 'unknown';
+
+    const prefix = namespaceMap.get(namespace) ?? 'unknown';
+    return `${prefix}:${name}`;
+  });
+
+  return processedPrompt;
+};