@eko-ai/eko 2.1.0-alpha.5 → 2.1.0-alpha.7

package/dist/config/index.d.ts

@@ -5,6 +5,7 @@ type GlobalConfig = {
     maxTokens: number;
     compressThreshold: number;
     largeTextLength: number;
+    maxDialogueImgFileNum: number;
 };
 declare const config: GlobalConfig;
 export default config;

package/dist/config/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/config/index.ts"],"names":[],"mappings":"AAAA,KAAK,YAAY,GAAG;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,SAAS,GAAG,KAAK,GAAG,OAAO,CAAC;IACtC,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,eAAe,EAAE,MAAM,CAAC;CACzB,CAAA;AAED,QAAA,MAAM,MAAM,EAAE,YAOb,CAAC;AAEF,eAAe,MAAM,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/config/index.ts"],"names":[],"mappings":"AAAA,KAAK,YAAY,GAAG;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,SAAS,GAAG,KAAK,GAAG,OAAO,CAAC;IACtC,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,eAAe,EAAE,MAAM,CAAC;IACxB,qBAAqB,EAAE,MAAM,CAAC;CAC/B,CAAA;AAED,QAAA,MAAM,MAAM,EAAE,YAQb,CAAC;AAEF,eAAe,MAAM,CAAC"}

package/dist/index.cjs.js

@@ -11,6 +11,7 @@ const config = {
     maxTokens: 16000,
     compressThreshold: 80,
     largeTextLength: 5000,
+    maxDialogueImgFileNum: 2,
 };
 
 var LogLevel;
@@ -17682,7 +17683,7 @@ async function compressAgentMessages(agentContext, rlm, messages, tools) {
             toolName: toolCall.toolName,
             params: args,
             toolResult: toolResult,
-        });
+        }, agentContext);
     }
     // handle original messages
     let firstToolIndex = 3;
@@ -17701,6 +17702,7 @@ async function compressAgentMessages(agentContext, rlm, messages, tools) {
 function handleLargeContextMessages(messages) {
     let imageNum = 0;
     let fileNum = 0;
+    let maxNum = config.maxDialogueImgFileNum;
     let longTextTools = {};
     for (let i = messages.length - 1; i >= 0; i--) {
         let message = messages[i];
@@ -17708,7 +17710,7 @@ function handleLargeContextMessages(messages) {
             for (let j = 0; j < message.content.length; j++) {
                 let content = message.content[j];
                 if (content.type == "image") {
-                    if (++imageNum == 1) {
+                    if (++imageNum <= maxNum) {
                         break;
                     }
                     content = {
@@ -17718,7 +17720,7 @@ function handleLargeContextMessages(messages) {
                     message.content[j] = content;
                 }
                 else if (content.type == "file") {
-                    if (++fileNum == 1) {
+                    if (++fileNum <= maxNum) {
                         break;
                     }
                     content = {
@@ -17739,7 +17741,7 @@ function handleLargeContextMessages(messages) {
                 for (let r = 0; r < toolContent.length; r++) {
                     let _content = toolContent[r];
                     if (_content.type == "image") {
-                        if (++imageNum == 1) {
+                        if (++imageNum <= maxNum) {
                             break;
                         }
                         _content = {
@@ -18385,7 +18387,7 @@ class Agent {
                     toolName: result.toolName,
                     params: result.args || {},
                     toolResult: toolResult,
-                });
+                }, agentContext);
             }
             let llmToolResult = this.convertToolResult(result, toolResult, user_messages);
             toolResults.push(llmToolResult);
@@ -18641,7 +18643,7 @@ async function callLLM(agentContext, rlm, messages, tools, noCompress, toolChoic
                         streamId,
                         streamDone: false,
                         text: streamText,
-                    });
+                    }, agentContext);
                     break;
                 }
                 case "reasoning": {
@@ -18654,7 +18656,7 @@ async function callLLM(agentContext, rlm, messages, tools, noCompress, toolChoic
                         streamId,
                         streamDone: false,
                         text: thinkText,
-                    });
+                    }, agentContext);
                     break;
                 }
                 case "tool-call-delta": {
@@ -18668,7 +18670,7 @@ async function callLLM(agentContext, rlm, messages, tools, noCompress, toolChoic
                             streamId,
                             streamDone: true,
                             text: streamText,
-                        });
+                        }, agentContext);
                     }
                     toolArgsText += chunk.argsTextDelta || "";
                     await streamCallback.onMessage({
@@ -18679,7 +18681,7 @@ async function callLLM(agentContext, rlm, messages, tools, noCompress, toolChoic
                         toolId: chunk.toolCallId,
                         toolName: chunk.toolName,
                         paramsText: toolArgsText,
-                    });
+                    }, agentContext);
                     break;
                 }
                 case "tool-call": {
@@ -18694,7 +18696,7 @@ async function callLLM(agentContext, rlm, messages, tools, noCompress, toolChoic
                         toolName: chunk.toolName,
                         params: args,
                     };
-                    await streamCallback.onMessage(message);
+                    await streamCallback.onMessage(message, agentContext);
                     toolParts.push({
                         type: "tool-call",
                         toolCallId: chunk.toolCallId,
@@ -18711,7 +18713,7 @@ async function callLLM(agentContext, rlm, messages, tools, noCompress, toolChoic
                         type: "file",
                         mimeType: chunk.mimeType,
                         data: chunk.data,
-                    });
+                    }, agentContext);
                     break;
                 }
                 case "error": {
@@ -18722,7 +18724,7 @@ async function callLLM(agentContext, rlm, messages, tools, noCompress, toolChoic
                         nodeId: agentNode.id,
                         type: "error",
                         error: chunk.error,
-                    });
+                    }, agentContext);
                     throw new Error("Plan Error");
                 }
                 case "finish": {
@@ -18736,7 +18738,7 @@ async function callLLM(agentContext, rlm, messages, tools, noCompress, toolChoic
                             streamId,
                             streamDone: true,
                             text: streamText,
-                        });
+                        }, agentContext);
                     }
                     await streamCallback.onMessage({
                         taskId: context.taskId,
@@ -18745,7 +18747,7 @@ async function callLLM(agentContext, rlm, messages, tools, noCompress, toolChoic
                         type: "finish",
                         finishReason: chunk.finishReason,
                         usage: chunk.usage,
-                    });
+                    }, agentContext);
                     if (chunk.finishReason === "length" &&
                         messages.length >= 10 &&
                         !noCompress &&