0pflow 0.1.0-dev.00aaa03
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/__tests__/agent-parser.test.d.ts +2 -0
- package/dist/__tests__/agent-parser.test.d.ts.map +1 -0
- package/dist/__tests__/agent-parser.test.js +64 -0
- package/dist/__tests__/agent-parser.test.js.map +1 -0
- package/dist/__tests__/agent.e2e.test.d.ts +2 -0
- package/dist/__tests__/agent.e2e.test.d.ts.map +1 -0
- package/dist/__tests__/agent.e2e.test.js +90 -0
- package/dist/__tests__/agent.e2e.test.js.map +1 -0
- package/dist/__tests__/agent.test.d.ts +2 -0
- package/dist/__tests__/agent.test.d.ts.map +1 -0
- package/dist/__tests__/agent.test.js +30 -0
- package/dist/__tests__/agent.test.js.map +1 -0
- package/dist/__tests__/context.test.d.ts +2 -0
- package/dist/__tests__/context.test.d.ts.map +1 -0
- package/dist/__tests__/context.test.js +53 -0
- package/dist/__tests__/context.test.js.map +1 -0
- package/dist/__tests__/dbos.test.d.ts +2 -0
- package/dist/__tests__/dbos.test.d.ts.map +1 -0
- package/dist/__tests__/dbos.test.js +35 -0
- package/dist/__tests__/dbos.test.js.map +1 -0
- package/dist/__tests__/factory.test.d.ts +2 -0
- package/dist/__tests__/factory.test.d.ts.map +1 -0
- package/dist/__tests__/factory.test.js +126 -0
- package/dist/__tests__/factory.test.js.map +1 -0
- package/dist/__tests__/integration.e2e.test.d.ts +2 -0
- package/dist/__tests__/integration.e2e.test.d.ts.map +1 -0
- package/dist/__tests__/integration.e2e.test.js +127 -0
- package/dist/__tests__/integration.e2e.test.js.map +1 -0
- package/dist/__tests__/integration.test.d.ts +2 -0
- package/dist/__tests__/integration.test.d.ts.map +1 -0
- package/dist/__tests__/integration.test.js +103 -0
- package/dist/__tests__/integration.test.js.map +1 -0
- package/dist/__tests__/model-config.test.d.ts +2 -0
- package/dist/__tests__/model-config.test.d.ts.map +1 -0
- package/dist/__tests__/model-config.test.js +48 -0
- package/dist/__tests__/model-config.test.js.map +1 -0
- package/dist/__tests__/node-registry.test.d.ts +2 -0
- package/dist/__tests__/node-registry.test.d.ts.map +1 -0
- package/dist/__tests__/node-registry.test.js +63 -0
- package/dist/__tests__/node-registry.test.js.map +1 -0
- package/dist/__tests__/node.test.d.ts +2 -0
- package/dist/__tests__/node.test.d.ts.map +1 -0
- package/dist/__tests__/node.test.js +48 -0
- package/dist/__tests__/node.test.js.map +1 -0
- package/dist/__tests__/registry.test.d.ts +2 -0
- package/dist/__tests__/registry.test.d.ts.map +1 -0
- package/dist/__tests__/registry.test.js +62 -0
- package/dist/__tests__/registry.test.js.map +1 -0
- package/dist/__tests__/web-tool.test.d.ts +2 -0
- package/dist/__tests__/web-tool.test.d.ts.map +1 -0
- package/dist/__tests__/web-tool.test.js +99 -0
- package/dist/__tests__/web-tool.test.js.map +1 -0
- package/dist/__tests__/workflow.test.d.ts +2 -0
- package/dist/__tests__/workflow.test.d.ts.map +1 -0
- package/dist/__tests__/workflow.test.js +49 -0
- package/dist/__tests__/workflow.test.js.map +1 -0
- package/dist/agent.d.ts +46 -0
- package/dist/agent.d.ts.map +1 -0
- package/dist/agent.js +75 -0
- package/dist/agent.js.map +1 -0
- package/dist/cli/__tests__/discovery.test.d.ts +2 -0
- package/dist/cli/__tests__/discovery.test.d.ts.map +1 -0
- package/dist/cli/__tests__/discovery.test.js +26 -0
- package/dist/cli/__tests__/discovery.test.js.map +1 -0
- package/dist/cli/__tests__/env.test.d.ts +2 -0
- package/dist/cli/__tests__/env.test.d.ts.map +1 -0
- package/dist/cli/__tests__/env.test.js +32 -0
- package/dist/cli/__tests__/env.test.js.map +1 -0
- package/dist/cli/__tests__/runs.test.d.ts +2 -0
- package/dist/cli/__tests__/runs.test.d.ts.map +1 -0
- package/dist/cli/__tests__/runs.test.js +46 -0
- package/dist/cli/__tests__/runs.test.js.map +1 -0
- package/dist/cli/app.d.ts +6 -0
- package/dist/cli/app.d.ts.map +1 -0
- package/dist/cli/app.js +21 -0
- package/dist/cli/app.js.map +1 -0
- package/dist/cli/discovery.d.ts +24 -0
- package/dist/cli/discovery.d.ts.map +1 -0
- package/dist/cli/discovery.js +88 -0
- package/dist/cli/discovery.js.map +1 -0
- package/dist/cli/env.d.ts +15 -0
- package/dist/cli/env.d.ts.map +1 -0
- package/dist/cli/env.js +54 -0
- package/dist/cli/env.js.map +1 -0
- package/dist/cli/index.d.ts +3 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +423 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/install.d.ts +54 -0
- package/dist/cli/install.d.ts.map +1 -0
- package/dist/cli/install.js +217 -0
- package/dist/cli/install.js.map +1 -0
- package/dist/cli/mcp/config.d.ts +5 -0
- package/dist/cli/mcp/config.d.ts.map +1 -0
- package/dist/cli/mcp/config.js +14 -0
- package/dist/cli/mcp/config.js.map +1 -0
- package/dist/cli/mcp/lib/templates.d.ts +12 -0
- package/dist/cli/mcp/lib/templates.d.ts.map +1 -0
- package/dist/cli/mcp/lib/templates.js +77 -0
- package/dist/cli/mcp/lib/templates.js.map +1 -0
- package/dist/cli/mcp/server.d.ts +5 -0
- package/dist/cli/mcp/server.d.ts.map +1 -0
- package/dist/cli/mcp/server.js +15 -0
- package/dist/cli/mcp/server.js.map +1 -0
- package/dist/cli/mcp/serverInfo.d.ts +7 -0
- package/dist/cli/mcp/serverInfo.d.ts.map +1 -0
- package/dist/cli/mcp/serverInfo.js +7 -0
- package/dist/cli/mcp/serverInfo.js.map +1 -0
- package/dist/cli/mcp/tools/createApp.d.ts +15 -0
- package/dist/cli/mcp/tools/createApp.d.ts.map +1 -0
- package/dist/cli/mcp/tools/createApp.js +81 -0
- package/dist/cli/mcp/tools/createApp.js.map +1 -0
- package/dist/cli/mcp/tools/createDatabase.d.ts +14 -0
- package/dist/cli/mcp/tools/createDatabase.d.ts.map +1 -0
- package/dist/cli/mcp/tools/createDatabase.js +66 -0
- package/dist/cli/mcp/tools/createDatabase.js.map +1 -0
- package/dist/cli/mcp/tools/index.d.ts +24 -0
- package/dist/cli/mcp/tools/index.d.ts.map +1 -0
- package/dist/cli/mcp/tools/index.js +11 -0
- package/dist/cli/mcp/tools/index.js.map +1 -0
- package/dist/cli/mcp/tools/setupAppSchema.d.ts +17 -0
- package/dist/cli/mcp/tools/setupAppSchema.d.ts.map +1 -0
- package/dist/cli/mcp/tools/setupAppSchema.js +157 -0
- package/dist/cli/mcp/tools/setupAppSchema.js.map +1 -0
- package/dist/cli/mcp/types.d.ts +3 -0
- package/dist/cli/mcp/types.d.ts.map +1 -0
- package/dist/cli/mcp/types.js +2 -0
- package/dist/cli/mcp/types.js.map +1 -0
- package/dist/cli/runs.d.ts +28 -0
- package/dist/cli/runs.d.ts.map +1 -0
- package/dist/cli/runs.js +70 -0
- package/dist/cli/runs.js.map +1 -0
- package/dist/cli/trace.d.ts +31 -0
- package/dist/cli/trace.d.ts.map +1 -0
- package/dist/cli/trace.js +264 -0
- package/dist/cli/trace.js.map +1 -0
- package/dist/context.d.ts +13 -0
- package/dist/context.d.ts.map +1 -0
- package/dist/context.js +25 -0
- package/dist/context.js.map +1 -0
- package/dist/dbos.d.ts +15 -0
- package/dist/dbos.d.ts.map +1 -0
- package/dist/dbos.js +27 -0
- package/dist/dbos.js.map +1 -0
- package/dist/factory.d.ts +6 -0
- package/dist/factory.d.ts.map +1 -0
- package/dist/factory.js +78 -0
- package/dist/factory.js.map +1 -0
- package/dist/index.d.ts +20 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +18 -0
- package/dist/index.js.map +1 -0
- package/dist/node.d.ts +19 -0
- package/dist/node.d.ts.map +1 -0
- package/dist/node.js +16 -0
- package/dist/node.js.map +1 -0
- package/dist/nodes/agent/executor.d.ts +70 -0
- package/dist/nodes/agent/executor.d.ts.map +1 -0
- package/dist/nodes/agent/executor.js +230 -0
- package/dist/nodes/agent/executor.js.map +1 -0
- package/dist/nodes/agent/index.d.ts +7 -0
- package/dist/nodes/agent/index.d.ts.map +1 -0
- package/dist/nodes/agent/index.js +5 -0
- package/dist/nodes/agent/index.js.map +1 -0
- package/dist/nodes/agent/model-config.d.ts +42 -0
- package/dist/nodes/agent/model-config.d.ts.map +1 -0
- package/dist/nodes/agent/model-config.js +58 -0
- package/dist/nodes/agent/model-config.js.map +1 -0
- package/dist/nodes/agent/parser.d.ts +22 -0
- package/dist/nodes/agent/parser.d.ts.map +1 -0
- package/dist/nodes/agent/parser.js +42 -0
- package/dist/nodes/agent/parser.js.map +1 -0
- package/dist/nodes/builtin/index.d.ts +9 -0
- package/dist/nodes/builtin/index.d.ts.map +1 -0
- package/dist/nodes/builtin/index.js +10 -0
- package/dist/nodes/builtin/index.js.map +1 -0
- package/dist/nodes/builtin/web.d.ts +25 -0
- package/dist/nodes/builtin/web.d.ts.map +1 -0
- package/dist/nodes/builtin/web.js +86 -0
- package/dist/nodes/builtin/web.js.map +1 -0
- package/dist/nodes/registry.d.ts +32 -0
- package/dist/nodes/registry.d.ts.map +1 -0
- package/dist/nodes/registry.js +58 -0
- package/dist/nodes/registry.js.map +1 -0
- package/dist/registry.d.ts +31 -0
- package/dist/registry.d.ts.map +1 -0
- package/dist/registry.js +44 -0
- package/dist/registry.js.map +1 -0
- package/dist/types.d.ts +63 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/workflow.d.ts +26 -0
- package/dist/workflow.d.ts.map +1 -0
- package/dist/workflow.js +50 -0
- package/dist/workflow.js.map +1 -0
- package/package.json +52 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"integration.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/integration.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,103 @@
|
|
|
1
|
+
// packages/core/src/__tests__/integration.test.ts
|
|
2
|
+
import { describe, it, expect, vi, beforeEach } from "vitest";
|
|
3
|
+
import { z } from "zod";
|
|
4
|
+
import { create0pflow, Workflow, Node } from "../index.js";
|
|
5
|
+
// Mock DBOS
|
|
6
|
+
vi.mock("@dbos-inc/dbos-sdk", () => ({
|
|
7
|
+
DBOS: {
|
|
8
|
+
setConfig: vi.fn(),
|
|
9
|
+
launch: vi.fn().mockResolvedValue(undefined),
|
|
10
|
+
shutdown: vi.fn().mockResolvedValue(undefined),
|
|
11
|
+
runStep: vi.fn().mockImplementation(async (fn) => fn()),
|
|
12
|
+
registerWorkflow: vi.fn().mockImplementation((fn) => fn),
|
|
13
|
+
logger: {
|
|
14
|
+
info: vi.fn(),
|
|
15
|
+
warn: vi.fn(),
|
|
16
|
+
error: vi.fn(),
|
|
17
|
+
debug: vi.fn(),
|
|
18
|
+
},
|
|
19
|
+
},
|
|
20
|
+
}));
|
|
21
|
+
describe("0pflow integration", () => {
|
|
22
|
+
beforeEach(() => {
|
|
23
|
+
vi.clearAllMocks();
|
|
24
|
+
});
|
|
25
|
+
it("complete workflow with multiple nodes", async () => {
|
|
26
|
+
// Define nodes
|
|
27
|
+
const fetchData = Node.create({
|
|
28
|
+
name: "fetch-data",
|
|
29
|
+
description: "Fetches data from a URL",
|
|
30
|
+
inputSchema: z.object({ url: z.string() }),
|
|
31
|
+
outputSchema: z.object({ title: z.string(), body: z.string() }),
|
|
32
|
+
execute: async (_ctx, inputs) => ({
|
|
33
|
+
title: `Page: ${inputs.url}`,
|
|
34
|
+
body: "Content here",
|
|
35
|
+
}),
|
|
36
|
+
});
|
|
37
|
+
const summarize = Node.create({
|
|
38
|
+
name: "summarize",
|
|
39
|
+
description: "Summarizes text",
|
|
40
|
+
inputSchema: z.object({ text: z.string() }),
|
|
41
|
+
outputSchema: z.object({ summary: z.string() }),
|
|
42
|
+
execute: async (_ctx, inputs) => ({
|
|
43
|
+
summary: `Summary of: ${inputs.text.slice(0, 20)}...`,
|
|
44
|
+
}),
|
|
45
|
+
});
|
|
46
|
+
// Define workflow
|
|
47
|
+
const researchWorkflow = Workflow.create({
|
|
48
|
+
name: "research",
|
|
49
|
+
description: "Researches a URL",
|
|
50
|
+
version: 1,
|
|
51
|
+
inputSchema: z.object({ url: z.string() }),
|
|
52
|
+
outputSchema: z.object({ title: z.string(), summary: z.string() }),
|
|
53
|
+
run: async (ctx, inputs) => {
|
|
54
|
+
ctx.log("Starting research workflow");
|
|
55
|
+
const data = await ctx.run(fetchData, { url: inputs.url });
|
|
56
|
+
ctx.log(`Fetched: ${data.title}`);
|
|
57
|
+
const result = await ctx.run(summarize, { text: data.body });
|
|
58
|
+
ctx.log("Summarization complete");
|
|
59
|
+
return { title: data.title, summary: result.summary };
|
|
60
|
+
},
|
|
61
|
+
});
|
|
62
|
+
// Create instance
|
|
63
|
+
const pflow = await create0pflow({
|
|
64
|
+
databaseUrl: "postgres://localhost/test",
|
|
65
|
+
workflows: { research: researchWorkflow },
|
|
66
|
+
nodes: { "fetch-data": fetchData, summarize },
|
|
67
|
+
});
|
|
68
|
+
// Execute
|
|
69
|
+
const result = await pflow.triggerWorkflow("research", {
|
|
70
|
+
url: "https://example.com",
|
|
71
|
+
});
|
|
72
|
+
expect(result).toEqual({
|
|
73
|
+
title: "Page: https://example.com",
|
|
74
|
+
summary: "Summary of: Content here...",
|
|
75
|
+
});
|
|
76
|
+
});
|
|
77
|
+
it("nested workflow calls", async () => {
|
|
78
|
+
const innerWorkflow = Workflow.create({
|
|
79
|
+
name: "inner",
|
|
80
|
+
description: "Inner workflow that doubles",
|
|
81
|
+
version: 1,
|
|
82
|
+
inputSchema: z.object({ value: z.number() }),
|
|
83
|
+
run: async (_ctx, inputs) => inputs.value * 2,
|
|
84
|
+
});
|
|
85
|
+
const outerWorkflow = Workflow.create({
|
|
86
|
+
name: "outer",
|
|
87
|
+
description: "Outer workflow that calls inner",
|
|
88
|
+
version: 1,
|
|
89
|
+
inputSchema: z.object({ value: z.number() }),
|
|
90
|
+
run: async (ctx, inputs) => {
|
|
91
|
+
const doubled = await ctx.run(innerWorkflow, { value: inputs.value });
|
|
92
|
+
return doubled + 1;
|
|
93
|
+
},
|
|
94
|
+
});
|
|
95
|
+
const pflow = await create0pflow({
|
|
96
|
+
databaseUrl: "postgres://localhost/test",
|
|
97
|
+
workflows: { outer: outerWorkflow, inner: innerWorkflow },
|
|
98
|
+
});
|
|
99
|
+
const result = await pflow.triggerWorkflow("outer", { value: 5 });
|
|
100
|
+
expect(result).toBe(11); // (5 * 2) + 1
|
|
101
|
+
});
|
|
102
|
+
});
|
|
103
|
+
//# sourceMappingURL=integration.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"integration.test.js","sourceRoot":"","sources":["../../src/__tests__/integration.test.ts"],"names":[],"mappings":"AAAA,kDAAkD;AAClD,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AAE3D,YAAY;AACZ,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,IAAI,EAAE;QACJ,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;QAClB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAC5C,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAC9C,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;QACvD,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;QACxD,MAAM,EAAE;YACN,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;YACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;YACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;SACf;KACF;CACF,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,eAAe;QACf,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC;YAC5B,IAAI,EAAE,YAAY;YAClB,WAAW,EAAE,yBAAyB;YACtC,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,YAAY,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/D,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;gBAChC,KAAK,EAAE,SAAS,MAAM,CAAC,GAAG,EAAE;gBAC5B,IAAI,EAAE,cAAc;aACrB,CAAC;SACH,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC;YAC5B,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,iBAAiB;YAC9B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,YAAY,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/C,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;gBAChC,OAAO,EAAE,eAAe,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK;aACtD,CAAC;SACH,CAAC,CAAC;QAEH,kBAAkB;QAClB,MAAM,gBAAgB,GAAG,QAAQ,CAAC,MAAM,CAAC;YACvC,IAAI,EAAE,UAAU;YAChB,WAAW,EAAE,kBAAkB;YAC/B,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,YAAY,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAClE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE;gBACzB,GAAG,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;gBAEtC,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC;gBAC3D,GAAG,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;gBAElC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC7D,GAAG,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;gBAElC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;YACxD,CAAC;SACF,CAAC,CAAC;QAEH,kBAAkB;QAClB,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC;YAC/B,WAAW,EAAE,2BAA2B;YACxC,SAAS,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;YACzC,KAAK,EAAE,EAAE,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE;SAC9C,CAAC,CAAC;QAEH,UAAU;QACV,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,UAAU,EAAE;YACrD,GAAG,EAAE,qBAAqB;SAC3B,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,KAAK,EAAE,2BAA2B;YAClC,OAAO,EAAE,6BAA6B;SACvC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;QACrC,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC;YACpC,IAAI,EAAE,OAAO;YACb,WAAW,EAAE,6BAA6B;YAC1C,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAC5C,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC;SAC9C,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC;YACpC,IAAI,EAAE,OAAO;YACb,WAAW,EAAE,iCAAiC;YAC9C,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAC5C,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE;gBACzB,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;gBACtE,OAAO,OAAO,GAAG,CAAC,CAAC;YACrB,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC;YAC/B,WAAW,EAAE,2BAA2B;YACxC,SAAS,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,aAAa,EAAE;SAC1D,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,cAAc;IACzC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-config.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/model-config.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
// packages/core/src/__tests__/model-config.test.ts
|
|
2
|
+
import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
|
|
3
|
+
import { getDefaultModelConfig, parseModelString, } from "../nodes/agent/model-config.js";
|
|
4
|
+
describe("model-config", () => {
|
|
5
|
+
const originalEnv = process.env;
|
|
6
|
+
beforeEach(() => {
|
|
7
|
+
vi.resetModules();
|
|
8
|
+
process.env = { ...originalEnv };
|
|
9
|
+
});
|
|
10
|
+
afterEach(() => {
|
|
11
|
+
process.env = originalEnv;
|
|
12
|
+
});
|
|
13
|
+
describe("getDefaultModelConfig()", () => {
|
|
14
|
+
it("returns openai/gpt-4o by default", () => {
|
|
15
|
+
delete process.env.PFLOW_MODEL_PROVIDER;
|
|
16
|
+
delete process.env.PFLOW_MODEL_ID;
|
|
17
|
+
const config = getDefaultModelConfig();
|
|
18
|
+
expect(config.provider).toBe("openai");
|
|
19
|
+
expect(config.modelId).toBe("gpt-4o");
|
|
20
|
+
});
|
|
21
|
+
it("reads from environment variables", () => {
|
|
22
|
+
process.env.PFLOW_MODEL_PROVIDER = "anthropic";
|
|
23
|
+
process.env.PFLOW_MODEL_ID = "claude-3-opus";
|
|
24
|
+
const config = getDefaultModelConfig();
|
|
25
|
+
expect(config.provider).toBe("anthropic");
|
|
26
|
+
expect(config.modelId).toBe("claude-3-opus");
|
|
27
|
+
});
|
|
28
|
+
});
|
|
29
|
+
describe("parseModelString()", () => {
|
|
30
|
+
it("parses provider/model format", () => {
|
|
31
|
+
const result = parseModelString("anthropic/claude-3-opus");
|
|
32
|
+
expect(result.provider).toBe("anthropic");
|
|
33
|
+
expect(result.modelId).toBe("claude-3-opus");
|
|
34
|
+
});
|
|
35
|
+
it("parses model-only format", () => {
|
|
36
|
+
const result = parseModelString("gpt-4-turbo");
|
|
37
|
+
expect(result.provider).toBeUndefined();
|
|
38
|
+
expect(result.modelId).toBe("gpt-4-turbo");
|
|
39
|
+
});
|
|
40
|
+
it("handles models with slashes in name", () => {
|
|
41
|
+
// Only splits on first slash
|
|
42
|
+
const result = parseModelString("openai/gpt-4/vision");
|
|
43
|
+
expect(result.provider).toBe("openai");
|
|
44
|
+
expect(result.modelId).toBe("gpt-4/vision");
|
|
45
|
+
});
|
|
46
|
+
});
|
|
47
|
+
});
|
|
48
|
+
//# sourceMappingURL=model-config.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-config.test.js","sourceRoot":"","sources":["../../src/__tests__/model-config.test.ts"],"names":[],"mappings":"AAAA,mDAAmD;AACnD,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EACL,qBAAqB,EACrB,gBAAgB,GACjB,MAAM,gCAAgC,CAAC;AAExC,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC;IAEhC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,CAAC,GAAG,GAAG,EAAE,GAAG,WAAW,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,GAAG,GAAG,WAAW,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,OAAO,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;YACxC,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;YAElC,MAAM,MAAM,GAAG,qBAAqB,EAAE,CAAC;YAEvC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,WAAW,CAAC;YAC/C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,eAAe,CAAC;YAE7C,MAAM,MAAM,GAAG,qBAAqB,EAAE,CAAC;YAEvC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,MAAM,GAAG,gBAAgB,CAAC,yBAAyB,CAAC,CAAC;YAE3D,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,MAAM,GAAG,gBAAgB,CAAC,aAAa,CAAC,CAAC;YAE/C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,6BAA6B;YAC7B,MAAM,MAAM,GAAG,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;YAEvD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"node-registry.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/node-registry.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
// packages/core/src/__tests__/node-registry.test.ts
|
|
2
|
+
import { describe, it, expect } from "vitest";
|
|
3
|
+
import { z } from "zod";
|
|
4
|
+
import { NodeRegistry } from "../nodes/registry.js";
|
|
5
|
+
import { Node } from "../node.js";
|
|
6
|
+
describe("NodeRegistry", () => {
|
|
7
|
+
const customNode = Node.create({
|
|
8
|
+
name: "custom.echo",
|
|
9
|
+
description: "Echo input",
|
|
10
|
+
inputSchema: z.object({ msg: z.string() }),
|
|
11
|
+
execute: async (_ctx, { msg }) => msg,
|
|
12
|
+
});
|
|
13
|
+
it("includes built-in nodes by default", () => {
|
|
14
|
+
const registry = new NodeRegistry();
|
|
15
|
+
expect(registry.hasNode("web_read")).toBe(true);
|
|
16
|
+
});
|
|
17
|
+
it("can add user-defined nodes", () => {
|
|
18
|
+
const registry = new NodeRegistry({
|
|
19
|
+
userNodes: { "custom.echo": customNode },
|
|
20
|
+
});
|
|
21
|
+
expect(registry.hasNode("custom.echo")).toBe(true);
|
|
22
|
+
expect(registry.getNode("custom.echo")).toBe(customNode);
|
|
23
|
+
});
|
|
24
|
+
it("user nodes can override built-in nodes", () => {
|
|
25
|
+
const overrideNode = Node.create({
|
|
26
|
+
name: "web_read",
|
|
27
|
+
description: "Overridden web_read",
|
|
28
|
+
inputSchema: z.object({ url: z.string() }),
|
|
29
|
+
execute: async () => ({ custom: true }),
|
|
30
|
+
});
|
|
31
|
+
const registry = new NodeRegistry({
|
|
32
|
+
userNodes: { "web_read": overrideNode },
|
|
33
|
+
});
|
|
34
|
+
const node = registry.getNode("web_read");
|
|
35
|
+
expect(node?.description).toBe("Overridden web_read");
|
|
36
|
+
});
|
|
37
|
+
it("getNode returns undefined for missing nodes", () => {
|
|
38
|
+
const registry = new NodeRegistry();
|
|
39
|
+
expect(registry.getNode("nonexistent")).toBeUndefined();
|
|
40
|
+
});
|
|
41
|
+
it("getNodes returns multiple nodes", () => {
|
|
42
|
+
const registry = new NodeRegistry({
|
|
43
|
+
userNodes: { "custom.echo": customNode },
|
|
44
|
+
});
|
|
45
|
+
const nodes = registry.getNodes(["web_read", "custom.echo"]);
|
|
46
|
+
expect(nodes).toHaveLength(2);
|
|
47
|
+
expect(nodes.map((n) => n.name)).toContain("web_read");
|
|
48
|
+
expect(nodes.map((n) => n.name)).toContain("custom.echo");
|
|
49
|
+
});
|
|
50
|
+
it("getNodes throws for missing nodes", () => {
|
|
51
|
+
const registry = new NodeRegistry();
|
|
52
|
+
expect(() => registry.getNodes(["web_read", "missing.node"])).toThrow("Nodes not found: missing.node");
|
|
53
|
+
});
|
|
54
|
+
it("listNodes returns all node names", () => {
|
|
55
|
+
const registry = new NodeRegistry({
|
|
56
|
+
userNodes: { "custom.echo": customNode },
|
|
57
|
+
});
|
|
58
|
+
const names = registry.listNodes();
|
|
59
|
+
expect(names).toContain("web_read");
|
|
60
|
+
expect(names).toContain("custom.echo");
|
|
61
|
+
});
|
|
62
|
+
});
|
|
63
|
+
//# sourceMappingURL=node-registry.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"node-registry.test.js","sourceRoot":"","sources":["../../src/__tests__/node-registry.test.ts"],"names":[],"mappings":"AAAA,oDAAoD;AACpD,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC;QAC7B,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,YAAY;QACzB,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;QAC1C,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,GAAG;KACtC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;QACpC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC;YAChC,SAAS,EAAE,EAAE,aAAa,EAAE,UAAU,EAAE;SACzC,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC;YAC/B,IAAI,EAAE,UAAU;YAChB,WAAW,EAAE,qBAAqB;YAClC,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;SACxC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC;YAChC,SAAS,EAAE,EAAE,UAAU,EAAE,YAAY,EAAE;SACxC,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC1C,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;QACpC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC;YAChC,SAAS,EAAE,EAAE,aAAa,EAAE,UAAU,EAAE;SACzC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC,CAAC;QAC7D,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACvD,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;QACpC,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,OAAO,CACnE,+BAA+B,CAChC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC;YAChC,SAAS,EAAE,EAAE,aAAa,EAAE,UAAU,EAAE;SACzC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,QAAQ,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"node.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/node.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
// packages/core/src/__tests__/node.test.ts
|
|
2
|
+
import { describe, it, expect } from "vitest";
|
|
3
|
+
import { z } from "zod";
|
|
4
|
+
import { Node } from "../node.js";
|
|
5
|
+
import { createWorkflowContext } from "../context.js";
|
|
6
|
+
describe("Node.create()", () => {
|
|
7
|
+
it("creates a node with correct properties", () => {
|
|
8
|
+
const inputSchema = z.object({ url: z.string() });
|
|
9
|
+
const outputSchema = z.object({ status: z.number() });
|
|
10
|
+
const node = Node.create({
|
|
11
|
+
name: "test.fetch",
|
|
12
|
+
description: "Fetch something",
|
|
13
|
+
inputSchema,
|
|
14
|
+
outputSchema,
|
|
15
|
+
execute: async () => ({ status: 200 }),
|
|
16
|
+
});
|
|
17
|
+
expect(node.name).toBe("test.fetch");
|
|
18
|
+
expect(node.type).toBe("node");
|
|
19
|
+
expect(node.description).toBe("Fetch something");
|
|
20
|
+
expect(node.inputSchema).toBe(inputSchema);
|
|
21
|
+
expect(node.outputSchema).toBe(outputSchema);
|
|
22
|
+
});
|
|
23
|
+
it("executes with context and inputs", async () => {
|
|
24
|
+
const node = Node.create({
|
|
25
|
+
name: "test.number",
|
|
26
|
+
description: "Returns a number",
|
|
27
|
+
inputSchema: z.object({ value: z.number() }),
|
|
28
|
+
execute: async (_ctx, { value }) => ({ doubled: value * 2 }),
|
|
29
|
+
});
|
|
30
|
+
const ctx = createWorkflowContext();
|
|
31
|
+
// Valid input
|
|
32
|
+
const result = await node.execute(ctx, { value: 5 });
|
|
33
|
+
expect(result).toEqual({ doubled: 10 });
|
|
34
|
+
});
|
|
35
|
+
it("works without outputSchema", async () => {
|
|
36
|
+
const node = Node.create({
|
|
37
|
+
name: "test.simple",
|
|
38
|
+
description: "Simple node",
|
|
39
|
+
inputSchema: z.object({ msg: z.string() }),
|
|
40
|
+
execute: async (_ctx, { msg }) => `Hello, ${msg}`,
|
|
41
|
+
});
|
|
42
|
+
const ctx = createWorkflowContext();
|
|
43
|
+
expect(node.outputSchema).toBeUndefined();
|
|
44
|
+
const result = await node.execute(ctx, { msg: "world" });
|
|
45
|
+
expect(result).toBe("Hello, world");
|
|
46
|
+
});
|
|
47
|
+
});
|
|
48
|
+
//# sourceMappingURL=node.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"node.test.js","sourceRoot":"","sources":["../../src/__tests__/node.test.ts"],"names":[],"mappings":"AAAA,2CAA2C;AAC3C,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAClC,OAAO,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAEtD,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAClD,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAEtD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC;YACvB,IAAI,EAAE,YAAY;YAClB,WAAW,EAAE,iBAAiB;YAC9B,WAAW;YACX,YAAY;YACZ,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;SACvC,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACrC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC/B,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACjD,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC3C,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC;YACvB,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,kBAAkB;YAC/B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAC5C,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,GAAG,CAAC,EAAE,CAAC;SAC7D,CAAC,CAAC;QAEH,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QAEpC,cAAc;QACd,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC;YACvB,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,aAAa;YAC1B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,UAAU,GAAG,EAAE;SAClD,CAAC,CAAC;QAEH,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"registry.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/registry.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,62 @@
|
|
|
1
|
+
// packages/core/src/__tests__/registry.test.ts
|
|
2
|
+
import { describe, it, expect } from "vitest";
|
|
3
|
+
import { z } from "zod";
|
|
4
|
+
import { Registry } from "../registry.js";
|
|
5
|
+
import { Node } from "../node.js";
|
|
6
|
+
import { Workflow } from "../workflow.js";
|
|
7
|
+
describe("Registry", () => {
|
|
8
|
+
it("registers and retrieves workflows", () => {
|
|
9
|
+
const workflow = Workflow.create({
|
|
10
|
+
name: "test-workflow",
|
|
11
|
+
description: "A test workflow",
|
|
12
|
+
version: 1,
|
|
13
|
+
inputSchema: z.object({}),
|
|
14
|
+
run: async () => "done",
|
|
15
|
+
});
|
|
16
|
+
const registry = new Registry({
|
|
17
|
+
workflows: { "test-workflow": workflow },
|
|
18
|
+
});
|
|
19
|
+
expect(registry.getWorkflow("test-workflow")).toBe(workflow);
|
|
20
|
+
expect(registry.listWorkflows()).toEqual(["test-workflow"]);
|
|
21
|
+
});
|
|
22
|
+
it("registers and retrieves nodes", () => {
|
|
23
|
+
const node = Node.create({
|
|
24
|
+
name: "test-node",
|
|
25
|
+
description: "A test node",
|
|
26
|
+
inputSchema: z.object({}),
|
|
27
|
+
execute: async () => "done",
|
|
28
|
+
});
|
|
29
|
+
const registry = new Registry({
|
|
30
|
+
nodes: { "test-node": node },
|
|
31
|
+
});
|
|
32
|
+
expect(registry.getExecutable("test-node")).toBe(node);
|
|
33
|
+
});
|
|
34
|
+
it("returns undefined for unknown executables", () => {
|
|
35
|
+
const registry = new Registry({});
|
|
36
|
+
expect(registry.getWorkflow("unknown")).toBeUndefined();
|
|
37
|
+
expect(registry.getExecutable("unknown")).toBeUndefined();
|
|
38
|
+
});
|
|
39
|
+
it("lists all registered executables", () => {
|
|
40
|
+
const node = Node.create({
|
|
41
|
+
name: "my-node",
|
|
42
|
+
description: "My node",
|
|
43
|
+
inputSchema: z.object({}),
|
|
44
|
+
execute: async () => { },
|
|
45
|
+
});
|
|
46
|
+
const workflow = Workflow.create({
|
|
47
|
+
name: "my-workflow",
|
|
48
|
+
description: "My workflow",
|
|
49
|
+
version: 1,
|
|
50
|
+
inputSchema: z.object({}),
|
|
51
|
+
run: async () => { },
|
|
52
|
+
});
|
|
53
|
+
const registry = new Registry({
|
|
54
|
+
nodes: { "my-node": node },
|
|
55
|
+
workflows: { "my-workflow": workflow },
|
|
56
|
+
});
|
|
57
|
+
const all = registry.listAll();
|
|
58
|
+
expect(all).toContain("my-node");
|
|
59
|
+
expect(all).toContain("my-workflow");
|
|
60
|
+
});
|
|
61
|
+
});
|
|
62
|
+
//# sourceMappingURL=registry.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"registry.test.js","sourceRoot":"","sources":["../../src/__tests__/registry.test.ts"],"names":[],"mappings":"AAAA,+CAA+C;AAC/C,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAClC,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC;YAC/B,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,iBAAiB;YAC9B,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;YACzB,GAAG,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM;SACxB,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;YAC5B,SAAS,EAAE,EAAE,eAAe,EAAE,QAAQ,EAAE;SACzC,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC;YACvB,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,aAAa;YAC1B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM;SAC5B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;YAC5B,KAAK,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE;SAC7B,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,CAAC,CAAC;QAElC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACxD,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC;YACvB,IAAI,EAAE,SAAS;YACf,WAAW,EAAE,SAAS;YACtB,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;SACxB,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC;YAC/B,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,aAAa;YAC1B,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;YACzB,GAAG,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;SACpB,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC;YAC5B,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE;YAC1B,SAAS,EAAE,EAAE,aAAa,EAAE,QAAQ,EAAE;SACvC,CAAC,CAAC;QAEH,MAAM,GAAG,GAAG,QAAQ,CAAC,OAAO,EAAE,CAAC;QAC/B,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACjC,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"web-tool.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/web-tool.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,99 @@
|
|
|
1
|
+
// packages/core/src/__tests__/web-tool.test.ts
|
|
2
|
+
import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
|
|
3
|
+
import { webRead } from "../nodes/builtin/web.js";
|
|
4
|
+
import { createWorkflowContext } from "../context.js";
|
|
5
|
+
describe("webRead tool", () => {
|
|
6
|
+
const originalFetch = global.fetch;
|
|
7
|
+
beforeEach(() => {
|
|
8
|
+
global.fetch = vi.fn();
|
|
9
|
+
});
|
|
10
|
+
afterEach(() => {
|
|
11
|
+
global.fetch = originalFetch;
|
|
12
|
+
});
|
|
13
|
+
it("has correct properties", () => {
|
|
14
|
+
expect(webRead.name).toBe("web_read");
|
|
15
|
+
expect(webRead.type).toBe("node");
|
|
16
|
+
expect(webRead.description).toContain("Readability");
|
|
17
|
+
});
|
|
18
|
+
it("extracts content using Readability", async () => {
|
|
19
|
+
const html = `
|
|
20
|
+
<!DOCTYPE html>
|
|
21
|
+
<html>
|
|
22
|
+
<head><title>Test Article</title></head>
|
|
23
|
+
<body>
|
|
24
|
+
<nav>Navigation here</nav>
|
|
25
|
+
<article>
|
|
26
|
+
<h1>Main Article Title</h1>
|
|
27
|
+
<p>This is the main content of the article. It has enough text to be recognized as the main content by Readability.</p>
|
|
28
|
+
<p>Here is another paragraph with more information about the topic being discussed.</p>
|
|
29
|
+
</article>
|
|
30
|
+
<footer>Footer content</footer>
|
|
31
|
+
</body>
|
|
32
|
+
</html>
|
|
33
|
+
`;
|
|
34
|
+
const mockResponse = {
|
|
35
|
+
ok: true,
|
|
36
|
+
status: 200,
|
|
37
|
+
text: vi.fn().mockResolvedValue(html),
|
|
38
|
+
};
|
|
39
|
+
global.fetch.mockResolvedValue(mockResponse);
|
|
40
|
+
const ctx = createWorkflowContext();
|
|
41
|
+
const result = await webRead.execute(ctx, {
|
|
42
|
+
url: "https://example.com/article",
|
|
43
|
+
headers: { Authorization: "Bearer token" },
|
|
44
|
+
});
|
|
45
|
+
expect(global.fetch).toHaveBeenCalledWith("https://example.com/article", {
|
|
46
|
+
method: "GET",
|
|
47
|
+
headers: { Authorization: "Bearer token" },
|
|
48
|
+
});
|
|
49
|
+
expect(result.status).toBe(200);
|
|
50
|
+
expect(result.title).toBe("Test Article");
|
|
51
|
+
expect(result.content).toContain("main content of the article");
|
|
52
|
+
// Navigation and footer should not be in the extracted content
|
|
53
|
+
expect(result.content).not.toContain("Navigation here");
|
|
54
|
+
});
|
|
55
|
+
it("returns null content for non-ok responses", async () => {
|
|
56
|
+
const mockResponse = {
|
|
57
|
+
ok: false,
|
|
58
|
+
status: 404,
|
|
59
|
+
text: vi.fn().mockResolvedValue("Not Found"),
|
|
60
|
+
};
|
|
61
|
+
global.fetch.mockResolvedValue(mockResponse);
|
|
62
|
+
const ctx = createWorkflowContext();
|
|
63
|
+
const result = await webRead.execute(ctx, {
|
|
64
|
+
url: "https://example.com/missing",
|
|
65
|
+
});
|
|
66
|
+
expect(global.fetch).toHaveBeenCalledWith("https://example.com/missing", {
|
|
67
|
+
method: "GET",
|
|
68
|
+
headers: undefined,
|
|
69
|
+
});
|
|
70
|
+
expect(result.status).toBe(404);
|
|
71
|
+
expect(result.content).toBeNull();
|
|
72
|
+
expect(result.title).toBeNull();
|
|
73
|
+
});
|
|
74
|
+
it("falls back to body text when Readability cannot extract", async () => {
|
|
75
|
+
// Minimal HTML that Readability can't extract as an article
|
|
76
|
+
const html = `
|
|
77
|
+
<!DOCTYPE html>
|
|
78
|
+
<html>
|
|
79
|
+
<head><title>Simple Page</title></head>
|
|
80
|
+
<body>Just some text</body>
|
|
81
|
+
</html>
|
|
82
|
+
`;
|
|
83
|
+
const mockResponse = {
|
|
84
|
+
ok: true,
|
|
85
|
+
status: 200,
|
|
86
|
+
text: vi.fn().mockResolvedValue(html),
|
|
87
|
+
};
|
|
88
|
+
global.fetch.mockResolvedValue(mockResponse);
|
|
89
|
+
const ctx = createWorkflowContext();
|
|
90
|
+
const result = await webRead.execute(ctx, {
|
|
91
|
+
url: "https://example.com/simple",
|
|
92
|
+
});
|
|
93
|
+
expect(result.status).toBe(200);
|
|
94
|
+
expect(result.title).toBe("Simple Page");
|
|
95
|
+
// Should fall back to body text content
|
|
96
|
+
expect(result.content).toContain("Just some text");
|
|
97
|
+
});
|
|
98
|
+
});
|
|
99
|
+
//# sourceMappingURL=web-tool.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"web-tool.test.js","sourceRoot":"","sources":["../../src/__tests__/web-tool.test.ts"],"names":[],"mappings":"AAAA,+CAA+C;AAC/C,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAClD,OAAO,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAEtD,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC;IAEnC,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,CAAC,KAAK,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,CAAC,KAAK,GAAG,aAAa,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;QAChC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAClC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,IAAI,GAAG;;;;;;;;;;;;;;KAcZ,CAAC;QAEF,MAAM,YAAY,GAAG;YACnB,EAAE,EAAE,IAAI;YACR,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;SACtC,CAAC;QAED,MAAM,CAAC,KAAkC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAE3E,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE;YACxC,GAAG,EAAE,6BAA6B;YAClC,OAAO,EAAE,EAAE,aAAa,EAAE,cAAc,EAAE;SAC3C,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,6BAA6B,EAAE;YACvE,MAAM,EAAE,KAAK;YACb,OAAO,EAAE,EAAE,aAAa,EAAE,cAAc,EAAE;SAC3C,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,6BAA6B,CAAC,CAAC;QAChE,+DAA+D;QAC/D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,YAAY,GAAG;YACnB,EAAE,EAAE,KAAK;YACT,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC;SAC7C,CAAC;QAED,MAAM,CAAC,KAAkC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAE3E,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE;YACxC,GAAG,EAAE,6BAA6B;SACnC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,6BAA6B,EAAE;YACvE,MAAM,EAAE,KAAK;YACb,OAAO,EAAE,SAAS;SACnB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,4DAA4D;QAC5D,MAAM,IAAI,GAAG;;;;;;KAMZ,CAAC;QAEF,MAAM,YAAY,GAAG;YACnB,EAAE,EAAE,IAAI;YACR,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;SACtC,CAAC;QAED,MAAM,CAAC,KAAkC,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAE3E,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE;YACxC,GAAG,EAAE,4BAA4B;SAClC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACzC,wCAAwC;QACxC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"workflow.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/workflow.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
// packages/core/src/__tests__/workflow.test.ts
|
|
2
|
+
import { describe, it, expect, vi } from "vitest";
|
|
3
|
+
import { z } from "zod";
|
|
4
|
+
// Mock DBOS SDK
|
|
5
|
+
vi.mock("@dbos-inc/dbos-sdk", () => ({
|
|
6
|
+
DBOS: {
|
|
7
|
+
registerWorkflow: vi.fn().mockImplementation((fn) => fn),
|
|
8
|
+
runStep: vi.fn().mockImplementation(async (fn) => fn()),
|
|
9
|
+
logger: {
|
|
10
|
+
info: vi.fn(),
|
|
11
|
+
warn: vi.fn(),
|
|
12
|
+
error: vi.fn(),
|
|
13
|
+
debug: vi.fn(),
|
|
14
|
+
},
|
|
15
|
+
},
|
|
16
|
+
}));
|
|
17
|
+
import { Workflow } from "../workflow.js";
|
|
18
|
+
describe("Workflow.create()", () => {
|
|
19
|
+
it("creates an executable with correct properties", () => {
|
|
20
|
+
const inputSchema = z.object({ url: z.string() });
|
|
21
|
+
const outputSchema = z.object({ status: z.string() });
|
|
22
|
+
const workflow = Workflow.create({
|
|
23
|
+
name: "fetch-status",
|
|
24
|
+
description: "Fetches status",
|
|
25
|
+
version: 1,
|
|
26
|
+
inputSchema,
|
|
27
|
+
outputSchema,
|
|
28
|
+
run: async (_ctx, _inputs) => ({ status: "ok" }),
|
|
29
|
+
});
|
|
30
|
+
expect(workflow.name).toBe("fetch-status");
|
|
31
|
+
expect(workflow.type).toBe("workflow");
|
|
32
|
+
expect(workflow.version).toBe(1);
|
|
33
|
+
expect(workflow.description).toBe("Fetches status");
|
|
34
|
+
expect(workflow.inputSchema).toBe(inputSchema);
|
|
35
|
+
});
|
|
36
|
+
it("execute calls run with context and inputs", async () => {
|
|
37
|
+
const workflow = Workflow.create({
|
|
38
|
+
name: "echo",
|
|
39
|
+
description: "Echoes input",
|
|
40
|
+
version: 1,
|
|
41
|
+
inputSchema: z.object({ message: z.string() }),
|
|
42
|
+
run: async (_ctx, inputs) => inputs.message,
|
|
43
|
+
});
|
|
44
|
+
const mockCtx = { run: async () => { }, log: () => { } };
|
|
45
|
+
const result = await workflow.execute(mockCtx, { message: "hello" });
|
|
46
|
+
expect(result).toBe("hello");
|
|
47
|
+
});
|
|
48
|
+
});
|
|
49
|
+
//# sourceMappingURL=workflow.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"workflow.test.js","sourceRoot":"","sources":["../../src/__tests__/workflow.test.ts"],"names":[],"mappings":"AAAA,+CAA+C;AAC/C,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,gBAAgB;AAChB,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,IAAI,EAAE;QACJ,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;QACxD,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;QACvD,MAAM,EAAE;YACN,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;YACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;YACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;SACf;KACF;CACF,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAClD,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAEtD,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC;YAC/B,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,gBAAgB;YAC7B,OAAO,EAAE,CAAC;YACV,WAAW;YACX,YAAY;YACZ,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;SACjD,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACvC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACpD,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC;YAC/B,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,cAAc;YAC3B,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9C,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO;SAC5C,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,EAAE,GAAG,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,GAAE,CAAC,EAAS,CAAC;QAC9D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QACrE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
package/dist/agent.d.ts
ADDED
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
import type { Executable } from "./types.js";
|
|
3
|
+
import type { AgentTools } from "./nodes/agent/executor.js";
|
|
4
|
+
import type { ModelConfig } from "./nodes/agent/model-config.js";
|
|
5
|
+
import type { NodeRegistry } from "./nodes/registry.js";
|
|
6
|
+
export type { AgentTool, AgentTools } from "./nodes/agent/executor.js";
|
|
7
|
+
/**
|
|
8
|
+
* Definition for creating an agent
|
|
9
|
+
*/
|
|
10
|
+
export interface AgentDefinition<TInput, TOutput> {
|
|
11
|
+
name: string;
|
|
12
|
+
description: string;
|
|
13
|
+
inputSchema: z.ZodType<TInput>;
|
|
14
|
+
outputSchema?: z.ZodType<TOutput>;
|
|
15
|
+
/** Tools available to this agent, keyed by name */
|
|
16
|
+
tools?: AgentTools;
|
|
17
|
+
/** Path to agent spec markdown file (for system prompt) */
|
|
18
|
+
specPath: string;
|
|
19
|
+
}
|
|
20
|
+
/**
|
|
21
|
+
* Extended executable interface for agents
|
|
22
|
+
*/
|
|
23
|
+
export interface AgentExecutable<TInput = unknown, TOutput = unknown> extends Executable<TInput, TOutput> {
|
|
24
|
+
readonly specPath: string;
|
|
25
|
+
readonly tools: AgentTools;
|
|
26
|
+
}
|
|
27
|
+
/**
|
|
28
|
+
* Runtime configuration for agent execution
|
|
29
|
+
* Set by create0pflow() factory
|
|
30
|
+
*/
|
|
31
|
+
interface AgentRuntimeConfig {
|
|
32
|
+
nodeRegistry: NodeRegistry;
|
|
33
|
+
modelConfig?: ModelConfig;
|
|
34
|
+
}
|
|
35
|
+
/**
|
|
36
|
+
* Configure the agent runtime (called by factory)
|
|
37
|
+
* @internal
|
|
38
|
+
*/
|
|
39
|
+
export declare function configureAgentRuntime(config: AgentRuntimeConfig): void;
|
|
40
|
+
/**
|
|
41
|
+
* Factory for creating agent executables
|
|
42
|
+
*/
|
|
43
|
+
export declare const Agent: {
|
|
44
|
+
create<TInput, TOutput = unknown>(definition: AgentDefinition<TInput, TOutput>): AgentExecutable<TInput, TOutput>;
|
|
45
|
+
};
|
|
46
|
+
//# sourceMappingURL=agent.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,KAAK,EAAE,UAAU,EAA6B,MAAM,YAAY,CAAC;AAGxE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,+BAA+B,CAAC;AACjE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAExD,YAAY,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAEvE;;GAEG;AACH,MAAM,WAAW,eAAe,CAAC,MAAM,EAAE,OAAO;IAC9C,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAC/B,YAAY,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAClC,mDAAmD;IACnD,KAAK,CAAC,EAAE,UAAU,CAAC;IACnB,2DAA2D;IAC3D,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,eAAe,CAAC,MAAM,GAAG,OAAO,EAAE,OAAO,GAAG,OAAO,CAClE,SAAQ,UAAU,CAAC,MAAM,EAAE,OAAO,CAAC;IACnC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAC;CAC5B;AAED;;;GAGG;AACH,UAAU,kBAAkB;IAC1B,YAAY,EAAE,YAAY,CAAC;IAC3B,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAID;;;GAGG;AACH,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,kBAAkB,GAAG,IAAI,CAEtE;AA6BD;;GAEG;AACH,eAAO,MAAM,KAAK;WACT,MAAM,EAAE,OAAO,wBACR,eAAe,CAAC,MAAM,EAAE,OAAO,CAAC,GAC3C,eAAe,CAAC,MAAM,EAAE,OAAO,CAAC;CAoDpC,CAAC"}
|