@cogitator-ai/tetsu 0.1.0

package/dist/schemas.js

@@ -0,0 +1,185 @@
+import { z } from 'zod';
+import { ERROR_STATUS_CODES } from '@cogitator-ai/types';
+const JsonObject = z.record(z.string(), z.unknown());
+export const NameParams = z.object({ name: z.string().min(1) });
+export const ThreadParams = z.object({ id: z.string().min(1) });
+export const RunBody = z.object({
+    input: z.string().min(1).describe('The message for the agent'),
+    context: JsonObject.optional().describe('Extra values passed to the run as context'),
+    threadId: z.string().min(1).optional().describe('Conversation thread kept in memory'),
+});
+export const SwarmRunBody = RunBody.extend({
+    timeout: z.number().positive().optional().describe('Run timeout in milliseconds'),
+});
+export const WorkflowRunBody = z.object({
+    input: JsonObject.optional().describe('Initial workflow state'),
+    options: z
+        .object({
+        maxConcurrency: z.number().int().positive().optional(),
+        maxIterations: z.number().int().positive().optional(),
+        checkpoint: z.boolean().optional(),
+    })
+        .optional(),
+});
+export const AddMessageBody = z.object({
+    role: z.enum(['user', 'assistant', 'system']),
+    content: z.string().min(1),
+    metadata: JsonObject.optional(),
+});
+export const ToolCallSchema = z.object({
+    id: z.string(),
+    name: z.string(),
+    arguments: JsonObject,
+});
+const ContentPartSchema = z.union([
+    z.object({ type: z.literal('text'), text: z.string() }),
+    z.object({
+        type: z.literal('image_url'),
+        image_url: z.object({ url: z.string(), detail: z.enum(['auto', 'low', 'high']).optional() }),
+    }),
+    z.object({
+        type: z.literal('image_base64'),
+        image_base64: z.object({ data: z.string(), media_type: z.string() }),
+    }),
+]);
+export const MessageSchema = z.object({
+    role: z.enum(['system', 'user', 'assistant', 'tool']),
+    content: z.union([z.string(), z.array(ContentPartSchema)]),
+    name: z.string().optional(),
+    toolCallId: z.string().optional(),
+    toolCalls: z.array(ToolCallSchema).optional(),
+});
+export const UsageSchema = z.object({
+    inputTokens: z.number(),
+    outputTokens: z.number(),
+    totalTokens: z.number(),
+});
+export const HealthResponse = z.object({
+    status: z.literal('ok'),
+    uptime: z.number().describe('Milliseconds since the controller was built'),
+    timestamp: z.number(),
+});
+export const ReadyResponse = z.object({ status: z.literal('ok') });
+export const AgentListResponse = z.object({
+    agents: z.array(z.object({
+        name: z.string(),
+        description: z.string().optional(),
+        tools: z.array(z.string()),
+    })),
+});
+export const AgentRunResponse = z.object({
+    output: z.string(),
+    structured: z.unknown().optional(),
+    threadId: z.string(),
+    usage: UsageSchema,
+    toolCalls: z.array(ToolCallSchema),
+});
+export const ThreadResponse = z.object({
+    id: z.string(),
+    messages: z.array(MessageSchema),
+    createdAt: z.number(),
+    updatedAt: z.number(),
+});
+export const AddMessageResponse = z.object({ success: z.literal(true) });
+export const ToolListResponse = z.object({
+    tools: z.array(z.object({
+        name: z.string(),
+        description: z.string().optional(),
+        parameters: JsonObject,
+    })),
+});
+export const WorkflowListResponse = z.object({
+    workflows: z.array(z.object({
+        name: z.string(),
+        entryPoint: z.string(),
+        nodes: z.array(z.string()),
+    })),
+});
+export const WorkflowRunResponse = z.object({
+    workflowId: z.string(),
+    workflowName: z.string(),
+    state: JsonObject,
+    duration: z.number(),
+    nodeResults: z.record(z.string(), z.object({ output: z.unknown(), duration: z.number() })),
+});
+export const SwarmListResponse = z.object({
+    swarms: z.array(z.object({
+        name: z.string(),
+        strategy: z.string(),
+        agents: z.array(z.string()),
+    })),
+});
+export const SwarmRunResponse = z.object({
+    swarmId: z.string(),
+    swarmName: z.string(),
+    strategy: z.string(),
+    output: z.unknown(),
+    agentResults: z.record(z.string(), z.object({
+        output: z.string(),
+        usage: UsageSchema.extend({ cost: z.number(), duration: z.number() }),
+    })),
+    usage: z.object({
+        totalTokens: z.number(),
+        totalCost: z.number(),
+        elapsedTime: z.number(),
+    }),
+});
+export const BlackboardResponse = z.object({ sections: JsonObject });
+export const SocketMessage = z.discriminatedUnion('type', [
+    z.object({ type: z.literal('ping'), id: z.string().optional() }),
+    z.object({ type: z.literal('stop'), id: z.string().optional() }),
+    z.object({
+        type: z.literal('run'),
+        id: z.string().optional(),
+        payload: z.object({
+            type: z.enum(['agent', 'workflow', 'swarm']),
+            name: z.string().min(1),
+            input: z.string().min(1),
+            context: JsonObject.optional(),
+            threadId: z.string().min(1).optional(),
+        }),
+    }),
+]);
+export function errorEnvelope(error, description) {
+    return z
+        .object({
+        status: z.number().int(),
+        message: z.string(),
+        error: z.literal(error),
+    })
+        .describe(description);
+}
+function codesFor(status) {
+    return Object.entries(ERROR_STATUS_CODES)
+        .filter(([, codeStatus]) => codeStatus === status)
+        .map(([code]) => code);
+}
+/**
+ * The envelope of every `CogitatorError` that maps to `status`, plus `extra` codes
+ * the adapter answers with itself.
+ */
+export function failureEnvelope(status, description, extra = []) {
+    const codes = [...new Set([...codesFor(status), ...extra])];
+    const [first, ...rest] = codes;
+    if (first === undefined)
+        throw new Error(`No error codes map to status ${status}`);
+    return z
+        .object({
+        status: z.number().int(),
+        message: z.string(),
+        error: z.enum([first, ...rest]),
+    })
+        .describe(description);
+}
+/** What a run of an agent, a workflow or a swarm can fail with, besides its own refusals. */
+export const RUN_FAILURES = {
+    400: failureEnvelope(400, 'The model, a tool or a guardrail refused the request (context too long, content filtered, invalid tool arguments, prompt injection)'),
+    409: failureEnvelope(409, 'The agent is already running'),
+    429: failureEnvelope(429, 'The model provider is rate limiting requests; see Retry-After'),
+    500: failureEnvelope(500, 'A tool, a workflow step or the sandbox failed'),
+    502: failureEnvelope(502, 'The model provider answered with something unusable'),
+    503: failureEnvelope(503, 'The model provider, the sandbox or a circuit breaker is unavailable'),
+    504: failureEnvelope(504, 'The model, a tool or the sandbox timed out'),
+    507: failureEnvelope(507, 'The sandbox ran out of memory'),
+};
+//# sourceMappingURL=schemas.js.map

