@minded-ai/mindedjs 2.0.13-beta.9 → 2.0.14-beta-1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/agent.d.ts.map +1 -1
- package/dist/agent.js +16 -6
- package/dist/agent.js.map +1 -1
- package/dist/browserTask/README.md +419 -0
- package/dist/browserTask/browserAgent.py +632 -0
- package/dist/browserTask/captcha_isolated.png +0 -0
- package/dist/browserTask/executeBrowserTask.d.ts +2 -2
- package/dist/browserTask/executeBrowserTask.d.ts.map +1 -1
- package/dist/browserTask/executeBrowserTask.js +7 -28
- package/dist/browserTask/executeBrowserTask.js.map +1 -1
- package/dist/browserTask/executeBrowserTask.ts +79 -0
- package/dist/browserTask/requirements.txt +8 -0
- package/dist/browserTask/setup.sh +144 -0
- package/dist/cli/index.js +7 -35
- package/dist/cli/index.js.map +1 -1
- package/dist/cli/lambdaHandlerTemplate.d.ts.map +1 -1
- package/dist/cli/lambdaHandlerTemplate.js +2 -1
- package/dist/cli/lambdaHandlerTemplate.js.map +1 -1
- package/dist/cli/lambdaHandlerTemplate.ts +3 -2
- package/dist/internalTools/retell.js +17 -7
- package/dist/internalTools/retell.js.map +1 -1
- package/dist/internalTools/sendPlaceholderMessage.js +17 -7
- package/dist/internalTools/sendPlaceholderMessage.js.map +1 -1
- package/dist/nodes/addBrowserTaskNode.js +2 -2
- package/dist/nodes/addBrowserTaskNode.js.map +1 -1
- package/dist/nodes/addBrowserTaskRunNode.d.ts.map +1 -1
- package/dist/nodes/addBrowserTaskRunNode.js +4 -1
- package/dist/nodes/addBrowserTaskRunNode.js.map +1 -1
- package/dist/toolsLibrary/classifier.d.ts.map +1 -1
- package/dist/toolsLibrary/classifier.js +103 -33
- package/dist/toolsLibrary/classifier.js.map +1 -1
- package/dist/types/LLM.types.js.map +1 -1
- package/dist/utils/extractStateMemoryResponse.d.ts +4 -0
- package/dist/utils/extractStateMemoryResponse.d.ts.map +1 -1
- package/dist/utils/extractStateMemoryResponse.js +89 -87
- package/dist/utils/extractStateMemoryResponse.js.map +1 -1
- package/package.json +1 -1
- package/src/agent.ts +15 -6
- package/src/browserTask/executeBrowserTask.ts +22 -47
- package/src/cli/index.ts +7 -34
- package/src/cli/lambdaHandlerTemplate.ts +3 -2
- package/src/nodes/addBrowserTaskNode.ts +2 -2
- package/src/nodes/addBrowserTaskRunNode.ts +5 -1
- package/src/toolsLibrary/classifier.ts +115 -37
- package/src/types/LLM.types.ts +2 -2
- package/dist/cli/localOperatorSetup.d.ts +0 -6
- package/dist/cli/localOperatorSetup.d.ts.map +0 -1
- package/dist/cli/localOperatorSetup.js +0 -300
- package/dist/cli/localOperatorSetup.js.map +0 -1
- package/dist/guidelines/guidelinesManager.d.ts +0 -37
- package/dist/guidelines/guidelinesManager.d.ts.map +0 -1
- package/dist/guidelines/guidelinesManager.js +0 -172
- package/dist/guidelines/guidelinesManager.js.map +0 -1
- package/dist/internalTools/documentExtraction.d.ts +0 -69
- package/dist/internalTools/documentExtraction.d.ts.map +0 -1
- package/dist/internalTools/documentExtraction.js +0 -65
- package/dist/internalTools/documentExtraction.js.map +0 -1
- package/dist/internalTools/managedDocumentParser/managedDocumentParser.d.ts +0 -10
- package/dist/internalTools/managedDocumentParser/managedDocumentParser.d.ts.map +0 -1
- package/dist/internalTools/managedDocumentParser/managedDocumentParser.js +0 -80
- package/dist/internalTools/managedDocumentParser/managedDocumentParser.js.map +0 -1
- package/dist/pii-gateway/gateway.d.ts +0 -17
- package/dist/pii-gateway/gateway.d.ts.map +0 -1
- package/dist/pii-gateway/gateway.js +0 -77
- package/dist/pii-gateway/gateway.js.map +0 -1
- package/dist/pii-gateway/index.d.ts +0 -4
- package/dist/pii-gateway/index.d.ts.map +0 -1
- package/dist/pii-gateway/index.js +0 -14
- package/dist/pii-gateway/index.js.map +0 -1
- package/dist/pii-gateway/proxy.d.ts +0 -80
- package/dist/pii-gateway/proxy.d.ts.map +0 -1
- package/dist/pii-gateway/proxy.js +0 -106
- package/dist/pii-gateway/proxy.js.map +0 -1
- package/dist/pii-gateway/types.d.ts +0 -27
- package/dist/pii-gateway/types.d.ts.map +0 -1
- package/dist/pii-gateway/types.js +0 -3
- package/dist/pii-gateway/types.js.map +0 -1
- package/dist/platform/mindedChatOpenAI.d.ts +0 -5
- package/dist/platform/mindedChatOpenAI.d.ts.map +0 -1
- package/dist/platform/mindedChatOpenAI.js +0 -23
- package/dist/platform/mindedChatOpenAI.js.map +0 -1
- package/dist/platform/mindedUtils.d.ts +0 -1
- package/dist/platform/mindedUtils.d.ts.map +0 -1
- package/dist/platform/mindedUtils.js +0 -2
- package/dist/platform/mindedUtils.js.map +0 -1
- package/dist/platform/parallelAzureChatOpenAI.d.ts +0 -39
- package/dist/platform/parallelAzureChatOpenAI.d.ts.map +0 -1
- package/dist/platform/parallelAzureChatOpenAI.js +0 -114
- package/dist/platform/parallelAzureChatOpenAI.js.map +0 -1
- package/dist/platform/parallelChatOpenAI.d.ts +0 -38
- package/dist/platform/parallelChatOpenAI.d.ts.map +0 -1
- package/dist/platform/parallelChatOpenAI.js +0 -111
- package/dist/platform/parallelChatOpenAI.js.map +0 -1
- package/dist/platform/parallelMindedChatOpenAI.d.ts +0 -39
- package/dist/platform/parallelMindedChatOpenAI.d.ts.map +0 -1
- package/dist/platform/parallelMindedChatOpenAI.js +0 -111
- package/dist/platform/parallelMindedChatOpenAI.js.map +0 -1
- package/dist/platform/piiGateway/proxy.d.ts +0 -20
- package/dist/platform/piiGateway/proxy.d.ts.map +0 -1
- package/dist/platform/piiGateway/proxy.js +0 -24
- package/dist/platform/piiGateway/proxy.js.map +0 -1
- package/dist/toolsLibrary/browser/extraction.d.ts +0 -83
- package/dist/toolsLibrary/browser/extraction.d.ts.map +0 -1
- package/dist/toolsLibrary/browser/extraction.js +0 -328
- package/dist/toolsLibrary/browser/extraction.js.map +0 -1
- package/dist/types/Agent.d.ts +0 -2
- package/dist/types/Agent.d.ts.map +0 -1
- package/dist/types/Agent.js +0 -18
- package/dist/types/Agent.js.map +0 -1
- package/dist/types/Flows.d.ts +0 -2
- package/dist/types/Flows.d.ts.map +0 -1
- package/dist/types/Flows.js +0 -18
- package/dist/types/Flows.js.map +0 -1
- package/dist/types/LLM.d.ts +0 -2
- package/dist/types/LLM.d.ts.map +0 -1
- package/dist/types/LLM.js +0 -18
- package/dist/types/LLM.js.map +0 -1
- package/dist/types/LangGraph.d.ts +0 -2
- package/dist/types/LangGraph.d.ts.map +0 -1
- package/dist/types/LangGraph.js +0 -18
- package/dist/types/LangGraph.js.map +0 -1
- package/dist/types/Platform.d.ts +0 -2
- package/dist/types/Platform.d.ts.map +0 -1
- package/dist/types/Platform.js +0 -18
- package/dist/types/Platform.js.map +0 -1
- package/dist/types/Tools.d.ts +0 -2
- package/dist/types/Tools.d.ts.map +0 -1
- package/dist/types/Tools.js +0 -18
- package/dist/types/Tools.js.map +0 -1
- package/dist/types/Triggers.d.ts +0 -2
- package/dist/types/Triggers.d.ts.map +0 -1
- package/dist/types/Triggers.js +0 -18
- package/dist/types/Triggers.js.map +0 -1
- package/dist/types/Triggers.types.d.ts +0 -25
- package/dist/types/Triggers.types.d.ts.map +0 -1
- package/dist/types/Triggers.types.js +0 -3
- package/dist/types/Triggers.types.js.map +0 -1
- package/dist/types/Voice.d.ts +0 -2
- package/dist/types/Voice.d.ts.map +0 -1
- package/dist/types/Voice.js +0 -18
- package/dist/types/Voice.js.map +0 -1
- package/dist/types/index.d.ts +0 -9
- package/dist/types/index.d.ts.map +0 -1
- package/dist/types/index.js +0 -25
- package/dist/types/index.js.map +0 -1
- package/dist/utils/LogicalConditionDebugger.d.ts +0 -57
- package/dist/utils/LogicalConditionDebugger.d.ts.map +0 -1
- package/dist/utils/LogicalConditionDebugger.js +0 -143
- package/dist/utils/LogicalConditionDebugger.js.map +0 -1
- package/dist/utils/documentProcessor.d.ts +0 -87
- package/dist/utils/documentProcessor.d.ts.map +0 -1
- package/dist/utils/documentProcessor.js +0 -726
- package/dist/utils/documentProcessor.js.map +0 -1
- package/dist/utils/extractToolGuidelineParams.d.ts +0 -4
- package/dist/utils/extractToolGuidelineParams.d.ts.map +0 -1
- package/dist/utils/extractToolGuidelineParams.js +0 -20
- package/dist/utils/extractToolGuidelineParams.js.map +0 -1
- package/dist/utils/extractToolMemoryResponse.d.ts +0 -4
- package/dist/utils/extractToolMemoryResponse.d.ts.map +0 -1
- package/dist/utils/extractToolMemoryResponse.js +0 -16
- package/dist/utils/extractToolMemoryResponse.js.map +0 -1
- package/dist/utils/parseAttachments.d.ts +0 -14
- package/dist/utils/parseAttachments.d.ts.map +0 -1
- package/dist/utils/parseAttachments.js +0 -54
- package/dist/utils/parseAttachments.js.map +0 -1
- package/src/cli/localOperatorSetup.ts +0 -308
- package/src/toolsLibrary/browser/extraction.ts +0 -393
|
@@ -9,6 +9,7 @@ const uuid_1 = require("uuid");
|
|
|
9
9
|
const executeBrowserTask_1 = require("../browserTask/executeBrowserTask");
|
|
10
10
|
const tools_1 = require("../platform/utils/tools");
|
|
11
11
|
const config_1 = require("../platform/config");
|
|
12
|
+
const compilePrompt_1 = require("./compilePrompt");
|
|
12
13
|
const buildBrowserTaskRunNodeName = (nodeName) => `${nodeName}${Flows_types_1.internalNodesSuffix.BROWSER_TASK_RUN}`;
|
|
13
14
|
exports.buildBrowserTaskRunNodeName = buildBrowserTaskRunNodeName;
|
|
14
15
|
const addBrowserTaskRunNode = async ({ graph, browserTaskNode, attachedToNodeName, agent }) => {
|
|
@@ -37,12 +38,14 @@ const addBrowserTaskRunNode = async ({ graph, browserTaskNode, attachedToNodeNam
|
|
|
37
38
|
description: tool.description,
|
|
38
39
|
inputParams: (0, tools_1.zodSchemaToParams)(tool.input),
|
|
39
40
|
}));
|
|
41
|
+
// We compile the env variables to avoid having to pass them to the platform in the tool input
|
|
42
|
+
const promptCompiledWithEnv = (0, compilePrompt_1.compilePrompt)(prompt, { env: process.env });
|
|
40
43
|
const { browserTaskMode } = (0, config_1.getConfig)();
|
|
41
44
|
// Invoke browser task via socket
|
|
42
45
|
const result = await (0, executeBrowserTask_1.invokeBrowserTask)({
|
|
43
46
|
sessionId,
|
|
44
47
|
cdpUrl,
|
|
45
|
-
task:
|
|
48
|
+
task: promptCompiledWithEnv,
|
|
46
49
|
keepAlive,
|
|
47
50
|
hooks,
|
|
48
51
|
browserTaskMode,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"addBrowserTaskRunNode.js","sourceRoot":"","sources":["../../src/nodes/addBrowserTaskRunNode.ts"],"names":[],"mappings":";;;AAEA,sDAAsF;AACtF,uDAAkE;AAElE,4CAAyC;AACzC,8CAAqD;AAErD,+BAAoC;AACpC,0EAAsE;AACtE,mDAA4D;AAC5D,+CAA+C;
|
|
1
|
+
{"version":3,"file":"addBrowserTaskRunNode.js","sourceRoot":"","sources":["../../src/nodes/addBrowserTaskRunNode.ts"],"names":[],"mappings":";;;AAEA,sDAAsF;AACtF,uDAAkE;AAElE,4CAAyC;AACzC,8CAAqD;AAErD,+BAAoC;AACpC,0EAAsE;AACtE,mDAA4D;AAC5D,+CAA+C;AAC/C,mDAAgD;AASzC,MAAM,2BAA2B,GAAG,CAAC,QAAgB,EAAE,EAAE,CAAC,GAAG,QAAQ,GAAG,iCAAmB,CAAC,gBAAgB,EAAE,CAAC;AAAzG,QAAA,2BAA2B,+BAA8E;AAE/G,MAAM,qBAAqB,GAAG,KAAK,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,kBAAkB,EAAE,KAAK,EAA+B,EAAE,EAAE;IAChI,MAAM,QAAQ,GAAiB,KAAK,EAAE,KAAmC,EAAE,EAAE;;QAC3E,eAAM,CAAC,IAAI,CAAC,uCAAuC,eAAe,CAAC,WAAW,EAAE,CAAC,CAAC;QAElF,MAAM,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAQ,CAAC;QACrE,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,CAAC;QACD,MAAM,sBAAsB,GAAG,MAAA,WAAW,CAAC,iBAAiB,0CAAE,cAAc,CAAC;QAC7E,MAAM,QAAQ,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,EAAE,MAAM,EAAE,GAAG,WAAW,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC;QACjD,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,GAAG,sBAAsB,CAAC;QAErF,IAAI,CAAC;YACH,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC;gBACtB,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACrC,CAAC;YAED,IAAI,CAAC,SAAS,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;YAC/E,CAAC;YAED,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK;iBAC5B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,KAAK,IAAI,CAAC;iBACtD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACd,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,WAAW,EAAE,IAAA,yBAAiB,EAAC,IAAI,CAAC,KAAK,CAAC;aAC3C,CAAC,CAAC,CAAC;YAEN,8FAA8F;YAC9F,MAAM,qBAAqB,GAAG,IAAA,6BAAa,EAAC,MAAM,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;YAE1E,MAAM,EAAE,eAAe,EAAE,GAAG,IAAA,kBAAS,GAAE,CAAC;YACxC,iCAAiC;YACjC,MAAM,MAAM,GAAG,MAAM,IAAA,sCAAiB,EAAC;gBACrC,SAAS;gBACT,MAAM;gBACN,IAAI,EAAE,qBAAqB;gBAC3B,SAAS;gBACT,KAAK;gBACL,eAAe;gBACf,WAAW;gBACX,YAAY;aACb,CAAC,CAAC;YAEH,eAAM,CAAC,KAAK,CAAC;gBACX,GAAG,EAAE,wBAAwB;gBAC7B,SAAS;gBACT,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM;gBAC1B,SAAS,EAAE,CAAA,MAAA,MAAM,CAAC,KAAK,0CAAE,MAAM,KAAI,CAAC;gBACpC,cAAc,EAAE,CAAA,MAAA,MAAM,CAAC,UAAU,0CAAE,MAAM,KAAI,CAAC;aAC/C,CAAC,CAAC;YAEH,sCAAsC;YACtC,MAAM,WAAW,GAAG,IAAI,sBAAW,CAAC;gBAClC,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;oBACtB,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,gCAAgC;oBACzD,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,EAAE;oBACzB,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,EAAE;oBACnC,WAAW,EAAE,WAAW;iBACzB,CAAC;gBACF,IAAI,EAAE,cAAc;gBACpB,YAAY,EAAE,QAAQ,CAAC,EAAE;gBACzB,MAAM,EAAE,SAAS;gBACjB,iBAAiB,EAAE;oBACjB,cAAc,EAAE;wBACd,QAAQ,EAAE,sBAAQ,CAAC,YAAY;wBAC/B,eAAe,EAAE,eAAe,CAAC,WAAW;wBAC5C,SAAS;wBACT,KAAK,EAAE,MAAM,CAAC,KAAK;wBACnB,UAAU,EAAE,MAAM,CAAC,UAAU;wBAC7B,WAAW,EAAE,WAAW;qBACzB;iBACF;aACF,CAAC,CAAC;YAEH,qEAAqE;YACrE,MAAM,gBAAgB,GAAG,IAAI,oBAAS,CAAC;gBACrC,EAAE,EAAE,WAAW,CAAC,EAAE;gBAClB,OAAO,EAAE,WAAW,CAAC,OAAO;gBAC5B,UAAU,EAAE,WAAW,CAAC,UAAU;gBAClC,iBAAiB,EAAE;oBACjB,GAAG,WAAW,CAAC,iBAAiB;oBAChC,cAAc,EAAE;wBACd,GAAG,MAAA,WAAW,CAAC,iBAAiB,0CAAE,cAAc;wBAChD,eAAe,EAAE,eAAe,CAAC,WAAW;wBAC5C,KAAK,EAAE,MAAM,CAAC,KAAK;wBACnB,UAAU,EAAE,MAAM,CAAC,UAAU;wBAC7B,MAAM,EAAE,WAAW;qBACpB;oBACD,MAAM,EAAE,IAAI,EAAE,mEAAmE;iBAClF;aACF,CAAC,CAAC;YAEH,KAAK,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC;YACrB,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,gBAAgB,EAAE,WAAW,CAAC,CAAC;YACnD,KAAK,CAAC,OAAO,CAAC,IAAI,CAChB,IAAA,2BAAiB,EAAc,KAAK,CAAC,OAAO,EAAE;gBAC5C,IAAI,EAAE,sBAAQ,CAAC,YAAY;gBAC3B,MAAM,EAAE,eAAe,CAAC,IAAI;gBAC5B,eAAe,EAAE,eAAe,CAAC,WAAW;gBAC5C,GAAG,EAAE,MAAM;gBACX,UAAU,EAAE,CAAC,WAAW,CAAC,EAAG,CAAC;aAC9B,CAAC,CACH,CAAC;YACF,OAAO,KAAK,CAAC;QACf,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,uCAAuC,EAAE,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YAExF,MAAM,gBAAgB,GAAG,IAAI,sBAAW,CAAC;gBACvC,EAAE,EAAE,IAAA,SAAM,GAAE;gBACZ,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;oBACtB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE;oBAChE,WAAW,EAAE,WAAW;oBACxB,OAAO,EAAE,gCAAgC;iBAC1C,CAAC;gBACF,MAAM,EAAE,OAAO;gBACf,IAAI,EAAE,cAAc;gBACpB,YAAY,EAAE,QAAQ,CAAC,EAAE;gBACzB,iBAAiB,EAAE;oBACjB,cAAc,EAAE;wBACd,QAAQ,EAAE,sBAAQ,CAAC,YAAY;wBAC/B,eAAe,EAAE,eAAe,CAAC,WAAW;wBAC5C,SAAS;wBACT,KAAK,EAAE,KAAK,CAAC,OAAO;wBACpB,WAAW,EAAE,WAAW;qBACzB;iBACF;aACF,CAAC,CAAC;YAEH,0CAA0C;YAC1C,MAAM,gBAAgB,GAAG,IAAI,oBAAS,CAAC;gBACrC,EAAE,EAAE,WAAW,CAAC,EAAE;gBAClB,OAAO,EAAE,WAAW,CAAC,OAAO;gBAC5B,UAAU,EAAE,WAAW,CAAC,UAAU;gBAClC,iBAAiB,EAAE;oBACjB,GAAG,WAAW,CAAC,iBAAiB;oBAChC,cAAc,EAAE;wBACd,GAAG,MAAA,WAAW,CAAC,iBAAiB,0CAAE,cAAc;wBAChD,eAAe,EAAE,eAAe,CAAC,WAAW;wBAC5C,MAAM,EAAE,QAAQ;wBAChB,KAAK,EAAE,KAAK,CAAC,OAAO;qBACrB;oBACD,MAAM,EAAE,IAAI;iBACb;aACF,CAAC,CAAC;YAEH,KAAK,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC;YACrB,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,gBAAgB,EAAE,gBAAgB,CAAC,CAAC;YACxD,KAAK,CAAC,OAAO,CAAC,IAAI,CAChB,IAAA,2BAAiB,EAAc,KAAK,CAAC,OAAO,EAAE;gBAC5C,IAAI,EAAE,sBAAQ,CAAC,YAAY;gBAC3B,MAAM,EAAE,eAAe,CAAC,IAAI;gBAC5B,eAAe,EAAE,eAAe,CAAC,WAAW;gBAC5C,GAAG,EAAE,KAAK;gBACV,UAAU,EAAE,EAAE;aACf,CAAC,CACH,CAAC;YACF,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC,CAAC;IAEF,KAAK,CAAC,OAAO,CAAC,IAAA,mCAA2B,EAAC,kBAAkB,CAAC,EAAE,QAAQ,CAAC,CAAC;IACzE,KAAK,CAAC,OAAO,CAAC,kBAAyB,EAAE,IAAA,mCAA2B,EAAC,kBAAkB,CAAQ,CAAC,CAAC;AACnG,CAAC,CAAC;AAtKW,QAAA,qBAAqB,yBAsKhC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"classifier.d.ts","sourceRoot":"","sources":["../../src/toolsLibrary/classifier.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAC;AAI5C,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AAGzE,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,eAAe,EAAE,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IAC/B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB;
|
|
1
|
+
{"version":3,"file":"classifier.d.ts","sourceRoot":"","sources":["../../src/toolsLibrary/classifier.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAC;AAI5C,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AAGzE,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,eAAe,EAAE,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IAC/B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB;AAsCD;;;;;;GAMG;AACH,wBAAsB,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,GAAG,EAAE,iBAAiB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CA8F/H;AAED;;;;;GAKG;AACH,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,aAazF,MAAM,OAAO,iBAAiB,mCAChD;AAGD,eAAO,MAAM,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAoBjB,CAAC;AAEH,QAAA,MAAM,cAAc,EAAE,IAAI,CAAC,OAAO,MAAM,EAAE,GAAG,CA6E5C,CAAC;AAEF,eAAe,cAAc,CAAC;AAG9B,wBAAsB,aAAa,CACjC,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,gBAAgB,GAAG;IAAE,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,EAClD,GAAG,EAAE,iBAAiB,GACrB,OAAO,CAAC,oBAAoB,EAAE,CAAC,CA6CjC"}
|
|
@@ -15,6 +15,30 @@ const DEFAULT_CONFIG = {
|
|
|
15
15
|
defaultClass: 'unknown',
|
|
16
16
|
defaultReason: 'Unable to determine classification',
|
|
17
17
|
};
|
|
18
|
+
/**
|
|
19
|
+
* Validates the classification result
|
|
20
|
+
* @param result The classification result to validate
|
|
21
|
+
* @param config The classifier configuration
|
|
22
|
+
* @returns True if the result is valid, false otherwise
|
|
23
|
+
*/
|
|
24
|
+
function validateClassificationResult(result, config) {
|
|
25
|
+
// Check if result exists and has a class property
|
|
26
|
+
if (!result || !result.class) {
|
|
27
|
+
return false;
|
|
28
|
+
}
|
|
29
|
+
// Check if the class is one of the configured classes
|
|
30
|
+
const validClasses = config.classes.map((c) => c.name);
|
|
31
|
+
if (!validClasses.includes(result.class)) {
|
|
32
|
+
return false;
|
|
33
|
+
}
|
|
34
|
+
// If we expect JSON format with confidence, validate it
|
|
35
|
+
if (config.outputFormat === 'json' && result.confidence !== undefined) {
|
|
36
|
+
if (typeof result.confidence !== 'number' || result.confidence < 0 || result.confidence > 1) {
|
|
37
|
+
return false;
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
return true;
|
|
41
|
+
}
|
|
18
42
|
/**
|
|
19
43
|
* Generic classifier utility that can be used standalone
|
|
20
44
|
* @param content The content to classify
|
|
@@ -24,43 +48,89 @@ const DEFAULT_CONFIG = {
|
|
|
24
48
|
*/
|
|
25
49
|
async function classify(content, config, llm) {
|
|
26
50
|
const mergedConfig = { ...DEFAULT_CONFIG, ...config };
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
51
|
+
const MAX_RETRIES = 3;
|
|
52
|
+
let lastError = null;
|
|
53
|
+
let lastInvalidResult = null;
|
|
54
|
+
for (let attempt = 1; attempt <= MAX_RETRIES; attempt++) {
|
|
55
|
+
try {
|
|
56
|
+
// Build the classification prompt
|
|
57
|
+
const classesDescription = mergedConfig.classes.map((c) => `${c.name}: ${c.description}`).join('\n');
|
|
58
|
+
const validClassNames = mergedConfig.classes.map((c) => c.name);
|
|
59
|
+
const basePrompt = mergedConfig.systemPrompt ||
|
|
60
|
+
'You are a classifier. Your task is to classify the given content into one of the following categories:';
|
|
61
|
+
let prompt = `${basePrompt}\n\n${classesDescription}\n\n`;
|
|
62
|
+
// Add retry feedback if this is not the first attempt
|
|
63
|
+
if (attempt > 1 && lastInvalidResult) {
|
|
64
|
+
prompt += `\n⚠️ IMPORTANT: Your previous classification attempt was INVALID. `;
|
|
65
|
+
if (lastInvalidResult.class && !validClassNames.includes(lastInvalidResult.class)) {
|
|
66
|
+
prompt += `You selected "${lastInvalidResult.class}" which is NOT in the list of valid classes. `;
|
|
67
|
+
}
|
|
68
|
+
else if (!lastInvalidResult.class) {
|
|
69
|
+
prompt += `You did not provide a class value. `;
|
|
70
|
+
}
|
|
71
|
+
prompt += `You MUST select ONLY from these valid classes:\n${validClassNames.join(', ')}\n\n`;
|
|
72
|
+
}
|
|
73
|
+
if (mergedConfig.outputFormat === 'json') {
|
|
74
|
+
prompt += `You should output the result in the following JSON format: {
|
|
75
|
+
"class": "<selected class name>",
|
|
76
|
+
${mergedConfig.includeReason ? '"reason": "<explanation for the classification>",' : ''}
|
|
77
|
+
"confidence": <confidence score between 0 and 1>
|
|
78
|
+
}.\nReturn JSON and nothing more.`;
|
|
79
|
+
}
|
|
80
|
+
else {
|
|
81
|
+
prompt += 'Return only the class name.';
|
|
82
|
+
}
|
|
83
|
+
prompt += `\n\nContent to classify:\n${content}`;
|
|
84
|
+
// Make the classification request
|
|
85
|
+
let result;
|
|
86
|
+
if (mergedConfig.outputFormat === 'json') {
|
|
87
|
+
const parser = new output_parsers_1.JsonOutputParser();
|
|
88
|
+
const parsedResult = await llm.pipe(parser).invoke([new messages_1.SystemMessage(prompt)]);
|
|
89
|
+
result = parsedResult;
|
|
90
|
+
}
|
|
91
|
+
else {
|
|
92
|
+
const llmResult = await llm.invoke([new messages_1.SystemMessage(prompt)]);
|
|
93
|
+
const classText = typeof llmResult.content === 'string' ? llmResult.content.trim() : '';
|
|
94
|
+
result = { class: classText };
|
|
95
|
+
}
|
|
96
|
+
// Validate the result
|
|
97
|
+
if (!validateClassificationResult(result, mergedConfig)) {
|
|
98
|
+
lastInvalidResult = result;
|
|
99
|
+
throw new Error(`Invalid classification result: ${JSON.stringify(result)}`);
|
|
100
|
+
}
|
|
101
|
+
// Success - return the valid result
|
|
47
102
|
return result;
|
|
48
103
|
}
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
104
|
+
catch (err) {
|
|
105
|
+
lastError = err;
|
|
106
|
+
logger_1.logger.warn({
|
|
107
|
+
message: `Classification attempt ${attempt} failed`,
|
|
108
|
+
attempt,
|
|
109
|
+
maxRetries: MAX_RETRIES,
|
|
110
|
+
error: lastError.message,
|
|
111
|
+
invalidResult: lastInvalidResult,
|
|
112
|
+
});
|
|
113
|
+
// If this is not the last attempt, continue to retry
|
|
114
|
+
if (attempt < MAX_RETRIES) {
|
|
115
|
+
continue;
|
|
116
|
+
}
|
|
53
117
|
}
|
|
54
118
|
}
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
}
|
|
119
|
+
// All retries exhausted - fallback to default
|
|
120
|
+
logger_1.logger.warn({
|
|
121
|
+
message: 'Classification failed after max retries, using default class',
|
|
122
|
+
maxRetries: MAX_RETRIES,
|
|
123
|
+
lastError: lastError === null || lastError === void 0 ? void 0 : lastError.message,
|
|
124
|
+
lastInvalidResult,
|
|
125
|
+
defaultClass: mergedConfig.defaultClass,
|
|
126
|
+
defaultReason: mergedConfig.defaultReason,
|
|
127
|
+
});
|
|
128
|
+
// Return default classification
|
|
129
|
+
return {
|
|
130
|
+
class: mergedConfig.defaultClass || 'unknown',
|
|
131
|
+
reason: mergedConfig.defaultReason || `Classification failed after ${MAX_RETRIES} attempts: ${lastError === null || lastError === void 0 ? void 0 : lastError.message}`,
|
|
132
|
+
confidence: 0,
|
|
133
|
+
};
|
|
64
134
|
}
|
|
65
135
|
/**
|
|
66
136
|
* Create a classifier from a simple class list
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"classifier.js","sourceRoot":"","sources":["../../src/toolsLibrary/classifier.ts"],"names":[],"mappings":";;;
|
|
1
|
+
{"version":3,"file":"classifier.js","sourceRoot":"","sources":["../../src/toolsLibrary/classifier.ts"],"names":[],"mappings":";;;AAwEA,4BA8FC;AAQD,4CAcC;AA2GD,sCAiDC;AAxVD,6BAAwB;AAExB,4CAAyC;AACzC,mEAAkE;AAClE,uDAAyD;AAyBzD,wBAAwB;AACxB,MAAM,cAAc,GAA8B;IAChD,YAAY,EAAE,MAAM;IACpB,aAAa,EAAE,IAAI;IACnB,YAAY,EAAE,SAAS;IACvB,aAAa,EAAE,oCAAoC;CACpD,CAAC;AAEF;;;;;GAKG;AACH,SAAS,4BAA4B,CAAC,MAA4B,EAAE,MAAwB;IAC1F,kDAAkD;IAClD,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAC7B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,sDAAsD;IACtD,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACvD,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACzC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,wDAAwD;IACxD,IAAI,MAAM,CAAC,YAAY,KAAK,MAAM,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QACtE,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,GAAG,CAAC,IAAI,MAAM,CAAC,UAAU,GAAG,CAAC,EAAE,CAAC;YAC5F,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;;;;GAMG;AACI,KAAK,UAAU,QAAQ,CAAC,OAAe,EAAE,MAAwB,EAAE,GAAsB;IAC9F,MAAM,YAAY,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;IACtD,MAAM,WAAW,GAAG,CAAC,CAAC;IACtB,IAAI,SAAS,GAAiB,IAAI,CAAC;IACnC,IAAI,iBAAiB,GAAgC,IAAI,CAAC;IAE1D,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;QACxD,IAAI,CAAC;YACH,kCAAkC;YAClC,MAAM,kBAAkB,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrG,MAAM,eAAe,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAEhE,MAAM,UAAU,GACd,YAAY,CAAC,YAAY;gBACzB,wGAAwG,CAAC;YAE3G,IAAI,MAAM,GAAG,GAAG,UAAU,OAAO,kBAAkB,MAAM,CAAC;YAE1D,sDAAsD;YACtD,IAAI,OAAO,GAAG,CAAC,IAAI,iBAAiB,EAAE,CAAC;gBACrC,MAAM,IAAI,oEAAoE,CAAC;gBAC/E,IAAI,iBAAiB,CAAC,KAAK,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;oBAClF,MAAM,IAAI,iBAAiB,iBAAiB,CAAC,KAAK,+CAA+C,CAAC;gBACpG,CAAC;qBAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;oBACpC,MAAM,IAAI,qCAAqC,CAAC;gBAClD,CAAC;gBACD,MAAM,IAAI,mDAAmD,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;YAChG,CAAC;YAED,IAAI,YAAY,CAAC,YAAY,KAAK,MAAM,EAAE,CAAC;gBACzC,MAAM,IAAI;;YAEN,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC,mDAAmD,CAAC,CAAC,CAAC,EAAE;;0CAEvD,CAAC;YACrC,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,6BAA6B,CAAC;YAC1C,CAAC;YAED,MAAM,IAAI,6BAA6B,OAAO,EAAE,CAAC;YAEjD,kCAAkC;YAClC,IAAI,MAA4B,CAAC;YACjC,IAAI,YAAY,CAAC,YAAY,KAAK,MAAM,EAAE,CAAC;gBACzC,MAAM,MAAM,GAAG,IAAI,iCAAgB,EAAE,CAAC;gBACtC,MAAM,YAAY,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,wBAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAChF,MAAM,GAAG,YAAoC,CAAC;YAChD,CAAC;iBAAM,CAAC;gBACN,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,wBAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAChE,MAAM,SAAS,GAAG,OAAO,SAAS,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACxF,MAAM,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;YAChC,CAAC;YAED,sBAAsB;YACtB,IAAI,CAAC,4BAA4B,CAAC,MAAM,EAAE,YAAY,CAAC,EAAE,CAAC;gBACxD,iBAAiB,GAAG,MAAM,CAAC;gBAC3B,MAAM,IAAI,KAAK,CAAC,kCAAkC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC9E,CAAC;YAED,oCAAoC;YACpC,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,GAAG,GAAY,CAAC;YACzB,eAAM,CAAC,IAAI,CAAC;gBACV,OAAO,EAAE,0BAA0B,OAAO,SAAS;gBACnD,OAAO;gBACP,UAAU,EAAE,WAAW;gBACvB,KAAK,EAAE,SAAS,CAAC,OAAO;gBACxB,aAAa,EAAE,iBAAiB;aACjC,CAAC,CAAC;YAEH,qDAAqD;YACrD,IAAI,OAAO,GAAG,WAAW,EAAE,CAAC;gBAC1B,SAAS;YACX,CAAC;QACH,CAAC;IACH,CAAC;IAED,8CAA8C;IAC9C,eAAM,CAAC,IAAI,CAAC;QACV,OAAO,EAAE,8DAA8D;QACvE,UAAU,EAAE,WAAW;QACvB,SAAS,EAAE,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,OAAO;QAC7B,iBAAiB;QACjB,YAAY,EAAE,YAAY,CAAC,YAAY;QACvC,aAAa,EAAE,YAAY,CAAC,aAAa;KAC1C,CAAC,CAAC;IAEH,gCAAgC;IAChC,OAAO;QACL,KAAK,EAAE,YAAY,CAAC,YAAY,IAAI,SAAS;QAC7C,MAAM,EAAE,YAAY,CAAC,aAAa,IAAI,+BAA+B,WAAW,cAAc,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,OAAO,EAAE;QAClH,UAAU,EAAE,CAAC;KACd,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,SAAgB,gBAAgB,CAAC,OAAsC,EAAE,OAAmC;IAC1G,MAAM,gBAAgB,GAAsB,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAC5D,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;QACtC,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,MAAM,MAAM,GAAqB;QAC/B,GAAG,OAAO;QACV,OAAO,EAAE,gBAAgB;KAC1B,CAAC;IAEF,OAAO,CAAC,OAAe,EAAE,GAAsB,EAAE,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;AACrF,CAAC;AAED,iCAAiC;AACpB,QAAA,MAAM,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7B,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;IACvD,OAAO,EAAE,OAAC;SACP,KAAK,CACJ,OAAC,CAAC,KAAK,CAAC;QACN,OAAC,CAAC,MAAM,EAAE;QACV,OAAC,CAAC,KAAK,CAAC,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QACjC,OAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;YAChB,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE;SACxB,CAAC;KACH,CAAC,CACH;SACA,QAAQ,EAAE;SACV,QAAQ,CAAC,4GAA4G,CAAC;IACzH,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;IACvF,aAAa,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,8CAA8C,CAAC;IAC5G,YAAY,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,sCAAsC,CAAC;IAClH,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8CAA8C,CAAC;IAC5F,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,+CAA+C,CAAC;CAC/F,CAAC,CAAC;AAEH,MAAM,cAAc,GAA6B;IAC/C,IAAI,EAAE,mBAAmB;IACzB,WAAW,EACT,4KAA4K;IAC9K,KAAK,EAAE,cAAM;IACb,QAAQ,EAAE,KAAK;IACf,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE;;QACzC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,YAAY,EAAE,YAAY,EAAE,aAAa,EAAE,GAAG,KAAK,CAAC;QAE3G,eAAM,CAAC,IAAI,CAAC;YACV,OAAO,EAAE,qBAAqB;YAC9B,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,aAAa,EAAE,OAAO,CAAC,MAAM;YAC7B,UAAU,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,MAAM;SAC5B,CAAC,CAAC;QAEH,kDAAkD;QAClD,IAAI,gBAAgB,GAAsB,EAAE,CAAC;QAC7C,IAAI,OAAO,EAAE,CAAC;YACZ,gBAAgB,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACnC,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE,CAAC;oBAC1B,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;gBACtC,CAAC;qBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC5B,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3C,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAoB,CAAC;gBAC9B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,uDAAuD;QACvD,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,KAAI,MAAA,MAAA,KAAK,CAAC,MAAM,0CAAE,gBAAgB,0CAAE,OAAO,CAAA,EAAE,CAAC;YAC7E,gBAAgB,GAAG,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC;QAC3D,CAAC;QAED,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,MAAM,GAAqB;YAC/B,OAAO,EAAE,gBAAgB;YACzB,YAAY;YACZ,aAAa;YACb,YAAY;YACZ,YAAY;YACZ,aAAa;SACd,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;YAE1D,eAAM,CAAC,IAAI,CAAC;gBACV,OAAO,EAAE,0BAA0B;gBACnC,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,UAAU,EAAE,MAAM,CAAC,UAAU;aAC9B,CAAC,CAAC;YAEH,KAAK,CAAC,MAAM,CAAC,kBAAkB,GAAG;gBAChC,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gBACxE,MAAM;gBACN,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC;YAEF,OAAO;gBACL,MAAM;aACP,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,eAAM,CAAC,KAAK,CAAC;gBACX,OAAO,EAAE,uBAAuB;gBAChC,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,GAAG;aACJ,CAAC,CAAC;YAEH,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;CACF,CAAC;AAEF,kBAAe,cAAc,CAAC;AAE9B,gDAAgD;AACzC,KAAK,UAAU,aAAa,CACjC,OAAe,EACf,MAAkD,EAClD,GAAsB;IAEtB,MAAM,YAAY,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;IACtD,MAAM,UAAU,GAAG,YAAY,CAAC,UAAU,IAAI,CAAC,CAAC;IAEhD,IAAI,CAAC;QACH,MAAM,kBAAkB,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAErG,MAAM,UAAU,GACd,YAAY,CAAC,YAAY,IAAI,2FAA2F,CAAC;QAE3H,MAAM,MAAM,GAAG,GAAG,UAAU,OAAO,kBAAkB;wDACD,UAAU;;;;MAI5D,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC,4BAA4B,CAAC,CAAC,CAAC,EAAE;;;;;;;;EAQlE,OAAO,EAAE,CAAC;QAER,MAAM,MAAM,GAAG,IAAI,iCAAgB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,wBAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAE1E,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,OAAO,MAAgC,CAAC;QAC1C,CAAC;QAED,2CAA2C;QAC3C,OAAO,CAAC,MAA8B,CAAC,CAAC;IAC1C,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,eAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,6BAA6B,EAAE,GAAG,EAAE,CAAC,CAAC;QAE9D,OAAO;YACL;gBACE,KAAK,EAAE,YAAY,CAAC,YAAY,IAAI,SAAS;gBAC7C,MAAM,EAAE,YAAY,CAAC,aAAa,IAAK,GAAa,CAAC,OAAO;gBAC5D,UAAU,EAAE,CAAC;aACd;SACF,CAAC;IACJ,CAAC;AACH,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"LLM.types.js","sourceRoot":"","sources":["../../src/types/LLM.types.ts"],"names":[],"mappings":";;;AAAA,8CAAgE;AAChE,0EAAuE;AAS1D,QAAA,YAAY,GAA6B;IACpD,UAAU,
|
|
1
|
+
{"version":3,"file":"LLM.types.js","sourceRoot":"","sources":["../../src/types/LLM.types.ts"],"names":[],"mappings":";;;AAAA,8CAAgE;AAChE,0EAAuE;AAS1D,QAAA,YAAY,GAA6B;IACpD,UAAU,EAAE,mBAAU;IACtB,eAAe,EAAE,wBAAe;IAChC,gBAAgB,EAAhB,mCAAgB;CACjB,CAAC"}
|
|
@@ -1 +1,5 @@
|
|
|
1
|
+
import { ToolMessage } from '@langchain/core/messages';
|
|
2
|
+
import { State } from '../types/LangGraph.types';
|
|
3
|
+
declare const extractToolStateResponse: (toolMessage: ToolMessage) => Partial<State>;
|
|
4
|
+
export default extractToolStateResponse;
|
|
1
5
|
//# sourceMappingURL=extractStateMemoryResponse.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"extractStateMemoryResponse.d.ts","sourceRoot":"","sources":["../../src/utils/extractStateMemoryResponse.ts"],"names":[],"mappings":""}
|
|
1
|
+
{"version":3,"file":"extractStateMemoryResponse.d.ts","sourceRoot":"","sources":["../../src/utils/extractStateMemoryResponse.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,WAAW,EAA0C,MAAM,0BAA0B,CAAC;AAE5G,OAAO,EAAE,KAAK,EAAE,MAAM,0BAA0B,CAAC;AAEjD,QAAA,MAAM,wBAAwB,gBAAiB,WAAW,KAAG,OAAO,CAAC,KAAK,CAazE,CAAC;AA4EF,eAAe,wBAAwB,CAAC"}
|
|
@@ -1,89 +1,91 @@
|
|
|
1
1
|
"use strict";
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
//
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
//
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
//
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
//
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
const messages_1 = require("@langchain/core/messages");
|
|
4
|
+
const logger_1 = require("./logger");
|
|
5
|
+
const extractToolStateResponse = (toolMessage) => {
|
|
6
|
+
try {
|
|
7
|
+
const parsed = JSON.parse(toolMessage.content);
|
|
8
|
+
if (typeof parsed === 'object' && parsed !== null && 'state' in parsed) {
|
|
9
|
+
if (parsed.state.messages) {
|
|
10
|
+
parsed.state.messages = constructMessagesFromToolMessage(parsed.state.messages);
|
|
11
|
+
}
|
|
12
|
+
return parsed.state;
|
|
13
|
+
}
|
|
14
|
+
}
|
|
15
|
+
catch (err) {
|
|
16
|
+
logger_1.logger.error({ message: 'Error parsing tool state response', err });
|
|
17
|
+
}
|
|
18
|
+
return {};
|
|
19
|
+
};
|
|
20
|
+
const constructMessagesFromToolMessage = (messages) => {
|
|
21
|
+
if (!Array.isArray(messages)) {
|
|
22
|
+
logger_1.logger.error({ msg: 'Expected messages to be an array', messages });
|
|
23
|
+
return [];
|
|
24
|
+
}
|
|
25
|
+
return messages
|
|
26
|
+
.map((messageObj) => {
|
|
27
|
+
try {
|
|
28
|
+
// Handle case where message is already a BaseMessage instance
|
|
29
|
+
if (messageObj instanceof messages_1.BaseMessage) {
|
|
30
|
+
return messageObj;
|
|
31
|
+
}
|
|
32
|
+
// Handle case where message is a plain object that needs reconstruction
|
|
33
|
+
const messageType = messageObj.id[2];
|
|
34
|
+
const content = messageObj.kwargs.content || '';
|
|
35
|
+
const id = messageObj.kwargs.id;
|
|
36
|
+
const additionalKwargs = messageObj.additional_kwargs || {};
|
|
37
|
+
switch (messageType) {
|
|
38
|
+
case 'AIMessage':
|
|
39
|
+
return new messages_1.AIMessage({
|
|
40
|
+
content,
|
|
41
|
+
id,
|
|
42
|
+
additional_kwargs: additionalKwargs,
|
|
43
|
+
tool_calls: messageObj.tool_calls || [],
|
|
44
|
+
invalid_tool_calls: messageObj.invalid_tool_calls || [],
|
|
45
|
+
});
|
|
46
|
+
case 'HumanMessage':
|
|
47
|
+
return new messages_1.HumanMessage({
|
|
48
|
+
content,
|
|
49
|
+
id,
|
|
50
|
+
additional_kwargs: additionalKwargs,
|
|
51
|
+
});
|
|
52
|
+
case 'SystemMessage':
|
|
53
|
+
return new messages_1.SystemMessage({
|
|
54
|
+
content,
|
|
55
|
+
id,
|
|
56
|
+
additional_kwargs: additionalKwargs,
|
|
57
|
+
});
|
|
58
|
+
case 'ToolMessage':
|
|
59
|
+
return new messages_1.ToolMessage({
|
|
60
|
+
content,
|
|
61
|
+
id,
|
|
62
|
+
additional_kwargs: additionalKwargs,
|
|
63
|
+
tool_call_id: messageObj.tool_call_id || '',
|
|
64
|
+
});
|
|
65
|
+
default:
|
|
66
|
+
// Default to HumanMessage for unknown types
|
|
67
|
+
logger_1.logger.warn({
|
|
68
|
+
msg: 'Unknown message type, defaulting to HumanMessage',
|
|
69
|
+
messageType,
|
|
70
|
+
messageObj,
|
|
71
|
+
});
|
|
72
|
+
return null;
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
catch (err) {
|
|
76
|
+
logger_1.logger.error({
|
|
77
|
+
msg: 'Error reconstructing message',
|
|
78
|
+
err,
|
|
79
|
+
messageObj,
|
|
80
|
+
});
|
|
81
|
+
// Return a fallback HumanMessage in case of error
|
|
82
|
+
return new messages_1.HumanMessage({
|
|
83
|
+
content: messageObj.content || 'Error reconstructing message',
|
|
84
|
+
id: messageObj.id,
|
|
85
|
+
});
|
|
86
|
+
}
|
|
87
|
+
})
|
|
88
|
+
.filter(Boolean);
|
|
89
|
+
};
|
|
90
|
+
exports.default = extractToolStateResponse;
|
|
89
91
|
//# sourceMappingURL=extractStateMemoryResponse.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"extractStateMemoryResponse.js","sourceRoot":"","sources":["../../src/utils/extractStateMemoryResponse.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"extractStateMemoryResponse.js","sourceRoot":"","sources":["../../src/utils/extractStateMemoryResponse.ts"],"names":[],"mappings":";;AAAA,uDAA4G;AAC5G,qCAAkC;AAGlC,MAAM,wBAAwB,GAAG,CAAC,WAAwB,EAAkB,EAAE;IAC5E,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAiB,CAAC,CAAC;QACzD,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;YACvE,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;gBAC1B,MAAM,CAAC,KAAK,CAAC,QAAQ,GAAG,gCAAgC,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAClF,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,eAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC;IACtE,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC,CAAC;AAEF,MAAM,gCAAgC,GAAG,CAAC,QAAe,EAAiB,EAAE;IAC1E,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC7B,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,kCAAkC,EAAE,QAAQ,EAAE,CAAC,CAAC;QACpE,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,QAAQ;SACZ,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;QAClB,IAAI,CAAC;YACH,8DAA8D;YAC9D,IAAI,UAAU,YAAY,sBAAW,EAAE,CAAC;gBACtC,OAAO,UAAU,CAAC;YACpB,CAAC;YAED,wEAAwE;YACxE,MAAM,WAAW,GAAG,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;YAChD,MAAM,EAAE,GAAG,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAChC,MAAM,gBAAgB,GAAG,UAAU,CAAC,iBAAiB,IAAI,EAAE,CAAC;YAE5D,QAAQ,WAAW,EAAE,CAAC;gBACpB,KAAK,WAAW;oBACd,OAAO,IAAI,oBAAS,CAAC;wBACnB,OAAO;wBACP,EAAE;wBACF,iBAAiB,EAAE,gBAAgB;wBACnC,UAAU,EAAE,UAAU,CAAC,UAAU,IAAI,EAAE;wBACvC,kBAAkB,EAAE,UAAU,CAAC,kBAAkB,IAAI,EAAE;qBACxD,CAAC,CAAC;gBACL,KAAK,cAAc;oBACjB,OAAO,IAAI,uBAAY,CAAC;wBACtB,OAAO;wBACP,EAAE;wBACF,iBAAiB,EAAE,gBAAgB;qBACpC,CAAC,CAAC;gBACL,KAAK,eAAe;oBAClB,OAAO,IAAI,wBAAa,CAAC;wBACvB,OAAO;wBACP,EAAE;wBACF,iBAAiB,EAAE,gBAAgB;qBACpC,CAAC,CAAC;gBACL,KAAK,aAAa;oBAChB,OAAO,IAAI,sBAAW,CAAC;wBACrB,OAAO;wBACP,EAAE;wBACF,iBAAiB,EAAE,gBAAgB;wBACnC,YAAY,EAAE,UAAU,CAAC,YAAY,IAAI,EAAE;qBAC5C,CAAC,CAAC;gBAEL;oBACE,4CAA4C;oBAC5C,eAAM,CAAC,IAAI,CAAC;wBACV,GAAG,EAAE,kDAAkD;wBACvD,WAAW;wBACX,UAAU;qBACX,CAAC,CAAC;oBACH,OAAO,IAAI,CAAC;YAChB,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,eAAM,CAAC,KAAK,CAAC;gBACX,GAAG,EAAE,8BAA8B;gBACnC,GAAG;gBACH,UAAU;aACX,CAAC,CAAC;YACH,kDAAkD;YAClD,OAAO,IAAI,uBAAY,CAAC;gBACtB,OAAO,EAAE,UAAU,CAAC,OAAO,IAAI,8BAA8B;gBAC7D,EAAE,EAAE,UAAU,CAAC,EAAE;aAClB,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC;SACD,MAAM,CAAC,OAAO,CAAkB,CAAC;AACtC,CAAC,CAAC;AAEF,kBAAe,wBAAwB,CAAC"}
|
package/package.json
CHANGED
package/src/agent.ts
CHANGED
|
@@ -178,12 +178,21 @@ export class Agent {
|
|
|
178
178
|
|
|
179
179
|
mindedConnection.on(mindedConnectionSocketMessageType.INVOKE, async (message) => {
|
|
180
180
|
const invokeMessage = message as InvokeMessage;
|
|
181
|
-
|
|
182
|
-
|
|
183
|
-
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
|
|
181
|
+
try {
|
|
182
|
+
const result = await this.invoke({
|
|
183
|
+
triggerBody: invokeMessage.triggerBody,
|
|
184
|
+
triggerName: invokeMessage.triggerName,
|
|
185
|
+
appName: invokeMessage.appName,
|
|
186
|
+
});
|
|
187
|
+
return result;
|
|
188
|
+
} catch (err: any) {
|
|
189
|
+
logger.error({ message: '[Agent] Error invoking trigger', err });
|
|
190
|
+
return {
|
|
191
|
+
errorMessage: err.message,
|
|
192
|
+
error: 'Error invoking the agent',
|
|
193
|
+
errorStack: err.stack,
|
|
194
|
+
};
|
|
195
|
+
}
|
|
187
196
|
});
|
|
188
197
|
|
|
189
198
|
mindedConnection.on(mindedConnectionSocketMessageType.TIMER_TRIGGER, async (timerTriggerMessage) => {
|
|
@@ -13,26 +13,17 @@ import {
|
|
|
13
13
|
} from '../platform/mindedConnectionTypes';
|
|
14
14
|
import { kill, getOrStartLocalCDP } from './localBrowserTask';
|
|
15
15
|
import { BrowserTaskMode, InvokeBrowserTaskOptions } from './types';
|
|
16
|
-
import { isLocalOperatorSetup } from '../cli/localOperatorSetup';
|
|
17
16
|
|
|
18
17
|
// Socket-based browser task functions
|
|
19
|
-
export const createBrowserSession = async ({
|
|
20
|
-
sessionId,
|
|
21
|
-
proxy,
|
|
22
|
-
browserTaskMode,
|
|
23
|
-
}: {
|
|
24
|
-
sessionId: string;
|
|
25
|
-
proxy?: string;
|
|
26
|
-
browserTaskMode: BrowserTaskMode;
|
|
27
|
-
}): Promise<CreateBrowserSessionResponse> => {
|
|
18
|
+
export const createBrowserSession = async ({ sessionId, proxy, browserTaskMode }: { sessionId: string, proxy?: string, browserTaskMode: BrowserTaskMode }): Promise<CreateBrowserSessionResponse> => {
|
|
28
19
|
logger.debug({ msg: 'Creating browser session via socket', proxy });
|
|
29
20
|
|
|
30
21
|
if (browserTaskMode === BrowserTaskMode.LOCAL) {
|
|
31
22
|
const { cdpUrl } = await getOrStartLocalCDP({ headless: false });
|
|
32
23
|
return {
|
|
33
24
|
sessionId,
|
|
34
|
-
cdpUrl
|
|
35
|
-
}
|
|
25
|
+
cdpUrl
|
|
26
|
+
}
|
|
36
27
|
}
|
|
37
28
|
|
|
38
29
|
const response = await mindedConnection.awaitEmit<CreateBrowserSessionRequest, CreateBrowserSessionResponse>(
|
|
@@ -59,15 +50,7 @@ export const createBrowserSession = async ({
|
|
|
59
50
|
return response;
|
|
60
51
|
};
|
|
61
52
|
|
|
62
|
-
export const destroyBrowserSession = async ({
|
|
63
|
-
sessionId,
|
|
64
|
-
onPrem,
|
|
65
|
-
localRun,
|
|
66
|
-
}: {
|
|
67
|
-
sessionId: string;
|
|
68
|
-
onPrem?: boolean;
|
|
69
|
-
localRun?: boolean;
|
|
70
|
-
}): Promise<DestroyBrowserSessionResponse> => {
|
|
53
|
+
export const destroyBrowserSession = async ({ sessionId, onPrem, localRun }: { sessionId: string, onPrem?: boolean, localRun?: boolean }): Promise<DestroyBrowserSessionResponse> => {
|
|
71
54
|
logger.debug({ msg: 'Destroying browser session via socket', sessionId, onPrem });
|
|
72
55
|
|
|
73
56
|
if (localRun) {
|
|
@@ -96,38 +79,30 @@ export const destroyBrowserSession = async ({
|
|
|
96
79
|
return response;
|
|
97
80
|
};
|
|
98
81
|
|
|
99
|
-
export const invokeBrowserTask = async (
|
|
82
|
+
export const invokeBrowserTask = async (
|
|
83
|
+
options: InvokeBrowserTaskOptions,
|
|
84
|
+
): Promise<InvokeBrowserTaskResponse> => {
|
|
85
|
+
|
|
100
86
|
try {
|
|
101
|
-
const {
|
|
87
|
+
const {
|
|
88
|
+
sessionId,
|
|
89
|
+
cdpUrl,
|
|
90
|
+
task,
|
|
91
|
+
keepAlive,
|
|
92
|
+
hooks,
|
|
93
|
+
browserTaskMode,
|
|
94
|
+
toolSchemas,
|
|
95
|
+
outputSchema,
|
|
96
|
+
} = options;
|
|
102
97
|
|
|
103
98
|
if (browserTaskMode === BrowserTaskMode.LOCAL) {
|
|
104
99
|
const pythonScriptPath = path.resolve(__dirname, 'executeBrowserTask.py');
|
|
105
100
|
|
|
106
|
-
|
|
107
|
-
let args: string[];
|
|
108
|
-
|
|
109
|
-
if (isLocalOperatorSetup()) {
|
|
110
|
-
// Use the Python interpreter directly from the virtual environment
|
|
111
|
-
command = path.join(process.cwd(), '.venv', 'bin', 'python');
|
|
112
|
-
args = [pythonScriptPath];
|
|
113
|
-
} else {
|
|
114
|
-
// Fallback to uv run for environments without local setup
|
|
115
|
-
command = 'uv';
|
|
116
|
-
args = ['run', pythonScriptPath];
|
|
117
|
-
}
|
|
101
|
+
const args = ['run', pythonScriptPath];
|
|
118
102
|
|
|
119
|
-
logger.info({ message: 'Spawning
|
|
120
|
-
|
|
121
|
-
const child = spawn(
|
|
122
|
-
stdio: ['pipe', 'pipe', 'pipe'],
|
|
123
|
-
env: isLocalOperatorSetup()
|
|
124
|
-
? {
|
|
125
|
-
...process.env,
|
|
126
|
-
VIRTUAL_ENV: path.join(process.cwd(), '.venv'),
|
|
127
|
-
PATH: `${path.join(process.cwd(), '.venv', 'bin')}:${process.env.PATH}`,
|
|
128
|
-
}
|
|
129
|
-
: process.env,
|
|
130
|
-
});
|
|
103
|
+
logger.info({ message: 'Spawning Python process', args });
|
|
104
|
+
|
|
105
|
+
const child = spawn('uv', args, { stdio: ['pipe', 'pipe', 'pipe'] });
|
|
131
106
|
|
|
132
107
|
// Build payload to send over stdin
|
|
133
108
|
const payload = {
|