@m6d/cortex-server 1.0.0

package/src/config.ts

@@ -0,0 +1,155 @@
+import type { UIMessage } from "ai";
+import type { DatabaseAdapter } from "./adapters/database";
+import type { StorageAdapter } from "./adapters/storage";
+import type { DomainDef } from "./graph/types.ts";
+
+export type KnowledgeConfig = {
+  swagger?: { url: string };
+  domains?: Record<string, DomainDef>;
+};
+
+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 = {
+  systemPrompt:
+    | string
+    | ((session: Record<string, unknown> | null) => string | Promise<string>);
+  tools?: Record<string, unknown>;
+  backendFetch?: {
+    baseUrl: string;
+    apiKey: string;
+    headers?: Record<string, string>;
+    transformRequestBody?: (
+      body: Record<string, unknown>,
+      context: { token: string },
+    ) => Promise<Record<string, unknown>>;
+  };
+  loadSessionData?: (token: string) => Promise<Record<string, unknown>>;
+  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;
+  };
+  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
+    | ((session: Record<string, unknown> | null) => string | Promise<string>);
+  tools?: Record<string, unknown>;
+  backendFetch?: {
+    baseUrl: string;
+    apiKey: string;
+    headers?: Record<string, string>;
+    transformRequestBody?: (
+      body: Record<string, unknown>,
+      context: { token: string },
+    ) => Promise<Record<string, unknown>>;
+  };
+  loadSessionData?: (token: string) => Promise<Record<string, unknown>>;
+  onToolCall?: (toolCall: {
+    toolName: string;
+    toolCallId: string;
+    args: Record<string, unknown>;
+  }) => void;
+  onStreamFinish?: (result: {
+    messages: UIMessage[];
+    isAborted: boolean;
+  }) => void;
+  knowledge?: KnowledgeConfig;
+};
+
+export type CortexConfig = {
+  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;
+  };
+  knowledge?: KnowledgeConfig;
+  agents: Record<string, CortexAgentDefinition>;
+};

package/src/db/migrate.ts

@@ -0,0 +1,21 @@
+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/20260309012148_cloudy_maria_hill/migration.sql

@@ -0,0 +1,36 @@
+CREATE SCHEMA [ai];
+--> statement-breakpoint
+CREATE TABLE [ai].[captured_files] (
+	[id] UNIQUEIDENTIFIER CONSTRAINT [captured_files_id_default] DEFAULT (NEWID()),
+	[name] nvarchar(2048) NOT NULL,
+	[user_id] UNIQUEIDENTIFIER NOT NULL,
+	[agent_generated_id] nvarchar(1024) NOT NULL,
+	[message_id] nvarchar(128) NOT NULL,
+	[tool_part] nvarchar(max) NOT NULL,
+	CONSTRAINT [captured_files_pkey] PRIMARY KEY([id])
+);
+--> statement-breakpoint
+CREATE TABLE [ai].[messages] (
+	[id] nvarchar(128),
+	[thread_id] UNIQUEIDENTIFIER NOT NULL,
+	[text] nvarchar(max),
+	[content] nvarchar(max) NOT NULL,
+	[role] nvarchar(16) NOT NULL,
+	[created_at] datetime2 NOT NULL,
+	[updated_at] datetime2 NOT NULL,
+	CONSTRAINT [messages_pkey] PRIMARY KEY([id])
+);
+--> statement-breakpoint
+CREATE TABLE [ai].[threads] (
+	[id] UNIQUEIDENTIFIER CONSTRAINT [threads_id_default] DEFAULT (NEWID()),
+	[user_id] UNIQUEIDENTIFIER NOT NULL,
+	[agent_id] nvarchar(128) NOT NULL CONSTRAINT [threads_agent_id_default] DEFAULT ('default'),
+	[title] nvarchar(256),
+	[session] nvarchar(max),
+	[created_at] datetime2 NOT NULL,
+	[updated_at] datetime2 NOT NULL,
+	CONSTRAINT [threads_pkey] PRIMARY KEY([id])
+);
+--> statement-breakpoint
+ALTER TABLE [ai].[captured_files] ADD CONSTRAINT [captured_files_message_id_messages_id_fk] FOREIGN KEY ([message_id]) REFERENCES [ai].[messages]([id]) ON DELETE CASCADE;--> statement-breakpoint
+ALTER TABLE [ai].[messages] ADD CONSTRAINT [messages_thread_id_threads_id_fk] FOREIGN KEY ([thread_id]) REFERENCES [ai].[threads]([id]) ON DELETE CASCADE;

package/src/db/migrations/20260309012148_cloudy_maria_hill/snapshot.json