package/dist/schemas.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schemas.js","sourceRoot":"","sources":["../src/schemas.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAEzD,MAAM,UAAU,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;AAErD,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;AAChE,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;AAEhE,MAAM,CAAC,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;IAC9D,OAAO,EAAE,UAAU,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2CAA2C,CAAC;IACpF,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,oCAAoC,CAAC;CACtF,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC;IACzC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;CAClF,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,KAAK,EAAE,UAAU,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;IAC/D,OAAO,EAAE,CAAC;SACP,MAAM,CAAC;QACN,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;QACtD,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;QACrD,UAAU,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;KACnC,CAAC;SACD,QAAQ,EAAE;CACd,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IACrC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;IAC7C,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IAC1B,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE;CAChC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IACrC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;IAChB,SAAS,EAAE,UAAU;CACtB,CAAC,CAAC;AAEH,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC;IAChC,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;IACvD,CAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC;QAC5B,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC;KAC7F,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC;QAC/B,YAAY,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;KACrE,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;IACrD,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;IAC1D,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC3B,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACjC,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,QAAQ,EAAE;CAC9C,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAClC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;IACvB,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;IACxB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;CACxB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IACrC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;IACvB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6CAA6C,CAAC;IAC1E,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAEnE,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,MAAM,EAAE,CAAC,CAAC,KAAK,CACb,CAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAClC,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC3B,CAAC,CACH;CACF,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,UAAU,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IAClC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,KAAK,EAAE,WAAW;IAClB,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC;CACnC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IACrC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;IACd,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC;IAChC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAEzE,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,KAAK,EAAE,CAAC,CAAC,KAAK,CACZ,CAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAClC,UAAU,EAAE,UAAU;KACvB,CAAC,CACH;CACF,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3C,SAAS,EAAE,CAAC,CAAC,KAAK,CAChB,CAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;QACtB,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC3B,CAAC,CACH;CACF,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC1C,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;IACtB,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;IACxB,KAAK,EAAE,UAAU;IACjB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;CAC3F,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,MAAM,EAAE,CAAC,CAAC,KAAK,CACb,CAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;QACpB,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC5B,CAAC,CACH;CACF,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;IACnB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;IACnB,YAAY,EAAE,CAAC,CAAC,MAAM,CACpB,CAAC,CAAC,MAAM,EAAE,EACV,CAAC,CAAC,MAAM,CAAC;QACP,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;QAClB,KAAK,EAAE,WAAW,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;KACtE,CAAC,CACH;IACD,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC;QACd,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;QACvB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;QACrB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;KACxB,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;AAErE,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,CAAC,kBAAkB,CAAC,MAAM,EAAE;IACxD,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC;IAChE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC;IAChE,CAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC;QACtB,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACzB,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC;YAChB,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;YAC5C,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;YACvB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;YACxB,OAAO,EAAE,UAAU,CAAC,QAAQ,EAAE;YAC9B,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE;SACvC,CAAC;KACH,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,UAAU,aAAa,CAAyB,KAAQ,EAAE,WAAmB;IACjF,OAAO,CAAC;SACL,MAAM,CAAC;QACN,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE;QACxB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;QACnB,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC;KACxB,CAAC;SACD,QAAQ,CAAC,WAAW,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,QAAQ,CAAC,MAAc;IAC9B,OAAO,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC;SACtC,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC,UAAU,KAAK,MAAM,CAAC;SACjD,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;AAC3B,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,MAAc,EAAE,WAAmB,EAAE,QAAkB,EAAE;IACvF,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,EAAE,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IAC5D,MAAM,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,GAAG,KAAK,CAAC;IAC/B,IAAI,KAAK,KAAK,SAAS;QAAE,MAAM,IAAI,KAAK,CAAC,gCAAgC,MAAM,EAAE,CAAC,CAAC;IACnF,OAAO,CAAC;SACL,MAAM,CAAC;QACN,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE;QACxB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;QACnB,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,CAAC;KAChC,CAAC;SACD,QAAQ,CAAC,WAAW,CAAC,CAAC;AAC3B,CAAC;AAED,6FAA6F;AAC7F,MAAM,CAAC,MAAM,YAAY,GAAG;IAC1B,GAAG,EAAE,eAAe,CAClB,GAAG,EACH,qIAAqI,CACtI;IACD,GAAG,EAAE,eAAe,CAAC,GAAG,EAAE,8BAA8B,CAAC;IACzD,GAAG,EAAE,eAAe,CAAC,GAAG,EAAE,+DAA+D,CAAC;IAC1F,GAAG,EAAE,eAAe,CAAC,GAAG,EAAE,+CAA+C,CAAC;IAC1E,GAAG,EAAE,eAAe,CAAC,GAAG,EAAE,qDAAqD,CAAC;IAChF,GAAG,EAAE,eAAe,CAAC,GAAG,EAAE,qEAAqE,CAAC;IAChG,GAAG,EAAE,eAAe,CAAC,GAAG,EAAE,4CAA4C,CAAC;IACvE,GAAG,EAAE,eAAe,CAAC,GAAG,EAAE,+BAA+B,CAAC;CAClD,CAAC"}

