@m6d/cortex-server 1.1.2 → 1.2.0

package/dist/src/ai/interceptors/{resolve-captured-files.d.ts → request-interceptor.d.ts}

@@ -4,8 +4,9 @@ export type ResolvedFile = {
     name: string;
     bytes: string;
 };
-export declare function createCapturedFileInterceptor(db: DatabaseAdapter, storage: StorageAdapter, options?: {
+export type RequestInterceptorOptions = {
     transformFile?: (file: ResolvedFile) => unknown;
-}): (body: Record<string, unknown>, context: {
+};
+export declare function createRequestInterceptor(db: DatabaseAdapter, storage: StorageAdapter, options?: RequestInterceptorOptions): (body: Record<string, unknown>, context: {
     token: string;
 }) => Promise<Record<string, unknown>>;

package/dist/src/config.d.ts

@@ -1,7 +1,8 @@
-import type { UIMessage } from "ai";
+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";
 export type KnowledgeConfig = {
     swagger?: {
         url: string;
@@ -22,7 +23,7 @@ export type StorageConfig = {
 };
 export type CortexAgentDefinition = {
     systemPrompt: string | ((session: Record<string, unknown> | null) => string | Promise<string>);
-    tools?: Record<string, unknown>;
+    tools?: ToolSet;
     backendFetch?: {
         baseUrl: string;
         apiKey: string;
@@ -30,6 +31,7 @@ export type CortexAgentDefinition = {
         transformRequestBody?: (body: Record<string, unknown>, context: {
             token: string;
         }) => Promise<Record<string, unknown>>;
+        interceptor?: RequestInterceptorOptions;
     };
     loadSessionData?: (token: string) => Promise<Record<string, unknown>>;
     onToolCall?: (toolCall: {
@@ -89,7 +91,7 @@ export type ResolvedCortexAgentConfig = {
         apiKey: string;
     };
     systemPrompt: string | ((session: Record<string, unknown> | null) => string | Promise<string>);
-    tools?: Record<string, unknown>;
+    tools?: ToolSet;
     backendFetch?: {
         baseUrl: string;
         apiKey: string;
@@ -97,6 +99,7 @@ export type ResolvedCortexAgentConfig = {
         transformRequestBody?: (body: Record<string, unknown>, context: {
             token: string;
         }) => Promise<Record<string, unknown>>;
+        interceptor?: RequestInterceptorOptions;
     };
     loadSessionData?: (token: string) => Promise<Record<string, unknown>>;
     onToolCall?: (toolCall: {

package/dist/src/index.d.ts

@@ -2,6 +2,8 @@ export type { CortexConfig, CortexAgentDefinition, KnowledgeConfig, DatabaseConf
 export type { Thread, AppEnv } from "./types";
 export type { CortexInstance } from "./factory";
 export { createCortex } from "./factory";
+export type { ResolvedFile, RequestInterceptorOptions, } from "./ai/interceptors/request-interceptor";
+export { createRequestInterceptor } from "./ai/interceptors/request-interceptor";
 export { captureFilesTool } from "./ai/tools/capture-files.tool";
 export { createQueryGraphTool } from "./ai/tools/query-graph.tool";
 export { createCallEndpointTool } from "./ai/tools/call-endpoint.tool";

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@m6d/cortex-server",
-    "version": "1.1.2",
+    "version": "1.2.0",
     "description": "Reusable AI agent chat server library for Hono + Bun",
     "license": "MIT",
     "type": "module",

package/src/adapters/mssql.ts

@@ -109,19 +109,15 @@ export function createMssqlAdapter(connectionString: string, storage: StorageAda
                 }
 
                 const existingMessages = messagesToInsert.filter((x) => existingIds.includes(x.id));
-                if (existingMessages.length) {
-                    await db.transaction(async (tx) => {
-                        for (const message of existingMessages) {
-                            await tx
-                                .update(messages)
-                                .set({
-                                    content: message,
-                                    text: message.parts.find((x) => x.type === "text")?.text,
-                                })
-                                .where(eq(messages.id, message.id))
-                                .execute();
-                        }
-                    });
+                for (const message of existingMessages) {
+                    await db
+                        .update(messages)
+                        .set({
+                            content: message,
+                            text: message.parts.find((x) => x.type === "text")?.text,
+                        })
+                        .where(eq(messages.id, message.id))
+                        .execute();
                 }
             },
         },

package/src/ai/index.ts

@@ -15,7 +15,7 @@ import { createQueryGraphTool } from "./tools/query-graph.tool.ts";
 import { createCallEndpointTool } from "./tools/call-endpoint.tool.ts";
 import { createExecuteCodeTool } from "./tools/execute-code.tool.ts";
 import { captureFilesTool } from "./tools/capture-files.tool.ts";
-import { createCapturedFileInterceptor } from "./interceptors/resolve-captured-files.ts";
+import { createRequestInterceptor } from "./interceptors/request-interceptor.ts";
 import { createNeo4jClient } from "../graph/neo4j.ts";
 import { resolveFromGraph } from "../graph/resolver.ts";
 import { notify } from "../ws/index.ts";
@@ -69,7 +69,11 @@ export async function stream(
             ...config.backendFetch,
             transformRequestBody:
                 config.backendFetch.transformRequestBody ??
-                createCapturedFileInterceptor(config.db, config.storage),
+                createRequestInterceptor(
+                    config.db,
+                    config.storage,
+                    config.backendFetch.interceptor,
+                ),
         };
 
         builtInTools["callEndpoint"] = createCallEndpointTool(backendFetchWithInterceptor, token);

package/src/ai/interceptors/{resolve-captured-files.ts → request-interceptor.ts}

@@ -7,14 +7,18 @@ export type ResolvedFile = {
     bytes: string;
 };
 
-export function createCapturedFileInterceptor(
+export type RequestInterceptorOptions = {
+    transformFile?: (file: ResolvedFile) => unknown;
+};
+
+export function createRequestInterceptor(
     db: DatabaseAdapter,
     storage: StorageAdapter,
-    options?: { transformFile?: (file: ResolvedFile) => unknown },
+    options?: RequestInterceptorOptions,
 ) {
     const transformFile = options?.transformFile ?? ((file: ResolvedFile) => file);
 
-    return async function resolveCapturedFiles(
+    return async function resolveRequestBody(
         body: Record<string, unknown>,
         context: { token: string },
     ) {

package/src/config.ts

@@ -1,7 +1,8 @@
-import type { UIMessage } from "ai";
+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";
 
 export type KnowledgeConfig = {
     swagger?: { url: string };
@@ -24,7 +25,7 @@ export type StorageConfig = {
 
 export type CortexAgentDefinition = {
     systemPrompt: string | ((session: Record<string, unknown> | null) => string | Promise<string>);
-    tools?: Record<string, unknown>;
+    tools?: ToolSet;
     backendFetch?: {
         baseUrl: string;
         apiKey: string;
@@ -33,6 +34,7 @@ export type CortexAgentDefinition = {
             body: Record<string, unknown>,
             context: { token: string },
         ) => Promise<Record<string, unknown>>;
+        interceptor?: RequestInterceptorOptions;
     };
     loadSessionData?: (token: string) => Promise<Record<string, unknown>>;
     onToolCall?: (toolCall: {
@@ -90,7 +92,7 @@ export type ResolvedCortexAgentConfig = {
         apiKey: string;
     };
     systemPrompt: string | ((session: Record<string, unknown> | null) => string | Promise<string>);
-    tools?: Record<string, unknown>;
+    tools?: ToolSet;
     backendFetch?: {
         baseUrl: string;
         apiKey: string;
@@ -99,6 +101,7 @@ export type ResolvedCortexAgentConfig = {
             body: Record<string, unknown>,
             context: { token: string },
         ) => Promise<Record<string, unknown>>;
+        interceptor?: RequestInterceptorOptions;
     };
     loadSessionData?: (token: string) => Promise<Record<string, unknown>>;
     onToolCall?: (toolCall: {

package/src/index.ts

@@ -14,6 +14,13 @@ export type { Thread, AppEnv } from "./types";
 export type { CortexInstance } from "./factory";
 export { createCortex } from "./factory";
 
+// Request interceptor (consumers may customize or replace the default)
+export type {
+    ResolvedFile,
+    RequestInterceptorOptions,
+} from "./ai/interceptors/request-interceptor";
+export { createRequestInterceptor } from "./ai/interceptors/request-interceptor";
+
 // Tools (consumers may register custom tools or use built-in ones)
 export { captureFilesTool } from "./ai/tools/capture-files.tool";
 export { createQueryGraphTool } from "./ai/tools/query-graph.tool";