@shuttle-ai/agent 0.0.1

package/README.md

@@ -0,0 +1,5 @@
+## `@shuttle-ai/agent`
+
+#### 说明
+
+该部分是shuttle-ai的智能体操作包

package/dist/cjs/callback/LLMMessage.cjs

@@ -0,0 +1 @@
+"use strict";Object.defineProperties(exports,{__esModule:{value:!0},[Symbol.toStringTag]:{value:"Module"}});const d=require("@langchain/core/callbacks/base"),o=require("node:crypto"),a=require("../cluster/instance.cjs");class l extends d.BaseCallbackHandler{constructor(t){super(),this.options=t}name="LLMMessage";handleLLMNewToken(t,n,s){t&&this.options.agentCluster.options.hooks.onChunk?.({role:"assistant_chunk",content:t,id:s,agentId:this.options.agentId,workId:this.options.agentCluster.id})}handleLLMEnd(t,n){const s=t.generations[0][0].message,i=s.tool_calls?.filter(e=>e.name!==a.default.CALL_SUB_AGENT_NAME)?.map(e=>({id:e.id||o.randomUUID(),name:e.name,args:e.args})),r=s.tool_calls?.filter(e=>e.name===a.default.CALL_SUB_AGENT_NAME)?.map(e=>({id:e.id||o.randomUUID(),name:e.args.subAgentName}));this.options.agentCluster.addMessage({role:"assistant",content:s.content,toolCalls:i,id:n,subAgents:r,agentId:this.options.agentId,workId:this.options.agentCluster.id})}}exports.default=l;

package/dist/cjs/cluster/fileMessageCollector.cjs

@@ -0,0 +1 @@
+"use strict";Object.defineProperties(exports,{__esModule:{value:!0},[Symbol.toStringTag]:{value:"Module"}});const r=require("node:path"),i=require("node:fs/promises");class g{constructor(s){this.dirPath=s}async saveMessage(s){const t=await this.getMessagesByAgentId(s.workId,s.agentId),e=r.join(this.dirPath,`${s.workId}`,`${s.agentId}.json`);t.push(s),await i.mkdir(r.dirname(e),{recursive:!0}),await i.writeFile(e,JSON.stringify(t,null,2))}async getMessagesByAgentId(s,t){const e=r.join(this.dirPath,`${s}`,`${t}.json`);try{const a=await i.readFile(e,"utf8");return JSON.parse(a)}catch{return[]}}async getSubAgentTasks(s,t){const e=[];for(const a of t){const n=(await this.getMessagesByAgentId(s,a)).find(c=>c.role==="user");n&&e.push(n)}return e}}exports.default=g;

package/dist/cjs/cluster/instance.cjs