package/dist/socket.d.ts

@@ -0,0 +1,40 @@
+import type { z } from 'zod';
+import type { callerHook } from './auth.js';
+import type { CogitatorDeps } from './types.js';
+/**
+ * The WebSocket endpoint of `cogitatorController()`.
+ *
+ * The handshake runs the same `auth` as the HTTP routes. Each socket runs one
+ * agent, workflow or swarm at a time; `stop` cancels it, and so does closing the
+ * socket.
+ */
+export declare function cogitatorSocket(deps: CogitatorDeps, caller: ReturnType<typeof callerHook>, path?: `/${string}`): import("@tetsujs/core").WsDef<`/${string}`, {
+    readonly message: z.ZodDiscriminatedUnion<[z.ZodObject<{
+        type: z.ZodLiteral<"ping">;
+        id: z.ZodOptional<z.ZodString>;
+    }, z.core.$strip>, z.ZodObject<{
+        type: z.ZodLiteral<"stop">;
+        id: z.ZodOptional<z.ZodString>;
+    }, z.core.$strip>, z.ZodObject<{
+        type: z.ZodLiteral<"run">;
+        id: z.ZodOptional<z.ZodString>;
+        payload: z.ZodObject<{
+            type: z.ZodEnum<{
+                agent: "agent";
+                workflow: "workflow";
+                swarm: "swarm";
+            }>;
+            name: z.ZodString;
+            input: z.ZodString;
+            context: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+            threadId: z.ZodOptional<z.ZodString>;
+        }, z.core.$strip>;
+    }, z.core.$strip>], "type">;
+}, {
+    readonly beforeParse: readonly [import("@tetsujs/core").Hook<"beforeParse", import("@tetsujs/core").BaseCtx & {
+        readonly params: {
+            [x: string]: string;
+        };
+    }, Pick<import("./auth.js").CallerFields, "cogitatorAuth">>];
+}>;
+//# sourceMappingURL=socket.d.ts.map

package/dist/socket.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"socket.d.ts","sourceRoot":"","sources":["../src/socket.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAC7B,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AAgB5C,OAAO,KAAK,EAAe,aAAa,EAA0B,MAAM,YAAY,CAAC;AAarF;;;;;;GAMG;AACH,wBAAgB,eAAe,CAC7B,IAAI,EAAE,aAAa,EACnB,MAAM,EAAE,UAAU,CAAC,OAAO,UAAU,CAAC,EACrC,IAAI,GAAE,IAAI,MAAM,EAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsDlC"}

package/dist/socket.js

