@inkeep/agents-core 0.59.4 → 0.61.0

package/dist/utils/apiKeys.js

@@ -1,7 +1,7 @@
 import { getLogger } from "./logger.js";
 import { generateId } from "./conversations.js";
-import { customAlphabet } from "nanoid";
 import { randomBytes, scrypt, timingSafeEqual } from "node:crypto";
+import { customAlphabet } from "nanoid";
 import { promisify } from "node:util";
 
 //#region src/utils/apiKeys.ts

package/dist/utils/conversations.d.ts

@@ -1,5 +1,13 @@
+import { BaseExecutionContext, ConversationMetadata } from "../types/utility.js";
+
 //#region src/utils/conversations.d.ts
 declare const generateId: (size?: number) => string;
+/**
+ * Derive a deterministic 32-character hex ID from natural key parts.
+ * Used for junction/relation table IDs so the same logical relation
+ * always produces the same ID regardless of which branch creates it.
+ */
+declare function deriveRelationId(...parts: string[]): string;
 /**
  * Generates a standardized conversation ID.
  *
@@ -15,5 +23,6 @@ declare const generateId: (size?: number) => string;
  * ```
  */
 declare function getConversationId(): string;
+declare function buildConversationMetadata(executionContext: BaseExecutionContext, userProperties?: Record<string, unknown>): ConversationMetadata | undefined;
 //#endregion
-export { generateId, getConversationId };
+export { buildConversationMetadata, deriveRelationId, generateId, getConversationId };

package/dist/utils/conversations.js