@@ -0,0 +1,5 @@
+"use strict";Object.defineProperties(exports,{__esModule:{value:!0},[Symbol.toStringTag]:{value:"Module"}});const _=require("@langchain/core/runnables"),u=require("langchain"),l=require("@langchain/core/messages"),d=require("@langchain/core/tools"),c=require("zod"),g=require("node:crypto"),p=require("../middleware/dynamicToolInterceptor.cjs"),h=require("../callback/LLMMessage.cjs");class i extends _.Runnable{constructor(t){super(),this.options=t,this.id=t.id||g.randomUUID(),t.single?.addEventListener("abort",()=>{this.stop()})}static MAIN_AGENT_NAME="main_agent";static CALL_SUB_AGENT_NAME="call_sub_agent";lc_namespace=["shuttle-ai","agent","cluster"];id;messages=[];abortController=new AbortController;addMessage(t){this.messages.push(t),this.options.messageCollector?.saveMessage(t)}async invoke(t,s){const e=await this.options.hooks.onAgentStart({agentId:this.id,agentName:i.MAIN_AGENT_NAME,content:t}),a=this.createAgent(e,this.id);this.addMessage({id:g.randomUUID(),workId:this.id,agentId:this.id,role:"user",content:t});const r=await this.revokeMessages(this.id),o=a.streamEvents({messages:[...r,{role:"user",content:t}]},{signal:this.abortController.signal,context:{...s?.context||{},_agentCluster:this,_agentId:this.id},callbacks:[new h.default({agentCluster:this,agentId:this.id})]});return await Array.fromAsync(o),this.options.hooks.onAgentEnd?.(this.id),this.getLastAiMessage(this.id)?.content||""}stop(t="stop"){this.abortController.abort(t)}createAgent({tools:t,subAgents:s,middleware:e,...a},r){let o=[];return t?.length&&(o=this.normalizationTools(t)),s?.length&&o.push(this.subAgentToDynamicTool(s,r)),u.createAgent({...a,middleware:[...e||[],p.default,u.todoListMiddleware()],tools:o})}normalizationTools(t){return t.map(s=>"lc_namespace"in s||"lc_name"in s?s:this.customToolToDynamicTool(s))}customToolToDynamicTool(t){return d.tool(s=>"",t)}subAgentToDynamicTool(t,s){return d.tool(async(e,a)=>{if(!t.find(b=>b.name===e.subAgentName))throw new Error(`Sub-agent ${e.subAgentName} not found.`);const o=this.getLastAiMessage(s),n=a.toolCall.id,m=await this.options.hooks.onAgentStart({agentId:n,agentName:e.subAgentName,beloneMessageId:o?.id,parentAgentId:s,content:e.request}),A=this.createAgent(m,n);this.addMessage({id:g.randomUUID(),workId:this.id,agentId:n,role:"user",content:e.request});const M=A.streamEvents({messages:[{role:"user",content:e.request}]},{signal:this.abortController.signal,context:{...a.context||{},_agentCluster:this,_agentId:n,_parentAgentId:s},callbacks:[new h.default({agentCluster:this,agentId:n})]});return await Array.fromAsync(M),this.options.hooks.onAgentEnd?.(n),this.getLastAiMessage(n)?.content||""},{name:i.CALL_SUB_AGENT_NAME,description:`Call a sub-agent to handle the user request.
+      The sub-agents are: 
+      ${t.map(e=>`- ${e.name}: ${e.description}`).join(`
+`)}
+      `,schema:c.z.object({subAgentName:c.z.literal(t.map(e=>e.name)).describe("The name of the sub-agent to call."),request:c.z.string().describe("The request to send to the sub-agent.")}),extras:{scope:"autoRun",skipReport:!0}})}getLastAiMessage(t){for(let s=this.messages.length-1;s>=0;s--){const e=this.messages[s];if(e.agentId===t&&e.role==="assistant")return e}}getMessages(){return this.messages}async revokeMessages(t){return this.options.messageCollector?(await this.options.messageCollector.getMessagesByAgentId(this.id,t)).map(e=>e.role==="user"?new l.HumanMessage(e.content):e.role==="assistant"?new l.AIMessage({content:e.content,tool_calls:e.toolCalls}):new l.ToolMessage({content:e.content||e.confirm?.result,tool_call_id:e.id})):[]}}exports.default=i;

package/dist/cjs/cluster/readableHook.cjs

@@ -0,0 +1,4 @@
+"use strict";Object.defineProperties(exports,{__esModule:{value:!0},[Symbol.toStringTag]:{value:"Module"}});const A=require("node:stream");function E(i,n){const r={},s={},c=new A.Readable({read(){}});let d=Date.now();function o(e){c.push(`${JSON.stringify(e)}
+
+`),d=Date.now()}const v=setInterval(()=>{Date.now()-d>20*1e3&&o({type:"ping",data:void 0})},30*1e3);function p(){c.push(null),clearInterval(v)}function y(e,t){if(!s[e])throw new Error(`not find confirm tool resolver: ${e}`);s[e](t),delete s[e]}function w(e,t){if(!r[e])throw new Error(`not find init agent resolver: ${e}`);r[e](t),delete r[e]}return{hooks:{onChunk(e){a("onChunk",[e],n),o({type:"chunk",data:{chunk:e}})},async onAgentStart(e){a("onAgentStart",[e],n),o({type:"agentStart",data:e});const{promise:t,resolve:m}=Promise.withResolvers();r[e.agentId]=m;const u=await t,l=await i(e.agentName),f=[l?.systemPrompt||"",u.systemPrompt||""].filter(Boolean),S=(u.tools||[]).map(g=>({...g,extras:{...g.extras,remote:!0}}));return{...l,systemPrompt:f.length>0?f.join(`
+`):void 0,tools:[...l?.tools||[],...S],subAgents:[...l?.subAgents||[],...u.subAgents||[]]}},onAgentEnd(e){a("onAgentEnd",[e],n),o({type:"agentEnd",data:{agentId:e}})},async onToolStart(e){if(a("onToolStart",[e],n),o({type:"toolStart",data:{tool:e}}),e.needConfirm){const{resolve:t,promise:m}=Promise.withResolvers();return s[e.toolCall.id]=t,m}return{type:"confirm"}},onToolEnd(e,t){a("onToolEnd",[e,t],n),o({type:"toolEnd",data:{toolPath:e,toolResult:t}})}},stream:c,send:o,close:p,resolveConfirmTool:y,resolveAgentStart:w}}function a(i,n,r){r?.forEach(s=>s[i]?.(...n))}exports.default=E;

package/dist/cjs/index.cjs

@@ -0,0 +1 @@
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const e=require("./cluster/instance.cjs"),r=require("./cluster/readableHook.cjs"),t=require("./cluster/fileMessageCollector.cjs"),o=require("./callback/LLMMessage.cjs"),l=require("./middleware/dynamicToolInterceptor.cjs");require("@shuttle-ai/type");exports.AgentCluster=e.default;exports.readableHook=r.default;exports.FileMessageCollector=t.default;exports.LLMMessage=o.default;exports.dynamicToolInterceptorMiddleware=l.default;