@@ -0,0 +1,120 @@
+import { ws } from '@tetsujs/core';
+import { describeError } from './errors.js';
+import { checkThreadAccess, executeSwarm, executeWorkflow, findAgent, findSwarm, findWorkflow, runAgent, toAgentRunResponse, toSwarmRunResponse, toWorkflowRunResponse, } from './operations.js';
+import { SocketMessage } from './schemas.js';
+import { resolveSignal } from './streaming.js';
+const OPEN = 1;
+const DEFAULT_PATH = '/ws';
+/**
+ * The WebSocket endpoint of `cogitatorController()`.
+ *
+ * The handshake runs the same `auth` as the HTTP routes. Each socket runs one
+ * agent, workflow or swarm at a time; `stop` cancels it, and so does closing the
+ * socket.
+ */
+export function cogitatorSocket(deps, caller, path = DEFAULT_PATH) {
+    const running = new WeakMap();
+    return ws({
+        path,
+        hooks: { beforeParse: [caller] },
+        schema: { message: SocketMessage },
+        docs: {
+            summary: 'Run agents, workflows and swarms over a WebSocket',
+            description: 'Send `{ type: "run", id, payload: { type, name, input } }`, `{ type: "stop" }` or `{ type: "ping" }`.',
+        },
+        until: () => resolveSignal(deps.until),
+        message: (socket, message) => {
+            switch (message.type) {
+                case 'ping':
+                    send(socket, { type: 'pong', id: message.id });
+                    return;
+                case 'stop':
+                    running.get(socket)?.abort();
+                    return;
+                case 'run': {
+                    if (running.has(socket)) {
+                        send(socket, {
+                            type: 'error',
+                            id: message.id,
+                            error: 'A run is already in progress',
+                            code: 'RUN_IN_PROGRESS',
+                        });
+                        return;
+                    }
+                    const controller = new AbortController();
+                    running.set(socket, controller);
+                    return handleRun(deps, socket, message.id, message.payload, socket.data.cogitatorAuth, {
+                        signal: controller.signal,
+                    }).finally(() => {
+                        if (running.get(socket) === controller)
+                            running.delete(socket);
+                    });
+                }
+            }
+        },
+        invalid: (socket, issues) => {
+            send(socket, {
+                type: 'error',
+                error: issues[0]?.message ?? 'Invalid message',
+                code: 'INVALID_MESSAGE',
+            });
+        },
+        close: (socket) => {
+            running.get(socket)?.abort();
+            running.delete(socket);
+        },
+    });
+}
+async function handleRun(deps, socket, id, payload, auth, { signal }) {
+    const emit = (event) => send(socket, { type: 'event', id, payload: event });
+    try {
+        const result = await execute(deps, payload, auth, signal, emit);
+        emit(signal.aborted ? { type: 'cancelled' } : { type: 'complete', result });
+    }
+    catch (error) {
+        if (signal.aborted) {
+            emit({ type: 'cancelled' });
+            return;
+        }
+        const described = describeError(error);
+        send(socket, { type: 'error', id, error: described.message, code: described.code });
+        if (described.unexpected)
+            throw error;
+    }
+}
+async function execute(deps, payload, auth, signal, emit) {
+    switch (payload.type) {
+        case 'agent': {
+            const agent = findAgent(deps, payload.name);
+            await checkThreadAccess(deps, auth, payload.threadId);
+            const result = await runAgent(deps, agent, {
+                input: payload.input,
+                context: payload.context,
+                threadId: payload.threadId,
+                stream: true,
+                signal,
+                onToken: (delta) => emit({ type: 'token', delta }),
+                onToolCall: (call) => emit({ type: 'tool-call', id: call.id, name: call.name, arguments: call.arguments }),
+                onToolResult: (toolResult) => emit({ type: 'tool-result', ...toolResult }),
+            }, auth);
+            return toAgentRunResponse(result);
+        }
+        case 'workflow': {
+            const workflow = findWorkflow(deps, payload.name);
+            const result = await executeWorkflow(deps, workflow, { input: payload.input }, {}, signal);
+            return toWorkflowRunResponse(result);
+        }
+        case 'swarm': {
+            const config = findSwarm(deps, payload.name);
+            await checkThreadAccess(deps, auth, payload.threadId);
+            const { swarm, result } = await executeSwarm(deps, config, { input: payload.input, context: payload.context, threadId: payload.threadId }, signal);
+            return toSwarmRunResponse(swarm, result);
+        }
+    }
+}
+function send(socket, message) {
+    if (socket.readyState !== OPEN)
+        return;
+    socket.send(JSON.stringify(message));
+}
+//# sourceMappingURL=socket.js.map

