@shuttle-ai/agent 0.0.4 → 0.0.5

package/dist/cjs/cluster/instance.cjs

@@ -1,4 +1,4 @@
-"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.
+"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.
       The sub-agents are: 
       ${t.map(e=>`- ${e.name}: ${e.description}`).join(`
 `)}

package/dist/es/cluster/instance.mjs

@@ -1,10 +1,10 @@
 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 { createAgent as p } from "langchain";
+import { HumanMessage as b, AIMessage as f, 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 w from "../middleware/dynamicToolInterceptor.mjs";
+import _ from "../middleware/dynamicToolInterceptor.mjs";
 import d from "../callback/LLMMessage.mjs";
 class g extends M {
   constructor(e) {
@@ -69,8 +69,7 @@ class g extends M {
       ...a,
       middleware: [
         ...t || [],
-        w,
-        b()
+        _
       ],
       tools: o
     });
@@ -156,10 +155,10 @@ class g extends M {
     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({
+    )).map((t) => t.role === "user" ? new b(t.content) : t.role === "assistant" ? new f({
       content: t.content,
       tool_calls: t.toolCalls
-    }) : new _({
+    }) : new T({
       content: t.content || t.confirm?.result,
       tool_call_id: t.id
     })) : [];

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

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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shuttle-ai/agent",
-  "version": "0.0.4",
+  "version": "0.0.5",
   "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.4"
+    "@shuttle-ai/type": "0.0.5"
   },
   "devDependencies": {
     "typescript": "^5.3.3",