@m6d/cortex-server 1.8.0 → 2.0.1

package/dist/src/lib/config.d.ts

@@ -1,10 +1,12 @@
-import type { 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";
-import type { Thread } from "./types.ts";
+import type { AnyServerTool } from "@tanstack/ai";
+import type { DatabaseAdapter } from "./adapters/database/index";
+import type { StorageAdapter } from "./adapters/storage/index";
+import type { DomainDef } from "../../contracts/graph";
+import type { RequestInterceptorOptions } from "./ai/interceptors/request-interceptor";
+import type { ContextConfig } from "./ai/context/types";
+import type { ChatMessage, Thread } from "./types";
+/** An agent's tools, as `chat()` takes them. */
+export type ToolSet = ReadonlyArray<AnyServerTool>;
 type ModelConfig = {
     baseURL: string;
     apiKey: string;
@@ -26,6 +28,10 @@ type RerankerConfig = {
     url: string;
     apiKey: string;
 };
+export type ControlCenterConfig = {
+    url: string;
+    apiKey: string;
+};
 export type KnowledgeConfig = {
     swagger?: {
         url: string;
@@ -50,9 +56,19 @@ export type ToolContext<TSession extends Record<string, unknown> = Record<string
 };
 export type ToolSetFactory<TSession extends Record<string, unknown> = Record<string, unknown>, TRequestContext extends Record<string, unknown> = Record<string, unknown>> = (context: ToolContext<TSession, TRequestContext>) => ToolSet;
 export type DatabaseConfig = {
-    type: "mssql";
+    type: "mssql" | "postgres";
     connectionString: string;
 };
+/**
+ * Optional. With Redis, live runs survive a producer restart and reach every
+ * instance: the chunk log, the thread→run claims, and abort fan-out all move
+ * there. Without it, an in-memory fallback covers a single process.
+ */
+export type RedisConfig = {
+    url: string;
+    /** How long a finished run's stream log stays replayable. Defaults to 24 h. */
+    streamTtlSeconds?: number;
+};
 export type StorageConfig = {
     endPoint: string;
     port: number;
@@ -61,18 +77,34 @@ export type StorageConfig = {
     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>);
+type BackendFetchConfig = {
+    baseUrl: string;
+    apiKey: string;
+    headers?: Record<string, string>;
+    transformRequestBody?: (body: Record<string, unknown>, context: {
+        token: string;
+    }) => Promise<Record<string, unknown>>;
+    interceptor?: RequestInterceptorOptions;
+};
+/** Model and knowledge-graph providers. Declared once at server level; an agent may override any of them. */
+type ProviderConfig = {
+    model: ModelConfig;
+    fastModel?: ModelConfig;
+    vision?: ModelConfig;
+    embedding?: EmbeddingConfig;
+    neo4j?: Neo4jConfig;
+    reranker?: RerankerConfig;
+};
+/**
+ * What an agent *does*, independent of where its providers came from. Shared
+ * verbatim by the authored definition and its resolved form — the two differ
+ * only in how providers and opt-outs are expressed, so everything else lives
+ * here and cannot drift between them.
+ */
+type AgentBehavior<TSession extends Record<string, unknown>, TRequestContext extends Record<string, unknown>> = {
+    systemPrompt?: string | ((context: PromptContext<TSession, TRequestContext>) => string | Promise<string>);
     tools?: ToolSet | ToolSetFactory<TSession, TRequestContext>;
-    backendFetch?: {
-        baseUrl: string;
-        apiKey: string;
-        headers?: Record<string, string>;
-        transformRequestBody?: (body: Record<string, unknown>, context: {
-            token: string;
-        }) => Promise<Record<string, unknown>>;
-        interceptor?: RequestInterceptorOptions;
-    };
+    backendFetch?: BackendFetchConfig;
     loadSessionData?: (token: string) => Promise<TSession>;
     resolveRequestContext?: (request: Request) => TRequestContext | Promise<TRequestContext>;
     onToolCall?: (toolCall: {
@@ -81,68 +113,53 @@ export type CortexAgentDefinition<TSession extends Record<string, unknown> = Rec
         args: Record<string, unknown>;
     }) => void;
     onStreamFinish?: (result: {
-        messages: UIMessage[];
+        messages: ChatMessage[];
         isAborted: boolean;
     }) => void;
-    model?: ModelConfig;
-    fastModel?: ModelConfig;
-    embedding?: EmbeddingConfig;
-    neo4j?: Neo4jConfig;
-    reranker?: RerankerConfig;
-    context?: Partial<ContextConfig>;
+};
+/**
+ * An agent as authored. Providers are optional because an unset one inherits
+ * from server level; `null` on knowledge means "opt out entirely" rather than
+ * "inherit". The Control Center is server-level only: which agents it backs is
+ * decided by publishing them there, not per entry here.
+ */
+export type CortexAgentDefinition<TSession extends Record<string, unknown> = Record<string, unknown>, TRequestContext extends Record<string, unknown> = Record<string, unknown>> = AgentBehavior<TSession, TRequestContext> & Partial<ProviderConfig> & {
     knowledge?: KnowledgeConfig | null;
+    context?: Partial<ContextConfig>;
 };
-export type ResolvedCortexAgentConfig = {
+/**
+ * An agent after the factory has merged it with server-level config. Every
+ * inherit-or-opt-out decision is already resolved, so nothing here is nullable
+ * and `model`/`context` are guaranteed present.
+ */
+export type ResolvedCortexAgentConfig = AgentBehavior<Record<string, unknown>, Record<string, unknown>> & ProviderConfig & {
+    agentId: string;
     db: DatabaseAdapter;
-    storage: StorageAdapter;
-    model: ModelConfig;
-    fastModel?: ModelConfig;
-    embedding: EmbeddingConfig;
-    neo4j?: Neo4jConfig;
-    reranker?: RerankerConfig;
-    systemPrompt: string | ((context: PromptContext) => string | Promise<string>);
-    tools?: ToolSet | ToolSetFactory;
-    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;
+    storage?: StorageAdapter;
+    controlCenter?: ControlCenterConfig;
     knowledge?: KnowledgeConfig;
+    context: ContextConfig;
 };
-export type CortexConfig = {
+export type CortexConfig = ProviderConfig & {
     port?: number;
     database: DatabaseConfig;
-    storage: StorageConfig;
+    storage?: StorageConfig;
+    redis?: RedisConfig;
     auth?: {
         jwksUri: string;
         issuer: string;
         tokenExtractor?: (req: Request) => string | null;
         cookieName?: string;
     };
-    model: ModelConfig;
-    fastModel?: ModelConfig;
-    embedding: EmbeddingConfig;
-    neo4j?: Neo4jConfig;
-    reranker?: RerankerConfig;
+    controlCenter?: ControlCenterConfig;
     context?: Partial<ContextConfig>;
     knowledge?: KnowledgeConfig;
-    agents: Record<string, CortexAgentDefinition>;
+    /**
+     * Locally-defined agents. With `controlCenter` set, agents published there
+     * are served automatically and need no entry here; an entry is only needed
+     * for local behavior (tools, backendFetch, provider overrides).
+     */
+    agents?: Record<string, CortexAgentDefinition>;
 };
 /**
  * Helper to define an agent with full type inference for `systemPrompt` context.

package/dist/src/lib/db/drizzle.config.mssql.d.ts

@@ -0,0 +1,2 @@
+declare const _default: import("drizzle-kit").Config;
+export default _default;

package/dist/src/lib/db/drizzle.config.pg.d.ts

@@ -0,0 +1,2 @@
+declare const _default: import("drizzle-kit").Config;
+export default _default;

package/dist/src/lib/db/migrate.d.ts

@@ -1 +1,7 @@
-export declare function runMigrations(connectionString: string): Promise<void>;
+import type { DatabaseConfig } from "../config";
+/**
+ * Each dialect keeps its own migration history — Postgres starts from a
+ * squashed initial migration rather than replaying MSSQL's. The driver imports
+ * are lazy so a server only loads the one it was configured for.
+ */
+export declare function runMigrations(database: DatabaseConfig): Promise<void>;

package/dist/src/lib/db/{schema.d.ts → schema.mssql.d.ts}

@@ -1,6 +1,5 @@
-import type { UIMessage } from "ai";
-import type { MessageMetadata, TokenUsage } from "../types.ts";
-import type { ThreadContextMeta } from "../ai/context/types.ts";
+import type { ChatMessage, TokenUsage } from "../types";
+import type { ThreadContextMeta } from "../ai/context/types";
 export declare const aiSchema: import("drizzle-orm/mssql-core").MsSqlSchema<"ai">;
 export declare const threads: import("drizzle-orm/mssql-core").MsSqlTableWithColumns<{
     name: "threads";
@@ -57,15 +56,15 @@ export declare const threads: import("drizzle-orm/mssql-core").MsSqlTableWithCol
         userId: import("drizzle-orm/mssql-core").MsSqlColumn<{
             name: string;
             tableName: "threads";
-            dataType: "custom";
+            dataType: "string";
             data: string;
-            driverParam: unknown;
+            driverParam: string | number;
             notNull: true;
             hasDefault: false;
             isPrimaryKey: false;
             isAutoincrement: false;
             hasRuntimeDefault: false;
-            enumValues: undefined;
+            enumValues: [string, ...string[]];
             baseColumn: never;
             identity: undefined;
             generated: undefined;
@@ -225,7 +224,7 @@ export declare const messages: import("drizzle-orm/mssql-core").MsSqlTableWithCo
             name: string;
             tableName: "messages";
             dataType: "object json";
-            data: UIMessage<MessageMetadata, import("ai").UIDataTypes, import("ai").UITools>;
+            data: ChatMessage;
             driverParam: string;
             notNull: true;
             hasDefault: false;
@@ -237,6 +236,22 @@ export declare const messages: import("drizzle-orm/mssql-core").MsSqlTableWithCo
             identity: undefined;
             generated: undefined;
         }, {}>;
+        ordinal: import("drizzle-orm/mssql-core").MsSqlColumn<{
+            name: string;
+            tableName: "messages";
+            dataType: "number int32";
+            data: number;
+            driverParam: number;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}>;
         role: import("drizzle-orm/mssql-core").MsSqlColumn<{
             name: string;
             tableName: "messages";
@@ -359,13 +374,45 @@ export declare const llmRequests: import("drizzle-orm/mssql-core").MsSqlTableWit
     };
     dialect: "mssql";
 }>;
-export declare const capturedFiles: import("drizzle-orm/mssql-core").MsSqlTableWithColumns<{
-    name: "captured_files";
+export declare const attachments: import("drizzle-orm/mssql-core").MsSqlTableWithColumns<{
+    name: "attachments";
     schema: "ai";
     columns: {
+        createdAt: import("drizzle-orm/mssql-core").MsSqlColumn<{
+            name: string;
+            tableName: "attachments";
+            dataType: "object date";
+            data: Date;
+            driverParam: string | Date;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: true;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}>;
+        updatedAt: import("drizzle-orm/mssql-core").MsSqlColumn<{
+            name: string;
+            tableName: "attachments";
+            dataType: "object date";
+            data: Date;
+            driverParam: string | Date;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: true;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}>;
         id: import("drizzle-orm/mssql-core").MsSqlColumn<{
             name: string;
-            tableName: "captured_files";
+            tableName: "attachments";
             dataType: "custom";
             data: string;
             driverParam: unknown;
@@ -379,9 +426,25 @@ export declare const capturedFiles: import("drizzle-orm/mssql-core").MsSqlTableW
             identity: undefined;
             generated: undefined;
         }, {}>;
-        name: import("drizzle-orm/mssql-core").MsSqlColumn<{
+        threadId: import("drizzle-orm/mssql-core").MsSqlColumn<{
             name: string;
-            tableName: "captured_files";
+            tableName: "attachments";
+            dataType: "custom";
+            data: string;
+            driverParam: unknown;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}>;
+        userId: import("drizzle-orm/mssql-core").MsSqlColumn<{
+            name: string;
+            tableName: "attachments";
             dataType: "string";
             data: string;
             driverParam: string | number;
@@ -395,25 +458,25 @@ export declare const capturedFiles: import("drizzle-orm/mssql-core").MsSqlTableW
             identity: undefined;
             generated: undefined;
         }, {}>;
-        userId: import("drizzle-orm/mssql-core").MsSqlColumn<{
+        messageId: import("drizzle-orm/mssql-core").MsSqlColumn<{
             name: string;
-            tableName: "captured_files";
-            dataType: "custom";
+            tableName: "attachments";
+            dataType: "string";
             data: string;
-            driverParam: unknown;
-            notNull: true;
+            driverParam: string | number;
+            notNull: false;
             hasDefault: false;
             isPrimaryKey: false;
             isAutoincrement: false;
             hasRuntimeDefault: false;
-            enumValues: undefined;
+            enumValues: [string, ...string[]];
             baseColumn: never;
             identity: undefined;
             generated: undefined;
         }, {}>;
-        agentGeneratedId: import("drizzle-orm/mssql-core").MsSqlColumn<{
+        filename: import("drizzle-orm/mssql-core").MsSqlColumn<{
             name: string;
-            tableName: "captured_files";
+            tableName: "attachments";
             dataType: "string";
             data: string;
             driverParam: string | number;
@@ -427,9 +490,9 @@ export declare const capturedFiles: import("drizzle-orm/mssql-core").MsSqlTableW
             identity: undefined;
             generated: undefined;
         }, {}>;
-        messageId: import("drizzle-orm/mssql-core").MsSqlColumn<{
+        contentType: import("drizzle-orm/mssql-core").MsSqlColumn<{
             name: string;
-            tableName: "captured_files";
+            tableName: "attachments";
             dataType: "string";
             data: string;
             driverParam: string | number;
@@ -443,14 +506,12 @@ export declare const capturedFiles: import("drizzle-orm/mssql-core").MsSqlTableW
             identity: undefined;
             generated: undefined;
         }, {}>;
-        toolPart: import("drizzle-orm/mssql-core").MsSqlColumn<{
+        sizeBytes: import("drizzle-orm/mssql-core").MsSqlColumn<{
             name: string;
-            tableName: "captured_files";
-            dataType: "object json";
-            data: {
-                type: `tool-${string}`;
-            } & import("ai").UIToolInvocation<import("ai").UITool>;
-            driverParam: string;
+            tableName: "attachments";
+            dataType: "number int32";
+            data: number;
+            driverParam: number;
             notNull: true;
             hasDefault: false;
             isPrimaryKey: false;
@@ -461,6 +522,54 @@ export declare const capturedFiles: import("drizzle-orm/mssql-core").MsSqlTableW
             identity: undefined;
             generated: undefined;
         }, {}>;
+        storageKey: import("drizzle-orm/mssql-core").MsSqlColumn<{
+            name: string;
+            tableName: "attachments";
+            dataType: "string";
+            data: string;
+            driverParam: string | number;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}>;
+        status: import("drizzle-orm/mssql-core").MsSqlColumn<{
+            name: string;
+            tableName: "attachments";
+            dataType: "string enum";
+            data: "pending" | "described" | "failed" | "skipped";
+            driverParam: string | number;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: ["pending", "described", "failed", "skipped"];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}>;
+        description: import("drizzle-orm/mssql-core").MsSqlColumn<{
+            name: string;
+            tableName: "attachments";
+            dataType: "string";
+            data: string;
+            driverParam: string | number;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}>;
     };
     dialect: "mssql";
 }>;