package/dist/socket.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"socket.js","sourceRoot":"","sources":["../src/socket.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,eAAe,CAAC;AAInC,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EACL,iBAAiB,EACjB,YAAY,EACZ,eAAe,EACf,SAAS,EACT,SAAS,EACT,YAAY,EACZ,QAAQ,EACR,kBAAkB,EAClB,kBAAkB,EAClB,qBAAqB,GACtB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAW/C,MAAM,IAAI,GAAG,CAAC,CAAC;AACf,MAAM,YAAY,GAAG,KAAc,CAAC;AAEpC;;;;;;GAMG;AACH,MAAM,UAAU,eAAe,CAC7B,IAAmB,EACnB,MAAqC,EACrC,OAAqB,YAAY;IAEjC,MAAM,OAAO,GAAG,IAAI,OAAO,EAA2B,CAAC;IAEvD,OAAO,EAAE,CAAC;QACR,IAAI;QACJ,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,MAAM,CAAC,EAAE;QAChC,MAAM,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE;QAClC,IAAI,EAAE;YACJ,OAAO,EAAE,mDAAmD;YAC5D,WAAW,EACT,uGAAuG;SAC1G;QACD,KAAK,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC;QACtC,OAAO,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;YAC3B,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;gBACrB,KAAK,MAAM;oBACT,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC/C,OAAO;gBACT,KAAK,MAAM;oBACT,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC;oBAC7B,OAAO;gBACT,KAAK,KAAK,CAAC,CAAC,CAAC;oBACX,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;wBACxB,IAAI,CAAC,MAAM,EAAE;4BACX,IAAI,EAAE,OAAO;4BACb,EAAE,EAAE,OAAO,CAAC,EAAE;4BACd,KAAK,EAAE,8BAA8B;4BACrC,IAAI,EAAE,iBAAiB;yBACxB,CAAC,CAAC;wBACH,OAAO;oBACT,CAAC;oBACD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;oBACzC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;oBAChC,OAAO,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,aAAa,EAAE;wBACrF,MAAM,EAAE,UAAU,CAAC,MAAM;qBAC1B,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;wBACd,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,UAAU;4BAAE,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBACjE,CAAC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,EAAE,CAAC,MAAM,EAAE,MAAkC,EAAE,EAAE;YACtD,IAAI,CAAC,MAAM,EAAE;gBACX,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,IAAI,iBAAiB;gBAC9C,IAAI,EAAE,iBAAiB;aACxB,CAAC,CAAC;QACL,CAAC;QACD,KAAK,EAAE,CAAC,MAAM,EAAE,EAAE;YAChB,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC;YAC7B,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACzB,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,SAAS,CACtB,IAAmB,EACnB,MAAkB,EAClB,EAAsB,EACtB,OAAmB,EACnB,IAA6B,EAC7B,EAAE,MAAM,EAA2B;IAEnC,MAAM,IAAI,GAAG,CAAC,KAA8B,EAAE,EAAE,CAC9C,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IAEtD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QAChE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC;IAC9E,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QACpF,IAAI,SAAS,CAAC,UAAU;YAAE,MAAM,KAAK,CAAC;IACxC,CAAC;AACH,CAAC;AAED,KAAK,UAAU,OAAO,CACpB,IAAmB,EACnB,OAAmB,EACnB,IAA6B,EAC7B,MAAmB,EACnB,IAA8C;IAE9C,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,MAAM,KAAK,GAAG,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAC5C,MAAM,iBAAiB,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;YACtD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAC3B,IAAI,EACJ,KAAK,EACL;gBACE,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,MAAM,EAAE,IAAI;gBACZ,MAAM;gBACN,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;gBAClD,UAAU,EAAE,CAAC,IAAI,EAAE,EAAE,CACnB,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC;gBACtF,YAAY,EAAE,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,UAAU,EAAE,CAAC;aAC3E,EACD,IAAI,CACL,CAAC;YACF,OAAO,kBAAkB,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC;QAED,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,IAAI,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC;YAC3F,OAAO,qBAAqB,CAAC,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAC7C,MAAM,iBAAiB,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;YACtD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,YAAY,CAC1C,IAAI,EACJ,MAAM,EACN,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,EAC9E,MAAM,CACP,CAAC;YACF,OAAO,kBAAkB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,IAAI,CAAC,MAAkB,EAAE,OAA+B;IAC/D,IAAI,MAAM,CAAC,UAAU,KAAK,IAAI;QAAE,OAAO;IACvC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;AACvC,CAAC"}

package/dist/streaming.d.ts

@@ -0,0 +1,16 @@
+import type { ServerSentEvent } from '@tetsujs/sse';
+import type { StreamEvent } from '@cogitator-ai/server-shared';
+import type { ShutdownSignal } from './types.js';
+/** The last event of every successful stream, kept for clients of the other Cogitator adapters. */
+export declare const DONE_EVENT: ServerSentEvent;
+export declare function sseEvent(event: StreamEvent): ServerSentEvent;
+export declare function resolveSignal(until: ShutdownSignal | undefined): AbortSignal | undefined;
+/**
+ * Runs `task`, yielding every event it emits as soon as the reader asks for it.
+ *
+ * Ends when the task settles and its events are drained, rethrowing its error, or
+ * at once when `signal` aborts: a task that was given the signal stops by itself,
+ * and nobody is left to read what it would emit.
+ */
+export declare function eventsOf<T>(signal: AbortSignal, task: (emit: (event: T) => void) => Promise<void>): AsyncGenerator<T, void, undefined>;
+//# sourceMappingURL=streaming.d.ts.map

