@m6d/cortex-server 1.7.0 → 2.0.0

package/src/config.ts

@@ -1,182 +0,0 @@
-import type { Tool, ToolSet, UIMessage } from "ai";
-import type { DatabaseAdapter } from "./adapters/database";
-import type { StorageAdapter } from "./adapters/storage";
-import type { DomainDef } from "./graph/types.ts";
-import type { RequestInterceptorOptions } from "./ai/interceptors/request-interceptor.ts";
-import type { ContextConfig } from "./ai/context/types.ts";
-
-export type KnowledgeConfig = {
-    swagger?: { url: string };
-    domains?: Record<string, DomainDef>;
-};
-
-export type PromptContext<
-    TSession extends Record<string, unknown> = Record<string, unknown>,
-    TRequestContext extends Record<string, unknown> = Record<string, unknown>,
-> = {
-    session: TSession | null;
-    requestContext: TRequestContext;
-};
-
-export type DatabaseConfig = {
-    type: "mssql";
-    connectionString: string;
-};
-
-export type StorageConfig = {
-    endPoint: string;
-    port: number;
-    useSSL: boolean;
-    accessKey: string;
-    secretKey: string;
-    bucketName?: string;
-};
-
-export type CortexAgentDefinition<
-    TSession extends Record<string, unknown> = Record<string, unknown>,
-    TRequestContext extends Record<string, unknown> = Record<string, unknown>,
-> = {
-    systemPrompt:
-        | string
-        | ((context: PromptContext<TSession, TRequestContext>) => string | Promise<string>);
-    tools?: ToolSet;
-    backendFetch?: {
-        baseUrl: string;
-        apiKey: string;
-        headers?: Record<string, string>;
-        transformRequestBody?: (
-            body: Record<string, unknown>,
-            context: { token: string },
-        ) => Promise<Record<string, unknown>>;
-        interceptor?: RequestInterceptorOptions;
-    };
-    loadSessionData?: (token: string) => Promise<TSession>;
-    resolveRequestContext?: (request: Request) => TRequestContext | Promise<TRequestContext>;
-    onToolCall?: (toolCall: {
-        toolName: string;
-        toolCallId: string;
-        args: Record<string, unknown>;
-    }) => void;
-    onStreamFinish?: (result: { messages: UIMessage[]; isAborted: boolean }) => void;
-    model?: {
-        baseURL: string;
-        apiKey: string;
-        modelName: string;
-        providerName?: string;
-    };
-    embedding?: {
-        baseURL: string;
-        apiKey: string;
-        modelName: string;
-        dimension: number;
-    };
-    neo4j?: {
-        url: string;
-        user: string;
-        password: string;
-    };
-    reranker?: {
-        url: string;
-        apiKey: string;
-    };
-    context?: Partial<ContextConfig>;
-    knowledge?: KnowledgeConfig | null;
-};
-
-export type ResolvedCortexAgentConfig = {
-    db: DatabaseAdapter;
-    storage: StorageAdapter;
-    model: {
-        baseURL: string;
-        apiKey: string;
-        modelName: string;
-        providerName?: string;
-    };
-    embedding: {
-        baseURL: string;
-        apiKey: string;
-        modelName: string;
-        dimension: number;
-    };
-    neo4j: {
-        url: string;
-        user: string;
-        password: string;
-    };
-    reranker?: {
-        url: string;
-        apiKey: string;
-    };
-    systemPrompt: string | ((context: PromptContext) => string | Promise<string>);
-    tools?: ToolSet;
-    backendFetch?: {
-        baseUrl: string;
-        apiKey: string;
-        headers?: Record<string, string>;
-        transformRequestBody?: (
-            body: Record<string, unknown>,
-            context: { token: string },
-        ) => Promise<Record<string, unknown>>;
-        interceptor?: RequestInterceptorOptions;
-    };
-    loadSessionData?: (token: string) => Promise<Record<string, unknown>>;
-    resolveRequestContext?: (
-        request: Request,
-    ) => Record<string, unknown> | Promise<Record<string, unknown>>;
-    onToolCall?: (toolCall: {
-        toolName: string;
-        toolCallId: string;
-        args: Record<string, unknown>;
-    }) => void;
-    onStreamFinish?: (result: { messages: UIMessage[]; isAborted: boolean }) => void;
-    context: ContextConfig;
-    knowledge?: KnowledgeConfig;
-};
-
-export type CortexConfig = {
-    port?: number;
-    database: DatabaseConfig;
-    storage: StorageConfig;
-    auth: {
-        jwksUri: string;
-        issuer: string;
-        tokenExtractor?: (req: Request) => string | null;
-        cookieName?: string;
-    };
-    model: {
-        baseURL: string;
-        apiKey: string;
-        modelName: string;
-        providerName?: string;
-    };
-    embedding: {
-        baseURL: string;
-        apiKey: string;
-        modelName: string;
-        dimension: number;
-    };
-    neo4j: {
-        url: string;
-        user: string;
-        password: string;
-    };
-    reranker?: {
-        url: string;
-        apiKey: string;
-    };
-    context?: Partial<ContextConfig>;
-    knowledge?: KnowledgeConfig;
-    agents: Record<string, CortexAgentDefinition>;
-};
-
-/**
- * Helper to define an agent with full type inference for `systemPrompt` context.
- * The `context.session` type is inferred from `loadSessionData`'s return type,
- * and `context.requestContext` is inferred from `resolveRequestContext`'s return type.
- */
-export function defineAgent<
-    TSession extends Record<string, unknown> = Record<string, unknown>,
-    TRequestContext extends Record<string, unknown> = Record<string, unknown>,
->(config: CortexAgentDefinition<TSession, TRequestContext>): CortexAgentDefinition {
-    return config as CortexAgentDefinition;
-}