package/dist/cjs/middleware/dynamicToolInterceptor.cjs

@@ -0,0 +1 @@
+"use strict";Object.defineProperties(exports,{__esModule:{value:!0},[Symbol.toStringTag]:{value:"Module"}});const C=require("langchain"),u=require("@langchain/core/tools"),s=require("@langchain/core/messages"),M=require("../cluster/instance.cjs"),p=C.createMiddleware({name:"dynamicToolInterceptorMiddleware",wrapToolCall:async(o,i)=>{const e=o.runtime.context,a={role:"tool",name:o.toolCall.name,id:o.toolCall.id||"",aiMessageId:"",agentId:e._agentId,workId:e._agentCluster.id};let t;if(o.tool instanceof u.DynamicTool||o.tool instanceof u.DynamicStructuredTool||o.tool&&"extras"in o.tool){const l=o.tool.extras||{};if(!l.skipReport){const f=!w(l,e._agentCluster),r=e._agentCluster.getLastAiMessage(e._agentId);a.aiMessageId=r?.id||"";const _=r?.toolCalls?.find(c=>c.id===o.toolCall.id),n=await e._agentCluster.options.hooks.onToolStart({role:"assistant_tool",toolCall:_,needConfirm:l.remote||f,id:r?.id||"",agentId:e._agentId,workId:e._agentCluster.id});if(a.confirm=n,n.type==="reject"||n.type==="confirm"&&l.remote)return e._agentCluster.addMessage(a),new s.ToolMessage({content:n.reason||"can not run tool",tool_call_id:o.toolCall.id||""});if(n.type==="confirmWithResult")return e._agentCluster.addMessage(a),new s.ToolMessage({content:typeof n.result=="object"?JSON.stringify(n.result):n.result||"success",tool_call_id:o.toolCall.id||""});const d={agentId:e._agentId,messageId:r?.id||"",toolId:o.toolCall.id||""};if(n.newArgs){o.toolCall.args=n.newArgs;const g=o.state.messages[o.state.messages.length-1].tool_calls?.find(m=>m.id===o.toolCall.id);g&&(g.args=n.newArgs)}try{t=await i(o),t instanceof s.ToolMessage||"content"in t?e._agentCluster.options.hooks.onToolEnd?.(d,t.content):e._agentCluster.options.hooks.onToolEnd?.(d,"{}")}catch(c){t=new s.ToolMessage({content:c.message||"can not run tool",tool_call_id:o.toolCall.id||""})}}}if(!t)try{t=await i(o)}catch(l){t=new s.ToolMessage({content:l.message||"can not run tool",tool_call_id:o.toolCall.id||""})}return"lg_name"in t&&t.lg_name==="Command"&&(t=t?.update?.messages?.[0]),t instanceof s.ToolMessage&&t.name!==M.default.CALL_SUB_AGENT_NAME&&(a.content=t.content,e._agentCluster.addMessage(a)),t}});function w(o,i){return o.scope==="autoRun"||i.options.autoRunScope==="always"||i.options.autoRunScope==="read"&&o.scope!=="write"}exports.default=p;

package/dist/es/callback/LLMMessage.mjs

@@ -0,0 +1,40 @@
+import { BaseCallbackHandler as d } from "@langchain/core/callbacks/base";
+import { randomUUID as o } from "node:crypto";
+import a from "../cluster/instance.mjs";
+class u extends d {
+  constructor(s) {
+    super(), this.options = s;
+  }
+  name = "LLMMessage";
+  handleLLMNewToken(s, n, e) {
+    s && this.options.agentCluster.options.hooks.onChunk?.({
+      role: "assistant_chunk",
+      content: s,
+      id: e,
+      agentId: this.options.agentId,
+      workId: this.options.agentCluster.id
+    });
+  }
+  handleLLMEnd(s, n) {
+    const e = s.generations[0][0].message, i = e.tool_calls?.filter((t) => t.name !== a.CALL_SUB_AGENT_NAME)?.map((t) => ({
+      id: t.id || o(),
+      name: t.name,
+      args: t.args
+    })), r = e.tool_calls?.filter((t) => t.name === a.CALL_SUB_AGENT_NAME)?.map((t) => ({
+      id: t.id || o(),
+      name: t.args.subAgentName
+    }));
+    this.options.agentCluster.addMessage({
+      role: "assistant",
+      content: e.content,
+      toolCalls: i,
+      id: n,
+      subAgents: r,
+      agentId: this.options.agentId,
+      workId: this.options.agentCluster.id
+    });
+  }
+}
+export {
+  u as default
+};

package/dist/es/cluster/fileMessageCollector.mjs