@@ -0,0 +1,305 @@
+{
+  "version": "2",
+  "dialect": "mssql",
+  "id": "d92d3e1a-9613-4d7e-a224-de2dd01b2718",
+  "prevIds": ["00000000-0000-0000-0000-000000000000"],
+  "ddl": [
+    {
+      "name": "ai",
+      "entityType": "schemas"
+    },
+    {
+      "name": "captured_files",
+      "entityType": "tables",
+      "schema": "ai"
+    },
+    {
+      "name": "messages",
+      "entityType": "tables",
+      "schema": "ai"
+    },
+    {
+      "name": "threads",
+      "entityType": "tables",
+      "schema": "ai"
+    },
+    {
+      "type": "UNIQUEIDENTIFIER",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "id",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "captured_files"
+    },
+    {
+      "type": "nvarchar(2048)",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "name",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "captured_files"
+    },
+    {
+      "type": "UNIQUEIDENTIFIER",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "user_id",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "captured_files"
+    },
+    {
+      "type": "nvarchar(1024)",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "agent_generated_id",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "captured_files"
+    },
+    {
+      "type": "nvarchar(128)",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "message_id",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "captured_files"
+    },
+    {
+      "type": "nvarchar(max)",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "tool_part",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "captured_files"
+    },
+    {
+      "type": "nvarchar(128)",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "id",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "messages"
+    },
+    {
+      "type": "UNIQUEIDENTIFIER",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "thread_id",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "messages"
+    },
+    {
+      "type": "nvarchar(max)",
+      "notNull": false,
+      "generated": null,
+      "identity": null,
+      "name": "text",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "messages"
+    },
+    {
+      "type": "nvarchar(max)",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "content",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "messages"
+    },
+    {
+      "type": "nvarchar(16)",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "role",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "messages"
+    },
+    {
+      "type": "datetime2",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "created_at",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "messages"
+    },
+    {
+      "type": "datetime2",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "updated_at",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "messages"
+    },
+    {
+      "type": "UNIQUEIDENTIFIER",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "id",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "threads"
+    },
+    {
+      "type": "UNIQUEIDENTIFIER",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "user_id",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "threads"
+    },
+    {
+      "type": "nvarchar(128)",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "agent_id",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "threads"
+    },
+    {
+      "type": "nvarchar(256)",
+      "notNull": false,
+      "generated": null,
+      "identity": null,
+      "name": "title",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "threads"
+    },
+    {
+      "type": "nvarchar(max)",
+      "notNull": false,
+      "generated": null,
+      "identity": null,
+      "name": "session",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "threads"
+    },
+    {
+      "type": "datetime2",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "created_at",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "threads"
+    },
+    {
+      "type": "datetime2",
+      "notNull": true,
+      "generated": null,
+      "identity": null,
+      "name": "updated_at",
+      "schema": "ai",
+      "entityType": "columns",
+      "table": "threads"
+    },
+    {
+      "columns": ["message_id"],
+      "nameExplicit": false,
+      "schemaTo": "ai",
+      "tableTo": "messages",
+      "columnsTo": ["id"],
+      "onUpdate": "NO ACTION",
+      "onDelete": "CASCADE",
+      "name": "captured_files_message_id_messages_id_fk",
+      "entityType": "fks",
+      "table": "captured_files",
+      "schema": "ai"
+    },
+    {
+      "columns": ["thread_id"],
+      "nameExplicit": false,
+      "schemaTo": "ai",
+      "tableTo": "threads",
+      "columnsTo": ["id"],
+      "onUpdate": "NO ACTION",
+      "onDelete": "CASCADE",
+      "name": "messages_thread_id_threads_id_fk",
+      "entityType": "fks",
+      "table": "messages",
+      "schema": "ai"
+    },
+    {
+      "nameExplicit": false,
+      "columns": ["id"],
+      "name": "captured_files_pkey",
+      "table": "captured_files",
+      "schema": "ai",
+      "entityType": "pks"
+    },
+    {
+      "nameExplicit": false,
+      "columns": ["id"],
+      "name": "messages_pkey",
+      "table": "messages",
+      "schema": "ai",
+      "entityType": "pks"
+    },
+    {
+      "nameExplicit": false,
+      "columns": ["id"],
+      "name": "threads_pkey",
+      "table": "threads",
+      "schema": "ai",
+      "entityType": "pks"
+    },
+    {
+      "column": "id",
+      "nameExplicit": false,
+      "default": "(NEWID())",
+      "name": "captured_files_id_default",
+      "entityType": "defaults",
+      "schema": "ai",
+      "table": "captured_files"
+    },
+    {
+      "column": "id",
+      "nameExplicit": false,
+      "default": "(NEWID())",
+      "name": "threads_id_default",
+      "entityType": "defaults",
+      "schema": "ai",
+      "table": "threads"
+    },
+    {
+      "column": "agent_id",
+      "nameExplicit": false,
+      "default": "('default')",
+      "name": "threads_agent_id_default",
+      "entityType": "defaults",
+      "schema": "ai",
+      "table": "threads"
+    }
+  ],
+  "renames": []
+}

package/src/db/schema.ts

@@ -0,0 +1,77 @@
+import type { ToolUIPart, UIMessage } from "ai";
+import { sql } from "drizzle-orm";
+import {
+  customType,
+  datetime2,
+  index,
+  mssqlSchema,
+  nvarchar,
+} from "drizzle-orm/mssql-core";
+
+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>
+  >(),
+  ...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,
+});
+
+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)],
+);