@shuttle-ai/agent 0.0.5 → 0.0.7

package/dist/cjs/cluster/instance.cjs

@@ -1,5 +1,5 @@
-"use strict";Object.defineProperties(exports,{__esModule:{value:!0},[Symbol.toStringTag]:{value:"Module"}});const b=require("@langchain/core/runnables"),_=require("langchain"),l=require("@langchain/core/messages"),u=require("@langchain/core/tools"),c=require("zod"),g=require("node:crypto"),p=require("../middleware/dynamicToolInterceptor.cjs"),d=require("../callback/LLMMessage.cjs");class i extends b.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 d.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)),_.createAgent({...a,middleware:[...e||[],p.default],tools:o})}normalizationTools(t){return t.map(s=>"lc_namespace"in s||"lc_name"in s?s:this.customToolToDynamicTool(s))}customToolToDynamicTool(t){return u.tool(s=>"",t)}subAgentToDynamicTool(t,s){return u.tool(async(e,a)=>{if(!t.find(M=>M.name===e.subAgentName))throw new Error(`Sub-agent ${e.subAgentName} not found.`);const o=this.getLastAiMessage(s),n=a.toolCall.id,h=await this.options.hooks.onAgentStart({agentId:n,agentName:e.subAgentName,beloneMessageId:o?.id,parentAgentId:s,content:e.request}),m=this.createAgent(h,n);this.addMessage({id:g.randomUUID(),workId:this.id,agentId:n,role:"user",content:e.request});const A=m.streamEvents({messages:[{role:"user",content:e.request}]},{signal:this.abortController.signal,context:{...a.context||{},_agentCluster:this,_agentId:n,_parentAgentId:s},callbacks:[new d.default({agentCluster:this,agentId:n})]});return await Array.fromAsync(A),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.
+"use strict";Object.defineProperties(exports,{__esModule:{value:!0},[Symbol.toStringTag]:{value:"Module"}});const b=require("@langchain/core/runnables"),p=require("langchain"),l=require("@langchain/core/messages"),u=require("@langchain/core/tools"),c=require("zod"),g=require("node:crypto"),T=require("../middleware/dynamicToolInterceptor.cjs"),d=require("../callback/LLMMessage.cjs");class i extends b.Runnable{constructor(e){super(),this.options=e,this.id=e.id||g.randomUUID(),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:i.MAIN_AGENT_NAME,content:e}),n=this.createAgent(t,this.id);this.addMessage({id:g.randomUUID(),workId:this.id,agentId:this.id,role:"user",content:e});const r=await this.revokeMessages(this.id),a=n.streamEvents({messages:[...r,{role:"user",content:e}]},{signal:this.abortController.signal,context:{...s?.context||{},_agentCluster:this,_agentId:this.id},callbacks:[new d.default({agentCluster:this,agentId:this.id})]});return await Array.fromAsync(a),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,...n},r){let a=[];return e?.length&&(a=this.normalizationTools(e)),s?.length&&a.push(this.subAgentToDynamicTool(s,r)),p.createAgent({...n,middleware:[...t||[],T.default],tools:a})}normalizationTools(e){return e.map(s=>"lc_namespace"in s||"lc_name"in s?s:this.customToolToDynamicTool(s))}customToolToDynamicTool(e){return u.tool(s=>"",e)}subAgentToDynamicTool(e,s){return u.tool(async(t,n)=>{if(!e.find(M=>M.name===t.subAgentName))throw new Error(`Sub-agent ${t.subAgentName} not found.`);const a=this.getLastAiMessage(s),o=n.toolCall.id,h=await this.options.hooks.onAgentStart({agentId:o,agentName:t.subAgentName,beloneMessageId:a?.id,parentAgentId:s,content:t.request}),m=this.createAgent(h,o);this.addMessage({id:g.randomUUID(),workId:this.id,agentId:o,role:"user",content:t.request});const A=m.streamEvents({messages:[{role:"user",content:t.request}]},{signal:this.abortController.signal,context:{...n.context||{},_agentCluster:this,_agentId:o,_parentAgentId:s},callbacks:[new d.default({agentCluster:this,agentId:o})]});return await Array.fromAsync(A),this.options.hooks.onAgentEnd?.(o),this.getLastAiMessage(o)?.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(`
+      ${e.map(t=>`- ${t.name}: ${t.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;
+      `,schema:c.z.object({subAgentName:c.z.literal(e.map(t=>t.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(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=>{if(t.role==="user")return new l.HumanMessage(t.content);if(t.role==="assistant")return new l.AIMessage({content:t.content,tool_calls:t.toolCalls});const n=t.result||t.confirm?.result;return new l.ToolMessage({content:(n?.type==="success"?this.anyToString(n.content):n?.reason)||"",tool_call_id:t.id})}):[]}anyToString(e){return typeof e=="object"?JSON.stringify(e):e}}exports.default=i;

package/dist/cjs/index.cjs

@@ -1 +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;
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const r=require("./cluster/instance.cjs"),o=require("./cluster/readableHook.cjs"),a=require("./cluster/fileMessageCollector.cjs"),l=require("./callback/LLMMessage.cjs"),c=require("./middleware/dynamicToolInterceptor.cjs"),t=require("@shuttle-ai/type");exports.AgentCluster=r.default;exports.readableHook=o.default;exports.FileMessageCollector=a.default;exports.LLMMessage=l.default;exports.dynamicToolInterceptorMiddleware=c.default;Object.keys(t).forEach(e=>{e!=="default"&&!Object.prototype.hasOwnProperty.call(exports,e)&&Object.defineProperty(exports,e,{enumerable:!0,get:()=>t[e]})});

package/dist/cjs/middleware/dynamicToolInterceptor.cjs

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

package/dist/es/cluster/instance.mjs

@@ -1,14 +1,14 @@
-import { Runnable as M } from "@langchain/core/runnables";
-import { createAgent as p } from "langchain";
-import { HumanMessage as b, AIMessage as f, ToolMessage as T } from "@langchain/core/messages";
+import { Runnable as p } from "@langchain/core/runnables";
+import { createAgent as M } from "langchain";
+import { HumanMessage as f, AIMessage as b, ToolMessage as T } 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 _ 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", () => {
+import h from "../callback/LLMMessage.mjs";
+class g extends p {
+  constructor(t) {
+    super(), this.options = t, this.id = t.id || l(), t.single?.addEventListener("abort", () => {
       this.stop();
     });
   }
@@ -18,25 +18,25 @@ class g extends M {
   id;
   messages = [];
   abortController = new AbortController();
-  addMessage(e) {
-    this.messages.push(e), this.options.messageCollector?.saveMessage(e);
+  addMessage(t) {
+    this.messages.push(t), this.options.messageCollector?.saveMessage(t);
   }
-  async invoke(e, s) {
-    const t = await this.options.hooks.onAgentStart({
+  async invoke(t, s) {
+    const e = await this.options.hooks.onAgentStart({
       agentId: this.id,
       agentName: g.MAIN_AGENT_NAME,
-      content: e
-    }), a = this.createAgent(t, this.id);
+      content: t
+    }), n = this.createAgent(e, this.id);
     this.addMessage({
       id: l(),
       workId: this.id,
       agentId: this.id,
       role: "user",
-      content: e
+      content: t
     });
-    const r = await this.revokeMessages(this.id), o = a.streamEvents(
+    const r = await this.revokeMessages(this.id), a = n.streamEvents(
       {
-        messages: [...r, { role: "user", content: e }]
+        messages: [...r, { role: "user", content: t }]
       },
       {
         signal: this.abortController.signal,
@@ -46,92 +46,92 @@ class g extends M {
           _agentId: this.id
         },
         callbacks: [
-          new d({
+          new h({
             agentCluster: this,
             agentId: this.id
           })
         ]
       }
     );
-    return await Array.fromAsync(o), this.options.hooks.onAgentEnd?.(this.id), this.getLastAiMessage(this.id)?.content || "";
+    return await Array.fromAsync(a), this.options.hooks.onAgentEnd?.(this.id), this.getLastAiMessage(this.id)?.content || "";
   }
-  stop(e = "stop") {
-    this.abortController.abort(e);
+  stop(t = "stop") {
+    this.abortController.abort(t);
   }
   createAgent({
-    tools: e,
+    tools: t,
     subAgents: s,
-    middleware: t,
-    ...a
+    middleware: e,
+    ...n
   }, r) {
-    let o = [];
-    return e?.length && (o = this.normalizationTools(e)), s?.length && o.push(this.subAgentToDynamicTool(s, r)), p({
-      ...a,
+    let a = [];
+    return t?.length && (a = this.normalizationTools(t)), s?.length && a.push(this.subAgentToDynamicTool(s, r)), M({
+      ...n,
       middleware: [
-        ...t || [],
+        ...e || [],
         _
       ],
-      tools: o
+      tools: a
     });
   }
-  normalizationTools(e) {
-    return e.map((s) => "lc_namespace" in s || "lc_name" in s ? s : this.customToolToDynamicTool(s));
+  normalizationTools(t) {
+    return t.map((s) => "lc_namespace" in s || "lc_name" in s ? s : this.customToolToDynamicTool(s));
   }
-  customToolToDynamicTool(e) {
-    return c((s) => "", e);
+  customToolToDynamicTool(t) {
+    return c((s) => "", t);
   }
-  subAgentToDynamicTool(e, s) {
+  subAgentToDynamicTool(t, s) {
     return c(
-      async (t, a) => {
-        if (!e.find(
-          (A) => A.name === t.subAgentName
+      async (e, n) => {
+        if (!t.find(
+          (A) => A.name === e.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,
+          throw new Error(`Sub-agent ${e.subAgentName} not found.`);
+        const a = this.getLastAiMessage(s), o = n.toolCall.id, u = await this.options.hooks.onAgentStart({
+          agentId: o,
+          agentName: e.subAgentName,
+          beloneMessageId: a?.id,
           parentAgentId: s,
-          content: t.request
-        }), u = this.createAgent(h, n);
+          content: e.request
+        }), d = this.createAgent(u, o);
         this.addMessage({
           id: l(),
           workId: this.id,
-          agentId: n,
+          agentId: o,
           role: "user",
-          content: t.request
+          content: e.request
         });
-        const m = u.streamEvents(
+        const m = d.streamEvents(
           {
-            messages: [{ role: "user", content: t.request }]
+            messages: [{ role: "user", content: e.request }]
           },
           {
             signal: this.abortController.signal,
             context: {
-              ...a.context || {},
+              ...n.context || {},
               _agentCluster: this,
-              _agentId: n,
+              _agentId: o,
               _parentAgentId: s
             },
             callbacks: [
-              new d({
+              new h({
                 agentCluster: this,
-                agentId: n
+                agentId: o
               })
             ]
           }
         );
-        return await Array.fromAsync(m), this.options.hooks.onAgentEnd?.(n), this.getLastAiMessage(n)?.content || "";
+        return await Array.fromAsync(m), this.options.hooks.onAgentEnd?.(o), this.getLastAiMessage(o)?.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(`
+      ${t.map((e) => `- ${e.name}: ${e.description}`).join(`
 `)}
       `,
         schema: i.object({
-          subAgentName: i.literal(e.map((t) => t.name)).describe("The name of the sub-agent to call."),
+          subAgentName: i.literal(t.map((e) => e.name)).describe("The name of the sub-agent to call."),
           request: i.string().describe("The request to send to the sub-agent.")
         }),
         extras: {
@@ -141,27 +141,37 @@ class g extends M {
       }
     );
   }
-  getLastAiMessage(e) {
+  getLastAiMessage(t) {
     for (let s = this.messages.length - 1; s >= 0; s--) {
-      const t = this.messages[s];
-      if (t.agentId === e && t.role === "assistant")
-        return t;
+      const e = this.messages[s];
+      if (e.agentId === t && e.role === "assistant")
+        return e;
     }
   }
   getMessages() {
     return this.messages;
   }
-  async revokeMessages(e) {
+  async revokeMessages(t) {
     return this.options.messageCollector ? (await this.options.messageCollector.getMessagesByAgentId(
       this.id,
-      e
-    )).map((t) => t.role === "user" ? new b(t.content) : t.role === "assistant" ? new f({
-      content: t.content,
-      tool_calls: t.toolCalls
-    }) : new T({
-      content: t.content || t.confirm?.result,
-      tool_call_id: t.id
-    })) : [];
+      t
+    )).map((e) => {
+      if (e.role === "user")
+        return new f(e.content);
+      if (e.role === "assistant")
+        return new b({
+          content: e.content,
+          tool_calls: e.toolCalls
+        });
+      const n = e.result || e.confirm?.result;
+      return new T({
+        content: (n?.type === "success" ? this.anyToString(n.content) : n?.reason) || "",
+        tool_call_id: e.id
+      });
+    }) : [];
+  }
+  anyToString(t) {
+    return typeof t == "object" ? JSON.stringify(t) : t;
   }
 }
 export {

package/dist/es/index.mjs

@@ -1,13 +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";
+import { default as r } from "./cluster/instance.mjs";
+import { default as t } from "./cluster/readableHook.mjs";
+import { default as f } from "./cluster/fileMessageCollector.mjs";
+import { default as d } from "./callback/LLMMessage.mjs";
+import { default as p } from "./middleware/dynamicToolInterceptor.mjs";
+export * from "@shuttle-ai/type";
 export {
-  a as AgentCluster,
-  s as FileMessageCollector,
-  m as LLMMessage,
-  u as dynamicToolInterceptorMiddleware,
-  l as readableHook
+  r as AgentCluster,
+  f as FileMessageCollector,
+  d as LLMMessage,
+  p as dynamicToolInterceptorMiddleware,
+  t as readableHook
 };

package/dist/es/middleware/dynamicToolInterceptor.mjs

@@ -1,8 +1,8 @@
-import { createMiddleware as C } from "langchain";
-import { DynamicTool as p, DynamicStructuredTool as u } from "@langchain/core/tools";
+import { createMiddleware as _ } from "langchain";
+import { DynamicTool as C, DynamicStructuredTool as w } from "@langchain/core/tools";
 import { ToolMessage as s } from "@langchain/core/messages";
-import w from "../cluster/instance.mjs";
-const h = C({
+import y from "../cluster/instance.mjs";
+const k = _({
   name: "dynamicToolInterceptorMiddleware",
   wrapToolCall: async (o, i) => {
     const e = o.runtime.context, a = {
@@ -13,19 +13,19 @@ const h = C({
       agentId: e._agentId,
       workId: e._agentCluster.id
     };
-    let t;
-    if (o.tool instanceof p || o.tool instanceof u || o.tool && "extras" in o.tool) {
+    let t, d = !1;
+    if (o.tool instanceof C || o.tool instanceof w || 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(
+        const p = r?.toolCalls?.find(
           (c) => c.id === o.toolCall.id
         ), n = await e._agentCluster.options.hooks.onToolStart({
           role: "assistant_tool",
-          toolCall: f,
+          toolCall: p,
           needConfirm: l.remote || m,
           id: r?.id || "",
           agentId: e._agentId,
@@ -41,27 +41,33 @@ const h = C({
             content: typeof n.result == "object" ? JSON.stringify(n.result) : n.result || "success",
             tool_call_id: o.toolCall.id || ""
           });
-        const d = {
+        const g = {
           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
+          const f = o.state.messages[o.state.messages.length - 1].tool_calls?.find(
+            (u) => u.id === o.toolCall.id
           );
-          g && (g.args = n.newArgs);
+          f && (f.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, "{}");
+          t = await i(o), t instanceof s || "content" in t ? e._agentCluster.options.hooks.onToolEnd?.(g, {
+            type: "success",
+            content: t.content
+          }) : e._agentCluster.options.hooks.onToolEnd?.(g, {
+            type: "success",
+            content: "{}"
+          });
         } catch (c) {
           t = new s({
             content: c.message || "can not run tool",
             tool_call_id: o.toolCall.id || ""
+          }), d = !0, e._agentCluster.options.hooks.onToolEnd?.(g, {
+            type: "fail",
+            reason: t.content
           });
         }
       }
@@ -70,17 +76,23 @@ const h = C({
       try {
         t = await i(o);
       } catch (l) {
-        t = new s({
+        d = !0, 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;
+    return "lg_name" in t && t.lg_name === "Command" && (t = t?.update?.messages?.[0]), t instanceof s && t.name !== y.CALL_SUB_AGENT_NAME && (a.result = d ? {
+      type: "fail",
+      reason: t.content
+    } : {
+      type: "success",
+      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
+  k as default
 };

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

@@ -19,16 +19,17 @@ export default class AgentCluster extends Runnable {
     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<any>;
+        subAgentName: z.ZodLiteral<string>;
         request: z.ZodString;
     }, z.core.$strip>, {
         subAgentName: string;
         request: string;
     }, {
-        subAgentName: any;
+        subAgentName: string;
         request: string;
-    }, any, string>;
-    getLastAiMessage(agentId: string): any;
+    }, string, string>;
+    getLastAiMessage(agentId: string): ShuttleAi.Message.AI | undefined;
     getMessages(): ShuttleAi.Message.Define[];
     private revokeMessages;
+    private anyToString;
 }

package/dist/types/type.d.ts

@@ -1,6 +1,7 @@
 import { ClientTool } from '@langchain/core/tools';
 import { CreateAgentParams } from 'langchain';
 import { default as AgentCluster } from './cluster/instance';
+export * from '@shuttle-ai/type';
 declare module '@shuttle-ai/type' {
     namespace ShuttleAi {
         namespace Cluster {
@@ -44,7 +45,7 @@ declare module '@shuttle-ai/type' {
                 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;
+                onToolEnd?: (toolPath: ShuttleAi.Tool.Path, result: Tool.Result) => void;
             }
             interface MessageCollector {
                 saveMessage: (message: ShuttleAi.Message.Define) => void;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shuttle-ai/agent",
-  "version": "0.0.5",
+  "version": "0.0.7",
   "author": "Mingbing-get <1508850533@qq.com>",
   "license": "MIT",
   "description": "Shuttle AI 智能体操作库",
@@ -24,7 +24,7 @@
     "@langchain/core": "^1.1.19",
     "@langchain/openai": "^1.2.4",
     "zod": "^4.3.5",
-    "@shuttle-ai/type": "0.0.5"
+    "@shuttle-ai/type": "0.0.7"
   },
   "devDependencies": {
     "typescript": "^5.3.3",