@@ -0,0 +1,40 @@
+import r from "node:path";
+import i from "node:fs/promises";
+class u {
+  constructor(s) {
+    this.dirPath = s;
+  }
+  async saveMessage(s) {
+    const e = await this.getMessagesByAgentId(
+      s.workId,
+      s.agentId
+    ), t = r.join(
+      this.dirPath,
+      `${s.workId}`,
+      `${s.agentId}.json`
+    );
+    e.push(s), await i.mkdir(r.dirname(t), { recursive: !0 }), await i.writeFile(t, JSON.stringify(e, null, 2));
+  }
+  async getMessagesByAgentId(s, e) {
+    const t = r.join(this.dirPath, `${s}`, `${e}.json`);
+    try {
+      const a = await i.readFile(t, "utf8");
+      return JSON.parse(a);
+    } catch {
+      return [];
+    }
+  }
+  async getSubAgentTasks(s, e) {
+    const t = [];
+    for (const a of e) {
+      const n = (await this.getMessagesByAgentId(s, a)).find(
+        (g) => g.role === "user"
+      );
+      n && t.push(n);
+    }
+    return t;
+  }
+}
+export {
+  u as default
+};

package/dist/es/cluster/instance.mjs

@@ -0,0 +1,170 @@
+import { Runnable as M } from "@langchain/core/runnables";
+import { createAgent as p, todoListMiddleware as b } from "langchain";
+import { HumanMessage as f, AIMessage as T, ToolMessage as _ } from "@langchain/core/messages";
+import { tool as c } from "@langchain/core/tools";
+import { z as i } from "zod";
+import { randomUUID as l } from "node:crypto";
+import w from "../middleware/dynamicToolInterceptor.mjs";
+import d from "../callback/LLMMessage.mjs";
+class g extends M {
+  constructor(e) {
+    super(), this.options = e, this.id = e.id || l(), e.single?.addEventListener("abort", () => {
+      this.stop();
+    });
+  }
+  static MAIN_AGENT_NAME = "main_agent";
+  static CALL_SUB_AGENT_NAME = "call_sub_agent";
+  lc_namespace = ["shuttle-ai", "agent", "cluster"];
+  id;
+  messages = [];
+  abortController = new AbortController();
+  addMessage(e) {
+    this.messages.push(e), this.options.messageCollector?.saveMessage(e);
+  }
+  async invoke(e, s) {
+    const t = await this.options.hooks.onAgentStart({
+      agentId: this.id,
+      agentName: g.MAIN_AGENT_NAME,
+      content: e
+    }), a = this.createAgent(t, this.id);
+    this.addMessage({
+      id: l(),
+      workId: this.id,
+      agentId: this.id,
+      role: "user",
+      content: e
+    });
+    const r = await this.revokeMessages(this.id), o = a.streamEvents(
+      {
+        messages: [...r, { role: "user", content: e }]
+      },
+      {
+        signal: this.abortController.signal,
+        context: {
+          ...s?.context || {},
+          _agentCluster: this,
+          _agentId: this.id
+        },
+        callbacks: [
+          new d({
+            agentCluster: this,
+            agentId: this.id
+          })
+        ]
+      }
+    );
+    return await Array.fromAsync(o), this.options.hooks.onAgentEnd?.(this.id), this.getLastAiMessage(this.id)?.content || "";
+  }
+  stop(e = "stop") {
+    this.abortController.abort(e);
+  }
+  createAgent({
+    tools: e,
+    subAgents: s,
+    middleware: t,
+    ...a
+  }, r) {
+    let o = [];
+    return e?.length && (o = this.normalizationTools(e)), s?.length && o.push(this.subAgentToDynamicTool(s, r)), p({
+      ...a,
+      middleware: [
+        ...t || [],
+        w,
+        b()
+      ],
+      tools: o
+    });
+  }
+  normalizationTools(e) {
+    return e.map((s) => "lc_namespace" in s || "lc_name" in s ? s : this.customToolToDynamicTool(s));
+  }
+  customToolToDynamicTool(e) {
+    return c((s) => "", e);
+  }
+  subAgentToDynamicTool(e, s) {
+    return c(
+      async (t, a) => {
+        if (!e.find(
+          (A) => A.name === t.subAgentName
+        ))
+          throw new Error(`Sub-agent ${t.subAgentName} not found.`);
+        const o = this.getLastAiMessage(s), n = a.toolCall.id, h = await this.options.hooks.onAgentStart({
+          agentId: n,
+          agentName: t.subAgentName,
+          beloneMessageId: o?.id,
+          parentAgentId: s,
+          content: t.request
+        }), u = this.createAgent(h, n);
+        this.addMessage({
+          id: l(),
+          workId: this.id,
+          agentId: n,
+          role: "user",
+          content: t.request
+        });
+        const m = u.streamEvents(
+          {
+            messages: [{ role: "user", content: t.request }]
+          },
+          {
+            signal: this.abortController.signal,
+            context: {
+              ...a.context || {},
+              _agentCluster: this,
+              _agentId: n,
+              _parentAgentId: s
+            },
+            callbacks: [
+              new d({
+                agentCluster: this,
+                agentId: n
+              })
+            ]
+          }
+        );
+        return await Array.fromAsync(m), this.options.hooks.onAgentEnd?.(n), this.getLastAiMessage(n)?.content || "";
+      },
+      {
+        name: g.CALL_SUB_AGENT_NAME,
+        description: `Call a sub-agent to handle the user request.
+      The sub-agents are: 
+      ${e.map((t) => `- ${t.name}: ${t.description}`).join(`
+`)}
+      `,
+        schema: i.object({
+          subAgentName: i.literal(e.map((t) => t.name)).describe("The name of the sub-agent to call."),
+          request: i.string().describe("The request to send to the sub-agent.")
+        }),
+        extras: {
+          scope: "autoRun",
+          skipReport: !0
+        }
+      }
+    );
+  }
+  getLastAiMessage(e) {
+    for (let s = this.messages.length - 1; s >= 0; s--) {
+      const t = this.messages[s];
+      if (t.agentId === e && t.role === "assistant")
+        return t;
+    }
+  }
+  getMessages() {
+    return this.messages;
+  }
+  async revokeMessages(e) {
+    return this.options.messageCollector ? (await this.options.messageCollector.getMessagesByAgentId(
+      this.id,
+      e
+    )).map((t) => t.role === "user" ? new f(t.content) : t.role === "assistant" ? new T({
+      content: t.content,
+      tool_calls: t.toolCalls
+    }) : new _({
+      content: t.content || t.confirm?.result,
+      tool_call_id: t.id
+    })) : [];
+  }
+}
+export {
+  g as default
+};