package/src/db/migrate.ts

@@ -1,21 +0,0 @@
-import mssql from "mssql";
-import { drizzle } from "drizzle-orm/node-mssql";
-import { migrate } from "drizzle-orm/node-mssql/migrator";
-import { resolve } from "path";
-
-export async function runMigrations(connectionString: string) {
-    const pool = new mssql.ConnectionPool(connectionString);
-    await pool.connect();
-
-    try {
-        const db = drizzle({ client: pool, casing: "snake_case" });
-        const migrationsFolder = resolve(import.meta.dir, "migrations");
-        console.log("[cortex-server] Running migrations from:", migrationsFolder);
-        await migrate(db, { migrationsFolder });
-        console.log("[cortex-server] Migrations complete");
-    } catch (e) {
-        console.log(e);
-    } finally {
-        await pool.close();
-    }
-}

package/src/db/migrations/20260315000000_add_context_meta/migration.sql

@@ -1 +0,0 @@
-ALTER TABLE [ai].[threads] ADD [context_meta] nvarchar(max);

package/src/db/schema.ts

@@ -1,70 +0,0 @@
-import type { ToolUIPart, UIMessage } from "ai";
-import { sql } from "drizzle-orm";
-import { customType, datetime2, index, mssqlSchema, nvarchar } from "drizzle-orm/mssql-core";
-import type { MessageMetadata } from "src/types";
-import type { ThreadContextMeta } from "../ai/context/types.ts";
-
-const uniqueIdentifier = customType<{ data: string }>({
-    dataType() {
-        return "UNIQUEIDENTIFIER";
-    },
-});
-
-export const aiSchema = mssqlSchema("ai");
-
-const auditColumns = {
-    createdAt: datetime2({ mode: "date" })
-        .$default(() => new Date())
-        .notNull(),
-    updatedAt: datetime2({ mode: "date" })
-        .$default(() => new Date())
-        .$onUpdate(() => new Date())
-        .notNull(),
-};
-
-export const threads = aiSchema.table("threads", {
-    id: uniqueIdentifier()
-        .default(sql`NEWID()`)
-        .primaryKey(),
-    userId: uniqueIdentifier().notNull(),
-    agentId: nvarchar({ length: 128 }).notNull().default("default"),
-    title: nvarchar({ length: 256 }),
-    session: nvarchar({ mode: "json", length: "max" }).$type<Record<string, unknown>>(),
-    contextMeta: nvarchar({ mode: "json", length: "max" }).$type<ThreadContextMeta>(),
-    ...auditColumns,
-});
-
-export const messages = aiSchema.table("messages", {
-    id: nvarchar({ length: 128 }).primaryKey(),
-    threadId: uniqueIdentifier()
-        .references(() => threads.id, { onDelete: "cascade" })
-        .notNull(),
-    text: nvarchar({ length: "max" }),
-    content: nvarchar({ mode: "json", length: "max" })
-        .notNull()
-        .$type<UIMessage<MessageMetadata>>(),
-    role: nvarchar({
-        enum: ["system", "user", "assistant", "tool"],
-        length: 16,
-    }).notNull(),
-    ...auditColumns,
-});
-
-export const capturedFiles = aiSchema.table(
-    "captured_files",
-    {
-        id: uniqueIdentifier()
-            .default(sql`NEWID()`)
-            .primaryKey(),
-        name: nvarchar({ length: 2048 }).notNull(),
-        userId: uniqueIdentifier().notNull(),
-        agentGeneratedId: nvarchar({ length: 1024 }).notNull(),
-        messageId: nvarchar({ length: 128 })
-            .references(() => messages.id, {
-                onDelete: "cascade",
-            })
-            .notNull(),
-        toolPart: nvarchar({ length: "max", mode: "json" }).notNull().$type<ToolUIPart>(),
-    },
-    (table) => [index(table.agentGeneratedId)],
-);

