update the skill system for inner abilities

Author: wangenius

src/agent/Agent.ts

@@ -27,6 +27,7 @@ export const DEFAULT_AGENT: AgentProps = {
   knowledges: [],
   workflows: [],
   agents: [],
+  skills: [],
 };
 
 export const AgentStore = new Echo<Record<string, AgentProps>>({}).indexed({

src/agent/engine/Engine.ts

@@ -93,6 +93,7 @@ export class Engine {
         ...(await ToolsHandler.transformKnowledgeToModelFormat(
           props.knowledges || [],
         )),
+        ...(await ToolsHandler.transformSkillToModelFormat(props.skills)),
       ]);
 
     // 初始化内存和上下文

src/agent/types/agent.ts

@@ -64,6 +64,8 @@ export interface AgentProps {
   workflows: string[];
   /* 允许调用的其他agents */
   agents: string[];
+  /* 可调用的技能 */
+  skills: string[];
   /* 配置 */
   configs?: {
     /* 温度 */

src/assets/const.ts

@@ -2,6 +2,7 @@ export const TOOL_NAME_SPLIT = "--";
 export const WORKFLOW_TOOL_NAME_PREFIX = "executeFlow";
 export const KNOWLEDGE_TOOL_NAME_PREFIX = "knowledge";
 export const AGENT_TOOL_NAME_PREFIX = "agent";
+export const SKILL_TOOL_NAME_PREFIX = "skill";
 export const WORKFLOW_BODY_DATABASE = "workflow_bodies";
 export const WORKFLOW_DATABASE = "workflows";
 export const KNOWLEDGE_DATABASE = "knowledge";

src/main.tsx

@@ -10,6 +10,7 @@ import "./model/chat/provider";
 import "./model/embedding/provider";
 import "./model/image/provider";
 import "./model/vision/provider";
+import "./skills/instance";
 import App from "./page/App";
 
 const element = document.getElementById("root") as HTMLElement;

src/model/chat/ChatModel.ts

@@ -267,7 +267,11 @@ export class ChatModel {
               created_at: Date.now(),
             },
           ]);
-          toolResult = await ToolsHandler.call(tool_call, this.otherModels);
+          toolResult = await ToolsHandler.call(
+            tool_call,
+            this.otherModels,
+            this,
+          );
           console.log(toolResult);
 
           if (

src/model/chat/ToolsHandler.ts

@@ -3,6 +3,7 @@ import { AgentMCPProps, AgentProps, AgentToolProps } from "@/agent/types/agent";
 import {
   AGENT_TOOL_NAME_PREFIX,
   KNOWLEDGE_TOOL_NAME_PREFIX,
+  SKILL_TOOL_NAME_PREFIX,
   TOOL_NAME_SPLIT,
   WORKFLOW_BODY_DATABASE,
   WORKFLOW_TOOL_NAME_PREFIX,
@@ -22,6 +23,8 @@ import {
   ToolRequestBody,
 } from "../types/chatModel";
 import { VisionModel } from "../vision/VisionModel";
+import { SkillManager } from "@/skills/SkillManager";
+import { ChatModel } from "./ChatModel";
 
 export class ToolsHandler {
   static async transformAgentToolToModelFormat(
@@ -235,9 +238,35 @@ export class ToolsHandler {
     return toolRequestBody;
   }
 
+  static async transformSkillToModelFormat(
+    skills: string[],
+  ): Promise<ToolRequestBody> {
+    const toolRequestBody: ToolRequestBody = [];
+    if (skills.length) {
+      /* 获取代理 */
+      const list = SkillManager.getSkills();
+
+      /* 将知识库变成工具 */
+      skills.forEach((skill) => {
+        const skillDoc = list[skill];
+        if (!skillDoc) return;
+        toolRequestBody.push({
+          type: "function",
+          function: {
+            name: `${SKILL_TOOL_NAME_PREFIX}${TOOL_NAME_SPLIT}${skill}`,
+            description: skillDoc.description,
+            parameters: skillDoc.params,
+          },
+        });
+      });
+    }
+    return toolRequestBody;
+  }
+
   static async call(
     tool_call: ToolCallReply,
     otherModels: AgentProps["models"],
+    chatModel: ChatModel,
   ): Promise<FunctionCallResult | undefined> {
     if (!tool_call) return;
 
@@ -343,7 +372,14 @@ export class ToolsHandler {
           result,
         };
       }
-
+      if (firstName === SKILL_TOOL_NAME_PREFIX) {
+        const result = await SkillManager.execute(secondName, query, chatModel);
+        return {
+          name: tool_call.function.name,
+          arguments: tool_call.function.arguments,
+          result,
+        };
+      }
       if (firstName === KNOWLEDGE_TOOL_NAME_PREFIX) {
         if (!query.query) {
           return {
@@ -402,6 +438,13 @@ export class ToolsHandler {
         name: `calling agent: ${list[secondName].name}`,
       };
     }
+    if (firstName === SKILL_TOOL_NAME_PREFIX) {
+      const list = SkillManager.getSkills();
+      return {
+        type: "skill",
+        name: `calling skill: ${list[secondName].name}`,
+      };
+    }
 
     if (firstName === WORKFLOW_TOOL_NAME_PREFIX) {
       const list = await WorkflowsStore.getCurrent();

src/page/agent/AgentEditor.tsx

@@ -26,6 +26,7 @@ import { TbListSearch, TbUpload } from "react-icons/tb";
 import { toast } from "sonner";
 import { MCPTool, MCP_Actived } from "../mcp/MCP";
 import { SettingItem } from "../settings/components/SettingItem";
+import { SkillManager } from "@/skills/SkillManager";
 
 export const AgentEditor = ({ agent }: { agent: Agent }) => {
   const list = KnowledgesStore.use();
@@ -296,6 +297,24 @@ export const AgentEditor = ({ agent }: { agent: Agent }) => {
                 multiple
                 placeholder="Select MCP..."
               />
+              <DrawerSelector
+                title="Select Skill"
+                value={props.skills}
+                items={Object.entries(SkillManager.getSkills()).map(
+                  ([id, skill]) => ({
+                    label: skill.name,
+                    value: id,
+                    description: skill.description,
+                  }),
+                )}
+                onSelect={(value) =>
+                  agent.update({
+                    skills: value,
+                  })
+                }
+                multiple
+                placeholder="Select Skill..."
+              />
               <DrawerSelector
                 title="Select Knowledge"
                 value={props.knowledges || []}

src/skills/Skill.ts

@@ -1,22 +1,42 @@
-import { Skill } from "./Skill";
+/* 技能方法接口 */
+import { ChatModel } from "@/model/chat/ChatModel";
+import { ToolParameters } from "@/plugin/types";
 
-export interface SkillProps {
+/** 方法元数据 */
+export interface Skill {
   name: string;
   description: string;
+  params: ToolParameters;
+  execute: (params: Record<string, any>, chatModel: ChatModel) => any;
 }
 
-/** Skill 管理器 */
-export class SkillManager {
-  /** 已注册的 Engine */
-  private static readonly skills: Record<string, Skill> = {};
+/** 方法管理器 */
+export const SkillManager = {
+  /** 已注册的方法 */
+  skills: {} as Record<string, Skill>,
 
-  /** 注册 Engine */
-  public static register(type: string, skill: Skill): void {
+  /** 获取所有已注册的技能 */
+  getSkills(): Record<string, Skill> {
+    return this.skills;
+  },
+  /** 注册技能 */
+  register(type: string, skill: Skill): void {
     this.skills[type] = skill;
-  }
-
-  /** 获取指定的 Engine */
-  public static getSkill(type: string): Skill | undefined {
+  },
+  /** 获取指定的技能 */
+  getSkill(type: string): Skill | undefined {
     return this.skills[type];
-  }
-}
+  },
+  /** 执行指定方法 */
+  execute(
+    skillType: string,
+    params: Record<string, any> = {},
+    chatModel: ChatModel,
+  ): any {
+    const skill = this.getSkill(skillType);
+    if (!skill) {
+      throw new Error(`技能 ${skillType} 不存在`);
+    }
+    return skill.execute(params, chatModel);
+  },
+};