package/dist/es/cluster/readableHook.mjs

@@ -0,0 +1,90 @@
+import { Readable as E } from "node:stream";
+function P(i, n) {
+  const r = {}, s = {}, m = new E({
+    read() {
+    }
+  });
+  let d = Date.now();
+  function o(t) {
+    m.push(`${JSON.stringify(t)}
+
+`), d = Date.now();
+  }
+  const p = setInterval(() => {
+    Date.now() - d > 20 * 1e3 && o({ type: "ping", data: void 0 });
+  }, 30 * 1e3);
+  function v() {
+    m.push(null), clearInterval(p);
+  }
+  function y(t, e) {
+    if (!s[t])
+      throw new Error(`not find confirm tool resolver: ${t}`);
+    s[t](e), delete s[t];
+  }
+  function w(t, e) {
+    if (!r[t])
+      throw new Error(`not find init agent resolver: ${t}`);
+    r[t](e), delete r[t];
+  }
+  return {
+    hooks: {
+      onChunk(t) {
+        a("onChunk", [t], n), o({ type: "chunk", data: { chunk: t } });
+      },
+      async onAgentStart(t) {
+        a("onAgentStart", [t], n), o({
+          type: "agentStart",
+          data: t
+        });
+        const { promise: e, resolve: c } = Promise.withResolvers();
+        r[t.agentId] = c;
+        const f = await e, l = await i(t.agentName), u = [
+          l?.systemPrompt || "",
+          f.systemPrompt || ""
+        ].filter(Boolean), A = (f.tools || []).map((g) => ({
+          ...g,
+          extras: {
+            ...g.extras,
+            remote: !0
+          }
+        }));
+        return {
+          ...l,
+          systemPrompt: u.length > 0 ? u.join(`
+`) : void 0,
+          tools: [...l?.tools || [], ...A],
+          subAgents: [
+            ...l?.subAgents || [],
+            ...f.subAgents || []
+          ]
+        };
+      },
+      onAgentEnd(t) {
+        a("onAgentEnd", [t], n), o({ type: "agentEnd", data: { agentId: t } });
+      },
+      async onToolStart(t) {
+        if (a("onToolStart", [t], n), o({ type: "toolStart", data: { tool: t } }), t.needConfirm) {
+          const { resolve: e, promise: c } = Promise.withResolvers();
+          return s[t.toolCall.id] = e, c;
+        }
+        return {
+          type: "confirm"
+        };
+      },
+      onToolEnd(t, e) {
+        a("onToolEnd", [t, e], n), o({ type: "toolEnd", data: { toolPath: t, toolResult: e } });
+      }
+    },
+    stream: m,
+    send: o,
+    close: v,
+    resolveConfirmTool: y,
+    resolveAgentStart: w
+  };
+}
+function a(i, n, r) {
+  r?.forEach((s) => s[i]?.(...n));
+}
+export {
+  P as default
+};

package/dist/es/index.mjs