package/src/factory.ts

@@ -1,171 +0,0 @@
-import { Hono } from "hono";
-import { websocket } from "hono/bun";
-import { createOpenAICompatible } from "@ai-sdk/openai-compatible";
-import type { CortexConfig, ResolvedCortexAgentConfig } from "./config.ts";
-import { DEFAULT_CONTEXT_CONFIG } from "./ai/context/types.ts";
-import type { ContextConfig } from "./ai/context/types.ts";
-import type { AppEnv, CortexAppEnv } from "./types.ts";
-import type { DomainDef } from "./graph/types.ts";
-import { createUserLoaderMiddleware } from "./auth/middleware.ts";
-import { createThreadRoutes } from "./routes/threads.ts";
-import { createChatRoutes } from "./routes/chat.ts";
-import { createFileRoutes } from "./routes/files.ts";
-import { createWsRoute } from "./routes/ws.ts";
-import { runMigrations } from "./db/migrate.ts";
-import { createMssqlAdapter } from "./adapters/mssql.ts";
-import { createMinioAdapter } from "./adapters/minio.ts";
-import { seedGraph as seedGraphFn } from "./graph/seed.ts";
-import { extractEndpoints as extractEndpointsFn } from "./cli/extract-endpoints.ts";
-
-export type CortexInstance = {
-    serve(): Promise<{
-        app: Hono<AppEnv>;
-        port: number;
-        fetch: Hono<AppEnv>["fetch"];
-        websocket: typeof websocket;
-    }>;
-    seedGraph(): Promise<void>;
-    extractEndpoints(options: { domainsDir: string; write?: boolean }): Promise<void>;
-};
-
-export function createCortex(config: CortexConfig) {
-    function collectAllDomains() {
-        const all: Record<string, DomainDef> = {};
-        if (config.knowledge?.domains) Object.assign(all, config.knowledge.domains);
-        for (const agent of Object.values(config.agents)) {
-            if (agent.knowledge && agent.knowledge.domains) {
-                Object.assign(all, agent.knowledge.domains);
-            }
-        }
-        return all;
-    }
-
-    function collectAllSwaggerUrls() {
-        const urls = new Set<string>();
-        if (config.knowledge?.swagger?.url) urls.add(config.knowledge.swagger.url);
-        for (const agent of Object.values(config.agents)) {
-            if (agent.knowledge && agent.knowledge.swagger?.url) {
-                urls.add(agent.knowledge.swagger.url);
-            }
-        }
-        return [...urls];
-    }
-
-    return {
-        async serve() {
-            const storage = createMinioAdapter(config.storage);
-            const db = createMssqlAdapter(config.database.connectionString, storage);
-
-            await runMigrations(config.database.connectionString);
-
-            const app = new Hono<AppEnv>();
-
-            const userLoader = createUserLoaderMiddleware(config.auth);
-            app.use("*", userLoader);
-
-            // Compatibility WebSocket route
-            app.route("/", createWsRoute());
-
-            // Agent-scoped routes
-            const agentApp = new Hono<CortexAppEnv>();
-
-            agentApp.use("*", async function (c, next) {
-                const agentId = c.req.param("agentId") as string;
-                const agentDef = config.agents[agentId];
-                if (!agentDef) return c.json({ error: "Agent not found" }, 404);
-
-                const resolvedContext: ContextConfig = {
-                    ...DEFAULT_CONTEXT_CONFIG,
-                    ...config.context,
-                    ...agentDef.context,
-                };
-
-                const resolvedConfig: ResolvedCortexAgentConfig = {
-                    db,
-                    storage,
-                    model: agentDef.model ?? config.model,
-                    embedding: agentDef.embedding ?? config.embedding,
-                    neo4j: agentDef.neo4j ?? config.neo4j,
-                    reranker: agentDef.reranker ?? config.reranker,
-                    knowledge:
-                        agentDef.knowledge === null
-                            ? undefined
-                            : (agentDef.knowledge ?? config.knowledge),
-                    systemPrompt: agentDef.systemPrompt,
-                    tools: agentDef.tools,
-                    backendFetch: agentDef.backendFetch,
-                    loadSessionData: agentDef.loadSessionData,
-                    resolveRequestContext: agentDef.resolveRequestContext,
-                    onToolCall: agentDef.onToolCall,
-                    onStreamFinish: agentDef.onStreamFinish,
-                    context: resolvedContext,
-                };
-
-                c.set("agentConfig", resolvedConfig);
-                c.set("agentId", agentId);
-                await next();
-            });
-
-            agentApp.route("/", createThreadRoutes());
-            agentApp.route("/", createChatRoutes());
-            agentApp.route("/", createFileRoutes());
-            agentApp.route("/", createWsRoute({ useAgentParam: true }));
-
-            app.route("/agents/:agentId", agentApp);
-
-            const port = config.port ?? 3331;
-
-            return {
-                app,
-                port,
-                fetch: app.fetch,
-                websocket,
-            };
-        },
-
-        async seedGraph() {
-            const domains = collectAllDomains();
-            if (!Object.keys(domains).length) {
-                throw new Error("No domains found in knowledge config");
-            }
-            if (!config.neo4j) {
-                throw new Error("neo4j config is required for seedGraph");
-            }
-            if (!config.embedding) {
-                throw new Error("embedding config is required for seedGraph");
-            }
-
-            const provider = createOpenAICompatible({
-                name: "embedding-provider",
-                baseURL: config.embedding.baseURL,
-                apiKey: config.embedding.apiKey,
-            });
-
-            await seedGraphFn(
-                {
-                    neo4j: config.neo4j,
-                    embedding: {
-                        model: provider.textEmbeddingModel(config.embedding.modelName),
-                        dimension: config.embedding.dimension,
-                    },
-                },
-                domains,
-            );
-        },
-
-        async extractEndpoints(options) {
-            const swaggerUrls = collectAllSwaggerUrls();
-            if (!swaggerUrls.length) {
-                throw new Error("No swagger URLs found in knowledge config");
-            }
-
-            for (const swaggerUrl of swaggerUrls) {
-                await extractEndpointsFn({
-                    swaggerUrl,
-                    domainsDir: options.domainsDir,
-                    write: options.write ?? true,
-                });
-            }
-        },
-    } satisfies CortexInstance;
-}