package/dist/streaming.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../src/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AACpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjD,mGAAmG;AACnG,eAAO,MAAM,UAAU,EAAE,eAAoC,CAAC;AAE9D,wBAAgB,QAAQ,CAAC,KAAK,EAAE,WAAW,GAAG,eAAe,CAE5D;AAED,wBAAgB,aAAa,CAAC,KAAK,EAAE,cAAc,GAAG,SAAS,GAAG,WAAW,GAAG,SAAS,CAExF;AA+CD;;;;;;GAMG;AACH,wBAAuB,QAAQ,CAAC,CAAC,EAC/B,MAAM,EAAE,WAAW,EACnB,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,IAAI,KAAK,OAAO,CAAC,IAAI,CAAC,GAChD,cAAc,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,CAAC,CAuBpC"}

package/dist/streaming.js

@@ -0,0 +1,82 @@
+/** The last event of every successful stream, kept for clients of the other Cogitator adapters. */
+export const DONE_EVENT = { data: '[DONE]' };
+export function sseEvent(event) {
+    return { data: event };
+}
+export function resolveSignal(until) {
+    return typeof until === 'function' ? until() : until;
+}
+/**
+ * A queue between callbacks that push events and a generator that pulls them.
+ *
+ * Callbacks of a run fire whenever the runtime gets there; the generator behind
+ * `sse()` asks for one event at a time, as the client reads. The queue holds what
+ * was pushed in between, and `end()` wakes a reader that is waiting.
+ */
+class EventQueue {
+    items = [];
+    ended = false;
+    wake;
+    push(item) {
+        if (this.ended)
+            return;
+        this.items.push(item);
+        this.notify();
+    }
+    end() {
+        this.ended = true;
+        this.notify();
+    }
+    async *drain() {
+        for (;;) {
+            const item = this.items.shift();
+            if (item !== undefined) {
+                yield item;
+            }
+            else if (this.ended) {
+                return;
+            }
+            else {
+                await new Promise((resolve) => {
+                    this.wake = resolve;
+                });
+            }
+        }
+    }
+    notify() {
+        const wake = this.wake;
+        this.wake = undefined;
+        wake?.();
+    }
+}
+/**
+ * Runs `task`, yielding every event it emits as soon as the reader asks for it.
+ *
+ * Ends when the task settles and its events are drained, rethrowing its error, or
+ * at once when `signal` aborts: a task that was given the signal stops by itself,
+ * and nobody is left to read what it would emit.
+ */
+export async function* eventsOf(signal, task) {
+    const queue = new EventQueue();
+    const stop = () => queue.end();
+    signal.addEventListener('abort', stop, { once: true });
+    if (signal.aborted)
+        stop();
+    let failure;
+    const settled = task((event) => queue.push(event)).then(() => queue.end(), (error) => {
+        failure = { error };
+        queue.end();
+    });
+    try {
+        yield* queue.drain();
+        if (signal.aborted)
+            return;
+        await settled;
+        if (failure)
+            throw failure.error;
+    }
+    finally {
+        signal.removeEventListener('abort', stop);
+    }
+}
+//# sourceMappingURL=streaming.js.map