@@ -1,8 +1,17 @@
+import { createHash } from "node:crypto";
 import { customAlphabet } from "nanoid";
 
 //#region src/utils/conversations.ts
 const generateId = customAlphabet("abcdefghijklmnopqrstuvwxyz0123456789", 21);
 /**
+* Derive a deterministic 32-character hex ID from natural key parts.
+* Used for junction/relation table IDs so the same logical relation
+* always produces the same ID regardless of which branch creates it.
+*/
+function deriveRelationId(...parts) {
+	return createHash("sha256").update(parts.join("\0")).digest("hex").slice(0, 32);
+}
+/**
 * Generates a standardized conversation ID.
 *
 * The generated ID follows these rules:
@@ -19,6 +28,15 @@ const generateId = customAlphabet("abcdefghijklmnopqrstuvwxyz0123456789", 21);
 function getConversationId() {
 	return generateId();
 }
+function buildConversationMetadata(executionContext, userProperties) {
+	const meta = executionContext.metadata;
+	const isAuthenticated = meta?.authMethod === "app_credential_web_client_authenticated";
+	const result = {};
+	if (userProperties) result.userContext = userProperties;
+	if (meta?.verifiedClaims) result.verifiedClaims = meta.verifiedClaims;
+	if (isAuthenticated && meta?.endUserId) result.externalUserId = meta.endUserId;
+	return Object.keys(result).length > 0 ? result : void 0;
+}
 
 //#endregion
-export { generateId, getConversationId };
+export { buildConversationMetadata, deriveRelationId, generateId, getConversationId };

package/dist/utils/error.d.ts

@@ -3,8 +3,8 @@ import { HTTPException } from "hono/http-exception";
 
 //#region src/utils/error.d.ts
 declare const ErrorCode: z.ZodEnum<{
-  unauthorized: "unauthorized";
   bad_request: "bad_request";
+  unauthorized: "unauthorized";
   forbidden: "forbidden";
   not_found: "not_found";
   conflict: "conflict";
@@ -20,8 +20,8 @@ declare const problemDetailsSchema: z.ZodObject<{
   instance: z.ZodOptional<z.ZodString>;
   requestId: z.ZodOptional<z.ZodString>;
   code: z.ZodEnum<{
-    unauthorized: "unauthorized";
     bad_request: "bad_request";
+    unauthorized: "unauthorized";
     forbidden: "forbidden";
     not_found: "not_found";
     conflict: "conflict";
@@ -35,8 +35,8 @@ type ErrorCodes = z.infer<typeof ErrorCode>;
 declare const errorResponseSchema: z.ZodObject<{
   error: z.ZodObject<{
     code: z.ZodEnum<{
-      unauthorized: "unauthorized";
       bad_request: "bad_request";
+      unauthorized: "unauthorized";
       forbidden: "forbidden";
       not_found: "not_found";
       conflict: "conflict";
@@ -74,12 +74,12 @@ declare const errorSchemaFactory: (code: ErrorCodes, description: string) => {
       schema: z.ZodObject<{
         instance: z.ZodOptional<z.ZodString>;
         requestId: z.ZodOptional<z.ZodString>;
-        code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+        code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
         detail: z.ZodString;
         title: z.ZodString;
         status: z.ZodNumber;
         error: z.ZodObject<{
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           message: z.ZodString;
         }, z.core.$strip>;
       }, z.core.$strip>;
@@ -96,12 +96,12 @@ declare const commonCreateErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -115,12 +115,12 @@ declare const commonCreateErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -134,12 +134,12 @@ declare const commonCreateErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -153,12 +153,12 @@ declare const commonCreateErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -172,12 +172,12 @@ declare const commonCreateErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -193,12 +193,12 @@ declare const commonUpdateErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -212,12 +212,12 @@ declare const commonUpdateErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -231,12 +231,12 @@ declare const commonUpdateErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -250,12 +250,12 @@ declare const commonUpdateErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -269,12 +269,12 @@ declare const commonUpdateErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -288,12 +288,12 @@ declare const commonUpdateErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -309,12 +309,12 @@ declare const commonGetErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -328,12 +328,12 @@ declare const commonGetErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -347,12 +347,12 @@ declare const commonGetErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -366,12 +366,12 @@ declare const commonGetErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -385,12 +385,12 @@ declare const commonGetErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -404,12 +404,12 @@ declare const commonGetErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -425,12 +425,12 @@ declare const commonDeleteErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -444,12 +444,12 @@ declare const commonDeleteErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -463,12 +463,12 @@ declare const commonDeleteErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -482,12 +482,12 @@ declare const commonDeleteErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -501,12 +501,12 @@ declare const commonDeleteErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;
@@ -520,12 +520,12 @@ declare const commonDeleteErrorResponses: {
         schema: z.ZodObject<{
           instance: z.ZodOptional<z.ZodString>;
           requestId: z.ZodOptional<z.ZodString>;
-          code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+          code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
           detail: z.ZodString;
           title: z.ZodString;
           status: z.ZodNumber;
           error: z.ZodObject<{
-            code: z.ZodLiteral<"unauthorized" | "bad_request" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
+            code: z.ZodLiteral<"bad_request" | "unauthorized" | "forbidden" | "not_found" | "conflict" | "too_many_requests" | "internal_server_error" | "unprocessable_entity">;
             message: z.ZodString;
           }, z.core.$strip>;
         }, z.core.$strip>;

package/dist/utils/index.d.ts

@@ -2,9 +2,10 @@ import { JwtVerifyResult, SignJwtOptions, VerifyJwtOptions, decodeJwtPayload, ex
 import { GenerateInternalServiceTokenParams, InternalServiceId, InternalServiceTokenPayload, InternalServices, VerifyInternalServiceTokenResult, generateInternalServiceToken, isInternalServiceToken, validateInternalServiceProjectAccess, validateInternalServiceTenantAccess, verifyInternalServiceAuthHeader, verifyInternalServiceToken } from "./internal-service-auth.js";
 import { LoggerFactoryConfig, PinoLogger, PinoLoggerConfig, getLogger, loggerFactory } from "./logger.js";
 import { McpClient, McpClientOptions, McpSSEConfig, McpServerConfig, McpStreamableHttpConfig, activeMcpClients } from "./mcp-client.js";
+import { GenerationType } from "../db/runtime/runtime-schema.js";
 import { ApiKeyGenerationResult, AppCredentialResult, extractPublicId, generateApiKey, generateAppCredential, getMetadataFromApiKey, hashApiKey, isApiKeyExpired, maskApiKey, sanitizeAppConfig, validateApiKey } from "./apiKeys.js";
 import { McpOAuthFlowResult, McpTokenExchangeResult, OAuthConfig, detectAuthenticationRequired, exchangeMcpAuthorizationCode, initiateMcpOAuthFlow } from "./auth-detection.js";
-import { generateId, getConversationId } from "./conversations.js";
+import { buildConversationMetadata, deriveRelationId, generateId, getConversationId } from "./conversations.js";
 import { getCredentialStoreLookupKeyFromRetrievalParams } from "./credential-store-utils.js";
 import { normalizeDateString, toISODateString } from "./date.js";
 import { validateOrigin } from "./domain-validation.js";
@@ -28,8 +29,12 @@ import { interpolateTemplate } from "./template-interpolation.js";
 import { ComposioAuthResult, CredentialScope, configureComposioMCPServer, deleteComposioConnectedAccount, extractComposioServerId, fetchComposioServers, fetchSingleComposioServer, getComposioInstance, getComposioOAuthRedirectUrl, getComposioUserId, isComposioMCPServerAuthenticated } from "./third-party-mcp-servers/composio-client.js";
 import { isThirdPartyMCPServerAuthenticated } from "./third-party-mcp-servers/third-party-check.js";
 import "./third-party-mcp-servers/index.js";
+import { AssembleResult, estimateTokens } from "./token-estimator.js";
 import { flushTraces, getTracer, setSpanWithError, unwrapError } from "./tracer-factory.js";
 import { HashedHeaderValue, SignatureVerificationErrorCode, SignatureVerificationResult, TriggerAuthResult, hashAuthenticationHeaders, hashTriggerHeaderValue, validateTriggerHeaderValue, verifySignatureWithConfig, verifyTriggerAuth } from "./trigger-auth.js";
+import { extractUsageTokens, gatewayCostMiddleware } from "./usage-cost-middleware.js";
+import "./usage-tracker.js";
+import { ValidatePublicKeyResult, validatePublicKey } from "./validate-public-key.js";
 import { _resetWaitUntilCache, getWaitUntil } from "./wait-until.js";
 import { TRUSTED_WORK_APP_MCP_PATHS, isTrustedWorkAppMcpUrl } from "./work-app-mcp.js";
-export { ApiKeyGenerationResult, AppCredentialResult, CommonCreateErrorResponses, CommonDeleteErrorResponses, CommonGetErrorResponses, CommonUpdateErrorResponses, ComposioAuthResult, CredentialScope, ERROR_DOCS_BASE_URL, ErrorCode, ErrorCodes, ErrorResponse, GenerateInternalServiceTokenParams, GenerateServiceTokenParams, HashedHeaderValue, InternalServiceId, InternalServiceTokenPayload, InternalServices, JsonTransformer, JwtVerifyResult, LLMMessage, LoggerFactoryConfig, McpClient, McpClientOptions, McpOAuthFlowResult, McpSSEConfig, McpServerConfig, McpStreamableHttpConfig, McpTokenExchangeResult, MockLanguageModel, ModelFactory, OAuthConfig, ParsedSSEResponse, PinoLogger, PinoLoggerConfig, PoWError, PoWResult, ProblemDetails, ServiceTokenPayload, SignJwtOptions, SignSlackLinkTokenParams, SignSlackUserTokenParams, SignatureVerificationErrorCode, SignatureVerificationResult, SignedTempToken, SlackAccessTokenPayload, SlackAccessTokenPayloadSchema, SlackLinkIntent, SlackLinkIntentSchema, SlackLinkTokenPayload, SlackLinkTokenPayloadSchema, TRUSTED_WORK_APP_MCP_PATHS, TempTokenPayload, TriggerAuthResult, VerifyInternalServiceTokenResult, VerifyJwtOptions, VerifyServiceTokenResult, VerifySlackLinkTokenResult, VerifySlackUserTokenResult, _resetWaitUntilCache, activeMcpClients, commonCreateErrorResponses, commonDeleteErrorResponses, commonGetErrorResponses, commonUpdateErrorResponses, configureComposioMCPServer, convertZodToJsonSchema, convertZodToJsonSchemaWithPreview, createApiError, createMockModel, decodeJwtPayload, deleteComposioConnectedAccount, detectAuthenticationRequired, errorResponseSchema, errorSchemaFactory, exchangeMcpAuthorizationCode, extractBearerToken, extractComposioServerId, extractPreviewFields, extractPublicId, fetchComposioServers, fetchSingleComposioServer, flushTraces, formatMessagesForLLM, formatMessagesForLLMContext, generateApiKey, generateAppCredential, generateId, generateInternalServiceToken, generateServiceToken, getComposioInstance, getComposioOAuthRedirectUrl, getComposioUserId, getConversationId, getCredentialStoreLookupKeyFromRetrievalParams, getInProcessFetch, getJwtSecret, getLogger, getMetadataFromApiKey, getPoWErrorMessage, getTracer, getWaitUntil, handleApiError, hasIssuer, hashApiKey, hashAuthenticationHeaders, hashTriggerHeaderValue, initiateMcpOAuthFlow, interpolateTemplate, isApiKeyExpired, isComposioMCPServerAuthenticated, isDevelopment, isInternalServiceToken, isPoWEnabled, isProduction, isSlackUserToken, isTest, isThirdPartyMCPServerAuthenticated, isTrustedWorkAppMcpUrl, isUniqueConstraintError, isZodSchema, loggerFactory, makeAllPropertiesRequired, maskApiKey, normalizeDataComponentSchema, normalizeDateString, parseEmbeddedJson, parseSSEResponse, preview, problemDetailsSchema, registerAppFetch, retryWithBackoff, sanitizeAppConfig, setSpanWithError, signJwt, signSlackLinkToken, signSlackUserToken, signTempToken, stripUnsupportedConstraints, throwIfUniqueConstraintError, toISODateString, unwrapError, validateApiKey, validateInternalServiceProjectAccess, validateInternalServiceTenantAccess, validateOrigin, validateTargetAgent, validateTenantId, validateTriggerHeaderValue, verifyAuthorizationHeader, verifyInternalServiceAuthHeader, verifyInternalServiceToken, verifyJwt, verifyPoW, verifyServiceToken, verifySignatureWithConfig, verifySlackLinkToken, verifySlackUserToken, verifyTempToken, verifyTriggerAuth };
+export { ApiKeyGenerationResult, AppCredentialResult, AssembleResult, CommonCreateErrorResponses, CommonDeleteErrorResponses, CommonGetErrorResponses, CommonUpdateErrorResponses, ComposioAuthResult, CredentialScope, ERROR_DOCS_BASE_URL, ErrorCode, ErrorCodes, ErrorResponse, GenerateInternalServiceTokenParams, GenerateServiceTokenParams, GenerationType, HashedHeaderValue, InternalServiceId, InternalServiceTokenPayload, InternalServices, JsonTransformer, JwtVerifyResult, LLMMessage, LoggerFactoryConfig, McpClient, McpClientOptions, McpOAuthFlowResult, McpSSEConfig, McpServerConfig, McpStreamableHttpConfig, McpTokenExchangeResult, MockLanguageModel, ModelFactory, OAuthConfig, ParsedSSEResponse, PinoLogger, PinoLoggerConfig, PoWError, PoWResult, ProblemDetails, ServiceTokenPayload, SignJwtOptions, SignSlackLinkTokenParams, SignSlackUserTokenParams, SignatureVerificationErrorCode, SignatureVerificationResult, SignedTempToken, SlackAccessTokenPayload, SlackAccessTokenPayloadSchema, SlackLinkIntent, SlackLinkIntentSchema, SlackLinkTokenPayload, SlackLinkTokenPayloadSchema, TRUSTED_WORK_APP_MCP_PATHS, TempTokenPayload, TriggerAuthResult, ValidatePublicKeyResult, VerifyInternalServiceTokenResult, VerifyJwtOptions, VerifyServiceTokenResult, VerifySlackLinkTokenResult, VerifySlackUserTokenResult, _resetWaitUntilCache, activeMcpClients, buildConversationMetadata, commonCreateErrorResponses, commonDeleteErrorResponses, commonGetErrorResponses, commonUpdateErrorResponses, configureComposioMCPServer, convertZodToJsonSchema, convertZodToJsonSchemaWithPreview, createApiError, createMockModel, decodeJwtPayload, deleteComposioConnectedAccount, deriveRelationId, detectAuthenticationRequired, errorResponseSchema, errorSchemaFactory, estimateTokens, exchangeMcpAuthorizationCode, extractBearerToken, extractComposioServerId, extractPreviewFields, extractPublicId, extractUsageTokens, fetchComposioServers, fetchSingleComposioServer, flushTraces, formatMessagesForLLM, formatMessagesForLLMContext, gatewayCostMiddleware, generateApiKey, generateAppCredential, generateId, generateInternalServiceToken, generateServiceToken, getComposioInstance, getComposioOAuthRedirectUrl, getComposioUserId, getConversationId, getCredentialStoreLookupKeyFromRetrievalParams, getInProcessFetch, getJwtSecret, getLogger, getMetadataFromApiKey, getPoWErrorMessage, getTracer, getWaitUntil, handleApiError, hasIssuer, hashApiKey, hashAuthenticationHeaders, hashTriggerHeaderValue, initiateMcpOAuthFlow, interpolateTemplate, isApiKeyExpired, isComposioMCPServerAuthenticated, isDevelopment, isInternalServiceToken, isPoWEnabled, isProduction, isSlackUserToken, isTest, isThirdPartyMCPServerAuthenticated, isTrustedWorkAppMcpUrl, isUniqueConstraintError, isZodSchema, loggerFactory, makeAllPropertiesRequired, maskApiKey, normalizeDataComponentSchema, normalizeDateString, parseEmbeddedJson, parseSSEResponse, preview, problemDetailsSchema, registerAppFetch, retryWithBackoff, sanitizeAppConfig, setSpanWithError, signJwt, signSlackLinkToken, signSlackUserToken, signTempToken, stripUnsupportedConstraints, throwIfUniqueConstraintError, toISODateString, unwrapError, validateApiKey, validateInternalServiceProjectAccess, validateInternalServiceTenantAccess, validateOrigin, validatePublicKey, validateTargetAgent, validateTenantId, validateTriggerHeaderValue, verifyAuthorizationHeader, verifyInternalServiceAuthHeader, verifyInternalServiceToken, verifyJwt, verifyPoW, verifyServiceToken, verifySignatureWithConfig, verifySlackLinkToken, verifySlackUserToken, verifyTempToken, verifyTriggerAuth };

package/dist/utils/index.js

@@ -7,7 +7,7 @@ import { configureComposioMCPServer, deleteComposioConnectedAccount, extractComp
 import { isThirdPartyMCPServerAuthenticated } from "./third-party-mcp-servers/third-party-check.js";
 import "./third-party-mcp-servers/index.js";
 import { getCredentialStoreLookupKeyFromRetrievalParams } from "./credential-store-utils.js";
-import { generateId, getConversationId } from "./conversations.js";
+import { buildConversationMetadata, deriveRelationId, generateId, getConversationId } from "./conversations.js";
 import { ERROR_DOCS_BASE_URL, ErrorCode, commonCreateErrorResponses, commonDeleteErrorResponses, commonGetErrorResponses, commonUpdateErrorResponses, createApiError, errorResponseSchema, errorSchemaFactory, handleApiError, isUniqueConstraintError, problemDetailsSchema, throwIfUniqueConstraintError } from "./error.js";
 import { extractPublicId, generateApiKey, generateAppCredential, getMetadataFromApiKey, hashApiKey, isApiKeyExpired, maskApiKey, sanitizeAppConfig, validateApiKey } from "./apiKeys.js";
 import { normalizeDateString, toISODateString } from "./date.js";
@@ -19,6 +19,7 @@ import { JsonTransformer } from "./JsonTransformer.js";
 import { parseEmbeddedJson } from "./json-parser.js";
 import { McpClient, activeMcpClients } from "./mcp-client.js";
 import { MockLanguageModel, createMockModel } from "./mock-provider.js";
+import { extractUsageTokens, gatewayCostMiddleware } from "./usage-cost-middleware.js";
 import { ModelFactory } from "./model-factory.js";
 import { getPoWErrorMessage, isPoWEnabled, verifyPoW } from "./pow.js";
 import { retryWithBackoff } from "./retry.js";
@@ -28,9 +29,11 @@ import { SlackAccessTokenPayloadSchema, isSlackUserToken, signSlackUserToken, ve
 import { parseSSEResponse } from "./sse-parser.js";
 import { signTempToken, verifyTempToken } from "./temp-jwt.js";
 import { interpolateTemplate } from "./template-interpolation.js";
+import { estimateTokens } from "./token-estimator.js";
 import { flushTraces, getTracer, setSpanWithError, unwrapError } from "./tracer-factory.js";
 import { hashAuthenticationHeaders, hashTriggerHeaderValue, validateTriggerHeaderValue, verifySignatureWithConfig, verifyTriggerAuth } from "./trigger-auth.js";
+import { validatePublicKey } from "./validate-public-key.js";
 import { _resetWaitUntilCache, getWaitUntil } from "./wait-until.js";
 import { TRUSTED_WORK_APP_MCP_PATHS, isTrustedWorkAppMcpUrl } from "./work-app-mcp.js";
 
-export { ERROR_DOCS_BASE_URL, ErrorCode, InternalServices, JsonTransformer, McpClient, MockLanguageModel, ModelFactory, PinoLogger, SlackAccessTokenPayloadSchema, SlackLinkIntentSchema, SlackLinkTokenPayloadSchema, TRUSTED_WORK_APP_MCP_PATHS, _resetWaitUntilCache, activeMcpClients, commonCreateErrorResponses, commonDeleteErrorResponses, commonGetErrorResponses, commonUpdateErrorResponses, configureComposioMCPServer, convertZodToJsonSchema, convertZodToJsonSchemaWithPreview, createApiError, createMockModel, decodeJwtPayload, deleteComposioConnectedAccount, detectAuthenticationRequired, errorResponseSchema, errorSchemaFactory, exchangeMcpAuthorizationCode, extractBearerToken, extractComposioServerId, extractPreviewFields, extractPublicId, fetchComposioServers, fetchSingleComposioServer, flushTraces, formatMessagesForLLM, formatMessagesForLLMContext, generateApiKey, generateAppCredential, generateId, generateInternalServiceToken, generateServiceToken, getComposioInstance, getComposioOAuthRedirectUrl, getComposioUserId, getConversationId, getCredentialStoreLookupKeyFromRetrievalParams, getInProcessFetch, getJwtSecret, getLogger, getMetadataFromApiKey, getPoWErrorMessage, getTracer, getWaitUntil, handleApiError, hasIssuer, hashApiKey, hashAuthenticationHeaders, hashTriggerHeaderValue, initiateMcpOAuthFlow, interpolateTemplate, isApiKeyExpired, isComposioMCPServerAuthenticated, isDevelopment, isInternalServiceToken, isPoWEnabled, isProduction, isSlackUserToken, isTest, isThirdPartyMCPServerAuthenticated, isTrustedWorkAppMcpUrl, isUniqueConstraintError, isZodSchema, loggerFactory, makeAllPropertiesRequired, maskApiKey, normalizeDataComponentSchema, normalizeDateString, parseEmbeddedJson, parseSSEResponse, preview, problemDetailsSchema, registerAppFetch, retryWithBackoff, sanitizeAppConfig, setSpanWithError, signJwt, signSlackLinkToken, signSlackUserToken, signTempToken, stripUnsupportedConstraints, throwIfUniqueConstraintError, toISODateString, unwrapError, validateApiKey, validateInternalServiceProjectAccess, validateInternalServiceTenantAccess, validateOrigin, validateTargetAgent, validateTenantId, validateTriggerHeaderValue, verifyAuthorizationHeader, verifyInternalServiceAuthHeader, verifyInternalServiceToken, verifyJwt, verifyPoW, verifyServiceToken, verifySignatureWithConfig, verifySlackLinkToken, verifySlackUserToken, verifyTempToken, verifyTriggerAuth };
+export { ERROR_DOCS_BASE_URL, ErrorCode, InternalServices, JsonTransformer, McpClient, MockLanguageModel, ModelFactory, PinoLogger, SlackAccessTokenPayloadSchema, SlackLinkIntentSchema, SlackLinkTokenPayloadSchema, TRUSTED_WORK_APP_MCP_PATHS, _resetWaitUntilCache, activeMcpClients, buildConversationMetadata, commonCreateErrorResponses, commonDeleteErrorResponses, commonGetErrorResponses, commonUpdateErrorResponses, configureComposioMCPServer, convertZodToJsonSchema, convertZodToJsonSchemaWithPreview, createApiError, createMockModel, decodeJwtPayload, deleteComposioConnectedAccount, deriveRelationId, detectAuthenticationRequired, errorResponseSchema, errorSchemaFactory, estimateTokens, exchangeMcpAuthorizationCode, extractBearerToken, extractComposioServerId, extractPreviewFields, extractPublicId, extractUsageTokens, fetchComposioServers, fetchSingleComposioServer, flushTraces, formatMessagesForLLM, formatMessagesForLLMContext, gatewayCostMiddleware, generateApiKey, generateAppCredential, generateId, generateInternalServiceToken, generateServiceToken, getComposioInstance, getComposioOAuthRedirectUrl, getComposioUserId, getConversationId, getCredentialStoreLookupKeyFromRetrievalParams, getInProcessFetch, getJwtSecret, getLogger, getMetadataFromApiKey, getPoWErrorMessage, getTracer, getWaitUntil, handleApiError, hasIssuer, hashApiKey, hashAuthenticationHeaders, hashTriggerHeaderValue, initiateMcpOAuthFlow, interpolateTemplate, isApiKeyExpired, isComposioMCPServerAuthenticated, isDevelopment, isInternalServiceToken, isPoWEnabled, isProduction, isSlackUserToken, isTest, isThirdPartyMCPServerAuthenticated, isTrustedWorkAppMcpUrl, isUniqueConstraintError, isZodSchema, loggerFactory, makeAllPropertiesRequired, maskApiKey, normalizeDataComponentSchema, normalizeDateString, parseEmbeddedJson, parseSSEResponse, preview, problemDetailsSchema, registerAppFetch, retryWithBackoff, sanitizeAppConfig, setSpanWithError, signJwt, signSlackLinkToken, signSlackUserToken, signTempToken, stripUnsupportedConstraints, throwIfUniqueConstraintError, toISODateString, unwrapError, validateApiKey, validateInternalServiceProjectAccess, validateInternalServiceTenantAccess, validateOrigin, validatePublicKey, validateTargetAgent, validateTenantId, validateTriggerHeaderValue, verifyAuthorizationHeader, verifyInternalServiceAuthHeader, verifyInternalServiceToken, verifyJwt, verifyPoW, verifyServiceToken, verifySignatureWithConfig, verifySlackLinkToken, verifySlackUserToken, verifyTempToken, verifyTriggerAuth };

package/dist/utils/model-factory.js

@@ -1,5 +1,7 @@
 import { getLogger } from "./logger.js";
 import { createMockModel } from "./mock-provider.js";
+import { gatewayCostMiddleware } from "./usage-cost-middleware.js";
+import { wrapLanguageModel } from "ai";
 import { anthropic, createAnthropic } from "@ai-sdk/anthropic";
 import { createAzure } from "@ai-sdk/azure";
 import { createGateway, gateway } from "@ai-sdk/gateway";
@@ -10,6 +12,11 @@ import { createOpenRouter, openrouter } from "@openrouter/ai-sdk-provider";
 
 //#region src/utils/model-factory.ts
 const logger = getLogger("ModelFactory");
+const GATEWAY_ROUTABLE_PROVIDERS = [
+	"anthropic",
+	"openai",
+	"google"
+];
 const nimDefault = createOpenAICompatible({
 	name: "nim",
 	baseURL: "https://integrate.api.nvidia.com/v1",
@@ -107,21 +114,39 @@ var ModelFactory = class ModelFactory {
 			hasProviderOptions: !!modelSettings.providerOptions
 		}, "Creating language model from config");
 		const providerConfig = ModelFactory.extractProviderConfig(modelSettings.providerOptions);
-		if (provider !== "mock" && (provider === "azure" || Object.keys(providerConfig).length > 0)) {
+		const shouldRouteViaGateway = !!process.env.AI_GATEWAY_API_KEY && GATEWAY_ROUTABLE_PROVIDERS.includes(provider) && Object.keys(providerConfig).length === 0;
+		let model;
+		if (shouldRouteViaGateway) model = gateway(`${provider}/${modelName}`);
+		else if (provider !== "mock" && (provider === "azure" || Object.keys(providerConfig).length > 0)) {
 			logger.info({ config: providerConfig }, `Applying custom ${provider} provider configuration`);
-			return ModelFactory.createProvider(provider, providerConfig).languageModel(modelName);
-		}
-		switch (provider) {
-			case "anthropic": return anthropic(modelName);
-			case "openai": return openai(modelName);
-			case "google": return google(modelName);
-			case "openrouter": return openrouter(modelName);
-			case "gateway": return gateway(modelName);
-			case "nim": return nimDefault(modelName);
+			model = ModelFactory.createProvider(provider, providerConfig).languageModel(modelName);
+		} else switch (provider) {
+			case "anthropic":
+				model = anthropic(modelName);
+				break;
+			case "openai":
+				model = openai(modelName);
+				break;
+			case "google":
+				model = google(modelName);
+				break;
+			case "openrouter":
+				model = openrouter(modelName);
+				break;
+			case "gateway":
+				model = gateway(modelName);
+				break;
+			case "nim":
+				model = nimDefault(modelName);
+				break;
 			case "mock": return createMockModel(modelName);
 			case "custom": throw new Error("Custom provider requires configuration. Please provide baseURL in providerOptions.baseURL");
 			default: throw new Error(`Unsupported provider: ${provider}. Supported providers are: ${ModelFactory.BUILT_IN_PROVIDERS.join(", ")}. To access other models, use OpenRouter (openrouter/model-id), Vercel AI Gateway (gateway/model-id), NVIDIA NIM (nim/model-id), or Custom OpenAI-compatible (custom/model-id).`);
 		}
+		return wrapLanguageModel({
+			model,
+			middleware: gatewayCostMiddleware
+		});
 	}
 	/**
 	* Built-in providers that have special handling