@m6d/cortex-server 1.1.0 → 1.1.2

package/src/db/schema.ts

@@ -1,77 +1,65 @@
 import type { ToolUIPart, UIMessage } from "ai";
 import { sql } from "drizzle-orm";
-import {
-  customType,
-  datetime2,
-  index,
-  mssqlSchema,
-  nvarchar,
-} from "drizzle-orm/mssql-core";
+import { customType, datetime2, index, mssqlSchema, nvarchar } from "drizzle-orm/mssql-core";
 
 const uniqueIdentifier = customType<{ data: string }>({
-  dataType() {
-    return "UNIQUEIDENTIFIER";
-  },
+    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(),
+    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>
-  >(),
-  ...auditColumns,
+    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>>(),
+    ...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>(),
-  role: nvarchar({
-    enum: ["system", "user", "assistant", "tool"],
-    length: 16,
-  }).notNull(),
-  ...auditColumns,
+    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>(),
+    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)],
+    "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

@@ -16,148 +16,145 @@ 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>;
+    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): CortexInstance {
-  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);
-      }
+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;
     }
-    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);
-      }
+
+    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 [...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);
-
-      // WebSocket route (global, not agent-scoped)
-      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 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,
-          onToolCall: agentDef.onToolCall,
-          onStreamFinish: agentDef.onStreamFinish,
-        };
-
-        c.set("agentConfig", resolvedConfig);
-        c.set("agentId", agentId);
-        await next();
-      });
-
-      agentApp.route("/", createThreadRoutes());
-      agentApp.route("/", createChatRoutes());
-      agentApp.route("/", createFileRoutes());
-
-      app.route("/agents/:agentId", agentApp);
-
-      const port = config.port ?? 3000;
-
-      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,
-          },
+
+    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);
+
+            // WebSocket route (global, not agent-scoped)
+            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 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,
+                    onToolCall: agentDef.onToolCall,
+                    onStreamFinish: agentDef.onStreamFinish,
+                };
+
+                c.set("agentConfig", resolvedConfig);
+                c.set("agentId", agentId);
+                await next();
+            });
+
+            agentApp.route("/", createThreadRoutes());
+            agentApp.route("/", createChatRoutes());
+            agentApp.route("/", createFileRoutes());
+
+            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,
+                });
+            }
         },
-        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;
 }