runtimeuse 0.2.0
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/.env.example +4 -0
- package/README.md +222 -0
- package/dist/agent-handler.d.ts +26 -0
- package/dist/agent-handler.d.ts.map +1 -0
- package/dist/agent-handler.js +2 -0
- package/dist/agent-handler.js.map +1 -0
- package/dist/artifact-manager.d.ts +27 -0
- package/dist/artifact-manager.d.ts.map +1 -0
- package/dist/artifact-manager.js +125 -0
- package/dist/artifact-manager.js.map +1 -0
- package/dist/artifact-manager.test.d.ts +2 -0
- package/dist/artifact-manager.test.d.ts.map +1 -0
- package/dist/artifact-manager.test.js +251 -0
- package/dist/artifact-manager.test.js.map +1 -0
- package/dist/claude-handler.d.ts +3 -0
- package/dist/claude-handler.d.ts.map +1 -0
- package/dist/claude-handler.js +76 -0
- package/dist/claude-handler.js.map +1 -0
- package/dist/cli.d.ts +3 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +87 -0
- package/dist/cli.js.map +1 -0
- package/dist/command-handler.d.ts +22 -0
- package/dist/command-handler.d.ts.map +1 -0
- package/dist/command-handler.js +75 -0
- package/dist/command-handler.js.map +1 -0
- package/dist/command-handler.test.d.ts +2 -0
- package/dist/command-handler.test.d.ts.map +1 -0
- package/dist/command-handler.test.js +267 -0
- package/dist/command-handler.test.js.map +1 -0
- package/dist/constants.d.ts +3 -0
- package/dist/constants.d.ts.map +1 -0
- package/dist/constants.js +13 -0
- package/dist/constants.js.map +1 -0
- package/dist/default-handler.d.ts +3 -0
- package/dist/default-handler.d.ts.map +1 -0
- package/dist/default-handler.js +76 -0
- package/dist/default-handler.js.map +1 -0
- package/dist/download-handler.d.ts +8 -0
- package/dist/download-handler.d.ts.map +1 -0
- package/dist/download-handler.js +36 -0
- package/dist/download-handler.js.map +1 -0
- package/dist/download-handler.test.d.ts +2 -0
- package/dist/download-handler.test.d.ts.map +1 -0
- package/dist/download-handler.test.js +123 -0
- package/dist/download-handler.test.js.map +1 -0
- package/dist/index.d.ts +20 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +21 -0
- package/dist/index.js.map +1 -0
- package/dist/logger.d.ts +8 -0
- package/dist/logger.d.ts.map +1 -0
- package/dist/logger.js +14 -0
- package/dist/logger.js.map +1 -0
- package/dist/openai-handler.d.ts +3 -0
- package/dist/openai-handler.d.ts.map +1 -0
- package/dist/openai-handler.js +86 -0
- package/dist/openai-handler.js.map +1 -0
- package/dist/server.d.ts +21 -0
- package/dist/server.d.ts.map +1 -0
- package/dist/server.js +52 -0
- package/dist/server.js.map +1 -0
- package/dist/session.d.ts +29 -0
- package/dist/session.d.ts.map +1 -0
- package/dist/session.js +244 -0
- package/dist/session.js.map +1 -0
- package/dist/session.test.d.ts +2 -0
- package/dist/session.test.d.ts.map +1 -0
- package/dist/session.test.js +339 -0
- package/dist/session.test.js.map +1 -0
- package/dist/storage.d.ts +3 -0
- package/dist/storage.d.ts.map +1 -0
- package/dist/storage.js +21 -0
- package/dist/storage.js.map +1 -0
- package/dist/types.d.ts +62 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/upload-tracker.d.ts +10 -0
- package/dist/upload-tracker.d.ts.map +1 -0
- package/dist/upload-tracker.js +27 -0
- package/dist/upload-tracker.js.map +1 -0
- package/dist/upload-tracker.test.d.ts +2 -0
- package/dist/upload-tracker.test.d.ts.map +1 -0
- package/dist/upload-tracker.test.js +89 -0
- package/dist/upload-tracker.test.js.map +1 -0
- package/dist/utils.d.ts +7 -0
- package/dist/utils.d.ts.map +1 -0
- package/dist/utils.js +32 -0
- package/dist/utils.js.map +1 -0
- package/dist/utils.test.d.ts +2 -0
- package/dist/utils.test.d.ts.map +1 -0
- package/dist/utils.test.js +92 -0
- package/dist/utils.test.js.map +1 -0
- package/package.json +40 -0
- package/scripts/dev-publish.sh +45 -0
- package/src/agent-handler.ts +26 -0
- package/src/artifact-manager.test.ts +320 -0
- package/src/artifact-manager.ts +170 -0
- package/src/claude-handler.ts +95 -0
- package/src/cli.ts +107 -0
- package/src/command-handler.test.ts +507 -0
- package/src/command-handler.ts +102 -0
- package/src/constants.ts +12 -0
- package/src/download-handler.test.ts +183 -0
- package/src/download-handler.ts +45 -0
- package/src/index.ts +59 -0
- package/src/logger.ts +20 -0
- package/src/openai-handler.ts +120 -0
- package/src/server.ts +68 -0
- package/src/session.test.ts +448 -0
- package/src/session.ts +319 -0
- package/src/storage.ts +28 -0
- package/src/types.ts +101 -0
- package/src/upload-tracker.test.ts +112 -0
- package/src/upload-tracker.ts +30 -0
- package/src/utils.test.ts +120 -0
- package/src/utils.ts +35 -0
- package/tsconfig.json +20 -0
- package/vitest.config.ts +7 -0
package/dist/index.js
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
// Built-in handlers
|
|
2
|
+
export { openaiHandler } from "./openai-handler.js";
|
|
3
|
+
export { claudeHandler } from "./claude-handler.js";
|
|
4
|
+
// Server
|
|
5
|
+
export { RuntimeUseServer } from "./server.js";
|
|
6
|
+
// Session
|
|
7
|
+
export { WebSocketSession } from "./session.js";
|
|
8
|
+
// Artifact management
|
|
9
|
+
export { ArtifactManager } from "./artifact-manager.js";
|
|
10
|
+
// Upload
|
|
11
|
+
export { UploadTracker } from "./upload-tracker.js";
|
|
12
|
+
export { uploadFile } from "./storage.js";
|
|
13
|
+
// Commands & downloads
|
|
14
|
+
export { default as CommandHandler } from "./command-handler.js";
|
|
15
|
+
export { default as DownloadHandler } from "./download-handler.js";
|
|
16
|
+
// Utilities
|
|
17
|
+
export { redactSecrets, sleep } from "./utils.js";
|
|
18
|
+
export { createLogger, defaultLogger } from "./logger.js";
|
|
19
|
+
// Constants
|
|
20
|
+
export { DEFAULT_ARTIFACTS_DIR, DEFAULT_ARTIFACT_IGNORE, } from "./constants.js";
|
|
21
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAQA,oBAAoB;AACpB,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,SAAS;AACT,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAG/C,UAAU;AACV,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAGhD,sBAAsB;AACtB,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAGxD,SAAS;AACT,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAE1C,uBAAuB;AACvB,OAAO,EAAE,OAAO,IAAI,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEjE,OAAO,EAAE,OAAO,IAAI,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAkBnE,YAAY;AACZ,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAG1D,YAAY;AACZ,OAAO,EACL,qBAAqB,EACrB,uBAAuB,GACxB,MAAM,gBAAgB,CAAC"}
|
package/dist/logger.d.ts
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
export interface Logger {
|
|
2
|
+
log(...args: unknown[]): void;
|
|
3
|
+
error(...args: unknown[]): void;
|
|
4
|
+
debug(...args: unknown[]): void;
|
|
5
|
+
}
|
|
6
|
+
export declare function createLogger(sourceId: string): Logger;
|
|
7
|
+
export declare const defaultLogger: Logger;
|
|
8
|
+
//# sourceMappingURL=logger.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,MAAM;IACrB,GAAG,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;IAC9B,KAAK,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;IAChC,KAAK,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;CACjC;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAOrD;AAED,eAAO,MAAM,aAAa,EAAE,MAI3B,CAAC"}
|
package/dist/logger.js
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
export function createLogger(sourceId) {
|
|
2
|
+
const prefix = `[${sourceId}]`;
|
|
3
|
+
return {
|
|
4
|
+
log: (...args) => console.log(prefix, ...args),
|
|
5
|
+
error: (...args) => console.error(prefix, ...args),
|
|
6
|
+
debug: (...args) => console.debug(prefix, ...args),
|
|
7
|
+
};
|
|
8
|
+
}
|
|
9
|
+
export const defaultLogger = {
|
|
10
|
+
log: (...args) => console.log(...args),
|
|
11
|
+
error: (...args) => console.error(...args),
|
|
12
|
+
debug: (...args) => console.debug(...args),
|
|
13
|
+
};
|
|
14
|
+
//# sourceMappingURL=logger.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"logger.js","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":"AAMA,MAAM,UAAU,YAAY,CAAC,QAAgB;IAC3C,MAAM,MAAM,GAAG,IAAI,QAAQ,GAAG,CAAC;IAC/B,OAAO;QACL,GAAG,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC;QACzD,KAAK,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC;QAC7D,KAAK,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC;KAC9D,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,aAAa,GAAW;IACnC,GAAG,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;IACjD,KAAK,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC;IACrD,KAAK,EAAE,CAAC,GAAG,IAAe,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC;CACtD,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-handler.d.ts","sourceRoot":"","sources":["../src/openai-handler.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EACV,YAAY,EAIb,MAAM,oBAAoB,CAAC;AAE5B,eAAO,MAAM,aAAa,EAAE,YAyE3B,CAAC"}
|
|
@@ -0,0 +1,86 @@
|
|
|
1
|
+
import zod from "zod";
|
|
2
|
+
import { Agent, run as runAgent, codeInterpreterTool, webSearchTool, } from "@openai/agents";
|
|
3
|
+
export const openaiHandler = {
|
|
4
|
+
async run(invocation, sender) {
|
|
5
|
+
Object.assign(process.env, invocation.env);
|
|
6
|
+
const strictSchema = ensureStrictSchema(invocation.outputFormat.schema);
|
|
7
|
+
const outputType = zod.fromJSONSchema(strictSchema);
|
|
8
|
+
const agent = new Agent({
|
|
9
|
+
name: "runtimeuse-agent",
|
|
10
|
+
instructions: invocation.systemPrompt,
|
|
11
|
+
outputType,
|
|
12
|
+
model: invocation.model,
|
|
13
|
+
tools: [codeInterpreterTool(), webSearchTool()],
|
|
14
|
+
});
|
|
15
|
+
const result = await runAgent(agent, invocation.userPrompt, {
|
|
16
|
+
signal: invocation.signal,
|
|
17
|
+
stream: true,
|
|
18
|
+
});
|
|
19
|
+
let currentText = "";
|
|
20
|
+
for await (const event of result) {
|
|
21
|
+
if (event.type === "raw_model_stream_event" &&
|
|
22
|
+
"delta" in event.data &&
|
|
23
|
+
typeof event.data.delta === "string") {
|
|
24
|
+
currentText += event.data.delta;
|
|
25
|
+
}
|
|
26
|
+
else if (event.type === "run_item_stream_event" &&
|
|
27
|
+
event.name === "message_output_created") {
|
|
28
|
+
if (currentText) {
|
|
29
|
+
sender.sendAssistantMessage([currentText]);
|
|
30
|
+
currentText = "";
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
await result.completed;
|
|
35
|
+
if (currentText) {
|
|
36
|
+
sender.sendAssistantMessage([currentText]);
|
|
37
|
+
}
|
|
38
|
+
let structuredOutput = {};
|
|
39
|
+
const finalOutput = result.finalOutput;
|
|
40
|
+
if (typeof finalOutput === "string") {
|
|
41
|
+
try {
|
|
42
|
+
structuredOutput = JSON.parse(finalOutput);
|
|
43
|
+
}
|
|
44
|
+
catch {
|
|
45
|
+
structuredOutput = { result: finalOutput };
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
else if (finalOutput != null && typeof finalOutput === "object") {
|
|
49
|
+
structuredOutput = finalOutput;
|
|
50
|
+
}
|
|
51
|
+
const metadata = {};
|
|
52
|
+
const usage = result.state?.usage;
|
|
53
|
+
if (usage) {
|
|
54
|
+
metadata.usage = {
|
|
55
|
+
inputTokens: usage.inputTokens,
|
|
56
|
+
outputTokens: usage.outputTokens,
|
|
57
|
+
totalTokens: usage.totalTokens,
|
|
58
|
+
};
|
|
59
|
+
}
|
|
60
|
+
return { structuredOutput, metadata };
|
|
61
|
+
},
|
|
62
|
+
};
|
|
63
|
+
function ensureStrictSchema(schema) {
|
|
64
|
+
const result = { ...schema };
|
|
65
|
+
if (result.type === "object") {
|
|
66
|
+
result.additionalProperties = false;
|
|
67
|
+
if (result.properties &&
|
|
68
|
+
typeof result.properties === "object" &&
|
|
69
|
+
!Array.isArray(result.properties)) {
|
|
70
|
+
const props = {};
|
|
71
|
+
for (const [key, value] of Object.entries(result.properties)) {
|
|
72
|
+
props[key] =
|
|
73
|
+
value && typeof value === "object" && !Array.isArray(value)
|
|
74
|
+
? ensureStrictSchema(value)
|
|
75
|
+
: value;
|
|
76
|
+
}
|
|
77
|
+
result.properties = props;
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
if (result.items && typeof result.items === "object") {
|
|
81
|
+
result.items = ensureStrictSchema(result.items);
|
|
82
|
+
}
|
|
83
|
+
delete result.title;
|
|
84
|
+
return result;
|
|
85
|
+
}
|
|
86
|
+
//# sourceMappingURL=openai-handler.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-handler.js","sourceRoot":"","sources":["../src/openai-handler.ts"],"names":[],"mappings":"AAAA,OAAO,GAAG,MAAM,KAAK,CAAC;AAEtB,OAAO,EACL,KAAK,EACL,GAAG,IAAI,QAAQ,EACf,mBAAmB,EACnB,aAAa,GAEd,MAAM,gBAAgB,CAAC;AAQxB,MAAM,CAAC,MAAM,aAAa,GAAiB;IACzC,KAAK,CAAC,GAAG,CACP,UAA2B,EAC3B,MAAqB;QAErB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;QAE3C,MAAM,YAAY,GAAG,kBAAkB,CAAC,UAAU,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QACxE,MAAM,UAAU,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,CAAoB,CAAC;QAEvE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;YACtB,IAAI,EAAE,kBAAkB;YACxB,YAAY,EAAE,UAAU,CAAC,YAAY;YACrC,UAAU;YACV,KAAK,EAAE,UAAU,CAAC,KAAK;YACvB,KAAK,EAAE,CAAC,mBAAmB,EAAE,EAAE,aAAa,EAAE,CAAC;SAChD,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAC,UAAU,EAAE;YAC1D,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,MAAM,EAAE,IAAI;SACb,CAAC,CAAC;QAEH,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IACE,KAAK,CAAC,IAAI,KAAK,wBAAwB;gBACvC,OAAO,IAAI,KAAK,CAAC,IAAI;gBACrB,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,KAAK,QAAQ,EACpC,CAAC;gBACD,WAAW,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC;YAClC,CAAC;iBAAM,IACL,KAAK,CAAC,IAAI,KAAK,uBAAuB;gBACtC,KAAK,CAAC,IAAI,KAAK,wBAAwB,EACvC,CAAC;gBACD,IAAI,WAAW,EAAE,CAAC;oBAChB,MAAM,CAAC,oBAAoB,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;oBAC3C,WAAW,GAAG,EAAE,CAAC;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,MAAM,CAAC,SAAS,CAAC;QAEvB,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,CAAC,oBAAoB,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,gBAAgB,GAA4B,EAAE,CAAC;QACnD,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QAEvC,IAAI,OAAO,WAAW,KAAK,QAAQ,EAAE,CAAC;YACpC,IAAI,CAAC;gBACH,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAC7C,CAAC;YAAC,MAAM,CAAC;gBACP,gBAAgB,GAAG,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;YAC7C,CAAC;QACH,CAAC;aAAM,IAAI,WAAW,IAAI,IAAI,IAAI,OAAO,WAAW,KAAK,QAAQ,EAAE,CAAC;YAClE,gBAAgB,GAAG,WAAsC,CAAC;QAC5D,CAAC;QAED,MAAM,QAAQ,GAA4B,EAAE,CAAC;QAC7C,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC;QAClC,IAAI,KAAK,EAAE,CAAC;YACV,QAAQ,CAAC,KAAK,GAAG;gBACf,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,WAAW,EAAE,KAAK,CAAC,WAAW;aAC/B,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,gBAAgB,EAAE,QAAQ,EAAE,CAAC;IACxC,CAAC;CACF,CAAC;AAEF,SAAS,kBAAkB,CACzB,MAA+B;IAE/B,MAAM,MAAM,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC;IAC7B,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,MAAM,CAAC,oBAAoB,GAAG,KAAK,CAAC;QACpC,IACE,MAAM,CAAC,UAAU;YACjB,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ;YACrC,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,EACjC,CAAC;YACD,MAAM,KAAK,GAA4B,EAAE,CAAC;YAC1C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CACvC,MAAM,CAAC,UAAqC,CAC7C,EAAE,CAAC;gBACF,KAAK,CAAC,GAAG,CAAC;oBACR,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;wBACzD,CAAC,CAAC,kBAAkB,CAAC,KAAgC,CAAC;wBACtD,CAAC,CAAC,KAAK,CAAC;YACd,CAAC;YACD,MAAM,CAAC,UAAU,GAAG,KAAK,CAAC;QAC5B,CAAC;IACH,CAAC;IACD,IAAI,MAAM,CAAC,KAAK,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrD,MAAM,CAAC,KAAK,GAAG,kBAAkB,CAAC,MAAM,CAAC,KAAgC,CAAC,CAAC;IAC7E,CAAC;IACD,OAAO,MAAM,CAAC,KAAK,CAAC;IACpB,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
package/dist/server.d.ts
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import type { AgentHandler } from "./agent-handler.js";
|
|
2
|
+
import { type Logger } from "./logger.js";
|
|
3
|
+
export interface RuntimeUseServerConfig {
|
|
4
|
+
handler: AgentHandler;
|
|
5
|
+
port?: number;
|
|
6
|
+
defaultModel?: string;
|
|
7
|
+
uploadTimeoutMs?: number;
|
|
8
|
+
artifactWaitMs?: number;
|
|
9
|
+
postInvocationDelayMs?: number;
|
|
10
|
+
logger?: Logger;
|
|
11
|
+
}
|
|
12
|
+
export declare class RuntimeUseServer {
|
|
13
|
+
private readonly wss;
|
|
14
|
+
private readonly config;
|
|
15
|
+
private readonly logger;
|
|
16
|
+
constructor(config: RuntimeUseServerConfig);
|
|
17
|
+
private handleConnection;
|
|
18
|
+
start(): Promise<void>;
|
|
19
|
+
stop(): Promise<void>;
|
|
20
|
+
}
|
|
21
|
+
//# sourceMappingURL=server.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGvD,OAAO,EAAiB,KAAK,MAAM,EAAE,MAAM,aAAa,CAAC;AAEzD,MAAM,WAAW,sBAAsB;IACrC,OAAO,EAAE,YAAY,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAkB;IACtC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAyB;IAChD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;gBAEpB,MAAM,EAAE,sBAAsB;IAU1C,OAAO,CAAC,gBAAgB;IAiBlB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAWtB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;CAQ5B"}
|
package/dist/server.js
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
import { WebSocketServer } from "ws";
|
|
2
|
+
import { WebSocketSession } from "./session.js";
|
|
3
|
+
import { UploadTracker } from "./upload-tracker.js";
|
|
4
|
+
import { defaultLogger } from "./logger.js";
|
|
5
|
+
export class RuntimeUseServer {
|
|
6
|
+
wss;
|
|
7
|
+
config;
|
|
8
|
+
logger;
|
|
9
|
+
constructor(config) {
|
|
10
|
+
this.config = config;
|
|
11
|
+
this.logger = config.logger ?? defaultLogger;
|
|
12
|
+
this.wss = new WebSocketServer({ port: config.port ?? 8080 });
|
|
13
|
+
this.wss.on("connection", (ws) => {
|
|
14
|
+
this.handleConnection(ws);
|
|
15
|
+
});
|
|
16
|
+
}
|
|
17
|
+
handleConnection(ws) {
|
|
18
|
+
const uploadTracker = new UploadTracker();
|
|
19
|
+
const sessionConfig = {
|
|
20
|
+
handler: this.config.handler,
|
|
21
|
+
uploadTracker,
|
|
22
|
+
defaultModel: this.config.defaultModel,
|
|
23
|
+
uploadTimeoutMs: this.config.uploadTimeoutMs,
|
|
24
|
+
artifactWaitMs: this.config.artifactWaitMs,
|
|
25
|
+
postInvocationDelayMs: this.config.postInvocationDelayMs,
|
|
26
|
+
logger: this.config.logger,
|
|
27
|
+
};
|
|
28
|
+
const session = new WebSocketSession(ws, sessionConfig);
|
|
29
|
+
session.run().catch((error) => {
|
|
30
|
+
this.logger.error("Session error:", error);
|
|
31
|
+
});
|
|
32
|
+
}
|
|
33
|
+
async start() {
|
|
34
|
+
return new Promise((resolve) => {
|
|
35
|
+
this.wss.on("listening", () => {
|
|
36
|
+
this.logger.log(`RuntimeUse server listening on port ${this.config.port ?? 8080}`);
|
|
37
|
+
resolve();
|
|
38
|
+
});
|
|
39
|
+
});
|
|
40
|
+
}
|
|
41
|
+
async stop() {
|
|
42
|
+
return new Promise((resolve, reject) => {
|
|
43
|
+
this.wss.close((err) => {
|
|
44
|
+
if (err)
|
|
45
|
+
reject(err);
|
|
46
|
+
else
|
|
47
|
+
resolve();
|
|
48
|
+
});
|
|
49
|
+
});
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
//# sourceMappingURL=server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAa,MAAM,IAAI,CAAC;AAEhD,OAAO,EAAE,gBAAgB,EAAsB,MAAM,cAAc,CAAC;AACpE,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAe,MAAM,aAAa,CAAC;AAYzD,MAAM,OAAO,gBAAgB;IACV,GAAG,CAAkB;IACrB,MAAM,CAAyB;IAC/B,MAAM,CAAS;IAEhC,YAAY,MAA8B;QACxC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,aAAa,CAAC;QAC7C,IAAI,CAAC,GAAG,GAAG,IAAI,eAAe,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC;QAE9D,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAa,EAAE,EAAE;YAC1C,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,gBAAgB,CAAC,EAAa;QACpC,MAAM,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;QAC1C,MAAM,aAAa,GAAkB;YACnC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;YAC5B,aAAa;YACb,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;YACtC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;YAC5C,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;YAC1C,qBAAqB,EAAE,IAAI,CAAC,MAAM,CAAC,qBAAqB;YACxD,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;SAC3B,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC;QACxD,OAAO,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YAC5B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK;QACT,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,EAAE,GAAG,EAAE;gBAC5B,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,uCAAuC,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,IAAI,EAAE,CAClE,CAAC;gBACF,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI;QACR,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACrB,IAAI,GAAG;oBAAE,MAAM,CAAC,GAAG,CAAC,CAAC;;oBAChB,OAAO,EAAE,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
import { WebSocket } from "ws";
|
|
2
|
+
import type { AgentHandler } from "./agent-handler.js";
|
|
3
|
+
import type { UploadTracker } from "./upload-tracker.js";
|
|
4
|
+
import { type Logger } from "./logger.js";
|
|
5
|
+
export interface SessionConfig {
|
|
6
|
+
handler: AgentHandler;
|
|
7
|
+
uploadTracker: UploadTracker;
|
|
8
|
+
defaultModel?: string;
|
|
9
|
+
uploadTimeoutMs?: number;
|
|
10
|
+
artifactWaitMs?: number;
|
|
11
|
+
postInvocationDelayMs?: number;
|
|
12
|
+
logger?: Logger;
|
|
13
|
+
}
|
|
14
|
+
export declare class WebSocketSession {
|
|
15
|
+
private readonly ws;
|
|
16
|
+
private readonly config;
|
|
17
|
+
private readonly abortController;
|
|
18
|
+
private artifactManager;
|
|
19
|
+
private invocationReceived;
|
|
20
|
+
private logger;
|
|
21
|
+
private readonly downloadHandler;
|
|
22
|
+
constructor(ws: WebSocket, config: SessionConfig);
|
|
23
|
+
run(): Promise<void>;
|
|
24
|
+
private handleMessage;
|
|
25
|
+
private executeInvocation;
|
|
26
|
+
private finalize;
|
|
27
|
+
private send;
|
|
28
|
+
}
|
|
29
|
+
//# sourceMappingURL=session.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../src/session.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAE/B,OAAO,KAAK,EAAE,YAAY,EAAiB,MAAM,oBAAoB,CAAC;AAEtE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAQzD,OAAO,EAA+B,KAAK,MAAM,EAAE,MAAM,aAAa,CAAC;AAKvE,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,YAAY,CAAC;IACtB,aAAa,EAAE,aAAa,CAAC;IAC7B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAY;IAC/B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAgB;IACvC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAyB;IACzD,OAAO,CAAC,eAAe,CAAgC;IACvD,OAAO,CAAC,kBAAkB,CAAS;IACnC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAkB;gBAEtC,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,aAAa;IAOhD,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;YAiCN,aAAa;YAiDb,iBAAiB;YA0KjB,QAAQ;IAgBtB,OAAO,CAAC,IAAI;CAKb"}
|
package/dist/session.js
ADDED
|
@@ -0,0 +1,244 @@
|
|
|
1
|
+
import fs from "fs";
|
|
2
|
+
import path from "path";
|
|
3
|
+
import { WebSocket } from "ws";
|
|
4
|
+
import { ArtifactManager } from "./artifact-manager.js";
|
|
5
|
+
import { sleep } from "./utils.js";
|
|
6
|
+
import { createLogger, defaultLogger } from "./logger.js";
|
|
7
|
+
import { DEFAULT_ARTIFACTS_DIR } from "./constants.js";
|
|
8
|
+
import CommandHandler from "./command-handler.js";
|
|
9
|
+
import DownloadHandler from "./download-handler.js";
|
|
10
|
+
export class WebSocketSession {
|
|
11
|
+
ws;
|
|
12
|
+
config;
|
|
13
|
+
abortController = new AbortController();
|
|
14
|
+
artifactManager = null;
|
|
15
|
+
invocationReceived = false;
|
|
16
|
+
logger;
|
|
17
|
+
downloadHandler;
|
|
18
|
+
constructor(ws, config) {
|
|
19
|
+
this.ws = ws;
|
|
20
|
+
this.config = config;
|
|
21
|
+
this.logger = config.logger ?? defaultLogger;
|
|
22
|
+
this.downloadHandler = new DownloadHandler(this.logger);
|
|
23
|
+
}
|
|
24
|
+
run() {
|
|
25
|
+
return new Promise((resolve) => {
|
|
26
|
+
this.ws.on("message", async (rawData) => {
|
|
27
|
+
this.logger.log("Received new WS message");
|
|
28
|
+
try {
|
|
29
|
+
const message = JSON.parse(rawData.toString());
|
|
30
|
+
await this.handleMessage(message, resolve);
|
|
31
|
+
}
|
|
32
|
+
catch (error) {
|
|
33
|
+
this.logger.error("Error processing message:", error);
|
|
34
|
+
this.send({
|
|
35
|
+
message_type: "error_message",
|
|
36
|
+
error: String(error),
|
|
37
|
+
metadata: {},
|
|
38
|
+
});
|
|
39
|
+
}
|
|
40
|
+
});
|
|
41
|
+
this.ws.on("close", async () => {
|
|
42
|
+
this.logger.log("WebSocket connection closed");
|
|
43
|
+
this.abortController.abort();
|
|
44
|
+
await this.artifactManager?.stopWatching();
|
|
45
|
+
await this.config.uploadTracker.waitForAll(this.config.uploadTimeoutMs ?? 30_000);
|
|
46
|
+
resolve();
|
|
47
|
+
});
|
|
48
|
+
this.ws.on("error", (error) => {
|
|
49
|
+
this.logger.error("WebSocket error:", error);
|
|
50
|
+
});
|
|
51
|
+
});
|
|
52
|
+
}
|
|
53
|
+
async handleMessage(message, resolve) {
|
|
54
|
+
if (!this.invocationReceived &&
|
|
55
|
+
message.message_type !== "invocation_message") {
|
|
56
|
+
throw new Error("Received non-invocation message before invocation message! Received: " +
|
|
57
|
+
JSON.stringify(message));
|
|
58
|
+
}
|
|
59
|
+
switch (message.message_type) {
|
|
60
|
+
case "artifact_upload_response_message":
|
|
61
|
+
try {
|
|
62
|
+
await this.artifactManager?.handleUploadResponse(message);
|
|
63
|
+
}
|
|
64
|
+
catch (error) {
|
|
65
|
+
this.logger.error("Error uploading artifact:", error);
|
|
66
|
+
this.send({
|
|
67
|
+
message_type: "error_message",
|
|
68
|
+
error: String(error),
|
|
69
|
+
metadata: {},
|
|
70
|
+
});
|
|
71
|
+
}
|
|
72
|
+
break;
|
|
73
|
+
case "cancel_message":
|
|
74
|
+
this.logger.log("Received cancel message. Aborting agent execution...");
|
|
75
|
+
this.abortController.abort();
|
|
76
|
+
this.ws.close();
|
|
77
|
+
break;
|
|
78
|
+
case "invocation_message":
|
|
79
|
+
if (this.invocationReceived) {
|
|
80
|
+
throw new Error("Received multiple invocation messages!");
|
|
81
|
+
}
|
|
82
|
+
this.invocationReceived = true;
|
|
83
|
+
await this.executeInvocation(message);
|
|
84
|
+
if (process.env.NODE_ENV !== "test") {
|
|
85
|
+
await sleep(this.config.postInvocationDelayMs ?? 3_000);
|
|
86
|
+
}
|
|
87
|
+
await this.finalize();
|
|
88
|
+
resolve();
|
|
89
|
+
break;
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
async executeInvocation(message) {
|
|
93
|
+
this.logger = createLogger(message.source_id);
|
|
94
|
+
this.config.uploadTracker.setLogger(this.logger);
|
|
95
|
+
const artifactsDir = message.artifacts_dir ?? DEFAULT_ARTIFACTS_DIR;
|
|
96
|
+
this.artifactManager = new ArtifactManager({
|
|
97
|
+
artifactsDir,
|
|
98
|
+
uploadTracker: this.config.uploadTracker,
|
|
99
|
+
send: (msg) => this.send(msg),
|
|
100
|
+
});
|
|
101
|
+
this.artifactManager.setLogger(this.logger);
|
|
102
|
+
const outputFormat = JSON.parse(message.output_format_json_schema_str);
|
|
103
|
+
const model = message.preferred_model ?? this.config.defaultModel ?? "default";
|
|
104
|
+
this.logger.log(`Received invocation: model=${model}`);
|
|
105
|
+
try {
|
|
106
|
+
if (message.runtime_environment_downloadables) {
|
|
107
|
+
this.logger.log("Downloading runtime environment downloadables...");
|
|
108
|
+
for (const downloadable of message.runtime_environment_downloadables) {
|
|
109
|
+
await this.downloadHandler.download(downloadable.download_url, downloadable.working_dir);
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
if (message.pre_agent_invocation_commands) {
|
|
113
|
+
const stdoutFile = path.join(artifactsDir, `cmd-stdout.out`);
|
|
114
|
+
const stderrFile = path.join(artifactsDir, `cmd-stderr.out`);
|
|
115
|
+
const stdoutStream = fs.createWriteStream(stdoutFile);
|
|
116
|
+
const stderrStream = fs.createWriteStream(stderrFile);
|
|
117
|
+
for (const command of message.pre_agent_invocation_commands) {
|
|
118
|
+
try {
|
|
119
|
+
this.logger.log(`Executing command: ${command.command} in directory: ${command.cwd}`);
|
|
120
|
+
const commandHandler = new CommandHandler(command, this.logger, stdoutStream, stderrStream, this.abortController, (stdout) => this.send({
|
|
121
|
+
message_type: "assistant_message",
|
|
122
|
+
text_blocks: [stdout],
|
|
123
|
+
}), (stderr) => this.send({
|
|
124
|
+
message_type: "assistant_message",
|
|
125
|
+
text_blocks: [stderr],
|
|
126
|
+
}));
|
|
127
|
+
try {
|
|
128
|
+
const result = await commandHandler.execute();
|
|
129
|
+
if (result.exitCode === 0) {
|
|
130
|
+
this.send({
|
|
131
|
+
message_type: "result_message",
|
|
132
|
+
structured_output: {
|
|
133
|
+
success: true,
|
|
134
|
+
steps: [],
|
|
135
|
+
summary: "Workflow completed successfully",
|
|
136
|
+
},
|
|
137
|
+
metadata: {},
|
|
138
|
+
});
|
|
139
|
+
return;
|
|
140
|
+
}
|
|
141
|
+
else {
|
|
142
|
+
this.logger.error("Command failed with exit code:", result.exitCode);
|
|
143
|
+
this.send({
|
|
144
|
+
message_type: "result_message",
|
|
145
|
+
structured_output: {
|
|
146
|
+
success: false,
|
|
147
|
+
steps: [],
|
|
148
|
+
summary: "Workflow completed with failure",
|
|
149
|
+
},
|
|
150
|
+
metadata: {},
|
|
151
|
+
});
|
|
152
|
+
return;
|
|
153
|
+
}
|
|
154
|
+
}
|
|
155
|
+
catch (error) {
|
|
156
|
+
this.logger.error("Error executing command:", error);
|
|
157
|
+
this.send({
|
|
158
|
+
message_type: "error_message",
|
|
159
|
+
error: String(error),
|
|
160
|
+
metadata: {},
|
|
161
|
+
});
|
|
162
|
+
this.send({
|
|
163
|
+
message_type: "result_message",
|
|
164
|
+
structured_output: {
|
|
165
|
+
success: false,
|
|
166
|
+
steps: [],
|
|
167
|
+
summary: "Workflow completed with failure",
|
|
168
|
+
},
|
|
169
|
+
metadata: {},
|
|
170
|
+
});
|
|
171
|
+
throw error;
|
|
172
|
+
}
|
|
173
|
+
}
|
|
174
|
+
catch (error) {
|
|
175
|
+
this.logger.error("Error executing command:", error);
|
|
176
|
+
throw error;
|
|
177
|
+
}
|
|
178
|
+
}
|
|
179
|
+
stderrStream.close();
|
|
180
|
+
stdoutStream.close();
|
|
181
|
+
}
|
|
182
|
+
const sender = {
|
|
183
|
+
sendAssistantMessage: (textBlocks) => {
|
|
184
|
+
this.send({
|
|
185
|
+
message_type: "assistant_message",
|
|
186
|
+
text_blocks: textBlocks,
|
|
187
|
+
});
|
|
188
|
+
},
|
|
189
|
+
sendErrorMessage: (error, metadata) => {
|
|
190
|
+
this.send({
|
|
191
|
+
message_type: "error_message",
|
|
192
|
+
error,
|
|
193
|
+
metadata: metadata ?? {},
|
|
194
|
+
});
|
|
195
|
+
},
|
|
196
|
+
};
|
|
197
|
+
const agentResult = await this.config.handler.run({
|
|
198
|
+
systemPrompt: message.system_prompt,
|
|
199
|
+
userPrompt: message.user_prompt,
|
|
200
|
+
outputFormat,
|
|
201
|
+
model,
|
|
202
|
+
secrets: message.secrets_to_redact,
|
|
203
|
+
env: message.agent_env ?? {},
|
|
204
|
+
signal: this.abortController.signal,
|
|
205
|
+
logger: this.logger,
|
|
206
|
+
}, sender);
|
|
207
|
+
const resultMessage = {
|
|
208
|
+
message_type: "result_message",
|
|
209
|
+
metadata: agentResult.metadata ?? {},
|
|
210
|
+
structured_output: agentResult.structuredOutput,
|
|
211
|
+
};
|
|
212
|
+
this.logger.log("Sending result message:", JSON.stringify(resultMessage));
|
|
213
|
+
this.send(resultMessage);
|
|
214
|
+
}
|
|
215
|
+
catch (error) {
|
|
216
|
+
if (this.abortController.signal.aborted) {
|
|
217
|
+
this.ws.close();
|
|
218
|
+
this.logger.log("Agent execution aborted.");
|
|
219
|
+
return;
|
|
220
|
+
}
|
|
221
|
+
this.logger.error("Error in agent execution:", error);
|
|
222
|
+
this.send({
|
|
223
|
+
message_type: "error_message",
|
|
224
|
+
error: String(error),
|
|
225
|
+
metadata: {},
|
|
226
|
+
});
|
|
227
|
+
}
|
|
228
|
+
}
|
|
229
|
+
async finalize() {
|
|
230
|
+
await this.artifactManager?.stopWatching();
|
|
231
|
+
if (!this.abortController.signal.aborted) {
|
|
232
|
+
await this.artifactManager?.waitForPendingRequests(this.config.artifactWaitMs ?? 60_000);
|
|
233
|
+
}
|
|
234
|
+
await this.config.uploadTracker.waitForAll(this.config.uploadTimeoutMs ?? 30_000);
|
|
235
|
+
this.logger.log("All artifacts uploaded.");
|
|
236
|
+
this.ws.close();
|
|
237
|
+
}
|
|
238
|
+
send(data) {
|
|
239
|
+
if (this.ws.readyState === WebSocket.OPEN) {
|
|
240
|
+
this.ws.send(JSON.stringify(data));
|
|
241
|
+
}
|
|
242
|
+
}
|
|
243
|
+
}
|
|
244
|
+
//# sourceMappingURL=session.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.js","sourceRoot":"","sources":["../src/session.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAG/B,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAQxD,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,YAAY,EAAE,aAAa,EAAe,MAAM,aAAa,CAAC;AACvE,OAAO,EAAE,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AACvD,OAAO,cAAc,MAAM,sBAAsB,CAAC;AAClD,OAAO,eAAe,MAAM,uBAAuB,CAAC;AAYpD,MAAM,OAAO,gBAAgB;IACV,EAAE,CAAY;IACd,MAAM,CAAgB;IACtB,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;IACjD,eAAe,GAA2B,IAAI,CAAC;IAC/C,kBAAkB,GAAG,KAAK,CAAC;IAC3B,MAAM,CAAS;IACN,eAAe,CAAkB;IAElD,YAAY,EAAa,EAAE,MAAqB;QAC9C,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,aAAa,CAAC;QAC7C,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC1D,CAAC;IAED,GAAG;QACD,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YACnC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBACtC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;gBAC3C,IAAI,CAAC;oBACH,MAAM,OAAO,GAAoB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;oBAChE,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBAC7C,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;oBACtD,IAAI,CAAC,IAAI,CAAC;wBACR,YAAY,EAAE,eAAe;wBAC7B,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;wBACpB,QAAQ,EAAE,EAAE;qBACb,CAAC,CAAC;gBACL,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE;gBAC7B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;gBAC/C,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;gBAC7B,MAAM,IAAI,CAAC,eAAe,EAAE,YAAY,EAAE,CAAC;gBAC3C,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,UAAU,CACxC,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,MAAM,CACtC,CAAC;gBACF,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBAC5B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,OAAwB,EACxB,OAAmB;QAEnB,IACE,CAAC,IAAI,CAAC,kBAAkB;YACxB,OAAO,CAAC,YAAY,KAAK,oBAAoB,EAC7C,CAAC;YACD,MAAM,IAAI,KAAK,CACb,uEAAuE;gBACrE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAC1B,CAAC;QACJ,CAAC;QAED,QAAQ,OAAO,CAAC,YAAY,EAAE,CAAC;YAC7B,KAAK,kCAAkC;gBACrC,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,eAAe,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAC;gBAC5D,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;oBACtD,IAAI,CAAC,IAAI,CAAC;wBACR,YAAY,EAAE,eAAe;wBAC7B,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;wBACpB,QAAQ,EAAE,EAAE;qBACb,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM;YAER,KAAK,gBAAgB;gBACnB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAC;gBACxE,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;gBAC7B,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;gBAChB,MAAM;YAER,KAAK,oBAAoB;gBACvB,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;oBAC5B,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;gBAC5D,CAAC;gBACD,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;gBAC/B,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;gBACtC,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;oBACpC,MAAM,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,qBAAqB,IAAI,KAAK,CAAC,CAAC;gBAC1D,CAAC;gBACD,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACtB,OAAO,EAAE,CAAC;gBACV,MAAM;QACV,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,OAA0B;QACxD,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEjD,MAAM,YAAY,GAAG,OAAO,CAAC,aAAa,IAAI,qBAAqB,CAAC;QACpE,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,CAAC;YACzC,YAAY;YACZ,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;YACxC,IAAI,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;SAC9B,CAAC,CAAC;QACH,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE5C,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,6BAA6B,CAGpE,CAAC;QACF,MAAM,KAAK,GACT,OAAO,CAAC,eAAe,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,SAAS,CAAC;QAEnE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,8BAA8B,KAAK,EAAE,CAAC,CAAC;QAEvD,IAAI,CAAC;YACH,IAAI,OAAO,CAAC,iCAAiC,EAAE,CAAC;gBAC9C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;gBACpE,KAAK,MAAM,YAAY,IAAI,OAAO,CAAC,iCAAiC,EAAE,CAAC;oBACrE,MAAM,IAAI,CAAC,eAAe,CAAC,QAAQ,CACjC,YAAY,CAAC,YAAY,EACzB,YAAY,CAAC,WAAW,CACzB,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,IAAI,OAAO,CAAC,6BAA6B,EAAE,CAAC;gBAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAC;gBAC7D,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,gBAAgB,CAAC,CAAC;gBAC7D,MAAM,YAAY,GAAG,EAAE,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;gBACtD,MAAM,YAAY,GAAG,EAAE,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;gBAEtD,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,6BAA6B,EAAE,CAAC;oBAC5D,IAAI,CAAC;wBACH,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,sBAAsB,OAAO,CAAC,OAAO,kBAAkB,OAAO,CAAC,GAAG,EAAE,CACrE,CAAC;wBACF,MAAM,cAAc,GAAG,IAAI,cAAc,CACvC,OAAO,EACP,IAAI,CAAC,MAAM,EACX,YAAY,EACZ,YAAY,EACZ,IAAI,CAAC,eAAe,EACpB,CAAC,MAAM,EAAE,EAAE,CACT,IAAI,CAAC,IAAI,CAAC;4BACR,YAAY,EAAE,mBAAmB;4BACjC,WAAW,EAAE,CAAC,MAAM,CAAC;yBACtB,CAAC,EACJ,CAAC,MAAM,EAAE,EAAE,CACT,IAAI,CAAC,IAAI,CAAC;4BACR,YAAY,EAAE,mBAAmB;4BACjC,WAAW,EAAE,CAAC,MAAM,CAAC;yBACtB,CAAC,CACL,CAAC;wBACF,IAAI,CAAC;4BACH,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,CAAC;4BAC9C,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;gCAC1B,IAAI,CAAC,IAAI,CAAC;oCACR,YAAY,EAAE,gBAAgB;oCAC9B,iBAAiB,EAAE;wCACjB,OAAO,EAAE,IAAI;wCACb,KAAK,EAAE,EAAE;wCACT,OAAO,EAAE,iCAAiC;qCAC3C;oCACD,QAAQ,EAAE,EAAE;iCACb,CAAC,CAAC;gCACH,OAAO;4BACT,CAAC;iCAAM,CAAC;gCACN,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,gCAAgC,EAChC,MAAM,CAAC,QAAQ,CAChB,CAAC;gCACF,IAAI,CAAC,IAAI,CAAC;oCACR,YAAY,EAAE,gBAAgB;oCAC9B,iBAAiB,EAAE;wCACjB,OAAO,EAAE,KAAK;wCACd,KAAK,EAAE,EAAE;wCACT,OAAO,EAAE,iCAAiC;qCAC3C;oCACD,QAAQ,EAAE,EAAE;iCACb,CAAC,CAAC;gCACH,OAAO;4BACT,CAAC;wBACH,CAAC;wBAAC,OAAO,KAAK,EAAE,CAAC;4BACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;4BACrD,IAAI,CAAC,IAAI,CAAC;gCACR,YAAY,EAAE,eAAe;gCAC7B,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;gCACpB,QAAQ,EAAE,EAAE;6BACb,CAAC,CAAC;4BACH,IAAI,CAAC,IAAI,CAAC;gCACR,YAAY,EAAE,gBAAgB;gCAC9B,iBAAiB,EAAE;oCACjB,OAAO,EAAE,KAAK;oCACd,KAAK,EAAE,EAAE;oCACT,OAAO,EAAE,iCAAiC;iCAC3C;gCACD,QAAQ,EAAE,EAAE;6BACb,CAAC,CAAC;4BACH,MAAM,KAAK,CAAC;wBACd,CAAC;oBACH,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;wBACrD,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC;gBACD,YAAY,CAAC,KAAK,EAAE,CAAC;gBACrB,YAAY,CAAC,KAAK,EAAE,CAAC;YACvB,CAAC;YAED,MAAM,MAAM,GAAkB;gBAC5B,oBAAoB,EAAE,CAAC,UAAoB,EAAE,EAAE;oBAC7C,IAAI,CAAC,IAAI,CAAC;wBACR,YAAY,EAAE,mBAAmB;wBACjC,WAAW,EAAE,UAAU;qBACxB,CAAC,CAAC;gBACL,CAAC;gBACD,gBAAgB,EAAE,CAChB,KAAa,EACb,QAAkC,EAClC,EAAE;oBACF,IAAI,CAAC,IAAI,CAAC;wBACR,YAAY,EAAE,eAAe;wBAC7B,KAAK;wBACL,QAAQ,EAAE,QAAQ,IAAI,EAAE;qBACzB,CAAC,CAAC;gBACL,CAAC;aACF,CAAC;YAEF,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAC/C;gBACE,YAAY,EAAE,OAAO,CAAC,aAAa;gBACnC,UAAU,EAAE,OAAO,CAAC,WAAW;gBAC/B,YAAY;gBACZ,KAAK;gBACL,OAAO,EAAE,OAAO,CAAC,iBAAiB;gBAClC,GAAG,EAAE,OAAO,CAAC,SAAS,IAAI,EAAE;gBAC5B,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM;gBACnC,MAAM,EAAE,IAAI,CAAC,MAAM;aACpB,EACD,MAAM,CACP,CAAC;YAEF,MAAM,aAAa,GAAkB;gBACnC,YAAY,EAAE,gBAAgB;gBAC9B,QAAQ,EAAE,WAAW,CAAC,QAAQ,IAAI,EAAE;gBACpC,iBAAiB,EAAE,WAAW,CAAC,gBAAgB;aAChD,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,yBAAyB,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC;YAC1E,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACxC,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;gBAChB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;gBAC5C,OAAO;YACT,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;YACtD,IAAI,CAAC,IAAI,CAAC;gBACR,YAAY,EAAE,eAAe;gBAC7B,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;gBACpB,QAAQ,EAAE,EAAE;aACb,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,QAAQ;QACpB,MAAM,IAAI,CAAC,eAAe,EAAE,YAAY,EAAE,CAAC;QAE3C,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACzC,MAAM,IAAI,CAAC,eAAe,EAAE,sBAAsB,CAChD,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,MAAM,CACrC,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,UAAU,CACxC,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,MAAM,CACtC,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QAC3C,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC;IAEO,IAAI,CAAC,IAAqB;QAChC,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YAC1C,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.test.d.ts","sourceRoot":"","sources":["../src/session.test.ts"],"names":[],"mappings":""}
|