package/dist/streaming.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.js","sourceRoot":"","sources":["../src/streaming.ts"],"names":[],"mappings":"AAIA,mGAAmG;AACnG,MAAM,CAAC,MAAM,UAAU,GAAoB,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;AAE9D,MAAM,UAAU,QAAQ,CAAC,KAAkB;IACzC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,KAAiC;IAC7D,OAAO,OAAO,KAAK,KAAK,UAAU,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;AACvD,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU;IACG,KAAK,GAAQ,EAAE,CAAC;IACzB,KAAK,GAAG,KAAK,CAAC;IACd,IAAI,CAA2B;IAEvC,IAAI,CAAC,IAAO;QACV,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO;QACvB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,IAAI,CAAC,MAAM,EAAE,CAAC;IAChB,CAAC;IAED,GAAG;QACD,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,MAAM,EAAE,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,CAAC,KAAK;QACV,SAAS,CAAC;YACR,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YAChC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBACvB,MAAM,IAAI,CAAC;YACb,CAAC;iBAAM,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACtB,OAAO;YACT,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;oBAClC,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC;gBACtB,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAEO,MAAM;QACZ,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACvB,IAAI,CAAC,IAAI,GAAG,SAAS,CAAC;QACtB,IAAI,EAAE,EAAE,CAAC;IACX,CAAC;CACF;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,QAAQ,CAC7B,MAAmB,EACnB,IAAiD;IAEjD,MAAM,KAAK,GAAG,IAAI,UAAU,EAAK,CAAC;IAClC,MAAM,IAAI,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;IAC/B,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACvD,IAAI,MAAM,CAAC,OAAO;QAAE,IAAI,EAAE,CAAC;IAE3B,IAAI,OAAuC,CAAC;IAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CACrD,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EACjB,CAAC,KAAc,EAAE,EAAE;QACjB,OAAO,GAAG,EAAE,KAAK,EAAE,CAAC;QACpB,KAAK,CAAC,GAAG,EAAE,CAAC;IACd,CAAC,CACF,CAAC;IAEF,IAAI,CAAC;QACH,KAAK,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,MAAM,CAAC,OAAO;YAAE,OAAO;QAC3B,MAAM,OAAO,CAAC;QACd,IAAI,OAAO;YAAE,MAAM,OAAO,CAAC,KAAK,CAAC;IACnC,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC5C,CAAC;AACH,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,84 @@
+import type { BaseCtx } from '@tetsujs/core';
+import type { CallerHook } from './auth.js';
+import type { z } from 'zod';
+import type { Agent, Cogitator } from '@cogitator-ai/core';
+import type { SwarmConfig, Workflow, WorkflowState } from '@cogitator-ai/types';
+import type { AddMessageBody, AgentListResponse, AgentRunResponse, BlackboardResponse, HealthResponse, RunBody, SocketMessage, SwarmListResponse, SwarmRunBody, SwarmRunResponse, ThreadResponse, ToolListResponse, WorkflowListResponse, WorkflowRunBody, WorkflowRunResponse } from './schemas.js';
+/**
+ * Who is calling, as the `auth` function established it.
+ *
+ * `userId` is passed to every agent run, so memory and cost tracking are
+ * attributed to the caller.
+ */
+export interface AuthContext {
+    userId?: string;
+    roles?: string[];
+    permissions?: string[];
+    metadata?: Record<string, unknown>;
+}
+/**
+ * Establishes the caller before the request body is read.
+ *
+ * Return the caller to let the request through, or `undefined` to refuse it
+ * with `401 UNAUTHORIZED`. Throw an `HttpError` to answer with another status.
+ * Return an empty object to accept anonymous callers.
+ */
+export type Authenticate = (ctx: BaseCtx) => AuthContext | undefined | Promise<AuthContext | undefined>;
+/**
+ * Decides whether the caller may read or write a memory thread.
+ *
+ * Checked on the thread routes and on every run, stream and WebSocket run
+ * that names a `threadId`. A refusal answers `403 THREAD_FORBIDDEN`.
+ */
+export type AuthorizeThread = (auth: AuthContext | undefined, threadId: string) => boolean | Promise<boolean>;
+/**
+ * A signal that ends open SSE streams and WebSocket connections, such as
+ * `draining` from `@tetsujs/lifecycle`, or a function that returns it.
+ */
+export type ShutdownSignal = AbortSignal | (() => AbortSignal | undefined);
+export interface WebSocketOptions {
+    /** Path of the handshake. Default: `/ws`. */
+    path?: `/${string}`;
+}
+/** What `cogitatorController()` is built from. */
+export interface CogitatorDeps {
+    cogitator: Cogitator;
+    agents?: Record<string, Agent>;
+    workflows?: Record<string, Workflow<WorkflowState>>;
+    swarms?: Record<string, SwarmConfig>;
+    /**
+     * Establishes the caller before every route except `/health` and `/ready`, and
+     * before the WebSocket handshake: a function, or a `callerHook()` wrapped in
+     * `secured()` from `@tetsujs/openapi` to document it.
+     */
+    auth?: Authenticate | CallerHook;
+    authorizeThread?: AuthorizeThread;
+    /** Serve the WebSocket endpoint. Default: `false`. */
+    websocket?: boolean | WebSocketOptions;
+    /** Ends streams and sockets when the server starts draining. */
+    until?: ShutdownSignal;
+}
+export type AgentRunRequest = z.input<typeof RunBody>;
+export type SwarmRunRequest = z.input<typeof SwarmRunBody>;
+export type WorkflowRunRequest = z.input<typeof WorkflowRunBody>;
+export type AddMessageRequest = z.input<typeof AddMessageBody>;
+export type HealthResponseBody = z.output<typeof HealthResponse>;
+export type AgentListResponseBody = z.output<typeof AgentListResponse>;
+export type AgentRunResponseBody = z.output<typeof AgentRunResponse>;
+export type ThreadResponseBody = z.output<typeof ThreadResponse>;
+export type ToolListResponseBody = z.output<typeof ToolListResponse>;
+export type WorkflowListResponseBody = z.output<typeof WorkflowListResponse>;
+export type WorkflowRunResponseBody = z.output<typeof WorkflowRunResponse>;
+export type SwarmListResponseBody = z.output<typeof SwarmListResponse>;
+export type SwarmRunResponseBody = z.output<typeof SwarmRunResponse>;
+export type BlackboardResponseBody = z.output<typeof BlackboardResponse>;
+export type WebSocketClientMessage = z.input<typeof SocketMessage>;
+/** A frame the server sends over the WebSocket. */
+export interface WebSocketServerMessage {
+    type: 'event' | 'error' | 'pong';
+    id?: string;
+    payload?: unknown;
+    error?: string;
+    code?: string;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AAC5C,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAC7B,OAAO,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,KAAK,EAAE,WAAW,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAChF,OAAO,KAAK,EACV,cAAc,EACd,iBAAiB,EACjB,gBAAgB,EAChB,kBAAkB,EAClB,cAAc,EACd,OAAO,EACP,aAAa,EACb,iBAAiB,EACjB,YAAY,EACZ,gBAAgB,EAChB,cAAc,EACd,gBAAgB,EAChB,oBAAoB,EACpB,eAAe,EACf,mBAAmB,EACpB,MAAM,cAAc,CAAC;AAEtB;;;;;GAKG;AACH,MAAM,WAAW,WAAW;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;;;;;GAMG;AACH,MAAM,MAAM,YAAY,GAAG,CACzB,GAAG,EAAE,OAAO,KACT,WAAW,GAAG,SAAS,GAAG,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC,CAAC;AAEhE;;;;;GAKG;AACH,MAAM,MAAM,eAAe,GAAG,CAC5B,IAAI,EAAE,WAAW,GAAG,SAAS,EAC7B,QAAQ,EAAE,MAAM,KACb,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;AAEhC;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG,WAAW,GAAG,CAAC,MAAM,WAAW,GAAG,SAAS,CAAC,CAAC;AAE3E,MAAM,WAAW,gBAAgB;IAC/B,6CAA6C;IAC7C,IAAI,CAAC,EAAE,IAAI,MAAM,EAAE,CAAC;CACrB;AAED,kDAAkD;AAClD,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,SAAS,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC/B,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;IACpD,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IACrC;;;;OAIG;IACH,IAAI,CAAC,EAAE,YAAY,GAAG,UAAU,CAAC;IACjC,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,sDAAsD;IACtD,SAAS,CAAC,EAAE,OAAO,GAAG,gBAAgB,CAAC;IACvC,gEAAgE;IAChE,KAAK,CAAC,EAAE,cAAc,CAAC;CACxB;AAED,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,OAAO,CAAC,CAAC;AACtD,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,YAAY,CAAC,CAAC;AAC3D,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,eAAe,CAAC,CAAC;AACjE,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAC;AAC/D,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,cAAc,CAAC,CAAC;AACjE,MAAM,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,iBAAiB,CAAC,CAAC;AACvE,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,gBAAgB,CAAC,CAAC;AACrE,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,cAAc,CAAC,CAAC;AACjE,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,gBAAgB,CAAC,CAAC;AACrE,MAAM,MAAM,wBAAwB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAC7E,MAAM,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAC3E,MAAM,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,iBAAiB,CAAC,CAAC;AACvE,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,gBAAgB,CAAC,CAAC;AACrE,MAAM,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,kBAAkB,CAAC,CAAC;AACzE,MAAM,MAAM,sBAAsB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AAEnE,mDAAmD;AACnD,MAAM,WAAW,sBAAsB;IACrC,IAAI,EAAE,OAAO,GAAG,OAAO,GAAG,MAAM,CAAC;IACjC,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,CAAC;CACf"}

package/dist/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}

