@cogitator-ai/express 0.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +346 -0
- package/dist/index.d.ts +9 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +7 -0
- package/dist/index.js.map +1 -0
- package/dist/middleware/auth.d.ts +4 -0
- package/dist/middleware/auth.d.ts.map +1 -0
- package/dist/middleware/auth.js +26 -0
- package/dist/middleware/auth.js.map +1 -0
- package/dist/middleware/cors.d.ts +4 -0
- package/dist/middleware/cors.d.ts.map +1 -0
- package/dist/middleware/cors.js +42 -0
- package/dist/middleware/cors.js.map +1 -0
- package/dist/middleware/error-handler.d.ts +4 -0
- package/dist/middleware/error-handler.d.ts.map +1 -0
- package/dist/middleware/error-handler.js +32 -0
- package/dist/middleware/error-handler.js.map +1 -0
- package/dist/middleware/index.d.ts +5 -0
- package/dist/middleware/index.d.ts.map +1 -0
- package/dist/middleware/index.js +5 -0
- package/dist/middleware/index.js.map +1 -0
- package/dist/middleware/rate-limit.d.ts +4 -0
- package/dist/middleware/rate-limit.d.ts.map +1 -0
- package/dist/middleware/rate-limit.js +50 -0
- package/dist/middleware/rate-limit.js.map +1 -0
- package/dist/routes/agents.d.ts +4 -0
- package/dist/routes/agents.d.ts.map +1 -0
- package/dist/routes/agents.js +116 -0
- package/dist/routes/agents.js.map +1 -0
- package/dist/routes/health.d.ts +4 -0
- package/dist/routes/health.d.ts.map +1 -0
- package/dist/routes/health.js +18 -0
- package/dist/routes/health.js.map +1 -0
- package/dist/routes/index.d.ts +7 -0
- package/dist/routes/index.d.ts.map +1 -0
- package/dist/routes/index.js +7 -0
- package/dist/routes/index.js.map +1 -0
- package/dist/routes/swarms.d.ts +4 -0
- package/dist/routes/swarms.d.ts.map +1 -0
- package/dist/routes/swarms.js +179 -0
- package/dist/routes/swarms.js.map +1 -0
- package/dist/routes/threads.d.ts +4 -0
- package/dist/routes/threads.d.ts.map +1 -0
- package/dist/routes/threads.js +108 -0
- package/dist/routes/threads.js.map +1 -0
- package/dist/routes/tools.d.ts +4 -0
- package/dist/routes/tools.d.ts.map +1 -0
- package/dist/routes/tools.js +29 -0
- package/dist/routes/tools.js.map +1 -0
- package/dist/routes/workflows.d.ts +4 -0
- package/dist/routes/workflows.d.ts.map +1 -0
- package/dist/routes/workflows.js +110 -0
- package/dist/routes/workflows.js.map +1 -0
- package/dist/server.d.ts +15 -0
- package/dist/server.d.ts.map +1 -0
- package/dist/server.js +90 -0
- package/dist/server.js.map +1 -0
- package/dist/streaming/express-stream-writer.d.ts +24 -0
- package/dist/streaming/express-stream-writer.d.ts.map +1 -0
- package/dist/streaming/express-stream-writer.js +73 -0
- package/dist/streaming/express-stream-writer.js.map +1 -0
- package/dist/streaming/helpers.d.ts +4 -0
- package/dist/streaming/helpers.d.ts.map +1 -0
- package/dist/streaming/helpers.js +14 -0
- package/dist/streaming/helpers.js.map +1 -0
- package/dist/streaming/index.d.ts +4 -0
- package/dist/streaming/index.d.ts.map +1 -0
- package/dist/streaming/index.js +4 -0
- package/dist/streaming/index.js.map +1 -0
- package/dist/streaming/protocol.d.ts +76 -0
- package/dist/streaming/protocol.d.ts.map +1 -0
- package/dist/streaming/protocol.js +37 -0
- package/dist/streaming/protocol.js.map +1 -0
- package/dist/swagger/generator.d.ts +3 -0
- package/dist/swagger/generator.d.ts.map +1 -0
- package/dist/swagger/generator.js +486 -0
- package/dist/swagger/generator.js.map +1 -0
- package/dist/swagger/index.d.ts +3 -0
- package/dist/swagger/index.d.ts.map +1 -0
- package/dist/swagger/index.js +3 -0
- package/dist/swagger/index.js.map +1 -0
- package/dist/swagger/ui.d.ts +4 -0
- package/dist/swagger/ui.d.ts.map +1 -0
- package/dist/swagger/ui.js +39 -0
- package/dist/swagger/ui.js.map +1 -0
- package/dist/types.d.ts +246 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/websocket/index.d.ts +2 -0
- package/dist/websocket/index.d.ts.map +1 -0
- package/dist/websocket/index.js +2 -0
- package/dist/websocket/index.js.map +1 -0
- package/dist/websocket/ws-handler.d.ts +6 -0
- package/dist/websocket/ws-handler.d.ts.map +1 -0
- package/dist/websocket/ws-handler.js +145 -0
- package/dist/websocket/ws-handler.js.map +1 -0
- package/package.json +62 -0
|
@@ -0,0 +1,110 @@
|
|
|
1
|
+
import { Router } from 'express';
|
|
2
|
+
import { ExpressStreamWriter, setupSSEHeaders, generateId } from '../streaming/index.js';
|
|
3
|
+
export function createWorkflowRoutes(ctx) {
|
|
4
|
+
const router = Router();
|
|
5
|
+
router.get('/workflows', (_req, res) => {
|
|
6
|
+
const workflowList = Object.entries(ctx.workflows).map(([name, workflow]) => ({
|
|
7
|
+
name,
|
|
8
|
+
entryPoint: workflow.entryPoint,
|
|
9
|
+
nodes: Array.from(workflow.nodes.keys()),
|
|
10
|
+
}));
|
|
11
|
+
const response = { workflows: workflowList };
|
|
12
|
+
res.json(response);
|
|
13
|
+
});
|
|
14
|
+
router.post('/workflows/:name/run', async (req, res) => {
|
|
15
|
+
const { name } = req.params;
|
|
16
|
+
const workflow = ctx.workflows[name];
|
|
17
|
+
if (!workflow) {
|
|
18
|
+
res.status(404).json({
|
|
19
|
+
error: { message: `Workflow '${name}' not found`, code: 'NOT_FOUND' },
|
|
20
|
+
});
|
|
21
|
+
return;
|
|
22
|
+
}
|
|
23
|
+
const body = req.body;
|
|
24
|
+
try {
|
|
25
|
+
const { WorkflowExecutor } = await import('@cogitator-ai/workflows');
|
|
26
|
+
const executor = new WorkflowExecutor(ctx.cogitator);
|
|
27
|
+
const result = await executor.execute(workflow, body?.input, body?.options);
|
|
28
|
+
const nodeResults = {};
|
|
29
|
+
for (const [nodeName, nodeResult] of result.nodeResults.entries()) {
|
|
30
|
+
nodeResults[nodeName] = nodeResult;
|
|
31
|
+
}
|
|
32
|
+
const response = {
|
|
33
|
+
workflowId: result.workflowId,
|
|
34
|
+
workflowName: result.workflowName,
|
|
35
|
+
state: result.state,
|
|
36
|
+
duration: result.duration,
|
|
37
|
+
nodeResults,
|
|
38
|
+
};
|
|
39
|
+
res.json(response);
|
|
40
|
+
}
|
|
41
|
+
catch (error) {
|
|
42
|
+
if (error instanceof Error && error.message.includes('Cannot find module')) {
|
|
43
|
+
res.status(501).json({
|
|
44
|
+
error: { message: 'Workflows package not installed', code: 'UNIMPLEMENTED' },
|
|
45
|
+
});
|
|
46
|
+
return;
|
|
47
|
+
}
|
|
48
|
+
const message = error instanceof Error ? error.message : 'Unknown error';
|
|
49
|
+
res.status(500).json({
|
|
50
|
+
error: { message, code: 'INTERNAL' },
|
|
51
|
+
});
|
|
52
|
+
}
|
|
53
|
+
});
|
|
54
|
+
router.post('/workflows/:name/stream', async (req, res) => {
|
|
55
|
+
const { name } = req.params;
|
|
56
|
+
const workflow = ctx.workflows[name];
|
|
57
|
+
if (!workflow) {
|
|
58
|
+
res.status(404).json({
|
|
59
|
+
error: { message: `Workflow '${name}' not found`, code: 'NOT_FOUND' },
|
|
60
|
+
});
|
|
61
|
+
return;
|
|
62
|
+
}
|
|
63
|
+
const body = req.body;
|
|
64
|
+
setupSSEHeaders(res);
|
|
65
|
+
const writer = new ExpressStreamWriter(res);
|
|
66
|
+
const messageId = generateId('wf');
|
|
67
|
+
req.on('close', () => {
|
|
68
|
+
writer.close();
|
|
69
|
+
});
|
|
70
|
+
try {
|
|
71
|
+
const { WorkflowExecutor } = await import('@cogitator-ai/workflows');
|
|
72
|
+
const executor = new WorkflowExecutor(ctx.cogitator);
|
|
73
|
+
writer.start(messageId);
|
|
74
|
+
const result = await executor.execute(workflow, body?.input, {
|
|
75
|
+
...body?.options,
|
|
76
|
+
onNodeStart: (node) => {
|
|
77
|
+
writer.workflowEvent('node_started', { nodeName: node, timestamp: Date.now() });
|
|
78
|
+
},
|
|
79
|
+
onNodeComplete: (node, output, duration) => {
|
|
80
|
+
writer.workflowEvent('node_completed', { nodeName: node, output, duration });
|
|
81
|
+
},
|
|
82
|
+
onNodeError: (node, error) => {
|
|
83
|
+
writer.workflowEvent('node_error', { nodeName: node, error: error.message });
|
|
84
|
+
},
|
|
85
|
+
onNodeProgress: (node, progress) => {
|
|
86
|
+
writer.workflowEvent('node_progress', { nodeName: node, progress });
|
|
87
|
+
},
|
|
88
|
+
});
|
|
89
|
+
writer.workflowEvent('workflow_completed', {
|
|
90
|
+
workflowId: result.workflowId,
|
|
91
|
+
duration: result.duration,
|
|
92
|
+
});
|
|
93
|
+
writer.finish(messageId);
|
|
94
|
+
}
|
|
95
|
+
catch (error) {
|
|
96
|
+
if (error instanceof Error && error.message.includes('Cannot find module')) {
|
|
97
|
+
writer.error('Workflows package not installed', 'UNIMPLEMENTED');
|
|
98
|
+
}
|
|
99
|
+
else {
|
|
100
|
+
const message = error instanceof Error ? error.message : 'Unknown error';
|
|
101
|
+
writer.error(message, 'INTERNAL');
|
|
102
|
+
}
|
|
103
|
+
}
|
|
104
|
+
finally {
|
|
105
|
+
writer.close();
|
|
106
|
+
}
|
|
107
|
+
});
|
|
108
|
+
return router;
|
|
109
|
+
}
|
|
110
|
+
//# sourceMappingURL=workflows.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"workflows.js","sourceRoot":"","sources":["../../src/routes/workflows.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AASjC,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEzF,MAAM,UAAU,oBAAoB,CAAC,GAAiB;IACpD,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;IAExB,MAAM,CAAC,GAAG,CAAC,YAAY,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;QACrC,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC;YAC5E,IAAI;YACJ,UAAU,EAAE,QAAQ,CAAC,UAAU;YAC/B,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;SACzC,CAAC,CAAC,CAAC;QAEJ,MAAM,QAAQ,GAAyB,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC;QACnE,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE,KAAK,EAAE,GAAqB,EAAE,GAAa,EAAE,EAAE;QACjF,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC;QAC5B,MAAM,QAAQ,GAAG,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAErC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACnB,KAAK,EAAE,EAAE,OAAO,EAAE,aAAa,IAAI,aAAa,EAAE,IAAI,EAAE,WAAW,EAAE;aACtE,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,MAAM,IAAI,GAAG,GAAG,CAAC,IAA0B,CAAC;QAE5C,IAAI,CAAC;YACH,MAAM,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;YACrE,MAAM,QAAQ,GAAG,IAAI,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAErD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;YAE5E,MAAM,WAAW,GAA0D,EAAE,CAAC;YAC9E,KAAK,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,WAAW,CAAC,OAAO,EAAE,EAAE,CAAC;gBAClE,WAAW,CAAC,QAAQ,CAAC,GAAG,UAAU,CAAC;YACrC,CAAC;YAED,MAAM,QAAQ,GAAwB;gBACpC,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,WAAW;aACZ,CAAC;YAEF,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;gBAC3E,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;oBACnB,KAAK,EAAE,EAAE,OAAO,EAAE,iCAAiC,EAAE,IAAI,EAAE,eAAe,EAAE;iBAC7E,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACzE,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACnB,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE;aACrC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE,KAAK,EAAE,GAAqB,EAAE,GAAa,EAAE,EAAE;QACpF,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC;QAC5B,MAAM,QAAQ,GAAG,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAErC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBACnB,KAAK,EAAE,EAAE,OAAO,EAAE,aAAa,IAAI,aAAa,EAAE,IAAI,EAAE,WAAW,EAAE;aACtE,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,MAAM,IAAI,GAAG,GAAG,CAAC,IAA0B,CAAC;QAE5C,eAAe,CAAC,GAAG,CAAC,CAAC;QACrB,MAAM,MAAM,GAAG,IAAI,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAC5C,MAAM,SAAS,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;QAEnC,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACnB,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;YACrE,MAAM,QAAQ,GAAG,IAAI,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAErD,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAExB,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE;gBAC3D,GAAG,IAAI,EAAE,OAAO;gBAChB,WAAW,EAAE,CAAC,IAAY,EAAE,EAAE;oBAC5B,MAAM,CAAC,aAAa,CAAC,cAAc,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBAClF,CAAC;gBACD,cAAc,EAAE,CAAC,IAAY,EAAE,MAAe,EAAE,QAAgB,EAAE,EAAE;oBAClE,MAAM,CAAC,aAAa,CAAC,gBAAgB,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;gBAC/E,CAAC;gBACD,WAAW,EAAE,CAAC,IAAY,EAAE,KAAY,EAAE,EAAE;oBAC1C,MAAM,CAAC,aAAa,CAAC,YAAY,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC/E,CAAC;gBACD,cAAc,EAAE,CAAC,IAAY,EAAE,QAAgB,EAAE,EAAE;oBACjD,MAAM,CAAC,aAAa,CAAC,eAAe,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;gBACtE,CAAC;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,aAAa,CAAC,oBAAoB,EAAE;gBACzC,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,QAAQ,EAAE,MAAM,CAAC,QAAQ;aAC1B,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;gBAC3E,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE,eAAe,CAAC,CAAC;YACnE,CAAC;iBAAM,CAAC;gBACN,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;gBACzE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
package/dist/server.d.ts
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import type { CogitatorServerConfig } from './types.js';
|
|
2
|
+
export declare class CogitatorServer {
|
|
3
|
+
private app;
|
|
4
|
+
private cogitator;
|
|
5
|
+
private agents;
|
|
6
|
+
private workflows;
|
|
7
|
+
private swarms;
|
|
8
|
+
private config;
|
|
9
|
+
private initialized;
|
|
10
|
+
constructor(options: CogitatorServerConfig);
|
|
11
|
+
init(): Promise<void>;
|
|
12
|
+
private setupSwagger;
|
|
13
|
+
get isInitialized(): boolean;
|
|
14
|
+
}
|
|
15
|
+
//# sourceMappingURL=server.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,qBAAqB,EAA6C,MAAM,YAAY,CAAC;AAwBnG,qBAAa,eAAe;IAC1B,OAAO,CAAC,GAAG,CAAS;IACpB,OAAO,CAAC,SAAS,CAAqC;IACtD,OAAO,CAAC,MAAM,CAAkC;IAChD,OAAO,CAAC,SAAS,CAAqC;IACtD,OAAO,CAAC,MAAM,CAAkC;IAChD,OAAO,CAAC,MAAM,CAAyD;IACvE,OAAO,CAAC,WAAW,CAAS;gBAEhB,OAAO,EAAE,qBAAqB;IAkBpC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;YAoDb,YAAY;IAe1B,IAAI,aAAa,IAAI,OAAO,CAE3B;CACF"}
|
package/dist/server.js
ADDED
|
@@ -0,0 +1,90 @@
|
|
|
1
|
+
import { Router, json } from 'express';
|
|
2
|
+
import { createAuthMiddleware, createRateLimitMiddleware, createCorsMiddleware, errorHandler, notFoundHandler, } from './middleware/index.js';
|
|
3
|
+
import { createHealthRoutes, createAgentRoutes, createThreadRoutes, createToolRoutes, createWorkflowRoutes, createSwarmRoutes, } from './routes/index.js';
|
|
4
|
+
const DEFAULT_CONFIG = {
|
|
5
|
+
basePath: '/cogitator',
|
|
6
|
+
enableWebSocket: false,
|
|
7
|
+
enableSwagger: true,
|
|
8
|
+
requestTimeout: 30000,
|
|
9
|
+
};
|
|
10
|
+
export class CogitatorServer {
|
|
11
|
+
app;
|
|
12
|
+
cogitator;
|
|
13
|
+
agents;
|
|
14
|
+
workflows;
|
|
15
|
+
swarms;
|
|
16
|
+
config;
|
|
17
|
+
initialized = false;
|
|
18
|
+
constructor(options) {
|
|
19
|
+
this.app = options.app;
|
|
20
|
+
this.cogitator = options.cogitator;
|
|
21
|
+
this.agents = options.agents || {};
|
|
22
|
+
this.workflows = options.workflows || {};
|
|
23
|
+
this.swarms = options.swarms || {};
|
|
24
|
+
this.config = {
|
|
25
|
+
...DEFAULT_CONFIG,
|
|
26
|
+
auth: options.config?.auth,
|
|
27
|
+
rateLimit: options.config?.rateLimit,
|
|
28
|
+
cors: options.config?.cors,
|
|
29
|
+
swagger: options.config?.swagger || {},
|
|
30
|
+
websocket: options.config?.websocket || {},
|
|
31
|
+
...options.config,
|
|
32
|
+
};
|
|
33
|
+
}
|
|
34
|
+
async init() {
|
|
35
|
+
if (this.initialized) {
|
|
36
|
+
return;
|
|
37
|
+
}
|
|
38
|
+
const router = Router();
|
|
39
|
+
const basePath = this.config.basePath;
|
|
40
|
+
router.use(json());
|
|
41
|
+
if (this.config.cors) {
|
|
42
|
+
router.use(createCorsMiddleware(this.config.cors));
|
|
43
|
+
}
|
|
44
|
+
router.use(createAuthMiddleware(this.config.auth));
|
|
45
|
+
if (this.config.rateLimit) {
|
|
46
|
+
router.use(createRateLimitMiddleware(this.config.rateLimit));
|
|
47
|
+
}
|
|
48
|
+
const ctx = {
|
|
49
|
+
cogitator: this.cogitator,
|
|
50
|
+
agents: this.agents,
|
|
51
|
+
workflows: this.workflows,
|
|
52
|
+
swarms: this.swarms,
|
|
53
|
+
config: this.config,
|
|
54
|
+
};
|
|
55
|
+
router.use(createHealthRoutes(ctx));
|
|
56
|
+
router.use(createAgentRoutes(ctx));
|
|
57
|
+
router.use(createThreadRoutes(ctx));
|
|
58
|
+
router.use(createToolRoutes(ctx));
|
|
59
|
+
router.use(createWorkflowRoutes(ctx));
|
|
60
|
+
router.use(createSwarmRoutes(ctx));
|
|
61
|
+
if (this.config.enableSwagger) {
|
|
62
|
+
await this.setupSwagger(router, ctx);
|
|
63
|
+
}
|
|
64
|
+
if (this.config.enableWebSocket) {
|
|
65
|
+
console.log('[CogitatorServer] WebSocket support enabled but requires separate setup');
|
|
66
|
+
}
|
|
67
|
+
router.use(notFoundHandler);
|
|
68
|
+
router.use(errorHandler);
|
|
69
|
+
this.app.use(basePath, router);
|
|
70
|
+
this.initialized = true;
|
|
71
|
+
console.log(`[CogitatorServer] Initialized at ${basePath}`);
|
|
72
|
+
}
|
|
73
|
+
async setupSwagger(router, ctx) {
|
|
74
|
+
try {
|
|
75
|
+
const { generateOpenAPISpec, serveSwaggerUI } = await import('./swagger/index.js');
|
|
76
|
+
const spec = generateOpenAPISpec(ctx, this.config.swagger || {});
|
|
77
|
+
router.get('/openapi.json', (_req, res) => {
|
|
78
|
+
res.json(spec);
|
|
79
|
+
});
|
|
80
|
+
router.get('/docs', serveSwaggerUI(spec));
|
|
81
|
+
}
|
|
82
|
+
catch {
|
|
83
|
+
console.warn('[CogitatorServer] Swagger setup failed, continuing without docs');
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
get isInitialized() {
|
|
87
|
+
return this.initialized;
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
//# sourceMappingURL=server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAEvC,OAAO,EACL,oBAAoB,EACpB,yBAAyB,EACzB,oBAAoB,EACpB,YAAY,EACZ,eAAe,GAChB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACL,kBAAkB,EAClB,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,EAChB,oBAAoB,EACpB,iBAAiB,GAClB,MAAM,mBAAmB,CAAC;AAE3B,MAAM,cAAc,GAAG;IACrB,QAAQ,EAAE,YAAY;IACtB,eAAe,EAAE,KAAK;IACtB,aAAa,EAAE,IAAI;IACnB,cAAc,EAAE,KAAK;CACtB,CAAC;AAEF,MAAM,OAAO,eAAe;IAClB,GAAG,CAAS;IACZ,SAAS,CAAqC;IAC9C,MAAM,CAAkC;IACxC,SAAS,CAAqC;IAC9C,MAAM,CAAkC;IACxC,MAAM,CAAyD;IAC/D,WAAW,GAAG,KAAK,CAAC;IAE5B,YAAY,OAA8B;QACxC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;QACnC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,EAAE,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;QAEnC,IAAI,CAAC,MAAM,GAAG;YACZ,GAAG,cAAc;YACjB,IAAI,EAAE,OAAO,CAAC,MAAM,EAAE,IAAI;YAC1B,SAAS,EAAE,OAAO,CAAC,MAAM,EAAE,SAA4B;YACvD,IAAI,EAAE,OAAO,CAAC,MAAM,EAAE,IAAkB;YACxC,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,IAAI,EAAE;YACtC,SAAS,EAAE,OAAO,CAAC,MAAM,EAAE,SAAS,IAAI,EAAE;YAC1C,GAAG,OAAO,CAAC,MAAM;SACwC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;QACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;QAEtC,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;QAEnB,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACrB,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QAEnD,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YAC1B,MAAM,CAAC,GAAG,CAAC,yBAAyB,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QAC/D,CAAC;QAED,MAAM,GAAG,GAAiB;YACxB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,IAAI,CAAC,MAAO;YACpB,SAAS,EAAE,IAAI,CAAC,SAAU;YAC1B,MAAM,EAAE,IAAI,CAAC,MAAO;YACpB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC;QAEF,MAAM,CAAC,GAAG,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,GAAG,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC;QAEnC,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YAC9B,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;YAChC,OAAO,CAAC,GAAG,CAAC,yEAAyE,CAAC,CAAC;QACzF,CAAC;QAED,MAAM,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAC5B,MAAM,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAEzB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAE/B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,OAAO,CAAC,GAAG,CAAC,oCAAoC,QAAQ,EAAE,CAAC,CAAC;IAC9D,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,GAAiB;QAC1D,IAAI,CAAC;YACH,MAAM,EAAE,mBAAmB,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;YACnF,MAAM,IAAI,GAAG,mBAAmB,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;YAEjE,MAAM,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;gBACxC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjB,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,IAAI,CAAC,iEAAiE,CAAC,CAAC;QAClF,CAAC;IACH,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;CACF"}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
import type { Response } from 'express';
|
|
2
|
+
import { type Usage } from './protocol.js';
|
|
3
|
+
export declare class ExpressStreamWriter {
|
|
4
|
+
private res;
|
|
5
|
+
private closed;
|
|
6
|
+
constructor(res: Response);
|
|
7
|
+
private write;
|
|
8
|
+
start(messageId: string): void;
|
|
9
|
+
textStart(id: string): void;
|
|
10
|
+
textDelta(id: string, delta: string): void;
|
|
11
|
+
textEnd(id: string): void;
|
|
12
|
+
toolCallStart(id: string, toolName: string): void;
|
|
13
|
+
toolCallDelta(id: string, argsTextDelta: string): void;
|
|
14
|
+
toolCallEnd(id: string): void;
|
|
15
|
+
toolResult(id: string, toolCallId: string, result: unknown): void;
|
|
16
|
+
workflowEvent(event: string, data: unknown): void;
|
|
17
|
+
swarmEvent(event: string, data: unknown): void;
|
|
18
|
+
error(message: string, code?: string): void;
|
|
19
|
+
finish(messageId: string, usage?: Usage): void;
|
|
20
|
+
close(): void;
|
|
21
|
+
get isClosed(): boolean;
|
|
22
|
+
}
|
|
23
|
+
export declare function setupSSEHeaders(res: Response): void;
|
|
24
|
+
//# sourceMappingURL=express-stream-writer.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"express-stream-writer.d.ts","sourceRoot":"","sources":["../../src/streaming/express-stream-writer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AAExC,OAAO,EAaL,KAAK,KAAK,EACX,MAAM,eAAe,CAAC;AAEvB,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,GAAG,CAAW;IACtB,OAAO,CAAC,MAAM,CAAS;gBAEX,GAAG,EAAE,QAAQ;IAIzB,OAAO,CAAC,KAAK;IAKb,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAI9B,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAI3B,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAK1C,OAAO,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAIzB,aAAa,CAAC,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,IAAI;IAIjD,aAAa,CAAC,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,IAAI;IAItD,WAAW,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAI7B,UAAU,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,GAAG,IAAI;IAIjE,aAAa,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,IAAI;IAIjD,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,IAAI;IAI9C,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI;IAI3C,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,KAAK,GAAG,IAAI;IAK9C,KAAK,IAAI,IAAI;IAQb,IAAI,QAAQ,IAAI,OAAO,CAEtB;CACF;AAED,wBAAgB,eAAe,CAAC,GAAG,EAAE,QAAQ,GAAG,IAAI,CAMnD"}
|
|
@@ -0,0 +1,73 @@
|
|
|
1
|
+
import { encodeSSE, encodeDone } from './helpers.js';
|
|
2
|
+
import { createStartEvent, createTextStartEvent, createTextDeltaEvent, createTextEndEvent, createToolCallStartEvent, createToolCallDeltaEvent, createToolCallEndEvent, createToolResultEvent, createErrorEvent, createFinishEvent, createWorkflowEvent, createSwarmEvent, } from './protocol.js';
|
|
3
|
+
export class ExpressStreamWriter {
|
|
4
|
+
res;
|
|
5
|
+
closed = false;
|
|
6
|
+
constructor(res) {
|
|
7
|
+
this.res = res;
|
|
8
|
+
}
|
|
9
|
+
write(data) {
|
|
10
|
+
if (this.closed)
|
|
11
|
+
return;
|
|
12
|
+
this.res.write(encodeSSE(data));
|
|
13
|
+
}
|
|
14
|
+
start(messageId) {
|
|
15
|
+
this.write(createStartEvent(messageId));
|
|
16
|
+
}
|
|
17
|
+
textStart(id) {
|
|
18
|
+
this.write(createTextStartEvent(id));
|
|
19
|
+
}
|
|
20
|
+
textDelta(id, delta) {
|
|
21
|
+
if (!delta)
|
|
22
|
+
return;
|
|
23
|
+
this.write(createTextDeltaEvent(id, delta));
|
|
24
|
+
}
|
|
25
|
+
textEnd(id) {
|
|
26
|
+
this.write(createTextEndEvent(id));
|
|
27
|
+
}
|
|
28
|
+
toolCallStart(id, toolName) {
|
|
29
|
+
this.write(createToolCallStartEvent(id, toolName));
|
|
30
|
+
}
|
|
31
|
+
toolCallDelta(id, argsTextDelta) {
|
|
32
|
+
this.write(createToolCallDeltaEvent(id, argsTextDelta));
|
|
33
|
+
}
|
|
34
|
+
toolCallEnd(id) {
|
|
35
|
+
this.write(createToolCallEndEvent(id));
|
|
36
|
+
}
|
|
37
|
+
toolResult(id, toolCallId, result) {
|
|
38
|
+
this.write(createToolResultEvent(id, toolCallId, result));
|
|
39
|
+
}
|
|
40
|
+
workflowEvent(event, data) {
|
|
41
|
+
this.write(createWorkflowEvent(event, data));
|
|
42
|
+
}
|
|
43
|
+
swarmEvent(event, data) {
|
|
44
|
+
this.write(createSwarmEvent(event, data));
|
|
45
|
+
}
|
|
46
|
+
error(message, code) {
|
|
47
|
+
this.write(createErrorEvent(message, code));
|
|
48
|
+
}
|
|
49
|
+
finish(messageId, usage) {
|
|
50
|
+
this.write(createFinishEvent(messageId, usage));
|
|
51
|
+
this.res.write(encodeDone());
|
|
52
|
+
}
|
|
53
|
+
close() {
|
|
54
|
+
if (this.closed)
|
|
55
|
+
return;
|
|
56
|
+
this.closed = true;
|
|
57
|
+
try {
|
|
58
|
+
this.res.end();
|
|
59
|
+
}
|
|
60
|
+
catch { }
|
|
61
|
+
}
|
|
62
|
+
get isClosed() {
|
|
63
|
+
return this.closed;
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
export function setupSSEHeaders(res) {
|
|
67
|
+
res.setHeader('Content-Type', 'text/event-stream');
|
|
68
|
+
res.setHeader('Cache-Control', 'no-cache');
|
|
69
|
+
res.setHeader('Connection', 'keep-alive');
|
|
70
|
+
res.setHeader('X-Accel-Buffering', 'no');
|
|
71
|
+
res.flushHeaders();
|
|
72
|
+
}
|
|
73
|
+
//# sourceMappingURL=express-stream-writer.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"express-stream-writer.js","sourceRoot":"","sources":["../../src/streaming/express-stream-writer.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AACrD,OAAO,EACL,gBAAgB,EAChB,oBAAoB,EACpB,oBAAoB,EACpB,kBAAkB,EAClB,wBAAwB,EACxB,wBAAwB,EACxB,sBAAsB,EACtB,qBAAqB,EACrB,gBAAgB,EAChB,iBAAiB,EACjB,mBAAmB,EACnB,gBAAgB,GAEjB,MAAM,eAAe,CAAC;AAEvB,MAAM,OAAO,mBAAmB;IACtB,GAAG,CAAW;IACd,MAAM,GAAG,KAAK,CAAC;IAEvB,YAAY,GAAa;QACvB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,IAAa;QACzB,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,SAAiB;QACrB,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,CAAC;IAC1C,CAAC;IAED,SAAS,CAAC,EAAU;QAClB,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC,CAAC;IACvC,CAAC;IAED,SAAS,CAAC,EAAU,EAAE,KAAa;QACjC,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC;IAC9C,CAAC;IAED,OAAO,CAAC,EAAU;QAChB,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC;IACrC,CAAC;IAED,aAAa,CAAC,EAAU,EAAE,QAAgB;QACxC,IAAI,CAAC,KAAK,CAAC,wBAAwB,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC,CAAC;IACrD,CAAC;IAED,aAAa,CAAC,EAAU,EAAE,aAAqB;QAC7C,IAAI,CAAC,KAAK,CAAC,wBAAwB,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC;IAC1D,CAAC;IAED,WAAW,CAAC,EAAU;QACpB,IAAI,CAAC,KAAK,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC,CAAC;IACzC,CAAC;IAED,UAAU,CAAC,EAAU,EAAE,UAAkB,EAAE,MAAe;QACxD,IAAI,CAAC,KAAK,CAAC,qBAAqB,CAAC,EAAE,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;IAC5D,CAAC;IAED,aAAa,CAAC,KAAa,EAAE,IAAa;QACxC,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC;IAC/C,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,IAAa;QACrC,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,IAAa;QAClC,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;IAC9C,CAAC;IAED,MAAM,CAAC,SAAiB,EAAE,KAAa;QACrC,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;QAChD,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK;QACH,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;QACjB,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACZ,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;CACF;AAED,MAAM,UAAU,eAAe,CAAC,GAAa;IAC3C,GAAG,CAAC,SAAS,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;IACnD,GAAG,CAAC,SAAS,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;IAC3C,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;IAC1C,GAAG,CAAC,SAAS,CAAC,mBAAmB,EAAE,IAAI,CAAC,CAAC;IACzC,GAAG,CAAC,YAAY,EAAE,CAAC;AACrB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"helpers.d.ts","sourceRoot":"","sources":["../../src/streaming/helpers.ts"],"names":[],"mappings":"AAEA,wBAAgB,UAAU,CAAC,MAAM,GAAE,MAAa,GAAG,MAAM,CAKxD;AAED,wBAAgB,SAAS,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM,CAE/C;AAED,wBAAgB,UAAU,IAAI,MAAM,CAEnC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
let counter = 0;
|
|
2
|
+
export function generateId(prefix = 'id') {
|
|
3
|
+
const timestamp = Date.now().toString(36);
|
|
4
|
+
const random = Math.random().toString(36).substring(2, 8);
|
|
5
|
+
const count = (counter++).toString(36);
|
|
6
|
+
return `${prefix}_${timestamp}${random}${count}`;
|
|
7
|
+
}
|
|
8
|
+
export function encodeSSE(data) {
|
|
9
|
+
return `data: ${JSON.stringify(data)}\n\n`;
|
|
10
|
+
}
|
|
11
|
+
export function encodeDone() {
|
|
12
|
+
return 'data: [DONE]\n\n';
|
|
13
|
+
}
|
|
14
|
+
//# sourceMappingURL=helpers.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"helpers.js","sourceRoot":"","sources":["../../src/streaming/helpers.ts"],"names":[],"mappings":"AAAA,IAAI,OAAO,GAAG,CAAC,CAAC;AAEhB,MAAM,UAAU,UAAU,CAAC,SAAiB,IAAI;IAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1D,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACvC,OAAO,GAAG,MAAM,IAAI,SAAS,GAAG,MAAM,GAAG,KAAK,EAAE,CAAC;AACnD,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,IAAa;IACrC,OAAO,SAAS,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC;AAC7C,CAAC;AAED,MAAM,UAAU,UAAU;IACxB,OAAO,kBAAkB,CAAC;AAC5B,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/streaming/index.ts"],"names":[],"mappings":"AAAA,cAAc,eAAe,CAAC;AAC9B,cAAc,cAAc,CAAC;AAC7B,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/streaming/index.ts"],"names":[],"mappings":"AAAA,cAAc,eAAe,CAAC;AAC9B,cAAc,cAAc,CAAC;AAC7B,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC"}
|
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
export interface Usage {
|
|
2
|
+
inputTokens: number;
|
|
3
|
+
outputTokens: number;
|
|
4
|
+
totalTokens: number;
|
|
5
|
+
}
|
|
6
|
+
export type StreamEvent = StartEvent | TextStartEvent | TextDeltaEvent | TextEndEvent | ToolCallStartEvent | ToolCallDeltaEvent | ToolCallEndEvent | ToolResultEvent | ErrorEvent | FinishEvent | WorkflowEvent | SwarmEvent;
|
|
7
|
+
export interface StartEvent {
|
|
8
|
+
type: 'start';
|
|
9
|
+
messageId: string;
|
|
10
|
+
}
|
|
11
|
+
export interface TextStartEvent {
|
|
12
|
+
type: 'text-start';
|
|
13
|
+
id: string;
|
|
14
|
+
}
|
|
15
|
+
export interface TextDeltaEvent {
|
|
16
|
+
type: 'text-delta';
|
|
17
|
+
id: string;
|
|
18
|
+
delta: string;
|
|
19
|
+
}
|
|
20
|
+
export interface TextEndEvent {
|
|
21
|
+
type: 'text-end';
|
|
22
|
+
id: string;
|
|
23
|
+
}
|
|
24
|
+
export interface ToolCallStartEvent {
|
|
25
|
+
type: 'tool-call-start';
|
|
26
|
+
id: string;
|
|
27
|
+
toolName: string;
|
|
28
|
+
}
|
|
29
|
+
export interface ToolCallDeltaEvent {
|
|
30
|
+
type: 'tool-call-delta';
|
|
31
|
+
id: string;
|
|
32
|
+
argsTextDelta: string;
|
|
33
|
+
}
|
|
34
|
+
export interface ToolCallEndEvent {
|
|
35
|
+
type: 'tool-call-end';
|
|
36
|
+
id: string;
|
|
37
|
+
}
|
|
38
|
+
export interface ToolResultEvent {
|
|
39
|
+
type: 'tool-result';
|
|
40
|
+
id: string;
|
|
41
|
+
toolCallId: string;
|
|
42
|
+
result: unknown;
|
|
43
|
+
}
|
|
44
|
+
export interface ErrorEvent {
|
|
45
|
+
type: 'error';
|
|
46
|
+
message: string;
|
|
47
|
+
code?: string;
|
|
48
|
+
}
|
|
49
|
+
export interface FinishEvent {
|
|
50
|
+
type: 'finish';
|
|
51
|
+
messageId: string;
|
|
52
|
+
usage?: Usage;
|
|
53
|
+
}
|
|
54
|
+
export interface WorkflowEvent {
|
|
55
|
+
type: 'workflow';
|
|
56
|
+
event: string;
|
|
57
|
+
data: unknown;
|
|
58
|
+
}
|
|
59
|
+
export interface SwarmEvent {
|
|
60
|
+
type: 'swarm';
|
|
61
|
+
event: string;
|
|
62
|
+
data: unknown;
|
|
63
|
+
}
|
|
64
|
+
export declare function createStartEvent(messageId: string): StartEvent;
|
|
65
|
+
export declare function createTextStartEvent(id: string): TextStartEvent;
|
|
66
|
+
export declare function createTextDeltaEvent(id: string, delta: string): TextDeltaEvent;
|
|
67
|
+
export declare function createTextEndEvent(id: string): TextEndEvent;
|
|
68
|
+
export declare function createToolCallStartEvent(id: string, toolName: string): ToolCallStartEvent;
|
|
69
|
+
export declare function createToolCallDeltaEvent(id: string, argsTextDelta: string): ToolCallDeltaEvent;
|
|
70
|
+
export declare function createToolCallEndEvent(id: string): ToolCallEndEvent;
|
|
71
|
+
export declare function createToolResultEvent(id: string, toolCallId: string, result: unknown): ToolResultEvent;
|
|
72
|
+
export declare function createErrorEvent(message: string, code?: string): ErrorEvent;
|
|
73
|
+
export declare function createFinishEvent(messageId: string, usage?: Usage): FinishEvent;
|
|
74
|
+
export declare function createWorkflowEvent(event: string, data: unknown): WorkflowEvent;
|
|
75
|
+
export declare function createSwarmEvent(event: string, data: unknown): SwarmEvent;
|
|
76
|
+
//# sourceMappingURL=protocol.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"protocol.d.ts","sourceRoot":"","sources":["../../src/streaming/protocol.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,KAAK;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,MAAM,WAAW,GACnB,UAAU,GACV,cAAc,GACd,cAAc,GACd,YAAY,GACZ,kBAAkB,GAClB,kBAAkB,GAClB,gBAAgB,GAChB,eAAe,GACf,UAAU,GACV,WAAW,GACX,aAAa,GACb,UAAU,CAAC;AAEf,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,OAAO,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,YAAY,CAAC;IACnB,EAAE,EAAE,MAAM,CAAC;CACZ;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,YAAY,CAAC;IACnB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,UAAU,CAAC;IACjB,EAAE,EAAE,MAAM,CAAC;CACZ;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,iBAAiB,CAAC;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,iBAAiB,CAAC;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,eAAe,CAAC;IACtB,EAAE,EAAE,MAAM,CAAC;CACZ;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,aAAa,CAAC;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,OAAO,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,QAAQ,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,KAAK,CAAC;CACf;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,UAAU,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,OAAO,CAAC;CACf;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,OAAO,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,OAAO,CAAC;CACf;AAED,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,UAAU,CAE9D;AAED,wBAAgB,oBAAoB,CAAC,EAAE,EAAE,MAAM,GAAG,cAAc,CAE/D;AAED,wBAAgB,oBAAoB,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,cAAc,CAE9E;AAED,wBAAgB,kBAAkB,CAAC,EAAE,EAAE,MAAM,GAAG,YAAY,CAE3D;AAED,wBAAgB,wBAAwB,CAAC,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,kBAAkB,CAEzF;AAED,wBAAgB,wBAAwB,CAAC,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,GAAG,kBAAkB,CAE9F;AAED,wBAAgB,sBAAsB,CAAC,EAAE,EAAE,MAAM,GAAG,gBAAgB,CAEnE;AAED,wBAAgB,qBAAqB,CACnC,EAAE,EAAE,MAAM,EACV,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,OAAO,GACd,eAAe,CAEjB;AAED,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,UAAU,CAE3E;AAED,wBAAgB,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,KAAK,GAAG,WAAW,CAE/E;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,aAAa,CAE/E;AAED,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,UAAU,CAEzE"}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
export function createStartEvent(messageId) {
|
|
2
|
+
return { type: 'start', messageId };
|
|
3
|
+
}
|
|
4
|
+
export function createTextStartEvent(id) {
|
|
5
|
+
return { type: 'text-start', id };
|
|
6
|
+
}
|
|
7
|
+
export function createTextDeltaEvent(id, delta) {
|
|
8
|
+
return { type: 'text-delta', id, delta };
|
|
9
|
+
}
|
|
10
|
+
export function createTextEndEvent(id) {
|
|
11
|
+
return { type: 'text-end', id };
|
|
12
|
+
}
|
|
13
|
+
export function createToolCallStartEvent(id, toolName) {
|
|
14
|
+
return { type: 'tool-call-start', id, toolName };
|
|
15
|
+
}
|
|
16
|
+
export function createToolCallDeltaEvent(id, argsTextDelta) {
|
|
17
|
+
return { type: 'tool-call-delta', id, argsTextDelta };
|
|
18
|
+
}
|
|
19
|
+
export function createToolCallEndEvent(id) {
|
|
20
|
+
return { type: 'tool-call-end', id };
|
|
21
|
+
}
|
|
22
|
+
export function createToolResultEvent(id, toolCallId, result) {
|
|
23
|
+
return { type: 'tool-result', id, toolCallId, result };
|
|
24
|
+
}
|
|
25
|
+
export function createErrorEvent(message, code) {
|
|
26
|
+
return { type: 'error', message, code };
|
|
27
|
+
}
|
|
28
|
+
export function createFinishEvent(messageId, usage) {
|
|
29
|
+
return { type: 'finish', messageId, usage };
|
|
30
|
+
}
|
|
31
|
+
export function createWorkflowEvent(event, data) {
|
|
32
|
+
return { type: 'workflow', event, data };
|
|
33
|
+
}
|
|
34
|
+
export function createSwarmEvent(event, data) {
|
|
35
|
+
return { type: 'swarm', event, data };
|
|
36
|
+
}
|
|
37
|
+
//# sourceMappingURL=protocol.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"protocol.js","sourceRoot":"","sources":["../../src/streaming/protocol.ts"],"names":[],"mappings":"AAyFA,MAAM,UAAU,gBAAgB,CAAC,SAAiB;IAChD,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,EAAU;IAC7C,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,EAAU,EAAE,KAAa;IAC5D,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,EAAU;IAC3C,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;AAClC,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,EAAU,EAAE,QAAgB;IACnE,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC;AACnD,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,EAAU,EAAE,aAAqB;IACxE,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,EAAU;IAC/C,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,EAAU,EACV,UAAkB,EAClB,MAAe;IAEf,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;AACzD,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,OAAe,EAAE,IAAa;IAC7D,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AAC1C,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,SAAiB,EAAE,KAAa;IAChE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,KAAa,EAAE,IAAa;IAC9D,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,KAAa,EAAE,IAAa;IAC3D,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;AACxC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generator.d.ts","sourceRoot":"","sources":["../../src/swagger/generator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE5E,wBAAgB,mBAAmB,CAAC,GAAG,EAAE,YAAY,EAAE,MAAM,EAAE,aAAa,GAAG,WAAW,CA8TzF"}
|