@tyvm/knowhow 0.0.20 → 0.0.22
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/package.json +3 -1
- package/src/agents/base/base.ts +16 -77
- package/src/agents/tools/executeScript/README.md +78 -0
- package/src/agents/tools/executeScript/definition.ts +73 -0
- package/src/agents/tools/executeScript/examples/quick-test.ts +80 -0
- package/src/agents/tools/executeScript/examples/serialization-test.ts +309 -0
- package/src/agents/tools/executeScript/examples/test-runner.ts +204 -0
- package/src/agents/tools/executeScript/index.ts +74 -0
- package/src/agents/tools/index.ts +1 -0
- package/src/agents/tools/list.ts +2 -1
- package/src/cli.ts +2 -6
- package/src/clients/index.ts +23 -9
- package/src/services/Tools.ts +150 -9
- package/src/services/script-execution/SandboxContext.ts +278 -0
- package/src/services/script-execution/ScriptExecutor.ts +337 -0
- package/src/services/script-execution/ScriptPolicy.ts +236 -0
- package/src/services/script-execution/ScriptTracer.ts +249 -0
- package/src/services/script-execution/types.ts +134 -0
- package/ts_build/src/agents/base/base.js +2 -53
- package/ts_build/src/agents/base/base.js.map +1 -1
- package/ts_build/src/agents/tools/executeScript/definition.d.ts +2 -0
- package/ts_build/src/agents/tools/executeScript/definition.js +70 -0
- package/ts_build/src/agents/tools/executeScript/definition.js.map +1 -0
- package/ts_build/src/agents/tools/executeScript/examples/quick-test.d.ts +3 -0
- package/ts_build/src/agents/tools/executeScript/examples/quick-test.js +68 -0
- package/ts_build/src/agents/tools/executeScript/examples/quick-test.js.map +1 -0
- package/ts_build/src/agents/tools/executeScript/examples/serialization-test.d.ts +15 -0
- package/ts_build/src/agents/tools/executeScript/examples/serialization-test.js +267 -0
- package/ts_build/src/agents/tools/executeScript/examples/serialization-test.js.map +1 -0
- package/ts_build/src/agents/tools/executeScript/examples/simple-example.d.ts +20 -0
- package/ts_build/src/agents/tools/executeScript/examples/simple-example.js +35 -0
- package/ts_build/src/agents/tools/executeScript/examples/simple-example.js.map +1 -0
- package/ts_build/src/agents/tools/executeScript/examples/test-runner.d.ts +4 -0
- package/ts_build/src/agents/tools/executeScript/examples/test-runner.js +202 -0
- package/ts_build/src/agents/tools/executeScript/examples/test-runner.js.map +1 -0
- package/ts_build/src/agents/tools/executeScript/handler.d.ts +27 -0
- package/ts_build/src/agents/tools/executeScript/handler.js +64 -0
- package/ts_build/src/agents/tools/executeScript/handler.js.map +1 -0
- package/ts_build/src/agents/tools/executeScript/index.d.ts +27 -0
- package/ts_build/src/agents/tools/executeScript/index.js +64 -0
- package/ts_build/src/agents/tools/executeScript/index.js.map +1 -0
- package/ts_build/src/agents/tools/executeScript.d.ts +29 -0
- package/ts_build/src/agents/tools/executeScript.js +124 -0
- package/ts_build/src/agents/tools/executeScript.js.map +1 -0
- package/ts_build/src/agents/tools/index.d.ts +1 -0
- package/ts_build/src/agents/tools/index.js +1 -0
- package/ts_build/src/agents/tools/index.js.map +1 -1
- package/ts_build/src/agents/tools/list.js +2 -0
- package/ts_build/src/agents/tools/list.js.map +1 -1
- package/ts_build/src/cli.js +2 -6
- package/ts_build/src/cli.js.map +1 -1
- package/ts_build/src/clients/index.d.ts +9 -2
- package/ts_build/src/clients/index.js +17 -4
- package/ts_build/src/clients/index.js.map +1 -1
- package/ts_build/src/services/Tools.d.ts +11 -1
- package/ts_build/src/services/Tools.js +94 -3
- package/ts_build/src/services/Tools.js.map +1 -1
- package/ts_build/src/services/script-execution/SandboxContext.d.ts +34 -0
- package/ts_build/src/services/script-execution/SandboxContext.js +188 -0
- package/ts_build/src/services/script-execution/SandboxContext.js.map +1 -0
- package/ts_build/src/services/script-execution/ScriptExecutor.d.ts +17 -0
- package/ts_build/src/services/script-execution/ScriptExecutor.js +207 -0
- package/ts_build/src/services/script-execution/ScriptExecutor.js.map +1 -0
- package/ts_build/src/services/script-execution/ScriptPolicy.d.ts +27 -0
- package/ts_build/src/services/script-execution/ScriptPolicy.js +150 -0
- package/ts_build/src/services/script-execution/ScriptPolicy.js.map +1 -0
- package/ts_build/src/services/script-execution/ScriptTracer.d.ts +19 -0
- package/ts_build/src/services/script-execution/ScriptTracer.js +186 -0
- package/ts_build/src/services/script-execution/ScriptTracer.js.map +1 -0
- package/ts_build/src/services/script-execution/types.d.ts +108 -0
- package/ts_build/src/services/script-execution/types.js +3 -0
- package/ts_build/src/services/script-execution/types.js.map +1 -0
package/ts_build/src/cli.js
CHANGED
|
@@ -34,7 +34,7 @@ const patcher_1 = require("./agents/patcher/patcher");
|
|
|
34
34
|
const developer_1 = require("./agents/developer/developer");
|
|
35
35
|
const services_1 = require("./services");
|
|
36
36
|
const list_1 = require("./agents/tools/list");
|
|
37
|
-
const allTools = __importStar(require("./agents/tools
|
|
37
|
+
const allTools = __importStar(require("./agents/tools"));
|
|
38
38
|
const Mcp_1 = require("./services/Mcp");
|
|
39
39
|
const login_1 = require("./login");
|
|
40
40
|
const worker_1 = require("./worker");
|
|
@@ -45,11 +45,7 @@ async function main() {
|
|
|
45
45
|
AgentService_1.Agents.registerAgent(patcher_1.Patcher);
|
|
46
46
|
AgentService_1.Agents.registerAgent(developer_1.Developer);
|
|
47
47
|
AgentService_1.Agents.loadAgentsFromConfig();
|
|
48
|
-
services_1.Tools.
|
|
49
|
-
const toolFunctions = Object.entries(allTools)
|
|
50
|
-
.filter(([_, value]) => typeof value === 'function')
|
|
51
|
-
.reduce((acc, [key, value]) => ({ ...acc, [key]: value }), {});
|
|
52
|
-
services_1.Tools.addFunctions(toolFunctions);
|
|
48
|
+
services_1.Tools.defineTools(list_1.includedTools, allTools);
|
|
53
49
|
await Mcp_1.Mcp.connectToConfigured(services_1.Tools);
|
|
54
50
|
await clients_1.Clients.registerConfiguredModels();
|
|
55
51
|
switch (command) {
|
package/ts_build/src/cli.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"cli.js","sourceRoot":"","sources":["../../src/cli.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AACA,uCAAqC;AACrC,mCAAwD;AACxD,qCAAgC;AAEhC,wBAAoC;AACpC,0DAAiD;AACjD,+DAA4D;AAC5D,sDAAmD;AAEnD,4DAAyD;AACzD,yCAAmC;AACnC,8CAAoD;AACpD
|
|
1
|
+
{"version":3,"file":"cli.js","sourceRoot":"","sources":["../../src/cli.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AACA,uCAAqC;AACrC,mCAAwD;AACxD,qCAAgC;AAEhC,wBAAoC;AACpC,0DAAiD;AACjD,+DAA4D;AAC5D,sDAAmD;AAEnD,4DAAyD;AACzD,yCAAmC;AACnC,8CAAoD;AACpD,yDAA2C;AAC3C,wCAAqC;AACrC,mCAAgC;AAChC,qCAAkC;AAClC,uCAAoC;AAEpC,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAEhC,KAAK,UAAU,IAAI;IACjB,qBAAM,CAAC,aAAa,CAAC,uBAAU,CAAC,CAAC;IACjC,qBAAM,CAAC,aAAa,CAAC,iBAAO,CAAC,CAAC;IAC9B,qBAAM,CAAC,aAAa,CAAC,qBAAS,CAAC,CAAC;IAChC,qBAAM,CAAC,oBAAoB,EAAE,CAAC;IAE9B,gBAAK,CAAC,WAAW,CAAC,oBAAa,EAAE,QAAQ,CAAC,CAAC;IAE3C,MAAM,SAAG,CAAC,mBAAmB,CAAC,gBAAK,CAAC,CAAC;IACrC,MAAM,iBAAO,CAAC,wBAAwB,EAAE,CAAC;IAKzC,QAAQ,OAAO,EAAE;QACf,KAAK,MAAM;YACT,MAAM,IAAA,aAAI,GAAE,CAAC;YACb,MAAM;QACR,KAAK,OAAO;YACV,MAAM,IAAA,aAAK,GAAE,CAAC;YACd,MAAM;QACR,KAAK,UAAU;YACb,MAAM,IAAA,gBAAQ,GAAE,CAAC;YACjB,MAAM;QACR,KAAK,OAAO;YACV,MAAM,IAAA,aAAK,GAAE,CAAC;YACd,MAAM;QACR,KAAK,aAAa;YAChB,MAAM,IAAA,QAAK,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM;QACR,KAAK,QAAQ;YACX,MAAM,IAAA,cAAM,GAAE,CAAC;YACf,MAAM;QACR,KAAK,UAAU;YACb,MAAM,IAAA,WAAQ,GAAE,CAAC;YACjB,MAAM;QACR,KAAK,MAAM;YACT,MAAM,IAAA,YAAI,GAAE,CAAC;YACb,MAAM;QACR,KAAK,QAAQ;YACX,MAAM,IAAA,eAAM,GAAE,CAAC;YACf,MAAM;QACR;YACE,OAAO,CAAC,GAAG,CACT,qHAAqH,CACtH,CAAC;YACF,MAAM;KACT;AACH,CAAC;AAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE;IAC3B,IAAI,EAAE;SACH,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;QACX,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACjB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC;SACD,IAAI,CAAC,GAAG,EAAE;QACT,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;CACN"}
|
|
@@ -35,8 +35,15 @@ export declare class AIClient {
|
|
|
35
35
|
registerConfiguredModels(): Promise<void>;
|
|
36
36
|
registerModels(provider: string, models: string[]): void;
|
|
37
37
|
registerEmbeddingModels(provider: string, models: string[]): void;
|
|
38
|
-
|
|
39
|
-
|
|
38
|
+
providerHasModel(provider: string, model: string): boolean;
|
|
39
|
+
findModel(modelPrefix: string): {
|
|
40
|
+
provider: string;
|
|
41
|
+
model: any;
|
|
42
|
+
};
|
|
43
|
+
detectProviderModel(provider: string, model?: string): {
|
|
44
|
+
provider: string;
|
|
45
|
+
model: any;
|
|
46
|
+
};
|
|
40
47
|
createCompletion(provider: string, options: CompletionOptions): Promise<CompletionResponse>;
|
|
41
48
|
createEmbedding(provider: string, options: EmbeddingOptions): Promise<EmbeddingResponse>;
|
|
42
49
|
getRegisteredModels(provider: string): string[];
|
|
@@ -126,6 +126,16 @@ class AIClient {
|
|
|
126
126
|
return false;
|
|
127
127
|
return models.includes(model);
|
|
128
128
|
}
|
|
129
|
+
findModel(modelPrefix) {
|
|
130
|
+
for (const provider of Object.keys(this.clientModels)) {
|
|
131
|
+
const models = this.clientModels[provider];
|
|
132
|
+
const foundModel = models.find((m) => m.startsWith(modelPrefix));
|
|
133
|
+
if (foundModel) {
|
|
134
|
+
return { provider, model: foundModel };
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
return undefined;
|
|
138
|
+
}
|
|
129
139
|
detectProviderModel(provider, model) {
|
|
130
140
|
if (this.providerHasModel(provider, model)) {
|
|
131
141
|
return { provider, model };
|
|
@@ -137,11 +147,14 @@ class AIClient {
|
|
|
137
147
|
if (this.providerHasModel(inferredProvider, inferredModel)) {
|
|
138
148
|
return { provider: inferredProvider, model: inferredModel };
|
|
139
149
|
}
|
|
150
|
+
const foundBySplit = this.findModel(inferredModel);
|
|
151
|
+
if (foundBySplit) {
|
|
152
|
+
return foundBySplit;
|
|
153
|
+
}
|
|
140
154
|
}
|
|
141
|
-
const
|
|
142
|
-
|
|
143
|
-
|
|
144
|
-
return { provider: foundProvider, model };
|
|
155
|
+
const foundByModel = this.findModel(model);
|
|
156
|
+
if (foundByModel) {
|
|
157
|
+
return foundByModel;
|
|
145
158
|
}
|
|
146
159
|
return { provider, model };
|
|
147
160
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/clients/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAOA,qCAA+C;AAC/C,2CAAqD;AACrD,qCAA+C;AAC/C,iCAAoC;AACpC,oCAAmD;AACnD,sCAAsC;AACtC,+BAAyC;AAEzC,SAAS,QAAQ,CAAC,GAAW;IAC3B,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC/B,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,KAAK,CAAC;KACd;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAa,QAAQ;IACnB,OAAO,GAAG;QACR,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,4BAAmB,EAAE,EAAE,CAAC;QAEpE,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI;YACnC,SAAS,EAAE,IAAI,kCAAsB,EAAE;SACxC,CAAC;QAEF,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,4BAAmB,EAAE,EAAE,CAAC;QACxE,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,sBAAgB,EAAE,EAAE,CAAC;KAChE,CAAC;IAEF,gBAAgB,GAAG;QACjB,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI;YAC5B,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,cAAM,CAAC,MAAM,CAAC;SACrC,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI;YACnC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,cAAM,CAAC,SAAS,CAAC;SAC3C,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI;YAChC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,cAAM,CAAC,MAAM,CAAC;SACrC,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,MAAM,CAAC,cAAM,CAAC,GAAG,CAAC,EAAE,CAAC;KACnE,CAAC;IAEF,eAAe,GAAG;QAChB,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI;YAC5B,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,uBAAe,CAAC,MAAM,CAAC;SAC9C,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI;YAChC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,uBAAe,CAAC,MAAM,CAAC;SAC9C,CAAC;KACH,CAAC;IAEF,YAAY,GAAG;QACb,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI;YAC5B,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC;SAC1E,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI;YACnC,SAAS,EAAE,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC;SAChD,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI;YAChC,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC;SAC1E,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC;KACnE,CAAC;IAEF,SAAS,CAAC,QAAgB,EAAE,KAAc;QACxC,IAAI,QAAQ,IAAI,CAAC,KAAK,EAAE;YACtB,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;SACvE;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAE3D,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC;QAC7B,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QAEvB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;YAC3B,MAAM,IAAI,KAAK,CACb,YAAY,QAAQ,cAAc,KAAK,yCAAyC,MAAM,CAAC,IAAI,CACzF,IAAI,CAAC,OAAO,CACb,EAAE,CACJ,CAAC;SACH;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAExD,IAAI,CAAC,QAAQ,EAAE;YACb,MAAM,IAAI,KAAK,CACb,SAAS,KAAK,gCAAgC,QAAQ,GAAG,CAC1D,CAAC;SACH;QAED,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;IAC7D,CAAC;IAED,cAAc,CAAC,QAAgB,EAAE,MAAqB;QACpD,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,wBAAwB;QAC5B,MAAM,MAAM,GAAG,MAAM,IAAA,kBAAS,GAAE,CAAC;QACjC,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC;QAEnD,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE;YAC1C,MAAM,MAAM,GAAG,IAAI,iBAAU,CAAC,aAAa,CAAC,GAAG,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC;YAExE,IAAI,aAAa,CAAC,OAAO,EAAE;gBACzB,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;aAC3C;YAED,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAEpD,IAAI;gBACF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;gBACxC,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC5C,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;aAClD;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CACX,0CAA0C,aAAa,CAAC,QAAQ,GAAG,EACnE,KAAK,CAAC,OAAO,CACd,CAAC;aACH;SACF;IACH,CAAC;IAED,cAAc,CAAC,QAAgB,EAAE,MAAgB;QAC/C,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACxD,MAAM,uBAAuB,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACtE,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC,IAAI,CACtC,IAAI,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CACtC,CAAC;QAGF,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC,IAAI,CAC1C,IAAI,GAAG,CAAC,uBAAuB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAChD,CAAC;IACJ,CAAC;IAED,uBAAuB,CAAC,QAAgB,EAAE,MAAgB;QACxD,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACxD,MAAM,sBAAsB,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAEpE,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC,IAAI,CACtC,IAAI,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CACtC,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC,IAAI,CACzC,IAAI,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CACtC,CAAC;IACJ,CAAC;
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/clients/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAOA,qCAA+C;AAC/C,2CAAqD;AACrD,qCAA+C;AAC/C,iCAAoC;AACpC,oCAAmD;AACnD,sCAAsC;AACtC,+BAAyC;AAEzC,SAAS,QAAQ,CAAC,GAAW;IAC3B,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC/B,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,KAAK,CAAC;KACd;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAa,QAAQ;IACnB,OAAO,GAAG;QACR,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,4BAAmB,EAAE,EAAE,CAAC;QAEpE,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI;YACnC,SAAS,EAAE,IAAI,kCAAsB,EAAE;SACxC,CAAC;QAEF,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,4BAAmB,EAAE,EAAE,CAAC;QACxE,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,sBAAgB,EAAE,EAAE,CAAC;KAChE,CAAC;IAEF,gBAAgB,GAAG;QACjB,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI;YAC5B,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,cAAM,CAAC,MAAM,CAAC;SACrC,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI;YACnC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,cAAM,CAAC,SAAS,CAAC;SAC3C,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI;YAChC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,cAAM,CAAC,MAAM,CAAC;SACrC,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,MAAM,CAAC,cAAM,CAAC,GAAG,CAAC,EAAE,CAAC;KACnE,CAAC;IAEF,eAAe,GAAG;QAChB,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI;YAC5B,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,uBAAe,CAAC,MAAM,CAAC;SAC9C,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI;YAChC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,uBAAe,CAAC,MAAM,CAAC;SAC9C,CAAC;KACH,CAAC;IAEF,YAAY,GAAG;QACb,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI;YAC5B,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC;SAC1E,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI;YACnC,SAAS,EAAE,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC;SAChD,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI;YAChC,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC;SAC1E,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC;KACnE,CAAC;IAEF,SAAS,CAAC,QAAgB,EAAE,KAAc;QACxC,IAAI,QAAQ,IAAI,CAAC,KAAK,EAAE;YACtB,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;SACvE;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAE3D,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC;QAC7B,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QAEvB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;YAC3B,MAAM,IAAI,KAAK,CACb,YAAY,QAAQ,cAAc,KAAK,yCAAyC,MAAM,CAAC,IAAI,CACzF,IAAI,CAAC,OAAO,CACb,EAAE,CACJ,CAAC;SACH;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAExD,IAAI,CAAC,QAAQ,EAAE;YACb,MAAM,IAAI,KAAK,CACb,SAAS,KAAK,gCAAgC,QAAQ,GAAG,CAC1D,CAAC;SACH;QAED,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;IAC7D,CAAC;IAED,cAAc,CAAC,QAAgB,EAAE,MAAqB;QACpD,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,wBAAwB;QAC5B,MAAM,MAAM,GAAG,MAAM,IAAA,kBAAS,GAAE,CAAC;QACjC,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC;QAEnD,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE;YAC1C,MAAM,MAAM,GAAG,IAAI,iBAAU,CAAC,aAAa,CAAC,GAAG,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC;YAExE,IAAI,aAAa,CAAC,OAAO,EAAE;gBACzB,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;aAC3C;YAED,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAEpD,IAAI;gBACF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;gBACxC,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC5C,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;aAClD;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CACX,0CAA0C,aAAa,CAAC,QAAQ,GAAG,EACnE,KAAK,CAAC,OAAO,CACd,CAAC;aACH;SACF;IACH,CAAC;IAED,cAAc,CAAC,QAAgB,EAAE,MAAgB;QAC/C,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACxD,MAAM,uBAAuB,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACtE,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC,IAAI,CACtC,IAAI,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CACtC,CAAC;QAGF,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC,IAAI,CAC1C,IAAI,GAAG,CAAC,uBAAuB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAChD,CAAC;IACJ,CAAC;IAED,uBAAuB,CAAC,QAAgB,EAAE,MAAgB;QACxD,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACxD,MAAM,sBAAsB,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAEpE,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC,IAAI,CACtC,IAAI,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CACtC,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC,IAAI,CACzC,IAAI,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CACtC,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,QAAgB,EAAE,KAAa;QAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM;YAAE,OAAO,KAAK,CAAC;QAC1B,OAAO,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IAED,SAAS,CAAC,WAAmB;QAC3B,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE;YACrD,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAC3C,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC;YACjE,IAAI,UAAU,EAAE;gBACd,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;aACxC;SACF;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,mBAAmB,CAAC,QAAgB,EAAE,KAAc;QAClD,IAAI,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE;YAC1C,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;SAC5B;QAED,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE;YACvB,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAE/B,MAAM,gBAAgB,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,aAAa,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAG/C,IAAI,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,EAAE,aAAa,CAAC,EAAE;gBAC1D,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;aAC7D;YAGD,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YACnD,IAAI,YAAY,EAAE;gBAChB,OAAO,YAAY,CAAC;aACrB;SACF;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,YAAY,EAAE;YAChB,OAAO,YAAY,CAAC;SACrB;QAED,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,QAAgB,EAChB,OAA0B;QAE1B,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;QAClE,OAAO,MAAM,CAAC,oBAAoB,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,QAAgB,EAChB,OAAyB;QAEzB,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;QAClE,OAAO,MAAM,CAAC,eAAe,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,mBAAmB,CAAC,QAAgB;QAClC,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC3C,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED,sBAAsB;QACpB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,uBAAuB;QACrB,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAC/B,CAAC;IAED,gBAAgB;QACd,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACxC,CAAC;CACF;AAvND,4BAuNC;AAEY,QAAA,OAAO,GAAG,IAAI,QAAQ,EAAE,CAAC;AAEtC,0CAAwB;AAExB,yCAAuB;AACvB,2CAAyB;AACzB,8CAA4B;AAC5B,4CAA0B;AAC1B,2CAAyB;AACzB,wCAAsB"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { Tool } from "../clients/types";
|
|
1
|
+
import { Tool, ToolCall } from "../clients/types";
|
|
2
2
|
import { ToolOverrideFunction, ToolWrapper } from "./types";
|
|
3
3
|
export declare class ToolsService {
|
|
4
4
|
tools: Tool[];
|
|
@@ -19,6 +19,16 @@ export declare class ToolsService {
|
|
|
19
19
|
addFunctions(fns: {
|
|
20
20
|
[fnName: string]: (...args: any) => any;
|
|
21
21
|
}): void;
|
|
22
|
+
defineTools(tools: Tool[], functions: {
|
|
23
|
+
[fnName: string]: ((...args: any) => any) | any;
|
|
24
|
+
}): void;
|
|
25
|
+
callTool(toolCall: ToolCall, enabledTools?: string[]): Promise<{
|
|
26
|
+
toolMessages: any[];
|
|
27
|
+
toolCallId: string;
|
|
28
|
+
functionName: string;
|
|
29
|
+
functionArgs: any;
|
|
30
|
+
functionResp: any;
|
|
31
|
+
}>;
|
|
22
32
|
registerOverride(pattern: string | RegExp, override: ToolOverrideFunction, priority?: number): void;
|
|
23
33
|
registerWrapper(pattern: string | RegExp, wrapper: ToolWrapper, priority?: number): void;
|
|
24
34
|
removeOverride(pattern: string | RegExp): void;
|
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.Tools = exports.ToolsService = void 0;
|
|
4
|
+
const utils_1 = require("../utils");
|
|
4
5
|
const types_1 = require("./types");
|
|
5
6
|
class ToolsService {
|
|
6
7
|
tools = [];
|
|
@@ -66,13 +67,103 @@ class ToolsService {
|
|
|
66
67
|
this.tools.push(tool);
|
|
67
68
|
}
|
|
68
69
|
addTools(tools) {
|
|
69
|
-
this.
|
|
70
|
+
const existingTools = this.getToolNames();
|
|
71
|
+
const filteredTools = tools.filter((tool) => !existingTools.includes(tool.function.name));
|
|
72
|
+
this.tools.push(...filteredTools);
|
|
70
73
|
}
|
|
71
74
|
addFunctions(fns) {
|
|
72
75
|
for (const fnName of Object.keys(fns)) {
|
|
76
|
+
if (typeof fns[fnName] !== "function") {
|
|
77
|
+
continue;
|
|
78
|
+
}
|
|
73
79
|
this.setFunction(fnName, fns[fnName]);
|
|
74
80
|
}
|
|
75
81
|
}
|
|
82
|
+
defineTools(tools, functions) {
|
|
83
|
+
this.addTools(tools);
|
|
84
|
+
this.addFunctions(functions);
|
|
85
|
+
}
|
|
86
|
+
async callTool(toolCall, enabledTools = this.getToolNames()) {
|
|
87
|
+
const functionName = toolCall.function.name;
|
|
88
|
+
const functionArgs = JSON.parse((0, utils_1.restoreEscapedNewLines)(toolCall.function.arguments));
|
|
89
|
+
try {
|
|
90
|
+
if (!enabledTools.includes(functionName)) {
|
|
91
|
+
const options = enabledTools.join(", ");
|
|
92
|
+
throw new Error(`Function ${functionName} not enabled, options are ${options}`);
|
|
93
|
+
}
|
|
94
|
+
const toolDefinition = this.getTool(functionName);
|
|
95
|
+
if (!toolDefinition) {
|
|
96
|
+
throw new Error(`Tool ${functionName} not found`);
|
|
97
|
+
}
|
|
98
|
+
const functionToCall = this.getFunction(functionName);
|
|
99
|
+
if (!functionToCall) {
|
|
100
|
+
const options = enabledTools.join(", ");
|
|
101
|
+
throw new Error(`Function ${functionName} not found, options are ${options}`);
|
|
102
|
+
}
|
|
103
|
+
const properties = toolDefinition?.function?.parameters?.properties || {};
|
|
104
|
+
const isPositional = toolDefinition?.function?.parameters?.positional || false;
|
|
105
|
+
const fnArgs = isPositional
|
|
106
|
+
? Object.keys(properties).map((p) => functionArgs[p])
|
|
107
|
+
: functionArgs;
|
|
108
|
+
console.log(`Calling function ${functionName} with args:`, JSON.stringify(fnArgs, null, 2));
|
|
109
|
+
const functionResponse = await Promise.resolve(isPositional ? functionToCall(...fnArgs) : functionToCall(fnArgs)).catch((e) => {
|
|
110
|
+
throw new Error("ERROR: " + e.message);
|
|
111
|
+
});
|
|
112
|
+
const toJsonIfObject = (arg) => {
|
|
113
|
+
if (typeof arg === "object") {
|
|
114
|
+
return JSON.stringify(arg, null, 2);
|
|
115
|
+
}
|
|
116
|
+
return arg;
|
|
117
|
+
};
|
|
118
|
+
let toolMessages = [];
|
|
119
|
+
if (functionName === "multi_tool_use.parallel") {
|
|
120
|
+
const args = fnArgs[0];
|
|
121
|
+
toolMessages = args.map((call, index) => {
|
|
122
|
+
return {
|
|
123
|
+
tool_call_id: toolCall.id + "_" + index,
|
|
124
|
+
role: "tool",
|
|
125
|
+
name: call.recipient_name.split(".").pop(),
|
|
126
|
+
content: toJsonIfObject(functionResponse[index]) || "Done",
|
|
127
|
+
};
|
|
128
|
+
});
|
|
129
|
+
}
|
|
130
|
+
else {
|
|
131
|
+
toolMessages = [
|
|
132
|
+
{
|
|
133
|
+
tool_call_id: toolCall.id,
|
|
134
|
+
role: "tool",
|
|
135
|
+
name: functionName,
|
|
136
|
+
content: toJsonIfObject(functionResponse) || "Done",
|
|
137
|
+
},
|
|
138
|
+
];
|
|
139
|
+
}
|
|
140
|
+
return {
|
|
141
|
+
toolMessages,
|
|
142
|
+
toolCallId: toolCall.id,
|
|
143
|
+
functionName,
|
|
144
|
+
functionArgs,
|
|
145
|
+
functionResp: functionResponse,
|
|
146
|
+
};
|
|
147
|
+
}
|
|
148
|
+
catch (error) {
|
|
149
|
+
console.log(error.message);
|
|
150
|
+
const toolMessages = [
|
|
151
|
+
{
|
|
152
|
+
tool_call_id: toolCall.id,
|
|
153
|
+
role: "tool",
|
|
154
|
+
name: "error",
|
|
155
|
+
content: error.message,
|
|
156
|
+
},
|
|
157
|
+
];
|
|
158
|
+
return {
|
|
159
|
+
toolMessages,
|
|
160
|
+
toolCallId: toolCall.id,
|
|
161
|
+
functionName,
|
|
162
|
+
functionArgs,
|
|
163
|
+
functionResp: undefined,
|
|
164
|
+
};
|
|
165
|
+
}
|
|
166
|
+
}
|
|
76
167
|
registerOverride(pattern, override, priority = 0) {
|
|
77
168
|
this.overrides.push({ pattern, override, priority });
|
|
78
169
|
this.overrides.sort((a, b) => b.priority - a.priority);
|
|
@@ -82,10 +173,10 @@ class ToolsService {
|
|
|
82
173
|
this.wrappers.sort((a, b) => b.priority - a.priority);
|
|
83
174
|
}
|
|
84
175
|
removeOverride(pattern) {
|
|
85
|
-
this.overrides = this.overrides.filter(reg => reg.pattern !== pattern);
|
|
176
|
+
this.overrides = this.overrides.filter((reg) => reg.pattern !== pattern);
|
|
86
177
|
}
|
|
87
178
|
removeWrapper(pattern) {
|
|
88
|
-
this.wrappers = this.wrappers.filter(reg => reg.pattern !== pattern);
|
|
179
|
+
this.wrappers = this.wrappers.filter((reg) => reg.pattern !== pattern);
|
|
89
180
|
}
|
|
90
181
|
findMatchingOverride(toolName) {
|
|
91
182
|
for (const registration of this.overrides) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"Tools.js","sourceRoot":"","sources":["../../../src/services/Tools.ts"],"names":[],"mappings":";;;
|
|
1
|
+
{"version":3,"file":"Tools.js","sourceRoot":"","sources":["../../../src/services/Tools.ts"],"names":[],"mappings":";;;AACA,oCAA0E;AAG1E,mCAMiB;AAEjB,MAAa,YAAY;IACvB,KAAK,GAAG,EAAY,CAAC;IACb,SAAS,GAA+B,EAAE,CAAC;IAC3C,QAAQ,GAA8B,EAAE,CAAC;IACzC,iBAAiB,GAA+C,EAAE,CAAC;IAE3E,SAAS,GAAG,EAAE,CAAC;IAEf,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,eAAe,CAAC,KAAe;QAC7B,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;IACzE,CAAC;IAED,aAAa,CAAC,SAAmB,EAAE,YAA0B;QAC3D,MAAM,KAAK,GAAG,YAAY,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QACtD,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAErB,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE;YAC5B,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;SACxD;IACH,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACtD,CAAC;IAED,OAAO,CAAC,IAAY;QAClB,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;IAChE,CAAC;IAED,WAAW,CAAC,IAAY;QAEtB,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC;IAED,WAAW,CAAC,IAAY,EAAE,IAA2B;QAEnD,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE;YACjC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;SACrC;QAGD,MAAM,QAAQ,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACjD,IAAI,QAAQ,EAAE;YACZ,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,KAAK,EAAE,GAAG,IAAW,EAAE,EAAE;gBAC9C,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBAChC,OAAO,MAAM,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC7C,CAAC,CAAC;YACF,OAAO;SACR;QAGD,MAAM,QAAQ,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACjD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;YACvB,IAAI,eAAe,GAAG,IAAI,CAAC;YAG3B,KAAK,MAAM,UAAU,IAAI,QAAQ,EAAE;gBACjC,MAAM,eAAe,GAAG,eAAe,CAAC;gBACxC,eAAe,GAAG,KAAK,EAAE,GAAG,IAAW,EAAE,EAAE;oBACzC,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;oBAChC,OAAO,MAAM,UAAU,CAAC,OAAO,CAAC,eAAe,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC/D,CAAC,CAAC;aACH;YAED,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC;YACvC,OAAO;SACR;QAGD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAC9B,CAAC;IAED,YAAY,CAAC,KAAe,EAAE,KAAgC;QAC5D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACrC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;SACtC;IACH,CAAC;IAED,OAAO,CAAC,IAAU;QAChB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAED,QAAQ,CAAC,KAAa;QAEpB,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QAC1C,MAAM,aAAa,GAAG,KAAK,CAAC,MAAM,CAChC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CACtD,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;IACpC,CAAC;IAED,YAAY,CAAC,GAAgD;QAC3D,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;YACrC,IAAI,OAAO,GAAG,CAAC,MAAM,CAAC,KAAK,UAAU,EAAE;gBAErC,SAAS;aACV;YACD,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;SACvC;IACH,CAAC;IAED,WAAW,CACT,KAAa,EACb,SAA8D;QAE9D,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACrB,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAkB,EAAE,YAAY,GAAG,IAAI,CAAC,YAAY,EAAE;QACnE,MAAM,YAAY,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;QAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAC7B,IAAA,8BAAsB,EAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CACpD,CAAC;QAEF,IAAI;YAEF,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;gBACxC,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACxC,MAAM,IAAI,KAAK,CACb,YAAY,YAAY,6BAA6B,OAAO,EAAE,CAC/D,CAAC;aACH;YAGD,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAClD,IAAI,CAAC,cAAc,EAAE;gBACnB,MAAM,IAAI,KAAK,CAAC,QAAQ,YAAY,YAAY,CAAC,CAAC;aACnD;YAGD,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;YACtD,IAAI,CAAC,cAAc,EAAE;gBACnB,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACxC,MAAM,IAAI,KAAK,CACb,YAAY,YAAY,2BAA2B,OAAO,EAAE,CAC7D,CAAC;aACH;YAGD,MAAM,UAAU,GAAG,cAAc,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,IAAI,EAAE,CAAC;YAC1E,MAAM,YAAY,GAChB,cAAc,EAAE,QAAQ,EAAE,UAAU,EAAE,UAAU,IAAI,KAAK,CAAC;YAC5D,MAAM,MAAM,GAAG,YAAY;gBACzB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;gBACrD,CAAC,CAAC,YAAY,CAAC;YAEjB,OAAO,CAAC,GAAG,CACT,oBAAoB,YAAY,aAAa,EAC7C,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAChC,CAAC;YAGF,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,OAAO,CAC5C,YAAY,CAAC,CAAC,CAAC,cAAc,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,CAClE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;gBACZ,MAAM,IAAI,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC;YACzC,CAAC,CAAC,CAAC;YAGH,MAAM,cAAc,GAAG,CAAC,GAAQ,EAAE,EAAE;gBAClC,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE;oBAC3B,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;iBACrC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,CAAC;YAEF,IAAI,YAAY,GAAG,EAAE,CAAC;YAGtB,IAAI,YAAY,KAAK,yBAAyB,EAAE;gBAC9C,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAGlB,CAAC;gBAEJ,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;oBACtC,OAAO;wBACL,YAAY,EAAE,QAAQ,CAAC,EAAE,GAAG,GAAG,GAAG,KAAK;wBACvC,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE;wBAC1C,OAAO,EAAE,cAAc,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,IAAI,MAAM;qBAC3D,CAAC;gBACJ,CAAC,CAAC,CAAC;aACJ;iBAAM;gBACL,YAAY,GAAG;oBACb;wBACE,YAAY,EAAE,QAAQ,CAAC,EAAE;wBACzB,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,YAAY;wBAClB,OAAO,EAAE,cAAc,CAAC,gBAAgB,CAAC,IAAI,MAAM;qBACpD;iBACF,CAAC;aACH;YAED,OAAO;gBACL,YAAY;gBACZ,UAAU,EAAE,QAAQ,CAAC,EAAE;gBACvB,YAAY;gBACZ,YAAY;gBACZ,YAAY,EAAE,gBAAgB;aAC/B,CAAC;SACH;QAAC,OAAO,KAAK,EAAE;YACd,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC3B,MAAM,YAAY,GAAG;gBACnB;oBACE,YAAY,EAAE,QAAQ,CAAC,EAAE;oBACzB,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,OAAO;oBACb,OAAO,EAAE,KAAK,CAAC,OAAO;iBACvB;aACF,CAAC;YAEF,OAAO;gBACL,YAAY;gBACZ,UAAU,EAAE,QAAQ,CAAC,EAAE;gBACvB,YAAY;gBACZ,YAAY;gBACZ,YAAY,EAAE,SAAS;aACxB,CAAC;SACH;IACH,CAAC;IAGD,gBAAgB,CACd,OAAwB,EACxB,QAA8B,EAC9B,WAAmB,CAAC;QAEpB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;QACrD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;IACzD,CAAC;IAED,eAAe,CACb,OAAwB,EACxB,OAAoB,EACpB,WAAmB,CAAC;QAEpB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;QACnD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;IACxD,CAAC;IAED,cAAc,CAAC,OAAwB;QACrC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,KAAK,OAAO,CAAC,CAAC;IAC3E,CAAC;IAED,aAAa,CAAC,OAAwB;QACpC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,KAAK,OAAO,CAAC,CAAC;IACzE,CAAC;IAEO,oBAAoB,CAC1B,QAAgB;QAEhB,KAAK,MAAM,YAAY,IAAI,IAAI,CAAC,SAAS,EAAE;YACzC,MAAM,OAAO,GAAG,IAAA,4BAAoB,EAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAC3D,IAAI,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;gBAC7B,OAAO,YAAY,CAAC;aACrB;SACF;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,oBAAoB,CAAC,QAAgB;QAC3C,MAAM,gBAAgB,GAA8B,EAAE,CAAC;QACvD,KAAK,MAAM,YAAY,IAAI,IAAI,CAAC,QAAQ,EAAE;YACxC,MAAM,OAAO,GAAG,IAAA,4BAAoB,EAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YAC3D,IAAI,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;gBAC7B,gBAAgB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;aACrC;SACF;QACD,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,mBAAmB,CAAC,IAAY;QAC9B,OAAO,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC;IAED,cAAc;QACZ,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;QACpB,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;QAEnB,KAAK,MAAM,CAAC,IAAI,EAAE,YAAY,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC,EAAE;YACzE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC;SACrC;IACH,CAAC;CACF;AAlSD,oCAkSC;AAEY,QAAA,KAAK,GAAG,IAAI,YAAY,EAAE,CAAC"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
import { Tools } from "../../services";
|
|
2
|
+
import { Clients } from "../../clients";
|
|
3
|
+
import { ScriptTracer } from "./ScriptTracer";
|
|
4
|
+
import { ScriptPolicyEnforcer } from "./ScriptPolicy";
|
|
5
|
+
import { Artifact, QuotaUsage } from "./types";
|
|
6
|
+
import { Message } from "../../clients/types";
|
|
7
|
+
export declare class SandboxContext {
|
|
8
|
+
private toolsService;
|
|
9
|
+
private clients;
|
|
10
|
+
private tracer;
|
|
11
|
+
private policyEnforcer;
|
|
12
|
+
private artifacts;
|
|
13
|
+
private consoleOutput;
|
|
14
|
+
constructor(toolsService: typeof Tools, clients: typeof Clients, tracer: ScriptTracer, policyEnforcer: ScriptPolicyEnforcer);
|
|
15
|
+
console: {
|
|
16
|
+
log: (...args: any[]) => void;
|
|
17
|
+
error: (...args: any[]) => void;
|
|
18
|
+
warn: (...args: any[]) => void;
|
|
19
|
+
info: (...args: any[]) => void;
|
|
20
|
+
};
|
|
21
|
+
callTool(toolName: string, parameters: any): Promise<any>;
|
|
22
|
+
llm(messages: Message[], options?: {
|
|
23
|
+
model?: string;
|
|
24
|
+
maxTokens?: number;
|
|
25
|
+
temperature?: number;
|
|
26
|
+
}): Promise<import("../../clients").CompletionResponse>;
|
|
27
|
+
getQuotaUsage(): QuotaUsage;
|
|
28
|
+
createArtifact(name: string, content: string, type?: "text" | "json" | "csv" | "html" | "markdown"): Promise<Artifact>;
|
|
29
|
+
sleep(ms: number): Promise<void>;
|
|
30
|
+
getArtifacts(): Artifact[];
|
|
31
|
+
getConsoleOutput(): string[];
|
|
32
|
+
private estimateTokens;
|
|
33
|
+
private sanitizeForLogging;
|
|
34
|
+
}
|
|
@@ -0,0 +1,188 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.SandboxContext = void 0;
|
|
4
|
+
const services_1 = require("../../services");
|
|
5
|
+
const clients_1 = require("../../clients");
|
|
6
|
+
class SandboxContext {
|
|
7
|
+
toolsService;
|
|
8
|
+
clients;
|
|
9
|
+
tracer;
|
|
10
|
+
policyEnforcer;
|
|
11
|
+
artifacts = [];
|
|
12
|
+
consoleOutput = [];
|
|
13
|
+
constructor(toolsService = services_1.Tools, clients = clients_1.Clients, tracer, policyEnforcer) {
|
|
14
|
+
this.toolsService = toolsService;
|
|
15
|
+
this.clients = clients;
|
|
16
|
+
this.tracer = tracer;
|
|
17
|
+
this.policyEnforcer = policyEnforcer;
|
|
18
|
+
}
|
|
19
|
+
console = {
|
|
20
|
+
log: (...args) => {
|
|
21
|
+
const message = args
|
|
22
|
+
.map((arg) => typeof arg === "object" ? JSON.stringify(arg) : String(arg))
|
|
23
|
+
.join(" ");
|
|
24
|
+
this.consoleOutput.push(`[LOG] ${message}`);
|
|
25
|
+
this.tracer.emitEvent("console_log", { message, args });
|
|
26
|
+
},
|
|
27
|
+
error: (...args) => {
|
|
28
|
+
const message = args
|
|
29
|
+
.map((arg) => typeof arg === "object" ? JSON.stringify(arg) : String(arg))
|
|
30
|
+
.join(" ");
|
|
31
|
+
this.consoleOutput.push(`[ERROR] ${message}`);
|
|
32
|
+
this.tracer.emitEvent("console_error", { message, args });
|
|
33
|
+
},
|
|
34
|
+
warn: (...args) => {
|
|
35
|
+
const message = args
|
|
36
|
+
.map((arg) => typeof arg === "object" ? JSON.stringify(arg) : String(arg))
|
|
37
|
+
.join(" ");
|
|
38
|
+
this.consoleOutput.push(`[WARN] ${message}`);
|
|
39
|
+
this.tracer.emitEvent("console_warn", { message, args });
|
|
40
|
+
},
|
|
41
|
+
info: (...args) => {
|
|
42
|
+
const message = args
|
|
43
|
+
.map((arg) => typeof arg === "object" ? JSON.stringify(arg) : String(arg))
|
|
44
|
+
.join(" ");
|
|
45
|
+
this.consoleOutput.push(`[INFO] ${message}`);
|
|
46
|
+
this.tracer.emitEvent("console_info", { message, args });
|
|
47
|
+
},
|
|
48
|
+
};
|
|
49
|
+
async callTool(toolName, parameters) {
|
|
50
|
+
if (!this.policyEnforcer.checkToolCall(toolName)) {
|
|
51
|
+
throw new Error(`Tool call '${toolName}' blocked by policy`);
|
|
52
|
+
}
|
|
53
|
+
this.tracer.emitEvent("tool_call_start", {
|
|
54
|
+
toolName,
|
|
55
|
+
parameters: this.sanitizeForLogging(parameters),
|
|
56
|
+
});
|
|
57
|
+
try {
|
|
58
|
+
this.policyEnforcer.recordToolCall();
|
|
59
|
+
const toolCall = {
|
|
60
|
+
id: `script-tool-${Date.now()}-${Math.random()
|
|
61
|
+
.toString(36)
|
|
62
|
+
.substr(2, 9)}`,
|
|
63
|
+
type: "function",
|
|
64
|
+
function: {
|
|
65
|
+
name: toolName,
|
|
66
|
+
arguments: JSON.stringify(parameters),
|
|
67
|
+
},
|
|
68
|
+
};
|
|
69
|
+
const result = await this.toolsService.callTool(toolCall);
|
|
70
|
+
this.tracer.emitEvent("tool_call_success", {
|
|
71
|
+
toolName,
|
|
72
|
+
result: this.sanitizeForLogging(result),
|
|
73
|
+
});
|
|
74
|
+
return result;
|
|
75
|
+
}
|
|
76
|
+
catch (error) {
|
|
77
|
+
this.tracer.emitEvent("tool_call_error", {
|
|
78
|
+
toolName,
|
|
79
|
+
error: error instanceof Error ? error.message : String(error),
|
|
80
|
+
});
|
|
81
|
+
throw error;
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
async llm(messages, options = {}) {
|
|
85
|
+
const estimatedTokens = this.estimateTokens(messages);
|
|
86
|
+
if (!this.policyEnforcer.checkTokenUsage(estimatedTokens)) {
|
|
87
|
+
throw new Error("Token quota would be exceeded");
|
|
88
|
+
}
|
|
89
|
+
this.tracer.emitEvent("llm_call_start", {
|
|
90
|
+
messageCount: messages.length,
|
|
91
|
+
estimatedTokens,
|
|
92
|
+
model: options.model,
|
|
93
|
+
options: this.sanitizeForLogging(options),
|
|
94
|
+
});
|
|
95
|
+
try {
|
|
96
|
+
this.policyEnforcer.recordTokenUsage(estimatedTokens);
|
|
97
|
+
const completionOptions = {
|
|
98
|
+
model: options.model,
|
|
99
|
+
messages,
|
|
100
|
+
max_tokens: options.maxTokens,
|
|
101
|
+
};
|
|
102
|
+
const response = await this.clients.createCompletion("", completionOptions);
|
|
103
|
+
this.tracer.emitEvent("llm_call_success", {
|
|
104
|
+
model: response.model,
|
|
105
|
+
usage: response.usage,
|
|
106
|
+
usdCost: response.usd_cost,
|
|
107
|
+
});
|
|
108
|
+
return response;
|
|
109
|
+
}
|
|
110
|
+
catch (error) {
|
|
111
|
+
this.tracer.emitEvent("llm_call_error", {
|
|
112
|
+
error: error instanceof Error ? error.message : String(error),
|
|
113
|
+
});
|
|
114
|
+
throw error;
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
getQuotaUsage() {
|
|
118
|
+
return this.policyEnforcer.getUsage();
|
|
119
|
+
}
|
|
120
|
+
async createArtifact(name, content, type = "text") {
|
|
121
|
+
const artifact = {
|
|
122
|
+
id: `artifact-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`,
|
|
123
|
+
name,
|
|
124
|
+
type,
|
|
125
|
+
content,
|
|
126
|
+
createdAt: new Date().toISOString(),
|
|
127
|
+
};
|
|
128
|
+
this.artifacts.push(artifact);
|
|
129
|
+
this.tracer.emitEvent("artifact_created", {
|
|
130
|
+
artifactId: artifact.id,
|
|
131
|
+
name,
|
|
132
|
+
type,
|
|
133
|
+
contentLength: content.length,
|
|
134
|
+
});
|
|
135
|
+
return artifact;
|
|
136
|
+
}
|
|
137
|
+
async sleep(ms) {
|
|
138
|
+
if (typeof ms !== "number" || ms < 0 || ms > 2000) {
|
|
139
|
+
throw new Error("Invalid sleep duration");
|
|
140
|
+
}
|
|
141
|
+
await new Promise((res) => setTimeout(res, ms));
|
|
142
|
+
this.tracer.emitEvent("sleep", { durationMs: ms });
|
|
143
|
+
}
|
|
144
|
+
getArtifacts() {
|
|
145
|
+
return [...this.artifacts];
|
|
146
|
+
}
|
|
147
|
+
getConsoleOutput() {
|
|
148
|
+
return [...this.consoleOutput];
|
|
149
|
+
}
|
|
150
|
+
estimateTokens(messages) {
|
|
151
|
+
let totalText = "";
|
|
152
|
+
for (const message of messages) {
|
|
153
|
+
if (typeof message === "string") {
|
|
154
|
+
totalText += message;
|
|
155
|
+
}
|
|
156
|
+
else if (message && typeof message.content === "string") {
|
|
157
|
+
totalText += message.content;
|
|
158
|
+
}
|
|
159
|
+
}
|
|
160
|
+
return Math.ceil(totalText.length / 4);
|
|
161
|
+
}
|
|
162
|
+
sanitizeForLogging(data) {
|
|
163
|
+
if (data === null || data === undefined) {
|
|
164
|
+
return data;
|
|
165
|
+
}
|
|
166
|
+
if (typeof data === "string") {
|
|
167
|
+
return data.length > 500 ? data.substring(0, 500) + "..." : data;
|
|
168
|
+
}
|
|
169
|
+
if (typeof data === "object") {
|
|
170
|
+
const sanitized = {};
|
|
171
|
+
for (const [key, value] of Object.entries(data)) {
|
|
172
|
+
if (key.toLowerCase().includes("password") ||
|
|
173
|
+
key.toLowerCase().includes("token") ||
|
|
174
|
+
key.toLowerCase().includes("secret") ||
|
|
175
|
+
key.toLowerCase().includes("key")) {
|
|
176
|
+
sanitized[key] = "[REDACTED]";
|
|
177
|
+
}
|
|
178
|
+
else {
|
|
179
|
+
sanitized[key] = this.sanitizeForLogging(value);
|
|
180
|
+
}
|
|
181
|
+
}
|
|
182
|
+
return sanitized;
|
|
183
|
+
}
|
|
184
|
+
return data;
|
|
185
|
+
}
|
|
186
|
+
}
|
|
187
|
+
exports.SandboxContext = SandboxContext;
|
|
188
|
+
//# sourceMappingURL=SandboxContext.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"SandboxContext.js","sourceRoot":"","sources":["../../../../src/services/script-execution/SandboxContext.ts"],"names":[],"mappings":";;;AAAA,6CAAuC;AACvC,2CAAwC;AASxC,MAAa,cAAc;IAKf;IACA;IACA;IACA;IAPF,SAAS,GAAe,EAAE,CAAC;IAC3B,aAAa,GAAa,EAAE,CAAC;IAErC,YACU,eAA6B,gBAAK,EAClC,UAA0B,iBAAO,EACjC,MAAoB,EACpB,cAAoC;QAHpC,iBAAY,GAAZ,YAAY,CAAsB;QAClC,YAAO,GAAP,OAAO,CAA0B;QACjC,WAAM,GAAN,MAAM,CAAc;QACpB,mBAAc,GAAd,cAAc,CAAsB;IAC3C,CAAC;IAKJ,OAAO,GAAG;QACR,GAAG,EAAE,CAAC,GAAG,IAAW,EAAE,EAAE;YACtB,MAAM,OAAO,GAAG,IAAI;iBACjB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACX,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAC5D;iBACA,IAAI,CAAC,GAAG,CAAC,CAAC;YACb,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,OAAO,EAAE,CAAC,CAAC;YAC5C,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,CAAC;QAED,KAAK,EAAE,CAAC,GAAG,IAAW,EAAE,EAAE;YACxB,MAAM,OAAO,GAAG,IAAI;iBACjB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACX,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAC5D;iBACA,IAAI,CAAC,GAAG,CAAC,CAAC;YACb,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,WAAW,OAAO,EAAE,CAAC,CAAC;YAC9C,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,eAAe,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,CAAC;QAED,IAAI,EAAE,CAAC,GAAG,IAAW,EAAE,EAAE;YACvB,MAAM,OAAO,GAAG,IAAI;iBACjB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACX,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAC5D;iBACA,IAAI,CAAC,GAAG,CAAC,CAAC;YACb,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,OAAO,EAAE,CAAC,CAAC;YAC7C,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3D,CAAC;QAED,IAAI,EAAE,CAAC,GAAG,IAAW,EAAE,EAAE;YACvB,MAAM,OAAO,GAAG,IAAI;iBACjB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACX,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAC5D;iBACA,IAAI,CAAC,GAAG,CAAC,CAAC;YACb,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,OAAO,EAAE,CAAC,CAAC;YAC7C,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3D,CAAC;KACF,CAAC;IAKF,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,UAAe;QAE9C,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE;YAChD,MAAM,IAAI,KAAK,CAAC,cAAc,QAAQ,qBAAqB,CAAC,CAAC;SAC9D;QAED,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,iBAAiB,EAAE;YACvC,QAAQ;YACR,UAAU,EAAE,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAAC;SAChD,CAAC,CAAC;QAEH,IAAI;YAEF,IAAI,CAAC,cAAc,CAAC,cAAc,EAAE,CAAC;YAGrC,MAAM,QAAQ,GAAG;gBACf,EAAE,EAAE,eAAe,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE;qBAC3C,QAAQ,CAAC,EAAE,CAAC;qBACZ,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;gBACjB,IAAI,EAAE,UAAmB;gBACzB,QAAQ,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;iBACtC;aACF,CAAC;YAGF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAE1D,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,mBAAmB,EAAE;gBACzC,QAAQ;gBACR,MAAM,EAAE,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC;aACxC,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;SACf;QAAC,OAAO,KAAK,EAAE;YACd,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,iBAAiB,EAAE;gBACvC,QAAQ;gBACR,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;SACb;IACH,CAAC;IAKD,KAAK,CAAC,GAAG,CACP,QAAmB,EACnB,UAII,EAAE;QAEN,MAAM,eAAe,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAGtD,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,eAAe,CAAC,EAAE;YACzD,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;SAClD;QAED,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,EAAE;YACtC,YAAY,EAAE,QAAQ,CAAC,MAAM;YAC7B,eAAe;YACf,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,OAAO,EAAE,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC;SAC1C,CAAC,CAAC;QAEH,IAAI;YAEF,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;YAGtD,MAAM,iBAAiB,GAAG;gBACxB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,QAAQ;gBACR,UAAU,EAAE,OAAO,CAAC,SAAS;aAC9B,CAAC;YAGF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAClD,EAAE,EACF,iBAAiB,CAClB,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,kBAAkB,EAAE;gBACxC,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACrB,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACrB,OAAO,EAAE,QAAQ,CAAC,QAAQ;aAC3B,CAAC,CAAC;YAEH,OAAO,QAAQ,CAAC;SACjB;QAAC,OAAO,KAAK,EAAE;YACd,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,EAAE;gBACtC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;SACb;IACH,CAAC;IAKD,aAAa;QACX,OAAO,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,CAAC;IACxC,CAAC;IAKD,KAAK,CAAC,cAAc,CAClB,IAAY,EACZ,OAAe,EACf,OAAsD,MAAM;QAE5D,MAAM,QAAQ,GAAa;YACzB,EAAE,EAAE,YAAY,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;YACvE,IAAI;YACJ,IAAI;YACJ,OAAO;YACP,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE9B,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,kBAAkB,EAAE;YACxC,UAAU,EAAE,QAAQ,CAAC,EAAE;YACvB,IAAI;YACJ,IAAI;YACJ,aAAa,EAAE,OAAO,CAAC,MAAM;SAC9B,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,EAAU;QACpB,IAAI,OAAO,EAAE,KAAK,QAAQ,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,GAAG,IAAI,EAAE;YACjD,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;SAC3C;QACD,MAAM,IAAI,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;IACrD,CAAC;IAKD,YAAY;QACV,OAAO,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7B,CAAC;IAKD,gBAAgB;QACd,OAAO,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC;IACjC,CAAC;IAKO,cAAc,CAAC,QAAe;QACpC,IAAI,SAAS,GAAG,EAAE,CAAC;QACnB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;YAC9B,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE;gBAC/B,SAAS,IAAI,OAAO,CAAC;aACtB;iBAAM,IAAI,OAAO,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE;gBACzD,SAAS,IAAI,OAAO,CAAC,OAAO,CAAC;aAC9B;SACF;QAED,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACzC,CAAC;IAKO,kBAAkB,CAAC,IAAS;QAClC,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,SAAS,EAAE;YACvC,OAAO,IAAI,CAAC;SACb;QAED,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE;YAE5B,OAAO,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;SAClE;QAED,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE;YAC5B,MAAM,SAAS,GAAQ,EAAE,CAAC;YAC1B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;gBAE/C,IACE,GAAG,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC;oBACtC,GAAG,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC;oBACnC,GAAG,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC;oBACpC,GAAG,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,EACjC;oBACA,SAAS,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC;iBAC/B;qBAAM;oBACL,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;iBACjD;aACF;YACD,OAAO,SAAS,CAAC;SAClB;QAED,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AA3QD,wCA2QC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import { Tools } from "../../services";
|
|
2
|
+
import { Clients } from "../../clients";
|
|
3
|
+
import { ExecutionRequest, ExecutionResult, ResourceQuotas, SecurityPolicy } from "./types";
|
|
4
|
+
export declare class ScriptExecutor {
|
|
5
|
+
private toolsService;
|
|
6
|
+
private clients;
|
|
7
|
+
private defaultQuotas;
|
|
8
|
+
private defaultPolicy;
|
|
9
|
+
constructor(toolsService?: typeof Tools | null, clients?: typeof Clients | null);
|
|
10
|
+
execute(request: ExecutionRequest): Promise<ExecutionResult>;
|
|
11
|
+
private executeWithTimeout;
|
|
12
|
+
private executeScriptSecure;
|
|
13
|
+
private setupIsolatedContext;
|
|
14
|
+
private executeScriptFallback;
|
|
15
|
+
getDefaultQuotas(): ResourceQuotas;
|
|
16
|
+
getDefaultPolicy(): SecurityPolicy;
|
|
17
|
+
}
|