@@ -0,0 +1,13 @@
+import { default as a } from "./cluster/instance.mjs";
+import { default as l } from "./cluster/readableHook.mjs";
+import { default as s } from "./cluster/fileMessageCollector.mjs";
+import { default as m } from "./callback/LLMMessage.mjs";
+import { default as u } from "./middleware/dynamicToolInterceptor.mjs";
+import "@shuttle-ai/type";
+export {
+  a as AgentCluster,
+  s as FileMessageCollector,
+  m as LLMMessage,
+  u as dynamicToolInterceptorMiddleware,
+  l as readableHook
+};

package/dist/es/middleware/dynamicToolInterceptor.mjs

@@ -0,0 +1,86 @@
+import { createMiddleware as C } from "langchain";
+import { DynamicTool as p, DynamicStructuredTool as u } from "@langchain/core/tools";
+import { ToolMessage as s } from "@langchain/core/messages";
+import w from "../cluster/instance.mjs";
+const h = C({
+  name: "dynamicToolInterceptorMiddleware",
+  wrapToolCall: async (o, i) => {
+    const e = o.runtime.context, a = {
+      role: "tool",
+      name: o.toolCall.name,
+      id: o.toolCall.id || "",
+      aiMessageId: "",
+      agentId: e._agentId,
+      workId: e._agentCluster.id
+    };
+    let t;
+    if (o.tool instanceof p || o.tool instanceof u || o.tool && "extras" in o.tool) {
+      const l = o.tool.extras || {};
+      if (!l.skipReport) {
+        const m = !M(l, e._agentCluster), r = e._agentCluster.getLastAiMessage(
+          e._agentId
+        );
+        a.aiMessageId = r?.id || "";
+        const f = r?.toolCalls?.find(
+          (c) => c.id === o.toolCall.id
+        ), n = await e._agentCluster.options.hooks.onToolStart({
+          role: "assistant_tool",
+          toolCall: f,
+          needConfirm: l.remote || m,
+          id: r?.id || "",
+          agentId: e._agentId,
+          workId: e._agentCluster.id
+        });
+        if (a.confirm = n, n.type === "reject" || n.type === "confirm" && l.remote)
+          return e._agentCluster.addMessage(a), new s({
+            content: n.reason || "can not run tool",
+            tool_call_id: o.toolCall.id || ""
+          });
+        if (n.type === "confirmWithResult")
+          return e._agentCluster.addMessage(a), new s({
+            content: typeof n.result == "object" ? JSON.stringify(n.result) : n.result || "success",
+            tool_call_id: o.toolCall.id || ""
+          });
+        const d = {
+          agentId: e._agentId,
+          messageId: r?.id || "",
+          toolId: o.toolCall.id || ""
+        };
+        if (n.newArgs) {
+          o.toolCall.args = n.newArgs;
+          const g = o.state.messages[o.state.messages.length - 1].tool_calls?.find(
+            (_) => _.id === o.toolCall.id
+          );
+          g && (g.args = n.newArgs);
+        }
+        try {
+          t = await i(o), t instanceof s || "content" in t ? e._agentCluster.options.hooks.onToolEnd?.(
+            d,
+            t.content
+          ) : e._agentCluster.options.hooks.onToolEnd?.(d, "{}");
+        } catch (c) {
+          t = new s({
+            content: c.message || "can not run tool",
+            tool_call_id: o.toolCall.id || ""
+          });
+        }
+      }
+    }
+    if (!t)
+      try {
+        t = await i(o);
+      } catch (l) {
+        t = new s({
+          content: l.message || "can not run tool",
+          tool_call_id: o.toolCall.id || ""
+        });
+      }
+    return "lg_name" in t && t.lg_name === "Command" && (t = t?.update?.messages?.[0]), t instanceof s && t.name !== w.CALL_SUB_AGENT_NAME && (a.content = t.content, e._agentCluster.addMessage(a)), t;
+  }
+});
+function M(o, i) {
+  return o.scope === "autoRun" || i.options.autoRunScope === "always" || i.options.autoRunScope === "read" && o.scope !== "write";
+}
+export {
+  h as default
+};

package/dist/types/callback/LLMMessage.d.ts

@@ -0,0 +1,15 @@
+import { BaseCallbackHandler, NewTokenIndices } from '@langchain/core/callbacks/base';
+import { LLMResult } from '@langchain/core/outputs';
+import { AgentCluster } from '../cluster';
+interface Options {
+    agentCluster: AgentCluster;
+    agentId: string;
+}
+export default class LLMMessage extends BaseCallbackHandler {
+    readonly options: Options;
+    readonly name = "LLMMessage";
+    constructor(options: Options);
+    handleLLMNewToken(token: string, idx: NewTokenIndices, runId: string): void;
+    handleLLMEnd(output: LLMResult, runId: string): void;
+}
+export {};

