@inkeep/agents-core 0.6.0 → 0.6.5

package/dist/index.cjs

@@ -1813,6 +1813,22 @@ var PaginationQueryParamsSchema = zodOpenapi.z.object({
 
 // src/context/ContextConfig.ts
 var logger = getLogger("context-config");
+var RequestContextSchemaBuilder = class {
+  constructor(options) {
+    __publicField(this, "schema");
+    this.schema = options.schema;
+  }
+  /** Template function for request context paths with type-safe autocomplete */
+  toTemplate(path2) {
+    return `{{requestContext.${path2}}}`;
+  }
+  getSchema() {
+    return this.schema;
+  }
+  getJsonSchema() {
+    return convertZodToJsonSchema(this.schema);
+  }
+};
 function convertZodToJsonSchema(zodSchema) {
   try {
     return zod.z.toJSONSchema(zodSchema, { target: "draft-7" });
@@ -1835,23 +1851,34 @@ var ContextConfigBuilder = class {
     this.tenantId = options.tenantId || "default";
     this.projectId = options.projectId || "default";
     this.baseURL = process.env.INKEEP_AGENTS_MANAGE_API_URL || "http://localhost:3002";
-    let requestContextSchema;
+    let requestContextSchema2;
     if (options.requestContextSchema) {
+      const actualSchema = options.requestContextSchema instanceof RequestContextSchemaBuilder ? options.requestContextSchema.getSchema() : options.requestContextSchema;
       logger.info(
         {
           requestContextSchema: options.requestContextSchema
         },
         "Converting request headers schema to JSON Schema for database storage"
       );
-      let schema = options.requestContextSchema;
-      if (schema instanceof zod.z.ZodObject) {
-        schema = schema.loose();
+      requestContextSchema2 = convertZodToJsonSchema(actualSchema);
+    }
+    const processedContextVariables = {};
+    if (options.contextVariables) {
+      for (const [key, definition] of Object.entries(options.contextVariables)) {
+        const { credentialReference, ...rest } = definition;
+        processedContextVariables[key] = {
+          ...rest,
+          responseSchema: convertZodToJsonSchema(definition.responseSchema),
+          credentialReferenceId: credentialReference?.id
+        };
         logger.debug(
-          { schemaType: "ZodObject" },
-          "Applied .loose() to ZodObject requestContextSchema for more permissive validation"
+          {
+            contextVariableKey: key,
+            originalSchema: definition.responseSchema
+          },
+          "Converting contextVariable responseSchema to JSON Schema for database storage"
         );
       }
-      requestContextSchema = convertZodToJsonSchema(schema);
     }
     this.config = {
       id: options.id,
@@ -1859,8 +1886,8 @@ var ContextConfigBuilder = class {
       projectId: this.projectId,
       name: options.name,
       description: options.description || "",
-      requestContextSchema,
-      contextVariables: options.contextVariables || {}
+      requestContextSchema: requestContextSchema2,
+      contextVariables: processedContextVariables
     };
     logger.info(
       {
@@ -1870,6 +1897,22 @@ var ContextConfigBuilder = class {
       "ContextConfig builder initialized"
     );
   }
+  /**
+   * Convert the builder to a plain object for database operations
+   */
+  toObject() {
+    return {
+      id: this.getId(),
+      tenantId: this.tenantId,
+      projectId: this.projectId,
+      name: this.getName(),
+      description: this.getDescription(),
+      requestContextSchema: this.getRequestContextSchema(),
+      contextVariables: this.getContextVariables(),
+      createdAt: (/* @__PURE__ */ new Date()).toISOString(),
+      updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+    };
+  }
   // Getter methods
   getId() {
     if (!this.config.id) {
@@ -1892,35 +1935,14 @@ var ContextConfigBuilder = class {
   getContextVariables() {
     return this.config.contextVariables || {};
   }
-  /**
-   * Convert the builder to a plain object for database operations
-   */
-  toObject() {
-    return {
-      id: this.getId(),
-      tenantId: this.tenantId,
-      projectId: this.projectId,
-      name: this.getName(),
-      description: this.getDescription(),
-      requestContextSchema: this.getRequestContextSchema(),
-      contextVariables: this.getContextVariables(),
-      createdAt: (/* @__PURE__ */ new Date()).toISOString(),
-      updatedAt: (/* @__PURE__ */ new Date()).toISOString()
-    };
-  }
   // Builder methods for fluent API
   withRequestContextSchema(schema) {
     this.config.requestContextSchema = schema;
     return this;
   }
-  withContextVariable(key, definition) {
-    this.config.contextVariables = this.config.contextVariables || {};
-    this.config.contextVariables[key] = definition;
-    return this;
-  }
-  withContextVariables(variables) {
-    this.config.contextVariables = variables;
-    return this;
+  /** 4) The function you ship: path autocomplete + validation, returns {{path}} */
+  toTemplate(path2) {
+    return `{{${path2}}}`;
   }
   // Validation method
   validate() {
@@ -2065,15 +2087,11 @@ var ContextConfigBuilder = class {
 function contextConfig(options) {
   return new ContextConfigBuilder(options);
 }
+function requestContextSchema(options) {
+  return new RequestContextSchemaBuilder(options);
+}
 function fetchDefinition(options) {
-  const fetchConfig = options.fetchConfig || {
-    url: options.url,
-    method: options.method,
-    headers: options.headers,
-    body: options.body,
-    transform: options.transform,
-    timeout: options.timeout
-  };
+  const fetchConfig = options.fetchConfig;
   return {
     id: options.id,
     name: options.name,
@@ -2086,9 +2104,9 @@ function fetchDefinition(options) {
       transform: fetchConfig.transform,
       timeout: fetchConfig.timeout
     },
-    responseSchema: options.responseSchema ? convertZodToJsonSchema(options.responseSchema) : void 0,
+    responseSchema: options.responseSchema,
     defaultValue: options.defaultValue,
-    credentialReferenceId: options.credential?.id
+    credentialReferenceId: options.credentialReference?.id
   };
 }
 var logger2 = getLogger("template-engine");
@@ -8121,7 +8139,7 @@ var McpClient = class {
     if (!activeTools) return;
     for (const item of activeTools) {
       if (tools2.includes(item)) continue;
-      throw new Error(`[Tools] Tool ${item} not found in tools`);
+      console.warn(`[Tools] Tool ${item} not found in tools`);
     }
   }
   async selectTools() {
@@ -8786,21 +8804,62 @@ var ContextResolver = class {
 var logger7 = getLogger("context-validation");
 var ajv = new Ajv__default.default({ allErrors: true, strict: false });
 var HTTP_REQUEST_PARTS = ["headers"];
+var MAX_SCHEMA_CACHE_SIZE = 1e3;
 var schemaCache = /* @__PURE__ */ new Map();
 function isValidHttpRequest(obj) {
   return obj != null && typeof obj === "object" && !Array.isArray(obj) && "headers" in obj;
 }
 function getCachedValidator(schema) {
   const key = JSON.stringify(schema);
-  if (!schemaCache.has(key)) {
-    schemaCache.set(key, ajv.compile(schema));
+  if (schemaCache.has(key)) {
+    const validator2 = schemaCache.get(key);
+    if (!validator2) {
+      throw new Error("Unexpected: validator not found in cache after has() check");
+    }
+    schemaCache.delete(key);
+    schemaCache.set(key, validator2);
+    return validator2;
   }
-  const validator = schemaCache.get(key);
-  if (!validator) {
-    throw new Error("Failed to compile JSON schema");
+  if (schemaCache.size >= MAX_SCHEMA_CACHE_SIZE) {
+    const firstKey = schemaCache.keys().next().value;
+    if (firstKey) {
+      schemaCache.delete(firstKey);
+    }
   }
+  const permissiveSchema = makeSchemaPermissive(schema);
+  const validator = ajv.compile(permissiveSchema);
+  schemaCache.set(key, validator);
   return validator;
 }
+function makeSchemaPermissive(schema) {
+  if (!schema || typeof schema !== "object") {
+    return schema;
+  }
+  const permissiveSchema = { ...schema };
+  if (permissiveSchema.type === "object") {
+    permissiveSchema.additionalProperties = true;
+    if (permissiveSchema.properties && typeof permissiveSchema.properties === "object") {
+      const newProperties = {};
+      for (const [key, value] of Object.entries(permissiveSchema.properties)) {
+        newProperties[key] = makeSchemaPermissive(value);
+      }
+      permissiveSchema.properties = newProperties;
+    }
+  }
+  if (permissiveSchema.type === "array" && permissiveSchema.items) {
+    permissiveSchema.items = makeSchemaPermissive(permissiveSchema.items);
+  }
+  if (permissiveSchema.oneOf) {
+    permissiveSchema.oneOf = permissiveSchema.oneOf.map(makeSchemaPermissive);
+  }
+  if (permissiveSchema.anyOf) {
+    permissiveSchema.anyOf = permissiveSchema.anyOf.map(makeSchemaPermissive);
+  }
+  if (permissiveSchema.allOf) {
+    permissiveSchema.allOf = permissiveSchema.allOf.map(makeSchemaPermissive);
+  }
+  return permissiveSchema;
+}
 function validationHelper(jsonSchema) {
   return getCachedValidator(jsonSchema);
 }
@@ -10702,6 +10761,7 @@ exports.projectsRelations = projectsRelations;
 exports.removeArtifactComponentFromAgent = removeArtifactComponentFromAgent;
 exports.removeDataComponentFromAgent = removeDataComponentFromAgent;
 exports.removeToolFromAgent = removeToolFromAgent;
+exports.requestContextSchema = requestContextSchema;
 exports.resourceIdSchema = resourceIdSchema;
 exports.setActiveAgentForConversation = setActiveAgentForConversation;
 exports.setActiveAgentForThread = setActiveAgentForThread;