@m6d/cortex-server 1.7.0 → 2.0.0

package/src/routes/files.ts

@@ -1,100 +0,0 @@
-import { Hono } from "hono";
-import { zValidator } from "@hono/zod-validator";
-import z from "zod";
-import { HTTPException } from "hono/http-exception";
-import { safeValidateUIMessages, type ToolUIPart, type UIMessage } from "ai";
-import type { CortexAppEnv } from "../types.ts";
-import { requireAuth } from "../auth/middleware.ts";
-
-export function createFileRoutes() {
-    const app = new Hono<CortexAppEnv>();
-
-    app.get("/files/:id", requireAuth, async function (c) {
-        const config = c.get("agentConfig");
-        const id = c.req.param("id");
-        const { id: userId } = c.get("user");
-        const file = await config.db.capturedFiles.getById(id, userId);
-
-        if (!file) {
-            throw new HTTPException(404, { message: "File not found" });
-        }
-
-        const fileStream = await config.storage.stream(`captured_files/${id}`);
-        return new Response(fileStream, {
-            headers: {
-                "Content-Type": "application/octet-stream",
-                "Content-Disposition": `attachment; filename="${file.name}"`,
-            },
-        });
-    });
-
-    app.post(
-        "/files",
-        requireAuth,
-        zValidator(
-            "json",
-            z.object({
-                toolCallId: z.string(),
-                message: z.unknown().transform(async (message, ctx) => {
-                    const result = await safeValidateUIMessages({
-                        messages: [message],
-                    });
-                    if (!result.success) {
-                        ctx.addIssue({
-                            code: "custom",
-                            path: ["message"],
-                            message: "Invalid message format",
-                        });
-                        return z.NEVER;
-                    }
-
-                    return result.data[0]!;
-                }),
-                files: z.array(
-                    z.object({
-                        id: z.string(),
-                        file: z.object({
-                            name: z.string(),
-                            bytes: z.string(),
-                        }),
-                    }),
-                ),
-            }),
-        ),
-        async function (c) {
-            const config = c.get("agentConfig");
-            const { message, toolCallId, files } = c.req.valid("json");
-            const { id: userId } = c.get("user");
-
-            function isTool(value: UIMessage["parts"][number]): value is ToolUIPart {
-                return value.type.startsWith("tool-");
-            }
-
-            const toolPart = message.parts.filter(isTool).find((x) => x.toolCallId === toolCallId);
-
-            if (!toolPart) {
-                throw new HTTPException(423, {
-                    message: "Invalid tool call id",
-                });
-            }
-
-            const result = await config.db.capturedFiles.create(
-                userId,
-                message.id,
-                toolPart,
-                files,
-            );
-
-            // Store file bytes
-            await Promise.all(
-                result.map(async (r, i) => {
-                    await config.storage.put(`captured_files/${r.uploadId}`, files[i]!.file.bytes);
-                }),
-            );
-
-            return c.json(result);
-        },
-    );
-
-    return app;
-}

package/src/routes/index.ts

@@ -1,4 +0,0 @@
-export { createThreadRoutes } from "./threads.ts";
-export { createChatRoutes } from "./chat.ts";
-export { createFileRoutes } from "./files.ts";
-export { createWsRoute } from "./ws.ts";

package/src/types.ts

@@ -1,74 +0,0 @@
-import type { ToolUIPart, UIMessage } from "ai";
-import type { ResolvedCortexAgentConfig } from "./config";
-import type { InferSelectModel } from "drizzle-orm";
-import type { messages, threads } from "./db/schema";
-
-export type Thread = InferSelectModel<typeof threads>;
-
-export type StoredMessage = InferSelectModel<typeof messages>;
-
-export type ThreadSummary = {
-    id: string;
-    title?: string;
-    createdAt: Date;
-    updatedAt: Date;
-    isRunning: boolean;
-};
-
-export type MessageMetadata = {
-    modelId: string;
-    providerMetadata: unknown;
-    isAborted?: boolean;
-    tokenUsage?: {
-        input: {
-            noCache: number;
-            cacheRead: number;
-            cacheWrite: number;
-            total: number;
-        };
-        output: {
-            reasoning: number;
-            text: number;
-            total: number;
-        };
-        total: number;
-    };
-};
-
-export type CapturedFileInput = {
-    id: string;
-    file: { name: string; bytes: string };
-};
-
-export type AppEnv = {
-    Bindings: {};
-    Variables: {
-        user: { id: string; token: string } | undefined;
-    };
-};
-
-export type AuthedAppEnv = {
-    Bindings: {};
-    Variables: {
-        user: { id: string; token: string };
-    };
-};
-
-export type CortexAppEnv = {
-    Bindings: {};
-    Variables: {
-        user: { id: string; token: string };
-        agentConfig: ResolvedCortexAgentConfig;
-        agentId: string;
-    };
-};
-
-export function toThreadSummary(thread: Thread, isRunning: boolean) {
-    return {
-        id: thread.id,
-        title: thread.title ?? undefined,
-        createdAt: thread.createdAt,
-        updatedAt: thread.updatedAt,
-        isRunning,
-    } satisfies ThreadSummary;
-}

package/src/ws/events.ts

@@ -1,39 +0,0 @@
-import type { ThreadSummary } from "../types.ts";
-
-export type ThreadCreatedEvent = {
-    type: "thread:created";
-    payload: { thread: ThreadSummary };
-};
-
-export type ThreadDeletedEvent = {
-    type: "thread:deleted";
-    payload: { threadId: string };
-};
-
-export type ThreadTitleUpdatedEvent = {
-    type: "thread:title-updated";
-    payload: { thread: ThreadSummary };
-};
-
-export type ThreadRunStartedEvent = {
-    type: "thread:run-started";
-    payload: { thread: ThreadSummary };
-};
-
-export type ThreadRunFinishedEvent = {
-    type: "thread:run-finished";
-    payload: { thread: ThreadSummary };
-};
-
-export type ThreadMessagesUpdatedEvent = {
-    type: "thread:messages-updated";
-    payload: { threadId: string; thread: ThreadSummary };
-};
-
-export type WsEvent =
-    | ThreadCreatedEvent
-    | ThreadDeletedEvent
-    | ThreadTitleUpdatedEvent
-    | ThreadRunStartedEvent
-    | ThreadRunFinishedEvent
-    | ThreadMessagesUpdatedEvent;

package/src/ws/index.ts

@@ -1,11 +0,0 @@
-export { addConnection, removeConnection, getConnections } from "./connections.ts";
-export { notify } from "./notify.ts";
-export type {
-    WsEvent,
-    ThreadCreatedEvent,
-    ThreadDeletedEvent,
-    ThreadTitleUpdatedEvent,
-    ThreadRunStartedEvent,
-    ThreadRunFinishedEvent,
-    ThreadMessagesUpdatedEvent,
-} from "./events.ts";

package/src/ws/notify.ts

@@ -1,9 +0,0 @@
-import { getConnections } from "./connections.ts";
-import type { WsEvent } from "./events.ts";
-
-export function notify(userId: string, agentId: string, event: WsEvent) {
-    const message = JSON.stringify(event);
-    for (const ws of getConnections(userId, agentId)) {
-        ws.send(message);
-    }
-}