package/dist/types/callback/index.d.ts

@@ -0,0 +1 @@
+export { default as LLMMessage } from './LLMMessage';

package/dist/types/cluster/fileMessageCollector.d.ts

@@ -0,0 +1,8 @@
+import { ShuttleAi } from '@shuttle-ai/type';
+export default class FileMessageCollector implements ShuttleAi.Cluster.MessageCollector {
+    private readonly dirPath;
+    constructor(dirPath: string);
+    saveMessage(message: ShuttleAi.Message.Define): Promise<void>;
+    getMessagesByAgentId(workId: string, agentId: string): Promise<ShuttleAi.Message.Define[]>;
+    getSubAgentTasks(workId: string, subAgentIds: string[]): Promise<ShuttleAi.Message.Define[]>;
+}

package/dist/types/cluster/index.d.ts

@@ -0,0 +1,3 @@
+export { default as AgentCluster } from './instance';
+export { default as readableHook } from './readableHook';
+export { default as FileMessageCollector } from './fileMessageCollector';

package/dist/types/cluster/instance.d.ts

@@ -0,0 +1,34 @@
+import { Runnable } from '@langchain/core/runnables';
+import { CreateAgentParams } from 'langchain';
+import { ClientTool } from '@langchain/core/tools';
+import { ShuttleAi } from '@shuttle-ai/type';
+import { z } from 'zod';
+export default class AgentCluster extends Runnable {
+    readonly options: ShuttleAi.Cluster.Options;
+    static MAIN_AGENT_NAME: string;
+    static CALL_SUB_AGENT_NAME: string;
+    readonly lc_namespace: string[];
+    readonly id: string;
+    private messages;
+    readonly abortController: AbortController;
+    constructor(options: ShuttleAi.Cluster.Options);
+    addMessage(message: ShuttleAi.Message.Define): void;
+    invoke(input: string, options?: Omit<ShuttleAi.Cluster.InvokeOptions, keyof ShuttleAi.Cluster.SystemContext>): Promise<string>;
+    stop(resign?: string): void;
+    createAgent({ tools, subAgents, middleware, ...extra }: ShuttleAi.Cluster.ToolsWithSubAgents & Omit<CreateAgentParams, 'tools'>, agentId: string): import('langchain').ReactAgent<import('langchain').AgentTypeConfig<import('langchain').ResponseFormatUndefined, undefined, import('langchain').AnyAnnotationRoot, any, readonly ClientTool[]>>;
+    normalizationTools(tools: (ShuttleAi.Tool.Define | ClientTool)[]): (ClientTool | import('langchain').DynamicStructuredTool<import('@langchain/core/utils/types').ZodObjectV3, Record<string, any> | undefined, any, string, string>)[];
+    customToolToDynamicTool(_tool: ShuttleAi.Tool.Define): import('langchain').DynamicStructuredTool<import('@langchain/core/utils/types').ZodObjectV3, Record<string, any> | undefined, any, string, string>;
+    subAgentToDynamicTool(subAgents: ShuttleAi.SubAgent.Define[], agentId: string): import('langchain').DynamicStructuredTool<z.ZodObject<{
+        subAgentName: z.ZodLiteral<string>;
+        request: z.ZodString;
+    }, z.core.$strip>, {
+        subAgentName: string;
+        request: string;
+    }, {
+        subAgentName: string;
+        request: string;
+    }, string, string>;
+    getLastAiMessage(agentId: string): ShuttleAi.Message.AI | undefined;
+    getMessages(): ShuttleAi.Message.Define[];
+    private revokeMessages;
+}

package/dist/types/cluster/readableHook.d.ts

@@ -0,0 +1,16 @@
+import { Readable } from 'stream';
+import { ShuttleAi } from '@shuttle-ai/type';
+import { CreateAgentParams } from 'langchain';
+interface ReadableHookLink extends Omit<ShuttleAi.Cluster.Hooks, 'onAgentStart' | 'onToolStart'> {
+    onAgentStart?: (options: ShuttleAi.Ask.AgentStart['data']) => void;
+    onToolStart?: (tool: ShuttleAi.Message.AITool) => void;
+}
+export default function createReadableHook(getAgentParamsFromServer: (agentName: string) => Promise<ShuttleAi.Cluster.ToolsWithSubAgents & Omit<CreateAgentParams, 'tools'>>, hookLinks?: ReadableHookLink[]): {
+    hooks: ShuttleAi.Cluster.Hooks;
+    stream: Readable;
+    send: (data: ShuttleAi.Ask.Define) => void;
+    close: () => void;
+    resolveConfirmTool: (id: string, value: ShuttleAi.Tool.ConfirmResult) => void;
+    resolveAgentStart: (id: string, value: ShuttleAi.Report.AgentStart["data"]["params"]) => void;
+};
+export {};

package/dist/types/index.d.ts

