@eko-ai/eko 4.0.9 → 4.1.1

package/dist/index.d.ts

@@ -10,14 +10,14 @@ import TaskContext, { AgentContext } from "./agent/agent-context";
 import { RetryLanguageModel, callLLM, callWithReAct } from "./llm";
 export default Eko;
 export { Eko, EkoMemory, Log, config, global, Chain, Planner, AgentChain, TaskContext, AgentContext, SimpleSseMcpClient, SimpleHttpMcpClient, RetryLanguageModel, TaskContext as Context, callLLM, callWithReAct, };
-export { ChatAgent, ChatContext, WebSearchTool, WebpageQaTool, DeepActionTool, TaskVariableStorageTool, } from "./chat";
+export { ChatAgent, ChatContext, AgentWrapTool, WebSearchTool, WebpageQaTool, DeepActionTool, TaskVariableStorageTool, } from "./chat";
 export { Agent, type AgentParams, BaseBrowserAgent, BaseBrowserLabelsAgent, BaseBrowserScreenAgent, } from "./agent";
 export { ForeachTaskTool, WatchTriggerTool, HumanInteractTool, TaskNodeStatusTool, VariableStorageTool, } from "./tools";
 export type { ChatService, BrowserService } from "./service";
-export { type LLMs, type LLMRequest, type HumanCallback, type Workflow, type EkoConfig, type WorkflowNode, type WorkflowAgent, type AgentStreamMessage, type AgentStreamCallback, type AgentStreamCallback as StreamCallback, type AgentStreamMessage as StreamCallbackMessage, } from "./types";
 export { sub, uuidv4, toFile, toImage, mergeTools, call_timeout, compressImageData, convertToolSchema, } from "./common/utils";
 export { parseWorkflow, resetWorkflowXml, buildSimpleAgentWorkflow, } from "./common/xml";
 export { buildAgentTree } from "./common/tree";
 export { PromptTemplate } from "./prompt/prompt-template";
 export { extract_page_content } from "./agent/browser/utils";
+export * from "./types";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,UAAU,CAAC;AAC9B,OAAO,GAAG,MAAM,cAAc,CAAC;AAC/B,OAAO,GAAG,MAAM,eAAe,CAAC;AAChC,OAAO,MAAM,MAAM,iBAAiB,CAAC;AACrC,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,KAAK,EAAE,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,OAAO,CAAC;AAChE,OAAO,WAAW,EAAE,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAClE,OAAO,EAAE,kBAAkB,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,OAAO,CAAC;AAEnE,eAAe,GAAG,CAAC;AAEnB,OAAO,EACL,GAAG,EACH,SAAS,EACT,GAAG,EACH,MAAM,EACN,MAAM,EACN,KAAK,EACL,OAAO,EACP,UAAU,EACV,WAAW,EACX,YAAY,EACZ,kBAAkB,EAClB,mBAAmB,EACnB,kBAAkB,EAClB,WAAW,IAAI,OAAO,EACtB,OAAO,EACP,aAAa,GACd,CAAC;AAEF,OAAO,EACL,SAAS,EACT,WAAW,EACX,aAAa,EACb,aAAa,EACb,cAAc,EACd,uBAAuB,GACxB,MAAM,QAAQ,CAAC;AAEhB,OAAO,EACL,KAAK,EACL,KAAK,WAAW,EAChB,gBAAgB,EAChB,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,SAAS,CAAC;AAEjB,OAAO,EACL,eAAe,EACf,gBAAgB,EAChB,iBAAiB,EACjB,kBAAkB,EAClB,mBAAmB,GACpB,MAAM,SAAS,CAAC;AAEjB,YAAY,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAE7D,OAAO,EACL,KAAK,IAAI,EACT,KAAK,UAAU,EACf,KAAK,aAAa,EAClB,KAAK,QAAQ,EACb,KAAK,SAAS,EACd,KAAK,YAAY,EACjB,KAAK,aAAa,EAClB,KAAK,kBAAkB,EACvB,KAAK,mBAAmB,EACxB,KAAK,mBAAmB,IAAI,cAAc,EAC1C,KAAK,kBAAkB,IAAI,qBAAqB,GACjD,MAAM,SAAS,CAAC;AAEjB,OAAO,EACL,GAAG,EACH,MAAM,EACN,MAAM,EACN,OAAO,EACP,UAAU,EACV,YAAY,EACZ,iBAAiB,EACjB,iBAAiB,GAClB,MAAM,gBAAgB,CAAC;AAExB,OAAO,EACL,aAAa,EACb,gBAAgB,EAChB,wBAAwB,GACzB,MAAM,cAAc,CAAC;AAEtB,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,UAAU,CAAC;AAC9B,OAAO,GAAG,MAAM,cAAc,CAAC;AAC/B,OAAO,GAAG,MAAM,eAAe,CAAC;AAChC,OAAO,MAAM,MAAM,iBAAiB,CAAC;AACrC,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,KAAK,EAAE,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,OAAO,CAAC;AAChE,OAAO,WAAW,EAAE,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAClE,OAAO,EAAE,kBAAkB,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,OAAO,CAAC;AAEnE,eAAe,GAAG,CAAC;AAEnB,OAAO,EACL,GAAG,EACH,SAAS,EACT,GAAG,EACH,MAAM,EACN,MAAM,EACN,KAAK,EACL,OAAO,EACP,UAAU,EACV,WAAW,EACX,YAAY,EACZ,kBAAkB,EAClB,mBAAmB,EACnB,kBAAkB,EAClB,WAAW,IAAI,OAAO,EACtB,OAAO,EACP,aAAa,GACd,CAAC;AAEF,OAAO,EACL,SAAS,EACT,WAAW,EACX,aAAa,EACb,aAAa,EACb,aAAa,EACb,cAAc,EACd,uBAAuB,GACxB,MAAM,QAAQ,CAAC;AAEhB,OAAO,EACL,KAAK,EACL,KAAK,WAAW,EAChB,gBAAgB,EAChB,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,SAAS,CAAC;AAEjB,OAAO,EACL,eAAe,EACf,gBAAgB,EAChB,iBAAiB,EACjB,kBAAkB,EAClB,mBAAmB,GACpB,MAAM,SAAS,CAAC;AAEjB,YAAY,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAE7D,OAAO,EACL,GAAG,EACH,MAAM,EACN,MAAM,EACN,OAAO,EACP,UAAU,EACV,YAAY,EACZ,iBAAiB,EACjB,iBAAiB,GAClB,MAAM,gBAAgB,CAAC;AAExB,OAAO,EACL,aAAa,EACb,gBAAgB,EAChB,wBAAwB,GACzB,MAAM,cAAc,CAAC;AAEtB,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAE7D,cAAc,SAAS,CAAC"}

