@m6d/cortex-server 1.6.0 → 1.7.0

package/dist/src/config.d.ts

@@ -10,9 +10,9 @@ export type KnowledgeConfig = {
     };
     domains?: Record<string, DomainDef>;
 };
-export type PromptContext = {
-    session: Record<string, unknown> | null;
-    requestContext: Record<string, unknown>;
+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";
@@ -26,8 +26,8 @@ export type StorageConfig = {
     secretKey: string;
     bucketName?: string;
 };
-export type CortexAgentDefinition = {
-    systemPrompt: string | ((context: PromptContext) => string | Promise<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;
@@ -38,8 +38,8 @@ export type CortexAgentDefinition = {
         }) => Promise<Record<string, unknown>>;
         interceptor?: RequestInterceptorOptions;
     };
-    loadSessionData?: (token: string) => Promise<Record<string, unknown>>;
-    resolveRequestContext?: (request: Request) => Record<string, unknown> | Promise<Record<string, unknown>>;
+    loadSessionData?: (token: string) => Promise<TSession>;
+    resolveRequestContext?: (request: Request) => TRequestContext | Promise<TRequestContext>;
     onToolCall?: (toolCall: {
         toolName: string;
         toolCallId: string;
@@ -157,3 +157,9 @@ export type CortexConfig = {
     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 declare function defineAgent<TSession extends Record<string, unknown> = Record<string, unknown>, TRequestContext extends Record<string, unknown> = Record<string, unknown>>(config: CortexAgentDefinition<TSession, TRequestContext>): CortexAgentDefinition;

package/dist/src/index.d.ts

@@ -1,4 +1,5 @@
 export type { CortexConfig, CortexAgentDefinition, KnowledgeConfig, DatabaseConfig, StorageConfig, PromptContext, } from "./config";
+export { defineAgent } from "./config";
 export type { ContextConfig, ThreadContextMeta } from "./ai/context/types";
 export type { Thread, AppEnv } from "./types";
 export type { CortexInstance } from "./factory";

package/package.json

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

package/src/config.ts

@@ -10,9 +10,12 @@ export type KnowledgeConfig = {
     domains?: Record<string, DomainDef>;
 };
 
-export type PromptContext = {
-    session: Record<string, unknown> | null;
-    requestContext: Record<string, unknown>;
+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 = {
@@ -29,8 +32,13 @@ export type StorageConfig = {
     bucketName?: string;
 };
 
-export type CortexAgentDefinition = {
-    systemPrompt: string | ((context: PromptContext) => string | Promise<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;
@@ -42,10 +50,8 @@ export type CortexAgentDefinition = {
         ) => Promise<Record<string, unknown>>;
         interceptor?: RequestInterceptorOptions;
     };
-    loadSessionData?: (token: string) => Promise<Record<string, unknown>>;
-    resolveRequestContext?: (
-        request: Request,
-    ) => Record<string, unknown> | Promise<Record<string, unknown>>;
+    loadSessionData?: (token: string) => Promise<TSession>;
+    resolveRequestContext?: (request: Request) => TRequestContext | Promise<TRequestContext>;
     onToolCall?: (toolCall: {
         toolName: string;
         toolCallId: string;
@@ -162,3 +168,15 @@ export type CortexConfig = {
     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/index.ts

@@ -7,6 +7,7 @@ export type {
     StorageConfig,
     PromptContext,
 } from "./config";
+export { defineAgent } from "./config";
 
 export type { ContextConfig, ThreadContextMeta } from "./ai/context/types";