@@ -0,0 +1,4 @@
+export * from './cluster';
+export * from './callback';
+export * from './middleware';
+export * from './type';

package/dist/types/middleware/dynamicToolInterceptor.d.ts

@@ -0,0 +1,2 @@
+declare const dynamicToolInterceptorMiddleware: import('langchain').AgentMiddleware<undefined, undefined, unknown, readonly (import('@langchain/core/tools').ClientTool | import('@langchain/core/tools').ServerTool)[]>;
+export default dynamicToolInterceptorMiddleware;

package/dist/types/middleware/index.d.ts

@@ -0,0 +1 @@
+export { default as dynamicToolInterceptorMiddleware } from './dynamicToolInterceptor';

package/dist/types/type.d.ts

@@ -0,0 +1,66 @@
+import { ClientTool } from '@langchain/core/tools';
+import { CreateAgentParams } from 'langchain';
+import { default as AgentCluster } from './cluster/instance';
+declare module '@shuttle-ai/type' {
+    namespace ShuttleAi {
+        namespace Cluster {
+            type AutoRunScope = 'always' | 'read' | 'none';
+            interface Options {
+                /**
+                 * The id of the agent cluster.
+                 */
+                id?: string;
+                /**
+                 * The scope of auto run.
+                 * - `always`: Always run the agent.
+                 * - `read`: Run the agent when the user asks a question.
+                 * - `none`: Never run the agent.
+                 * Default: `none`
+                 */
+                autoRunScope?: AutoRunScope;
+                /**
+                 * The hooks of the agent.
+                 */
+                hooks: Hooks;
+                messageCollector?: MessageCollector;
+                /**
+                 * The messages of the agent.
+                 */
+                messages?: ShuttleAi.Message.Define[];
+                single?: AbortSignal;
+            }
+            interface ToolsWithSubAgents {
+                /**
+                 * The tools of the agent.
+                 */
+                tools?: (ShuttleAi.Tool.Define | ClientTool)[];
+                /**
+                 * The sub-agents of the agent.
+                 */
+                subAgents?: ShuttleAi.SubAgent.Define[];
+            }
+            interface Hooks {
+                onChunk?: (chunk: ShuttleAi.Message.AIChunk) => void;
+                onAgentStart: (options: ShuttleAi.Ask.AgentStart['data']) => Promise<ToolsWithSubAgents & Omit<CreateAgentParams, 'tools'>>;
+                onAgentEnd?: (agentId: string) => void;
+                onToolStart: (tool: ShuttleAi.Message.AITool) => Promise<ShuttleAi.Tool.ConfirmResult>;
+                onToolEnd?: (toolPath: ShuttleAi.Tool.Path, result: any) => void;
+            }
+            interface MessageCollector {
+                saveMessage: (message: ShuttleAi.Message.Define) => void;
+                getMessagesByAgentId: (workId: string, agentId: string) => Promise<ShuttleAi.Message.Define[]>;
+            }
+            interface SystemContext {
+                _agentCluster: AgentCluster;
+                _agentId: string;
+                _parentAgentId?: string;
+            }
+            interface Context extends SystemContext {
+            }
+            interface InvokeOptions {
+                context?: Context;
+                [x: string]: any;
+            }
+        }
+    }
+}

package/package.json

@@ -0,0 +1,56 @@
+{
+  "name": "@shuttle-ai/agent",
+  "version": "0.0.1",
+  "author": "Mingbing-get <1508850533@qq.com>",
+  "license": "MIT",
+  "description": "Shuttle AI 智能体操作库",
+  "main": "./src/index.ts",
+  "module": "./src/index.ts",
+  "types": "./src/index.ts",
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./src/index.ts",
+      "import": "./src/index.ts",
+      "require": "./src/index.ts"
+    }
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "scripts": {
+    "build": "vite build",
+    "clean": "rm -rf dist",
+    "type-check": "tsc --noEmit",
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "dependencies": {
+    "@shuttle-ai/type": "workspace:*",
+    "langchain": "^1.2.17",
+    "@langchain/core": "^1.1.19",
+    "@langchain/openai": "^1.2.4",
+    "zod": "^4.3.5"
+  },
+  "devDependencies": {
+    "typescript": "^5.3.3",
+    "@types/node": "^20.11.0",
+    "mysql2": "^3.9.1",
+    "dotenv": "^17.2.3"
+  },
+  "access": "public",
+  "publishConfig": {
+    "access": "public",
+    "main": "dist/cjs/index.cjs",
+    "module": "dist/es/index.mjs",
+    "types": "dist/types/index.d.ts",
+    "type": "module",
+    "exports": {
+      ".": {
+        "types": "./dist/types/index.d.ts",
+        "import": "./dist/es/index.mjs",
+        "require": "./dist/cjs/index.cjs"
+      }
+    }
+  }
+}