package/dist/index.esm.js

@@ -6,6 +6,7 @@ const config$1 = {
     maxOutputTokens: 16000,
     maxRetryNum: 3,
     agentParallel: false,
+    workflowConfirm: false,
     compressThreshold: 80,
     compressTokensThreshold: 80000,
     largeTextLength: 8000,
@@ -43182,7 +43183,7 @@ class EkoMemory {
         this.memoryConfig = memoryConfig;
     }
     genMessageId() {
-        return uuidv4();
+        return "msg-" + uuidv4();
     }
     async import(data) {
         this.messages = [...data.messages];
@@ -44263,10 +44264,10 @@ class DeepActionTool {
         const taskDescription = args.taskDescription;
         const tabIds = args.tabIds;
         const dependentVariables = args.dependentVariables;
-        const config = this.chatContext.getConfig();
+        const chatConfig = this.chatContext.getConfig();
         const globalVariables = this.chatContext.getGlobalVariables();
         const eko = new Eko({
-            ...config,
+            ...chatConfig,
             callback: this.params.callback?.taskCallback,
         }, chatId);
         this.chatContext.addEko(messageId, eko);
@@ -44282,7 +44283,7 @@ class DeepActionTool {
         }
         const attachments = this.params.user
             .filter((part) => part.type === "file")
-            .filter((part) => part.data && part.data.length < 500)
+            .filter((part) => part.data && part.data.length < 1000)
             .map((part) => {
             return {
                 file_name: part.filename,
@@ -44302,7 +44303,30 @@ class DeepActionTool {
             datetime: this.params.datetime || new Date().toLocaleString(),
         });
         const context = eko.getTask(messageId);
-        console.log("==> workflow", workflow);
+        Log.info("==> workflow", workflow);
+        if (config$1.workflowConfirm && this.params.callback?.taskCallback) {
+            const result = await new Promise((resolve) => {
+                this.params.callback.taskCallback?.onMessage({
+                    streamType: "agent",
+                    chatId: context.chatId,
+                    taskId: context.taskId,
+                    agentName: "",
+                    type: "workflow_confirm",
+                    workflow: workflow,
+                    resolve,
+                });
+            });
+            if (result == "cancel") {
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: "User has canceled the execution.",
+                        },
+                    ],
+                };
+            }
+        }
         const result = await eko.execute(messageId);
         const variableNames = [];
         if (context.variables && context.variables.size > 0) {
@@ -44507,16 +44531,14 @@ function getTabsInfo(tabs) {
 }
 
 class ChatAgent {
-    constructor(config, chatId = uuidv4(), memory, tools) {
+    constructor(config, chatId = "chat-" + uuidv4(), memory, tools) {
+        this.maxReactLoopNum = 15;
         this.tools = tools ?? [];
         this.memory = memory ?? new EkoMemory();
         this.chatContext = new ChatContext(chatId, config);
         global.chatMap.set(chatId, this.chatContext);
     }
     async chat(params) {
-        return this.doChat(params, false);
-    }
-    async doChat(params, segmentedExecution) {
         const runStartTime = Date.now();
         let reactLoopNum = 0;
         let errorInfo = null;
@@ -44534,7 +44556,7 @@ class ChatAgent {
             await this.addUserMessage(params.messageId, params.user);
             const config = this.chatContext.getConfig();
             const rlm = new RetryLanguageModel(config.llms, config.chatLlms);
-            for (; reactLoopNum < 15; reactLoopNum++) {
+            for (; reactLoopNum < this.maxReactLoopNum; reactLoopNum++) {
                 const messages = this.memory.buildMessages();
                 const results = await callChatLLM(this.chatContext.getChatId(), params.messageId, rlm, messages, convertTools(chatTools), undefined, params.callback, params.signal);
                 const finalResult = await this.handleCallResult(params.messageId, chatTools, results, params.callback);
@@ -44729,5 +44751,169 @@ class ChatAgent {
     }
 }
 
-export { Agent, AgentChain, AgentContext, BaseBrowserAgent, BaseBrowserLabelsAgent, BaseBrowserScreenAgent, Chain, ChatAgent, ChatContext, TaskContext as Context, DeepActionTool, Eko, EkoMemory, ForeachTaskTool, HumanInteractTool, Log, Planner, PromptTemplate, RetryLanguageModel, SimpleHttpMcpClient, SimpleSseMcpClient, TaskContext, TaskNodeStatusTool, TaskVariableStorageTool, VariableStorageTool, WatchTriggerTool, WebSearchTool, WebpageQaTool, buildAgentTree, buildSimpleAgentWorkflow, callLLM, callWithReAct, call_timeout, compressImageData, config$1 as config, convertToolSchema, Eko as default, extract_page_content, global, mergeTools, parseWorkflow, resetWorkflowXml, sub, toFile, toImage, uuidv4 };
+class AgentWrapTool {
+    constructor(chatContext, params, agent, extra) {
+        this.agent = agent;
+        this.params = params;
+        this.chatContext = chatContext;
+        this.name =
+            extra?.name ||
+                agent.Name.substring(0, 1).toLowerCase() + agent.Name.substring(1);
+        this.description =
+            extra?.description ||
+                agent.PlanDescription ||
+                sub(agent.Description, 600, true, false);
+        this.parameters = extra?.parameters || {
+            type: "object",
+            properties: {
+                language: {
+                    type: "string",
+                    description: "User language used, eg: English",
+                },
+                taskDescription: {
+                    type: "string",
+                    description: "Task description, please do not omit any information from the user's instructions, and use the same language as the user's question.",
+                },
+                tabIds: {
+                    type: "array",
+                    description: "Browser Tab IDs associated with this task, When user says 'left side' or 'current', it means current active tab",
+                    items: { type: "integer" },
+                },
+                dependentVariables: {
+                    type: "array",
+                    description: "The current task relies on variable data from prerequisite execution outputs. Provide the name of the dependent variable.",
+                    items: {
+                        type: "string",
+                    },
+                },
+            },
+            required: ["language", "taskDescription"],
+        };
+    }
+    async execute(args, toolCall, messageId) {
+        const chatId = this.chatContext.getChatId();
+        const language = args.language;
+        const taskDescription = args.taskDescription;
+        const tabIds = args.tabIds;
+        const dependentVariables = args.dependentVariables;
+        const globalVariables = this.chatContext.getGlobalVariables();
+        const attachments = this.params.user
+            .filter((part) => part.type === "file")
+            .filter((part) => part.data && part.data.length < 500)
+            .map((part) => {
+            return {
+                file_name: part.filename,
+                file_path: part.filePath,
+                file_url: part.data,
+            };
+        });
+        const taskWebsite = await this.getTaskWebsite(tabIds);
+        const config = this.chatContext.getConfig();
+        const contextParams = {
+            ...(this.params.extra || {}),
+            ...globalVariables,
+            tabIds: tabIds,
+            language: language,
+            attachments: attachments,
+            taskWebsite: taskWebsite,
+            dependentVariables: dependentVariables,
+            datetime: this.params.datetime || new Date().toLocaleString(),
+        };
+        const agents = [this.agent];
+        const chain = new Chain(taskDescription);
+        const taskId = messageId;
+        const context = new TaskContext(chatId, taskId, {
+            ...config,
+            callback: this.params.callback?.taskCallback,
+        }, agents, chain);
+        if (contextParams) {
+            Object.keys(contextParams).forEach((key) => context.variables.set(key, contextParams[key]));
+        }
+        context.workflow = buildSimpleAgentWorkflow({
+            taskId,
+            name: "Task",
+            agentName: this.agent.Name,
+            task: taskDescription,
+        });
+        try {
+            global.taskMap.set(taskId, context);
+            const agentNode = context.workflow.agents[0];
+            const resultText = await this.runAgent(context, this.agent, {
+                type: "normal",
+                agent: agentNode,
+            }, new AgentChain(agentNode));
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: resultText || "",
+                    },
+                ],
+            };
+        }
+        finally {
+            global.taskMap.delete(taskId);
+        }
+    }
+    async runAgent(context, agent, agentNode, agentChain) {
+        const callback = this.params.callback?.taskCallback;
+        try {
+            agentNode.agent.status = "running";
+            callback &&
+                (await callback.onMessage({
+                    streamType: "agent",
+                    chatId: context.chatId,
+                    taskId: context.taskId,
+                    agentName: agentNode.agent.name,
+                    nodeId: agentNode.agent.id,
+                    type: "agent_start",
+                    agentNode: agentNode.agent,
+                }, agent.AgentContext));
+            agentNode.result = await agent.run(context, agentChain);
+            agentNode.agent.status = "done";
+            callback &&
+                (await callback.onMessage({
+                    streamType: "agent",
+                    chatId: context.chatId,
+                    taskId: context.taskId,
+                    agentName: agentNode.agent.name,
+                    nodeId: agentNode.agent.id,
+                    type: "agent_result",
+                    agentNode: agentNode.agent,
+                    result: agentNode.result,
+                }, agent.AgentContext));
+            return agentNode.result;
+        }
+        catch (e) {
+            agentNode.agent.status = "error";
+            callback &&
+                (await callback.onMessage({
+                    streamType: "agent",
+                    chatId: context.chatId,
+                    taskId: context.taskId,
+                    agentName: agentNode.agent.name,
+                    nodeId: agentNode.agent.id,
+                    type: "agent_result",
+                    agentNode: agentNode.agent,
+                    error: e,
+                }, agent.AgentContext));
+            throw e;
+        }
+    }
+    async getTaskWebsite(tabIds) {
+        if (!global.browserService) {
+            return [];
+        }
+        const tabs = await global.browserService.loadTabs(this.chatContext.getChatId(), tabIds);
+        return tabs.map((tab) => {
+            return {
+                tabId: tab.tabId,
+                title: tab.title,
+                url: sub(tab.url, 300),
+            };
+        });
+    }
+}
+
+export { Agent, AgentChain, AgentContext, AgentWrapTool, BaseBrowserAgent, BaseBrowserLabelsAgent, BaseBrowserScreenAgent, Chain, ChatAgent, ChatContext, TaskContext as Context, DeepActionTool, Eko, EkoMemory, ForeachTaskTool, GlobalPromptKey, HumanInteractTool, Log, Planner, PromptTemplate, RetryLanguageModel, SimpleHttpMcpClient, SimpleSseMcpClient, TaskContext, TaskNodeStatusTool, TaskVariableStorageTool, VariableStorageTool, WatchTriggerTool, WebSearchTool, WebpageQaTool, buildAgentTree, buildSimpleAgentWorkflow, callLLM, callWithReAct, call_timeout, compressImageData, config$1 as config, convertToolSchema, Eko as default, extract_page_content, global, mergeTools, parseWorkflow, resetWorkflowXml, sub, toFile, toImage, uuidv4 };
 //# sourceMappingURL=index.esm.js.map