@ekairos/dataset 1.8.3 → 1.8.5-beta.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/dist/clearDataset.tool.d.ts +1 -13
- package/dist/clearDataset.tool.d.ts.map +1 -1
- package/dist/clearDataset.tool.js.map +1 -1
- package/dist/completeDataset.tool.d.ts +1 -18
- package/dist/completeDataset.tool.d.ts.map +1 -1
- package/dist/completeDataset.tool.js.map +1 -1
- package/dist/executeCommand.tool.d.ts +1 -28
- package/dist/executeCommand.tool.d.ts.map +1 -1
- package/dist/executeCommand.tool.js.map +1 -1
- package/dist/file/file-dataset.agent.d.ts +19 -3
- package/dist/file/file-dataset.agent.d.ts.map +1 -1
- package/dist/file/file-dataset.agent.js +117 -131
- package/dist/file/file-dataset.agent.js.map +1 -1
- package/dist/file/generateSchema.tool.d.ts +1 -20
- package/dist/file/generateSchema.tool.d.ts.map +1 -1
- package/dist/file/generateSchema.tool.js +1 -1
- package/dist/file/generateSchema.tool.js.map +1 -1
- package/dist/file/scripts/preview_head_excel.py +30 -30
- package/dist/file/scripts/preview_mid_excel.py +39 -39
- package/dist/file/scripts/preview_tail_excel.py +29 -29
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +2 -0
- package/dist/index.js.map +1 -1
- package/dist/schema.d.ts +1 -1
- package/dist/schema.js +1 -1
- package/dist/schema.js.map +1 -1
- package/dist/service.d.ts +7 -1
- package/dist/service.d.ts.map +1 -1
- package/dist/service.js +2 -2
- package/dist/service.js.map +1 -1
- package/dist/transform/transform-dataset.agent.d.ts +20 -4
- package/dist/transform/transform-dataset.agent.d.ts.map +1 -1
- package/dist/transform/transform-dataset.agent.js +119 -129
- package/dist/transform/transform-dataset.agent.js.map +1 -1
- package/package.json +3 -3
|
@@ -5,18 +5,6 @@ interface ClearDatasetToolParams {
|
|
|
5
5
|
datasetId: string;
|
|
6
6
|
sandbox: Sandbox;
|
|
7
7
|
}
|
|
8
|
-
export declare function createClearDatasetTool({ service, datasetId, sandbox }: ClearDatasetToolParams):
|
|
9
|
-
reason: string;
|
|
10
|
-
}, {
|
|
11
|
-
success: boolean;
|
|
12
|
-
error: string;
|
|
13
|
-
deletedRecords?: undefined;
|
|
14
|
-
message?: undefined;
|
|
15
|
-
} | {
|
|
16
|
-
success: boolean;
|
|
17
|
-
deletedRecords: number;
|
|
18
|
-
message: string;
|
|
19
|
-
error?: undefined;
|
|
20
|
-
}>;
|
|
8
|
+
export declare function createClearDatasetTool({ service, datasetId, sandbox }: ClearDatasetToolParams): any;
|
|
21
9
|
export {};
|
|
22
10
|
//# sourceMappingURL=clearDataset.tool.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"clearDataset.tool.d.ts","sourceRoot":"","sources":["../src/clearDataset.tool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAA;AAEzC,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAE1C,UAAU,sBAAsB;IAC5B,OAAO,EAAE,cAAc,CAAA;IACvB,SAAS,EAAE,MAAM,CAAA;IACjB,OAAO,EAAE,OAAO,CAAA;CACnB;AAED,wBAAgB,sBAAsB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,sBAAsB
|
|
1
|
+
{"version":3,"file":"clearDataset.tool.d.ts","sourceRoot":"","sources":["../src/clearDataset.tool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAA;AAEzC,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAE1C,UAAU,sBAAsB;IAC5B,OAAO,EAAE,cAAc,CAAA;IACvB,SAAS,EAAE,MAAM,CAAA;IACjB,OAAO,EAAE,OAAO,CAAA;CACnB;AAED,wBAAgB,sBAAsB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,sBAAsB,OA0D7F"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"clearDataset.tool.js","sourceRoot":"","sources":["../src/clearDataset.tool.ts"],"names":[],"mappings":";;AAYA,wDA0DC;AAtED,2BAAyB;AACzB,6BAAuB;AAEvB,iDAAqD;AASrD,SAAgB,sBAAsB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAA0B;IAC1F,OAAO,IAAA,SAAI,EAAC;QACR,WAAW,EAAE,6HAA6H;QAC1I,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;YAClB,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qCAAqC,CAAC;SACrE,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"clearDataset.tool.js","sourceRoot":"","sources":["../src/clearDataset.tool.ts"],"names":[],"mappings":";;AAYA,wDA0DC;AAtED,2BAAyB;AACzB,6BAAuB;AAEvB,iDAAqD;AASrD,SAAgB,sBAAsB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAA0B;IAC1F,OAAO,IAAA,SAAI,EAAC;QACR,WAAW,EAAE,6HAA6H;QAC1I,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;YAClB,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qCAAqC,CAAC;SACrE,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,MAAM,EAAsB,EAAE,EAAE;YAC9C,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,4CAA4C,CAAC,CAAA;YAC9E,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,sBAAsB,CAAC,CAAA;YACxD,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,aAAa,MAAM,EAAE,CAAC,CAAA;YACvD,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,4CAA4C,CAAC,CAAA;YAE9E,MAAM,UAAU,GAAG,IAAA,mCAAoB,EAAC,SAAS,CAAC,CAAA;YAElD,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,gCAAgC,CAAC,CAAA;YAClE,IAAI,CAAC;gBACD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC;oBACpC,GAAG,EAAE,IAAI;oBACT,IAAI,EAAE,CAAC,IAAI,EAAE,UAAU,CAAC;iBAC3B,CAAC,CAAA;gBAEF,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;oBACxB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,MAAM,EAAE,CAAA;oBACpC,OAAO,CAAC,IAAI,CAAC,YAAY,SAAS,mCAAmC,MAAM,EAAE,CAAC,CAAA;gBAClF,CAAC;qBACI,CAAC;oBACF,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,yBAAyB,CAAC,CAAA;gBAC/D,CAAC;YACL,CAAC;YACD,OAAO,KAAK,EAAE,CAAC;gBACX,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;gBACtE,OAAO,CAAC,IAAI,CAAC,YAAY,SAAS,iCAAiC,OAAO,EAAE,CAAC,CAAA;YACjF,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,oCAAoC,CAAC,CAAA;YACtE,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAA;YAEzD,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC;gBAClB,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,8BAA8B,WAAW,CAAC,KAAK,EAAE,CAAC,CAAA;gBACrF,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,WAAW,CAAC,KAAK;iBAC3B,CAAA;YACL,CAAC;YAED,MAAM,YAAY,GAAG,WAAW,CAAC,IAAI,CAAC,YAAY,CAAA;YAClD,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,eAAe,YAAY,UAAU,CAAC,CAAA;YAEvE,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,gCAAgC,CAAC,CAAA;YAClE,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,4CAA4C,CAAC,CAAA;YAE9E,OAAO;gBACH,OAAO,EAAE,IAAI;gBACb,cAAc,EAAE,YAAY;gBAC5B,OAAO,EAAE,yCAAyC,YAAY,sCAAsC,MAAM,EAAE;aAC/G,CAAA;QACL,CAAC;KACJ,CAAC,CAAA;AACN,CAAC"}
|
|
@@ -5,23 +5,6 @@ interface CompleteDatasetToolParams {
|
|
|
5
5
|
datasetId: string;
|
|
6
6
|
sandbox: Sandbox;
|
|
7
7
|
}
|
|
8
|
-
export declare function createCompleteDatasetTool({ service, datasetId, sandbox }: CompleteDatasetToolParams):
|
|
9
|
-
summary: string;
|
|
10
|
-
}, {
|
|
11
|
-
success: boolean;
|
|
12
|
-
validation?: Array<{
|
|
13
|
-
index: number;
|
|
14
|
-
valid: boolean;
|
|
15
|
-
errors?: string[];
|
|
16
|
-
}>;
|
|
17
|
-
validRowCount?: number;
|
|
18
|
-
error?: string;
|
|
19
|
-
} | {
|
|
20
|
-
success: boolean;
|
|
21
|
-
validRows: number | undefined;
|
|
22
|
-
fileId: string;
|
|
23
|
-
storagePath: string;
|
|
24
|
-
message: string;
|
|
25
|
-
}>;
|
|
8
|
+
export declare function createCompleteDatasetTool({ service, datasetId, sandbox }: CompleteDatasetToolParams): any;
|
|
26
9
|
export {};
|
|
27
10
|
//# sourceMappingURL=completeDataset.tool.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"completeDataset.tool.d.ts","sourceRoot":"","sources":["../src/completeDataset.tool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAA;AAKzC,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAe1C,UAAU,yBAAyB;IAC/B,OAAO,EAAE,cAAc,CAAA;IACvB,SAAS,EAAE,MAAM,CAAA;IACjB,OAAO,EAAE,OAAO,CAAA;CACnB;AAED,wBAAgB,yBAAyB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,yBAAyB
|
|
1
|
+
{"version":3,"file":"completeDataset.tool.d.ts","sourceRoot":"","sources":["../src/completeDataset.tool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAA;AAKzC,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAe1C,UAAU,yBAAyB;IAC/B,OAAO,EAAE,cAAc,CAAA;IACvB,SAAS,EAAE,MAAM,CAAA;IACjB,OAAO,EAAE,OAAO,CAAA;CACnB;AAED,wBAAgB,yBAAyB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,yBAAyB,OA2JnG"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"completeDataset.tool.js","sourceRoot":"","sources":["../src/completeDataset.tool.ts"],"names":[],"mappings":";;;;;AA4BA,8DA2JC;AAvLD,2BAAyB;AACzB,6BAAuB;AAEvB,8CAA2C;AAC3C,iDAEuB;AAGvB,IAAI,WAAW,GAAe,IAAI,CAAA;AAElC,SAAS,MAAM;IACX,IAAI,CAAC,WAAW,EAChB,CAAC;QACG,WAAW,GAAG,IAAI,aAAG,CAAC;YAClB,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,KAAK;SAChB,CAAC,CAAA;IACN,CAAC;IACD,OAAO,WAAW,CAAA;AACtB,CAAC;AAQD,SAAgB,yBAAyB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAA6B;IAChG,OAAO,IAAA,SAAI,EAAC;QACR,WAAW,EAAE,wHAAwH;QACrI,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;YAClB,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uEAAuE,CAAC;SACxG,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,
|
|
1
|
+
{"version":3,"file":"completeDataset.tool.js","sourceRoot":"","sources":["../src/completeDataset.tool.ts"],"names":[],"mappings":";;;;;AA4BA,8DA2JC;AAvLD,2BAAyB;AACzB,6BAAuB;AAEvB,8CAA2C;AAC3C,iDAEuB;AAGvB,IAAI,WAAW,GAAe,IAAI,CAAA;AAElC,SAAS,MAAM;IACX,IAAI,CAAC,WAAW,EAChB,CAAC;QACG,WAAW,GAAG,IAAI,aAAG,CAAC;YAClB,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,KAAK;SAChB,CAAC,CAAA;IACN,CAAC;IACD,OAAO,WAAW,CAAA;AACtB,CAAC;AAQD,SAAgB,yBAAyB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAA6B;IAChG,OAAO,IAAA,SAAI,EAAC;QACR,WAAW,EAAE,wHAAwH;QACrI,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;YAClB,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uEAAuE,CAAC;SACxG,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,EAAuB,EAAE,EAAE;YAChD,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,4CAA4C,CAAC,CAAA;YAC9E,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,yBAAyB,CAAC,CAAA;YAC3D,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,cAAc,OAAO,EAAE,CAAC,CAAA;YACzD,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,4CAA4C,CAAC,CAAA;YAE9E,MAAM,UAAU,GAAG,IAAA,mCAAoB,EAAC,SAAS,CAAC,CAAA;YAElD,IAAI,CAAC;gBACD,MAAM,gBAAgB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAA;YAC/C,CAAC;YACD,OAAO,KAAK,EAAE,CAAC;gBACX,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;gBACtE,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,wBAAwB,EAAE,OAAO,CAAC,CAAA;gBACrE,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,OAAO;iBACjB,CAAA;YACL,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,0CAA0C,CAAC,CAAA;YAE5E,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC,CAAA;YAC7D,IAAI,CAAC,aAAa,CAAC,EAAE,EAAE,CAAC;gBACpB,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,KAAK,aAAa,CAAC,KAAK,EAAE,CAAC,CAAA;gBAC9D,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,aAAa,CAAC,KAAK;iBAC7B,CAAA;YACL,CAAC;YAED,MAAM,aAAa,GAAG,aAAa,CAAC,IAAI,CAAA;YACxC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;gBACxB,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,gCAAgC,CAAC,CAAA;gBACpE,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,6DAA6D;iBACvE,CAAA;YACL,CAAC;YAED,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,MAAM,CAAA;YAE9C,IAAI,SAA2B,CAAA;YAC/B,IAAI,CAAC;gBACD,SAAS,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,CAAA;YAC5C,CAAC;YACD,OAAO,KAAK,EAAE,CAAC;gBACX,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;gBACtE,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,6BAA6B,EAAE,OAAO,CAAC,CAAA;gBAC1E,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,6BAA6B,OAAO,EAAE;iBAChD,CAAA;YACL,CAAC;YAED,MAAM,gBAAgB,GAAG,MAAM,iBAAiB,CAAC;gBAC7C,OAAO;gBACP,UAAU;gBACV,SAAS;gBACT,SAAS;aACZ,CAAC,CAAA;YAEF,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;gBAC5B,OAAO,gBAAgB,CAAA;YAC3B,CAAC;YAED,MAAM,cAAc,GAAG,gBAAgB,CAAC,aAAa,CAAA;YAErD,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,mCAAmC,CAAC,CAAA;YACrE,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAA;YAE/D,IAAI,CAAC,UAAU,EAAE,CAAC;gBACd,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,uBAAuB,CAAC,CAAA;gBAC3D,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,kCAAkC;iBAC5C,CAAA;YACL,CAAC;YAED,MAAM,MAAM,GAAa,EAAE,CAAA;YAC3B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,UAAuE,EAAE,CAAC;gBAChG,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;oBAC5B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAA;gBAC5C,CAAC;qBACI,IAAI,KAAK,YAAY,UAAU,EAAE,CAAC;oBACnC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAA;gBACnC,CAAC;qBACI,IAAI,KAAK,YAAY,WAAW,EAAE,CAAC;oBACpC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAA;gBACnC,CAAC;qBACI,IAAI,KAAK,EAAE,CAAC;oBACb,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAA;gBACnC,CAAC;YACL,CAAC;YAED,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACtB,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,sBAAsB,CAAC,CAAA;gBAC1D,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,oBAAoB;iBAC9B,CAAA;YACL,CAAC;YAED,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;YAExC,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,uCAAuC,CAAC,CAAA;YAEzE,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,uBAAuB,CAAC;gBACvD,SAAS;gBACT,UAAU;aACb,CAAC,CAAA;YAEF,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,CAAC;gBACnB,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,yBAAyB,YAAY,CAAC,KAAK,EAAE,CAAC,CAAA;gBACjF,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,YAAY,CAAC,KAAK;iBAC5B,CAAA;YACL,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,iCAAiC,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAA;YAE7F,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,mBAAmB,CAAC;gBACnD,SAAS;gBACT,MAAM,EAAE,WAAW;gBACnB,mBAAmB,EAAE,cAAc;gBACnC,uBAAuB,EAAE,cAAc;aAC1C,CAAC,CAAA;YAEF,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,CAAC;gBACnB,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,8BAA8B,YAAY,CAAC,KAAK,EAAE,CAAC,CAAA;gBACtF,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,YAAY,CAAC,KAAK;iBAC5B,CAAA;YACL,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,kCAAkC,cAAc,cAAc,CAAC,CAAA;YAChG,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,4CAA4C,CAAC,CAAA;YAE9E,OAAO;gBACH,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,cAAc;gBACzB,MAAM,EAAE,YAAY,CAAC,IAAI,CAAC,MAAM;gBAChC,WAAW,EAAE,YAAY,CAAC,IAAI,CAAC,WAAW;gBAC1C,OAAO,EAAE,oDAAoD;aAChE,CAAA;QACL,CAAC;KACJ,CAAC,CAAA;AACN,CAAC;AAID,KAAK,UAAU,gBAAgB,CAAC,OAAgB,EAAE,IAAY;IAC1D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC;QACpC,GAAG,EAAE,MAAM;QACX,IAAI,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC;KACrB,CAAC,CAAA;IAEF,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAA;IACvD,CAAC;AACL,CAAC;AASD,KAAK,UAAU,iBAAiB,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAA2B;IAMnG,MAAM,UAAU,GAAgE,EAAE,CAAA;IAClF,IAAI,aAAa,GAAG,CAAC,CAAA;IAErB,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,kDAAkD,CAAC,CAAA;IAEpF,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAA;IAE/D,IAAI,CAAC,UAAU,EAAE,CAAC;QACd,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,qBAAqB,CAAC,CAAA;QACvD,OAAO;YACH,OAAO,EAAE,IAAI;YACb,UAAU;YACV,aAAa,EAAE,CAAC;SACnB,CAAA;IACL,CAAC;IAED,MAAM,MAAM,GAAa,EAAE,CAAA;IAC3B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,UAAuE,EAAE,CAAC;QAChG,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC5B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAA;QAC5C,CAAC;aACI,IAAI,KAAK,YAAY,UAAU,EAAE,CAAC;YACnC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAA;QACnC,CAAC;aACI,IAAI,KAAK,YAAY,WAAW,EAAE,CAAC;YACpC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAA;QACnC,CAAC;aACI,IAAI,KAAK,EAAE,CAAC;YACb,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAA;QACnC,CAAC;IACL,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACtB,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,qBAAqB,CAAC,CAAA;QACvD,OAAO;YACH,OAAO,EAAE,IAAI;YACb,UAAU;YACV,aAAa,EAAE,CAAC;SACnB,CAAA;IACL,CAAC;IAED,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAA;IACpD,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;IACrC,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,gBAAgB,KAAK,CAAC,MAAM,QAAQ,CAAC,CAAA;IAEtE,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,EACjD,CAAC;QACG,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,CAAA;QACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAA;QAC3B,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,SAAQ;QACZ,CAAC;QAED,IAAI,MAAW,CAAA;QACf,IAAI,CAAC;YACD,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;QAChC,CAAC;QACD,OAAO,KAAK,EAAE,CAAC;YACX,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;YACtE,UAAU,CAAC,IAAI,CAAC;gBACZ,KAAK;gBACL,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,CAAC,iBAAiB,OAAO,EAAE,CAAC;aACvC,CAAC,CAAA;YACF,SAAQ;QACZ,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;YACxB,SAAQ;QACZ,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAA;QACxB,IAAI,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YACtC,UAAU,CAAC,IAAI,CAAC;gBACZ,KAAK;gBACL,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,CAAC,sBAAsB,CAAC;aACnC,CAAC,CAAA;YACF,SAAQ;QACZ,CAAC;QAED,MAAM,KAAK,GAAG,SAAS,CAAC,IAAI,CAAC,CAAA;QAC7B,IAAI,CAAC,KAAK,EAAE,CAAC;YACT,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC;gBAC1C,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,IAAI,0BAA0B,CAAC;gBAC1E,CAAC,CAAC,CAAC,0BAA0B,CAAC,CAAA;YAClC,UAAU,CAAC,IAAI,CAAC;gBACZ,KAAK;gBACL,KAAK,EAAE,KAAK;gBACZ,MAAM;aACT,CAAC,CAAA;YACF,SAAQ;QACZ,CAAC;QAED,UAAU,CAAC,IAAI,CAAC;YACZ,KAAK;YACL,KAAK,EAAE,IAAI;SACd,CAAC,CAAA;QACF,aAAa,EAAE,CAAA;IACnB,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,2BAA2B,aAAa,aAAa,CAAC,CAAA;IAEvF,OAAO;QACH,OAAO,EAAE,IAAI;QACb,UAAU;QACV,aAAa;KAChB,CAAA;AACL,CAAC"}
|
|
@@ -5,33 +5,6 @@ interface ExecuteCommandToolParams {
|
|
|
5
5
|
datasetId: string;
|
|
6
6
|
sandbox: Sandbox;
|
|
7
7
|
}
|
|
8
|
-
export declare function createExecuteCommandTool({ service, datasetId, sandbox }: ExecuteCommandToolParams):
|
|
9
|
-
pythonCode: string;
|
|
10
|
-
scriptName: string;
|
|
11
|
-
}, {
|
|
12
|
-
success: boolean;
|
|
13
|
-
exitCode: number;
|
|
14
|
-
stdout: string;
|
|
15
|
-
stderr: string;
|
|
16
|
-
scriptPath: string;
|
|
17
|
-
error: string;
|
|
18
|
-
stdoutTruncated: boolean;
|
|
19
|
-
stderrTruncated: boolean;
|
|
20
|
-
stdoutOriginalLength: number;
|
|
21
|
-
stderrOriginalLength: number;
|
|
22
|
-
message?: undefined;
|
|
23
|
-
} | {
|
|
24
|
-
success: boolean;
|
|
25
|
-
exitCode: number;
|
|
26
|
-
stdout: string;
|
|
27
|
-
stderr: string;
|
|
28
|
-
scriptPath: string;
|
|
29
|
-
message: string;
|
|
30
|
-
stdoutTruncated: boolean;
|
|
31
|
-
stderrTruncated: boolean;
|
|
32
|
-
stdoutOriginalLength: number;
|
|
33
|
-
stderrOriginalLength: number;
|
|
34
|
-
error?: undefined;
|
|
35
|
-
}>;
|
|
8
|
+
export declare function createExecuteCommandTool({ service, datasetId, sandbox }: ExecuteCommandToolParams): any;
|
|
36
9
|
export {};
|
|
37
10
|
//# sourceMappingURL=executeCommand.tool.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"executeCommand.tool.d.ts","sourceRoot":"","sources":["../src/executeCommand.tool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAA;AAEzC,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAO1C,UAAU,wBAAwB;IAC9B,OAAO,EAAE,cAAc,CAAA;IACvB,SAAS,EAAE,MAAM,CAAA;IACjB,OAAO,EAAE,OAAO,CAAA;CACnB;AAED,wBAAgB,wBAAwB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,wBAAwB
|
|
1
|
+
{"version":3,"file":"executeCommand.tool.d.ts","sourceRoot":"","sources":["../src/executeCommand.tool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAA;AAEzC,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAA;AAO1C,UAAU,wBAAwB;IAC9B,OAAO,EAAE,cAAc,CAAA;IACvB,SAAS,EAAE,MAAM,CAAA;IACjB,OAAO,EAAE,OAAO,CAAA;CACnB;AAED,wBAAgB,wBAAwB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,EAAE,wBAAwB,OAkIjG"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"executeCommand.tool.js","sourceRoot":"","sources":["../src/executeCommand.tool.ts"],"names":[],"mappings":";;AAiBA,4DAkIC;AAnJD,2BAAyB;AACzB,6BAAuB;AAEvB,iDAAsD;AAGtD,8EAA8E;AAC9E,kFAAkF;AAClF,MAAM,gBAAgB,GAAG,KAAK,CAAA;AAC9B,MAAM,gBAAgB,GAAG,IAAI,CAAA;AAQ7B,SAAgB,wBAAwB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAA4B;IAC9F,OAAO,IAAA,SAAI,EAAC;QACR,WAAW,EAAE,4dAA4d;QACze,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;YAClB,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kRAAkR,CAAC;YACnT,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wIAAwI,CAAC;SAC5K,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,UAAU,
|
|
1
|
+
{"version":3,"file":"executeCommand.tool.js","sourceRoot":"","sources":["../src/executeCommand.tool.ts"],"names":[],"mappings":";;AAiBA,4DAkIC;AAnJD,2BAAyB;AACzB,6BAAuB;AAEvB,iDAAsD;AAGtD,8EAA8E;AAC9E,kFAAkF;AAClF,MAAM,gBAAgB,GAAG,KAAK,CAAA;AAC9B,MAAM,gBAAgB,GAAG,IAAI,CAAA;AAQ7B,SAAgB,wBAAwB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAA4B;IAC9F,OAAO,IAAA,SAAI,EAAC;QACR,WAAW,EAAE,4dAA4d;QACze,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;YAClB,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kRAAkR,CAAC;YACnT,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wIAAwI,CAAC;SAC5K,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,UAAU,EAA8C,EAAE,EAAE;YACtF,MAAM,IAAI,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAA;YAC1E,MAAM,WAAW,GAAG,IAAA,oCAAqB,EAAC,SAAS,CAAC,CAAA;YACpD,MAAM,UAAU,GAAG,GAAG,WAAW,IAAI,UAAU,IAAI,IAAI,KAAK,CAAA;YAE5D,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,4CAA4C,CAAC,CAAA;YAC9E,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,wBAAwB,CAAC,CAAA;YAC1D,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,aAAa,UAAU,EAAE,CAAC,CAAA;YAC3D,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,WAAW,UAAU,EAAE,CAAC,CAAA;YACzD,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,kBAAkB,UAAU,CAAC,MAAM,QAAQ,CAAC,CAAA;YAC7E,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,4CAA4C,CAAC,CAAA;YAE9E,IAAI,CAAC;gBACD,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC;wBACtB,IAAI,EAAE,UAAU;wBAChB,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC;qBAC5C,CAAC,CAAC,CAAA;gBAEH,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,wBAAwB,UAAU,EAAE,CAAC,CAAA;gBACtE,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,uBAAuB,UAAU,EAAE,CAAC,CAAA;gBAErE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC;oBACpC,GAAG,EAAE,QAAQ;oBACb,IAAI,EAAE,CAAC,UAAU,CAAC;iBACrB,CAAC,CAAA;gBAEF,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,MAAM,EAAE,CAAA;gBACvC,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,MAAM,EAAE,CAAA;gBACvC,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAA;gBAEhC,MAAM,MAAM,GAAG,SAAS,IAAI,EAAE,CAAA;gBAC9B,MAAM,MAAM,GAAG,SAAS,IAAI,EAAE,CAAA;gBAE9B,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,GAAG,gBAAgB,CAAA;gBAC1D,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,GAAG,gBAAgB,CAAA;gBAE1D,MAAM,YAAY,GAAG,iBAAiB,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAA;gBACnF,MAAM,YAAY,GAAG,iBAAiB,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAA;gBAEnF,IAAI,QAAQ,KAAK,CAAC,EAClB,CAAC;oBACG,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,qCAAqC,QAAQ,EAAE,CAAC,CAAA;oBACnF,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,WAAW,EAAE,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAA;oBAC/E,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,4CAA4C,CAAC,CAAA;oBAEhF,OAAO;wBACH,OAAO,EAAE,KAAK;wBACd,QAAQ;wBACR,MAAM,EAAE,YAAY;wBACpB,MAAM,EAAE,YAAY;wBACpB,UAAU,EAAE,UAAU;wBACtB,KAAK,EAAE,iCAAiC,QAAQ,EAAE;wBAClD,eAAe,EAAE,iBAAiB;wBAClC,eAAe,EAAE,iBAAiB;wBAClC,oBAAoB,EAAE,MAAM,CAAC,MAAM;wBACnC,oBAAoB,EAAE,MAAM,CAAC,MAAM;qBACtC,CAAA;gBACL,CAAC;gBAED,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,EACtF,CAAC;oBACG,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,2BAA2B,CAAC,CAAA;oBAC/D,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,WAAW,EAAE,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAA;oBAC/E,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,4CAA4C,CAAC,CAAA;oBAEhF,OAAO;wBACH,OAAO,EAAE,KAAK;wBACd,QAAQ;wBACR,MAAM,EAAE,YAAY;wBACpB,MAAM,EAAE,YAAY;wBACpB,UAAU,EAAE,UAAU;wBACtB,KAAK,EAAE,iCAAiC;wBACxC,eAAe,EAAE,iBAAiB;wBAClC,eAAe,EAAE,iBAAiB;wBAClC,oBAAoB,EAAE,MAAM,CAAC,MAAM;wBACnC,oBAAoB,EAAE,MAAM,CAAC,MAAM;qBACtC,CAAA;gBACL,CAAC;gBAED,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,mCAAmC,CAAC,CAAA;gBACrE,IAAI,MAAM,EACV,CAAC;oBACG,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,oBAAoB,MAAM,CAAC,MAAM,QAAQ,CAAC,CAAA;oBAC3E,IAAI,iBAAiB,EACrB,CAAC;wBACG,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,6BAA6B,gBAAgB,QAAQ,CAAC,CAAA;oBAC3F,CAAC;gBACL,CAAC;gBACD,OAAO,CAAC,GAAG,CAAC,YAAY,SAAS,4CAA4C,CAAC,CAAA;gBAE9E,OAAO;oBACH,OAAO,EAAE,IAAI;oBACb,QAAQ;oBACR,MAAM,EAAE,YAAY;oBACpB,MAAM,EAAE,YAAY;oBACpB,UAAU,EAAE,UAAU;oBACtB,OAAO,EAAE,+BAA+B;oBACxC,eAAe,EAAE,iBAAiB;oBAClC,eAAe,EAAE,iBAAiB;oBAClC,oBAAoB,EAAE,MAAM,CAAC,MAAM;oBACnC,oBAAoB,EAAE,MAAM,CAAC,MAAM;iBACtC,CAAA;YACL,CAAC;YACD,OAAO,KAAc,EAAE,CAAC;gBACpB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAA;gBAC7E,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,0BAA0B,EAAE,YAAY,CAAC,CAAA;gBAC5E,OAAO,CAAC,KAAK,CAAC,YAAY,SAAS,4CAA4C,CAAC,CAAA;gBAEhF,OAAO;oBACH,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,YAAY;oBACnB,MAAM,EAAE,EAAE;oBACV,MAAM,EAAE,EAAE;oBACV,QAAQ,EAAE,CAAC,CAAC;oBACZ,UAAU,EAAE,UAAU;oBACtB,eAAe,EAAE,KAAK;oBACtB,eAAe,EAAE,KAAK;oBACtB,oBAAoB,EAAE,CAAC;oBACvB,oBAAoB,EAAE,CAAC;iBAC1B,CAAA;YACL,CAAC;QACL,CAAC;KACJ,CAAC,CAAA;AACN,CAAC"}
|
|
@@ -1,7 +1,10 @@
|
|
|
1
|
-
import { AgentOptions } from "@ekairos/story";
|
|
2
1
|
import { Sandbox } from "@vercel/sandbox";
|
|
3
2
|
import { FilePreviewContext } from "./filepreview";
|
|
3
|
+
import { InstantAdminDatabase } from "@instantdb/admin";
|
|
4
4
|
import { DatasetService } from "../service";
|
|
5
|
+
import { SchemaOf } from "@ekairos/domain";
|
|
6
|
+
import { datasetDomain } from "../schema";
|
|
7
|
+
import { storyDomain } from "@ekairos/story";
|
|
5
8
|
export type FileDatasetContext = {
|
|
6
9
|
datasetId: string;
|
|
7
10
|
fileId: string;
|
|
@@ -17,12 +20,22 @@ export type FileDatasetContext = {
|
|
|
17
20
|
iterationCount: number;
|
|
18
21
|
filePreview?: FilePreviewContext;
|
|
19
22
|
};
|
|
20
|
-
|
|
23
|
+
type DatasetSchemaType = SchemaOf<typeof datasetDomain> & SchemaOf<typeof storyDomain>;
|
|
24
|
+
export type FileDatasetAgentParams = {
|
|
21
25
|
fileId: string;
|
|
22
26
|
instructions: string;
|
|
23
27
|
sandbox: Sandbox;
|
|
28
|
+
db: InstantAdminDatabase<DatasetSchemaType>;
|
|
29
|
+
};
|
|
30
|
+
export type FileDatasetAgentBuilder = {
|
|
31
|
+
build: () => any;
|
|
32
|
+
datasetId: string;
|
|
24
33
|
service: DatasetService;
|
|
25
|
-
}
|
|
34
|
+
};
|
|
35
|
+
/**
|
|
36
|
+
* Creates a FileDatasetAgent using the builder pattern
|
|
37
|
+
*/
|
|
38
|
+
export declare function createFileDatasetAgent(params: FileDatasetAgentParams): FileDatasetAgentBuilder;
|
|
26
39
|
export type DatasetResult = {
|
|
27
40
|
id: string;
|
|
28
41
|
status?: string;
|
|
@@ -40,12 +53,15 @@ export declare class FileDatasetAgent {
|
|
|
40
53
|
private sandbox;
|
|
41
54
|
private service;
|
|
42
55
|
private agentService;
|
|
56
|
+
private db;
|
|
43
57
|
constructor(params: {
|
|
44
58
|
fileId: string;
|
|
45
59
|
instructions: string;
|
|
46
60
|
sandbox: Sandbox;
|
|
61
|
+
db: InstantAdminDatabase<DatasetSchemaType>;
|
|
47
62
|
});
|
|
48
63
|
getDataset(): Promise<DatasetResult>;
|
|
49
64
|
followUp(datasetId: string, feedback: string): Promise<DatasetResult>;
|
|
50
65
|
}
|
|
66
|
+
export {};
|
|
51
67
|
//# sourceMappingURL=file-dataset.agent.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"file-dataset.agent.d.ts","sourceRoot":"","sources":["../../src/file/file-dataset.agent.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"file-dataset.agent.d.ts","sourceRoot":"","sources":["../../src/file/file-dataset.agent.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAA;AAMzC,OAAO,EAAuB,kBAAkB,EAAiC,MAAM,eAAe,CAAA;AACtG,OAAO,EAAM,oBAAoB,EAAE,MAAM,kBAAkB,CAAA;AAE3D,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAC3C,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAA;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAA;AACzC,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAA;AAE5C,MAAM,MAAM,kBAAkB,GAAG;IAC7B,SAAS,EAAE,MAAM,CAAA;IACjB,MAAM,EAAE,MAAM,CAAA;IACd,YAAY,EAAE,MAAM,CAAA;IACpB,aAAa,EAAE;QACX,QAAQ,EAAE,MAAM,CAAA;KACnB,CAAA;IACD,QAAQ,EAAE,GAAG,EAAE,CAAA;IACf,MAAM,EAAE,GAAG,GAAG,IAAI,CAAA;IAClB,IAAI,EAAE,GAAG,GAAG,IAAI,CAAA;IAChB,eAAe,EAAE,GAAG,GAAG,IAAI,CAAA;IAC3B,MAAM,EAAE,MAAM,EAAE,CAAA;IAChB,cAAc,EAAE,MAAM,CAAA;IACtB,WAAW,CAAC,EAAE,kBAAkB,CAAA;CACnC,CAAA;AAED,KAAK,iBAAiB,GAAG,QAAQ,CAAC,OAAO,aAAa,CAAC,GAAG,QAAQ,CAAC,OAAO,WAAW,CAAC,CAAC;AAEvF,MAAM,MAAM,sBAAsB,GAAG;IACjC,MAAM,EAAE,MAAM,CAAA;IACd,YAAY,EAAE,MAAM,CAAA;IACpB,OAAO,EAAE,OAAO,CAAA;IAChB,EAAE,EAAE,oBAAoB,CAAC,iBAAiB,CAAC,CAAA;CAC9C,CAAA;AA4ED,MAAM,MAAM,uBAAuB,GAAG;IAClC,KAAK,EAAE,MAAM,GAAG,CAAA;IAChB,SAAS,EAAE,MAAM,CAAA;IACjB,OAAO,EAAE,cAAc,CAAA;CAC1B,CAAA;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,sBAAsB,GAAG,uBAAuB,CA4F9F;AAED,MAAM,MAAM,aAAa,GAAG;IACxB,EAAE,EAAE,MAAM,CAAA;IACV,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,MAAM,CAAC,EAAE,GAAG,CAAA;IACZ,QAAQ,CAAC,EAAE,GAAG,CAAA;IACd,mBAAmB,CAAC,EAAE,MAAM,CAAA;IAC5B,uBAAuB,CAAC,EAAE,MAAM,CAAA;IAChC,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,SAAS,CAAC,EAAE,MAAM,CAAA;CACrB,CAAA;AAED,qBAAa,gBAAgB;IACzB,OAAO,CAAC,MAAM,CAAQ;IACtB,OAAO,CAAC,YAAY,CAAQ;IAC5B,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,OAAO,CAAgB;IAC/B,OAAO,CAAC,YAAY,CAAc;IAClC,OAAO,CAAC,EAAE,CAAyC;gBAEvC,MAAM,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,OAAO,CAAC;QAAC,EAAE,EAAE,oBAAoB,CAAC,iBAAiB,CAAC,CAAA;KAAE;IASrH,UAAU,IAAI,OAAO,CAAC,aAAa,CAAC;IAuDpC,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;CA4D9E"}
|
|
@@ -1,8 +1,8 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.FileDatasetAgent = void 0;
|
|
4
|
+
exports.createFileDatasetAgent = createFileDatasetAgent;
|
|
4
5
|
const story_1 = require("@ekairos/story");
|
|
5
|
-
const story_2 = require("@ekairos/story");
|
|
6
6
|
const generateSchema_tool_1 = require("./generateSchema.tool");
|
|
7
7
|
const completeDataset_tool_1 = require("../completeDataset.tool");
|
|
8
8
|
const executeCommand_tool_1 = require("../executeCommand.tool");
|
|
@@ -10,181 +10,165 @@ const clearDataset_tool_1 = require("../clearDataset.tool");
|
|
|
10
10
|
const prompts_1 = require("./prompts");
|
|
11
11
|
const filepreview_1 = require("./filepreview");
|
|
12
12
|
const admin_1 = require("@instantdb/admin");
|
|
13
|
-
const story_3 = require("@ekairos/story");
|
|
14
13
|
const datasetFiles_1 = require("../datasetFiles");
|
|
15
14
|
const service_1 = require("../service");
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
this.isSandboxInitialized = false;
|
|
20
|
-
this.sandboxFilePath = "";
|
|
21
|
-
this.datasetId = (0, admin_1.id)();
|
|
22
|
-
this.fileId = opts.fileId;
|
|
23
|
-
this.instructions = opts.instructions;
|
|
24
|
-
this.sandbox = opts.sandbox;
|
|
25
|
-
this.service = opts.service;
|
|
15
|
+
async function initializeSandbox(sandbox, datasetId, fileId, db, state) {
|
|
16
|
+
if (state.initialized) {
|
|
17
|
+
return state.filePath;
|
|
26
18
|
}
|
|
27
|
-
|
|
28
|
-
|
|
19
|
+
console.log(`[FileDatasetAgent ${datasetId}] Initializing sandbox...`);
|
|
20
|
+
await (0, filepreview_1.ensurePreviewScriptsAvailable)(sandbox);
|
|
21
|
+
console.log(`[FileDatasetAgent ${datasetId}] Installing Python dependencies...`);
|
|
22
|
+
const pipInstall = await sandbox.runCommand({
|
|
23
|
+
cmd: "python",
|
|
24
|
+
args: ["-m", "pip", "install", "pandas", "openpyxl", "--quiet", "--upgrade"],
|
|
25
|
+
});
|
|
26
|
+
const installStderr = await pipInstall.stderr();
|
|
27
|
+
if (installStderr && (installStderr.includes("ERROR") || installStderr.includes("FAILED"))) {
|
|
28
|
+
throw new Error(`pip install failed: ${installStderr.substring(0, 300)}`);
|
|
29
29
|
}
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
console.log(`[FileDatasetAgent ${this.datasetId}] Installing Python dependencies...`);
|
|
38
|
-
const pipInstall = await this.sandbox.runCommand({
|
|
39
|
-
cmd: "python",
|
|
40
|
-
args: ["-m", "pip", "install", "pandas", "openpyxl", "--quiet", "--upgrade"],
|
|
41
|
-
});
|
|
42
|
-
const installStderr = await pipInstall.stderr();
|
|
43
|
-
if (installStderr && (installStderr.includes("ERROR") || installStderr.includes("FAILED"))) {
|
|
44
|
-
throw new Error(`pip install failed: ${installStderr.substring(0, 300)}`);
|
|
45
|
-
}
|
|
46
|
-
console.log(`[FileDatasetAgent ${this.datasetId}] Fetching file from InstantDB...`);
|
|
47
|
-
const fileQuery = await this.db.query({
|
|
48
|
-
$files: { $: { where: { id: this.fileId }, limit: 1 } },
|
|
49
|
-
});
|
|
50
|
-
const fileRecord = fileQuery.$files?.[0];
|
|
51
|
-
if (!fileRecord || !fileRecord.url) {
|
|
52
|
-
throw new Error(`File not found: ${this.fileId}`);
|
|
53
|
-
}
|
|
54
|
-
console.log(`[FileDatasetAgent ${this.datasetId}] Creating dataset workstation...`);
|
|
55
|
-
const workstation = (0, datasetFiles_1.getDatasetWorkstation)(this.datasetId);
|
|
56
|
-
await this.sandbox.runCommand({
|
|
57
|
-
cmd: "mkdir",
|
|
58
|
-
args: ["-p", workstation],
|
|
59
|
-
});
|
|
60
|
-
const fileBuffer = await fetch(fileRecord.url).then((response) => response.arrayBuffer());
|
|
61
|
-
const fileName = fileRecord["content-disposition"];
|
|
62
|
-
const fileExtension = fileName.includes(".") ? fileName.substring(fileName.lastIndexOf(".")) : "";
|
|
63
|
-
const sandboxFilePath = `${workstation}/${this.fileId}${fileExtension}`;
|
|
64
|
-
await this.sandbox.writeFiles([
|
|
65
|
-
{
|
|
66
|
-
path: sandboxFilePath,
|
|
67
|
-
content: Buffer.from(fileBuffer),
|
|
68
|
-
},
|
|
69
|
-
]);
|
|
70
|
-
console.log(`[FileDatasetAgent ${this.datasetId}] ✅ Workstation created: ${workstation}`);
|
|
71
|
-
console.log(`[FileDatasetAgent ${this.datasetId}] ✅ File saved: ${sandboxFilePath}`);
|
|
72
|
-
this.sandboxFilePath = sandboxFilePath;
|
|
73
|
-
this.isSandboxInitialized = true;
|
|
74
|
-
return sandboxFilePath;
|
|
75
|
-
}
|
|
76
|
-
catch (error) {
|
|
77
|
-
const msg = error instanceof Error ? error.message : String(error);
|
|
78
|
-
console.error(`[FileDatasetAgent ${this.datasetId}] ❌ Failed to initialize sandbox:`, msg);
|
|
79
|
-
throw error;
|
|
80
|
-
}
|
|
30
|
+
console.log(`[FileDatasetAgent ${datasetId}] Fetching file from InstantDB...`);
|
|
31
|
+
const fileQuery = await db.query({
|
|
32
|
+
$files: { $: { where: { id: fileId }, limit: 1 } },
|
|
33
|
+
});
|
|
34
|
+
const fileRecord = fileQuery.$files?.[0];
|
|
35
|
+
if (!fileRecord || !fileRecord.url) {
|
|
36
|
+
throw new Error(`File not found: ${fileId}`);
|
|
81
37
|
}
|
|
82
|
-
|
|
83
|
-
|
|
38
|
+
console.log(`[FileDatasetAgent ${datasetId}] Creating dataset workstation...`);
|
|
39
|
+
const workstation = (0, datasetFiles_1.getDatasetWorkstation)(datasetId);
|
|
40
|
+
await sandbox.runCommand({
|
|
41
|
+
cmd: "mkdir",
|
|
42
|
+
args: ["-p", workstation],
|
|
43
|
+
});
|
|
44
|
+
const fileBuffer = await fetch(fileRecord.url).then((response) => response.arrayBuffer());
|
|
45
|
+
const fileName = fileRecord["content-disposition"];
|
|
46
|
+
const fileExtension = fileName.includes(".") ? fileName.substring(fileName.lastIndexOf(".")) : "";
|
|
47
|
+
const sandboxFilePath = `${workstation}/${fileId}${fileExtension}`;
|
|
48
|
+
await sandbox.writeFiles([
|
|
49
|
+
{
|
|
50
|
+
path: sandboxFilePath,
|
|
51
|
+
content: Buffer.from(fileBuffer),
|
|
52
|
+
},
|
|
53
|
+
]);
|
|
54
|
+
console.log(`[FileDatasetAgent ${datasetId}] ✅ Workstation created: ${workstation}`);
|
|
55
|
+
console.log(`[FileDatasetAgent ${datasetId}] ✅ File saved: ${sandboxFilePath}`);
|
|
56
|
+
state.filePath = sandboxFilePath;
|
|
57
|
+
state.initialized = true;
|
|
58
|
+
return sandboxFilePath;
|
|
59
|
+
}
|
|
60
|
+
/**
|
|
61
|
+
* Creates a FileDatasetAgent using the builder pattern
|
|
62
|
+
*/
|
|
63
|
+
function createFileDatasetAgent(params) {
|
|
64
|
+
const { fileId, instructions, sandbox, db } = params;
|
|
65
|
+
const datasetId = (0, admin_1.id)();
|
|
66
|
+
const service = new service_1.DatasetService(db);
|
|
67
|
+
const sandboxState = { initialized: false, filePath: "" };
|
|
68
|
+
const builder = (0, story_1.createAgent)(db)
|
|
69
|
+
.context(async (stored) => {
|
|
70
|
+
const previous = stored?.content;
|
|
71
|
+
if (previous && previous.datasetId === datasetId) {
|
|
72
|
+
return previous;
|
|
73
|
+
}
|
|
74
|
+
const sandboxFilePath = await initializeSandbox(sandbox, datasetId, fileId, db, sandboxState);
|
|
84
75
|
let filePreview = undefined;
|
|
85
76
|
try {
|
|
86
|
-
filePreview = await (0, filepreview_1.generateFilePreview)(
|
|
77
|
+
filePreview = await (0, filepreview_1.generateFilePreview)(sandbox, sandboxFilePath, datasetId);
|
|
87
78
|
}
|
|
88
79
|
catch (error) {
|
|
89
|
-
console.error(`[Dataset ${
|
|
80
|
+
console.error(`[Dataset ${datasetId}] Failed to generate preview:`, error);
|
|
90
81
|
}
|
|
91
82
|
let schema = null;
|
|
92
|
-
const datasetResult = await
|
|
83
|
+
const datasetResult = await service.getDatasetById(datasetId);
|
|
93
84
|
if (datasetResult.ok && datasetResult.data.schema) {
|
|
94
85
|
schema = datasetResult.data.schema;
|
|
95
|
-
console.log(`[FileDatasetAgent ${
|
|
86
|
+
console.log(`[FileDatasetAgent ${datasetId}] ✅ Schema loaded from database`);
|
|
96
87
|
}
|
|
97
88
|
else {
|
|
98
|
-
console.log(`[FileDatasetAgent ${
|
|
89
|
+
console.log(`[FileDatasetAgent ${datasetId}] ℹ️ No schema found in database yet`);
|
|
99
90
|
}
|
|
100
91
|
return {
|
|
101
|
-
datasetId
|
|
102
|
-
fileId
|
|
103
|
-
instructions
|
|
104
|
-
sandboxConfig: {
|
|
105
|
-
filePath: sandboxFilePath,
|
|
106
|
-
},
|
|
92
|
+
datasetId,
|
|
93
|
+
fileId,
|
|
94
|
+
instructions,
|
|
95
|
+
sandboxConfig: { filePath: sandboxFilePath },
|
|
107
96
|
analysis: [],
|
|
108
|
-
schema
|
|
97
|
+
schema,
|
|
109
98
|
plan: null,
|
|
110
99
|
executionResult: null,
|
|
111
100
|
errors: [],
|
|
112
101
|
iterationCount: 0,
|
|
113
102
|
filePreview,
|
|
114
103
|
};
|
|
115
|
-
}
|
|
116
|
-
|
|
117
|
-
|
|
118
|
-
console.log(`[FileDatasetAgent ${
|
|
119
|
-
console.log(`[FileDatasetAgent ${
|
|
120
|
-
console.log(`[FileDatasetAgent ${
|
|
121
|
-
|
|
122
|
-
|
|
104
|
+
})
|
|
105
|
+
.systemPrompt(async ({ content }) => {
|
|
106
|
+
const ctx = content;
|
|
107
|
+
console.log(`[FileDatasetAgent ${ctx.datasetId}] Building system prompt...`);
|
|
108
|
+
console.log(`[FileDatasetAgent ${ctx.datasetId}] Schema present: ${!!ctx.schema}`);
|
|
109
|
+
console.log(`[FileDatasetAgent ${ctx.datasetId}] ExecutionResult present: ${!!ctx.executionResult}`);
|
|
110
|
+
console.log(`[FileDatasetAgent ${ctx.datasetId}] Iteration count: ${ctx.iterationCount}`);
|
|
111
|
+
const prompt = (0, prompts_1.buildFileDatasetPrompt)(ctx);
|
|
112
|
+
console.log(`[FileDatasetAgent ${ctx.datasetId}] Prompt length: ${prompt.length} chars`);
|
|
123
113
|
return prompt;
|
|
124
|
-
}
|
|
125
|
-
|
|
126
|
-
const ctx =
|
|
114
|
+
})
|
|
115
|
+
.tools(async ({ content }, dataStream) => {
|
|
116
|
+
const ctx = content;
|
|
127
117
|
return {
|
|
128
118
|
executeCommand: (0, executeCommand_tool_1.createExecuteCommandTool)({
|
|
129
|
-
service
|
|
119
|
+
service,
|
|
130
120
|
datasetId: ctx.datasetId,
|
|
131
|
-
sandbox
|
|
121
|
+
sandbox,
|
|
132
122
|
}),
|
|
133
123
|
generateSchema: (0, generateSchema_tool_1.createGenerateSchemaTool)({
|
|
134
|
-
service
|
|
124
|
+
service,
|
|
135
125
|
datasetId: ctx.datasetId,
|
|
136
|
-
sandbox
|
|
137
|
-
fileId
|
|
126
|
+
sandbox,
|
|
127
|
+
fileId,
|
|
138
128
|
}),
|
|
139
129
|
completeDataset: (0, completeDataset_tool_1.createCompleteDatasetTool)({
|
|
140
|
-
service
|
|
130
|
+
service,
|
|
141
131
|
datasetId: ctx.datasetId,
|
|
142
|
-
sandbox
|
|
132
|
+
sandbox,
|
|
143
133
|
}),
|
|
144
134
|
clearDataset: (0, clearDataset_tool_1.createClearDatasetTool)({
|
|
145
|
-
service
|
|
135
|
+
service,
|
|
146
136
|
datasetId: ctx.datasetId,
|
|
147
|
-
sandbox
|
|
137
|
+
sandbox,
|
|
148
138
|
}),
|
|
149
139
|
};
|
|
150
|
-
}
|
|
151
|
-
|
|
152
|
-
|
|
153
|
-
|
|
154
|
-
|
|
155
|
-
|
|
156
|
-
|
|
157
|
-
|
|
158
|
-
return ["completeDataset"];
|
|
159
|
-
}
|
|
160
|
-
async onEnd(lastEvent) {
|
|
161
|
-
console.log(`[FileDatasetAgent ${this.datasetId}] On end called`);
|
|
162
|
-
return { end: false }; // dont stop on error, only when finished
|
|
163
|
-
}
|
|
164
|
-
async onToolCallExecuted(executionEvent) {
|
|
165
|
-
console.log(`[FileDatasetAgent ${this.datasetId}] Tool call executed: ${executionEvent.toolCall.name}`);
|
|
166
|
-
}
|
|
140
|
+
})
|
|
141
|
+
.includeBaseTools({ createMessage: false, requestDirection: false, end: false })
|
|
142
|
+
.model("gpt-5-codex");
|
|
143
|
+
return {
|
|
144
|
+
build: () => builder.build(),
|
|
145
|
+
datasetId,
|
|
146
|
+
service,
|
|
147
|
+
};
|
|
167
148
|
}
|
|
168
149
|
class FileDatasetAgent {
|
|
169
150
|
constructor(params) {
|
|
170
151
|
this.fileId = params.fileId;
|
|
171
152
|
this.instructions = params.instructions;
|
|
172
153
|
this.sandbox = params.sandbox;
|
|
173
|
-
this.
|
|
174
|
-
this.
|
|
154
|
+
this.db = params.db;
|
|
155
|
+
this.service = new service_1.DatasetService(params.db);
|
|
156
|
+
this.agentService = new story_1.AgentService(params.db);
|
|
175
157
|
}
|
|
176
158
|
async getDataset() {
|
|
177
|
-
const
|
|
159
|
+
const agentResult = createFileDatasetAgent({
|
|
178
160
|
fileId: this.fileId,
|
|
179
161
|
instructions: this.instructions,
|
|
180
162
|
sandbox: this.sandbox,
|
|
181
|
-
|
|
163
|
+
db: this.db,
|
|
182
164
|
});
|
|
183
|
-
const
|
|
165
|
+
const agent = agentResult.build();
|
|
166
|
+
const datasetId = agentResult.datasetId;
|
|
167
|
+
const service = agentResult.service;
|
|
184
168
|
const userEvent = {
|
|
185
169
|
id: (0, admin_1.id)(),
|
|
186
|
-
type:
|
|
187
|
-
channel:
|
|
170
|
+
type: story_1.USER_MESSAGE_TYPE,
|
|
171
|
+
channel: story_1.WEB_CHANNEL,
|
|
188
172
|
content: {
|
|
189
173
|
parts: [
|
|
190
174
|
{
|
|
@@ -195,13 +179,13 @@ class FileDatasetAgent {
|
|
|
195
179
|
},
|
|
196
180
|
createdAt: new Date().toISOString(),
|
|
197
181
|
};
|
|
198
|
-
const reaction = await
|
|
182
|
+
const reaction = await agent.progressStream(userEvent, null);
|
|
199
183
|
const stream = reaction.stream;
|
|
200
184
|
const streamResult = await this.agentService.readEventStream(stream);
|
|
201
185
|
if (streamResult.persistedEvent?.status !== "completed") {
|
|
202
186
|
throw new Error(`Dataset generation failed with status: ${streamResult.persistedEvent?.status}`);
|
|
203
187
|
}
|
|
204
|
-
const datasetResult = await
|
|
188
|
+
const datasetResult = await service.getDatasetById(datasetId);
|
|
205
189
|
if (!datasetResult.ok) {
|
|
206
190
|
throw new Error(datasetResult.error);
|
|
207
191
|
}
|
|
@@ -219,16 +203,18 @@ class FileDatasetAgent {
|
|
|
219
203
|
};
|
|
220
204
|
}
|
|
221
205
|
async followUp(datasetId, feedback) {
|
|
222
|
-
const
|
|
206
|
+
const agentResult = createFileDatasetAgent({
|
|
223
207
|
fileId: this.fileId,
|
|
224
208
|
instructions: this.instructions,
|
|
225
209
|
sandbox: this.sandbox,
|
|
226
|
-
|
|
210
|
+
db: this.db,
|
|
227
211
|
});
|
|
212
|
+
const agent = agentResult.build();
|
|
213
|
+
const service = agentResult.service;
|
|
228
214
|
const userEvent = {
|
|
229
215
|
id: (0, admin_1.id)(),
|
|
230
|
-
type:
|
|
231
|
-
channel:
|
|
216
|
+
type: story_1.USER_MESSAGE_TYPE,
|
|
217
|
+
channel: story_1.WEB_CHANNEL,
|
|
232
218
|
content: {
|
|
233
219
|
parts: [
|
|
234
220
|
{
|
|
@@ -244,13 +230,13 @@ class FileDatasetAgent {
|
|
|
244
230
|
throw new Error(contextResult.error);
|
|
245
231
|
}
|
|
246
232
|
const contextId = contextResult.data.id;
|
|
247
|
-
const reaction = await
|
|
233
|
+
const reaction = await agent.progressStream(userEvent, { id: contextId });
|
|
248
234
|
const stream = reaction.stream;
|
|
249
235
|
const streamResult = await this.agentService.readEventStream(stream);
|
|
250
236
|
if (streamResult.persistedEvent?.status !== "completed") {
|
|
251
237
|
throw new Error(`Dataset iteration failed with status: ${streamResult.persistedEvent?.status}`);
|
|
252
238
|
}
|
|
253
|
-
const datasetResult = await
|
|
239
|
+
const datasetResult = await service.getDatasetById(datasetId);
|
|
254
240
|
if (!datasetResult.ok) {
|
|
255
241
|
throw new Error(datasetResult.error);
|
|
256
242
|
}
|