package/package.json

@@ -0,0 +1,72 @@
+{
+  "name": "@cogitator-ai/tetsu",
+  "version": "0.1.0",
+  "description": "Tetsu server adapter for Cogitator AI runtime — a controller with agents, threads, workflows, swarms, SSE and WebSocket on Bun",
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist"
+  ],
+  "engines": {
+    "bun": ">=1.4"
+  },
+  "dependencies": {
+    "@cogitator-ai/core": "0.21.1",
+    "@cogitator-ai/memory": "0.8.0",
+    "@cogitator-ai/types": "0.24.0",
+    "@cogitator-ai/server-shared": "0.2.0",
+    "zod": "^4.6.5"
+  },
+  "peerDependencies": {
+    "@tetsujs/core": "^0.6.1",
+    "@tetsujs/sse": "^0.6.1"
+  },
+  "optionalDependencies": {
+    "@cogitator-ai/workflows": "0.7.1",
+    "@cogitator-ai/swarms": "0.6.1"
+  },
+  "devDependencies": {
+    "@cogitator-ai/workflows": "0.7.1",
+    "@cogitator-ai/swarms": "0.6.1",
+    "@tetsujs/core": "^0.6.1",
+    "@tetsujs/openapi": "^0.6.1",
+    "@tetsujs/sse": "^0.6.1",
+    "@types/bun": "^1.4.2",
+    "typescript": "^6.0.3"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/cogitator-ai/Cogitator-AI.git",
+    "directory": "packages/tetsu"
+  },
+  "license": "MIT",
+  "keywords": [
+    "cogitator",
+    "ai",
+    "agents",
+    "tetsu",
+    "bun",
+    "server",
+    "api",
+    "streaming",
+    "sse",
+    "websocket",
+    "openapi",
+    "llm"
+  ],
+  "scripts": {
+    "build": "tsc && tsx ../../scripts/fix-esm-imports.ts .",
+    "dev": "tsc --watch",
+    "test": "bun test",
+    "test:watch": "bun test --watch",
+    "typecheck": "tsc --noEmit -p tsconfig.test.json",
+    "clean": "rm -rf dist"
+  }
+}