@superblocksteam/vite-plugin-file-sync 2.0.42-next.9 → 2.0.43-next.0

package/dist/ai-service/integrations/metadata/databricks.js

@@ -0,0 +1,174 @@
+import { DatabaseIntegrationMetadata } from "./database.js";
+export class DatabricksIntegrationMetadata extends DatabaseIntegrationMetadata {
+    static async refineMetadataForPrompt({ metadata, prompt, llmProvider, }) {
+        // If no prompt provided, just do structural transformation without filtering
+        if (!prompt || prompt.trim() === "") {
+            return this.transformToHierarchical(metadata);
+        }
+        // First, normalize the Databricks table names to full qualified format
+        // so the LLM filtering works correctly
+        const normalizedMetadata = this.normalizeTableNames(metadata);
+        // Apply the parent's refinement (LLM-based table filtering, etc.)
+        // This works on the flat structure with normalized names
+        const filteredMetadata = await super.refineMetadataForPrompt({
+            metadata: normalizedMetadata,
+            prompt,
+            llmProvider,
+        });
+        // Transform the filtered result into hierarchical catalog structure
+        return this.transformToHierarchical(filteredMetadata);
+    }
+    /**
+     * Normalize Databricks table names to full qualified format (catalog.schema.table)
+     * This makes them usable by the parent's LLM filtering logic
+     */
+    static normalizeTableNames(metadata) {
+        const schema = metadata.schema || metadata.dbSchema;
+        if (!schema || !schema.tables || !Array.isArray(schema.tables)) {
+            return metadata;
+        }
+        const normalizedTables = schema.tables.map((table) => {
+            // In the current structure:
+            // - table.schema contains the CATALOG name (e.g., "instacart")
+            // - table.name contains SCHEMA.TABLE (e.g., "default.customers")
+            const catalogName = table.schema || "default";
+            const nameParts = table.name.split(".");
+            let schemaName;
+            let tableName;
+            if (nameParts.length === 2) {
+                schemaName = nameParts[0] || "default";
+                tableName = nameParts[1] || table.name;
+            }
+            else {
+                schemaName = "default";
+                tableName = table.name;
+            }
+            // Create the full qualified name
+            const fullName = `${catalogName}.${schemaName}.${tableName}`;
+            // Return table with normalized name for filtering
+            return {
+                ...table,
+                name: fullName, // Use full qualified name for filtering
+                _originalCatalog: catalogName,
+                _originalSchema: schemaName,
+                _originalTableName: tableName,
+            };
+        });
+        return {
+            schema: {
+                tables: normalizedTables,
+                schemas: schema.schemas,
+            },
+        };
+    }
+    static transformToHierarchical(metadata) {
+        const schema = metadata.schema || metadata.dbSchema;
+        if (!schema || !schema.tables || !Array.isArray(schema.tables)) {
+            return metadata;
+        }
+        // Group tables by catalog and schema
+        const catalogMap = new Map();
+        for (const table of schema.tables) {
+            // If the table has been normalized, use the _original* fields
+            // Otherwise, parse from the existing structure
+            const catalogName = table._originalCatalog || table.schema || "default";
+            const schemaName = table._originalSchema || "default";
+            const tableName = table._originalTableName || table.name;
+            // Get or create catalog entry
+            if (!catalogMap.has(catalogName)) {
+                catalogMap.set(catalogName, new Map());
+            }
+            const schemaMap = catalogMap.get(catalogName);
+            // Get or create schema entry within catalog
+            if (!schemaMap.has(schemaName)) {
+                schemaMap.set(schemaName, []);
+            }
+            const tables = schemaMap.get(schemaName);
+            // Create the full qualified name for convenience
+            const fullName = `${catalogName}.${schemaName}.${tableName}`;
+            // Clean up the table object and remove temporary fields
+            const cleanTable = { ...table };
+            delete cleanTable._originalCatalog;
+            delete cleanTable._originalSchema;
+            delete cleanTable._originalTableName;
+            // Add table with updated structure
+            tables.push({
+                ...cleanTable,
+                name: tableName, // Just the table name
+                schema: schemaName, // Just the schema name
+                catalog: catalogName, // Add catalog field
+                fullName, // Full qualified name for easy reference
+            });
+        }
+        // Build the hierarchical structure
+        const catalogs = [];
+        for (const [catalogName, schemaMap] of catalogMap.entries()) {
+            const schemas = [];
+            for (const [schemaName, tables] of schemaMap.entries()) {
+                schemas.push({
+                    name: schemaName,
+                    tables,
+                });
+            }
+            catalogs.push({
+                name: catalogName,
+                schemas,
+            });
+        }
+        return {
+            schema: {
+                catalogs,
+            },
+        };
+    }
+    /**
+     * Override to keep full qualified names without catalog-stripping normalization
+     */
+    static async dbTablesForPrompt(dbSchema, prompt, llmProvider) {
+        const { buildDbSchemaIntentPrompt } = await import("../../prompt-builder-service/builders/dbschema-intent.js");
+        const { generateObject } = await import("ai");
+        const { z } = await import("zod");
+        const { traceLLM } = await import("../../llmobs/helpers.js");
+        const { getLogger } = await import("../../../util/logger.js");
+        const { prompt: refineSchemaPrompt } = buildDbSchemaIntentPrompt(prompt, dbSchema);
+        const model = llmProvider.modelForTask("openApiIntent");
+        const flexibleSchema = z.object({
+            tables: z
+                .array(z.string())
+                .describe("The tables that the user wants to access"),
+        });
+        const result = await traceLLM("refineDbSchema", model.modelId, model.provider, async () => {
+            return await generateObject({
+                model,
+                schema: flexibleSchema,
+                prompt: refineSchemaPrompt,
+            });
+        });
+        if (!result?.object?.tables) {
+            getLogger().warn(`No tables returned from model. Result: ${JSON.stringify(result)}`);
+            return [];
+        }
+        // For Databricks, don't normalize - keep full qualified names (catalog.schema.table)
+        const validChunkTableNames = new Set(dbSchema.tables.map((t) => t.name).filter(Boolean));
+        // Filter to only include tables that exist in this chunk
+        // Match exactly on full qualified names
+        const filteredTables = result.object.tables.filter((tableName) => {
+            return validChunkTableNames.has(tableName);
+        });
+        return filteredTables;
+    }
+    /**
+     * Override to handle matching with full qualified names
+     */
+    static trimSchema(tables, toKeep) {
+        // For Databricks, match against full qualified names
+        const toKeepSet = new Set(toKeep);
+        const refinedTables = tables.filter((table) => {
+            // The table.name should be the full qualified name at this point
+            // (e.g., "instacart.default.customers")
+            return toKeepSet.has(table.name);
+        });
+        return refinedTables;
+    }
+}
+//# sourceMappingURL=databricks.js.map

package/dist/ai-service/integrations/metadata/databricks.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"databricks.js","sourceRoot":"","sources":["../../../../src/ai-service/integrations/metadata/databricks.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,2BAA2B,EAAE,MAAM,eAAe,CAAC;AAc5D,MAAM,OAAO,6BAA8B,SAAQ,2BAA2B;IACrE,MAAM,CAAU,KAAK,CAAC,uBAAuB,CAAC,EACnD,QAAQ,EACR,MAAM,EACN,WAAW,GAKZ;QACC,6EAA6E;QAC7E,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YACpC,OAAO,IAAI,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC;QAChD,CAAC;QAED,uEAAuE;QACvE,uCAAuC;QACvC,MAAM,kBAAkB,GAAG,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAE9D,kEAAkE;QAClE,yDAAyD;QACzD,MAAM,gBAAgB,GAAG,MAAM,KAAK,CAAC,uBAAuB,CAAC;YAC3D,QAAQ,EAAE,kBAAkB;YAC5B,MAAM;YACN,WAAW;SACZ,CAAC,CAAC;QAEH,oEAAoE;QACpE,OAAO,IAAI,CAAC,uBAAuB,CAAC,gBAAgB,CAAC,CAAC;IACxD,CAAC;IAED;;;OAGG;IACK,MAAM,CAAC,mBAAmB,CAChC,QAA6B;QAE7B,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,QAAQ,CAAC;QAEpD,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/D,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,gBAAgB,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAA0B,EAAE,EAAE;YACxE,4BAA4B;YAC5B,+DAA+D;YAC/D,iEAAiE;YACjE,MAAM,WAAW,GAAG,KAAK,CAAC,MAAM,IAAI,SAAS,CAAC;YAC9C,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAExC,IAAI,UAAkB,CAAC;YACvB,IAAI,SAAiB,CAAC;YAEtB,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3B,UAAU,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;gBACvC,SAAS,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC;YACzC,CAAC;iBAAM,CAAC;gBACN,UAAU,GAAG,SAAS,CAAC;gBACvB,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC;YACzB,CAAC;YAED,iCAAiC;YACjC,MAAM,QAAQ,GAAG,GAAG,WAAW,IAAI,UAAU,IAAI,SAAS,EAAE,CAAC;YAE7D,kDAAkD;YAClD,OAAO;gBACL,GAAG,KAAK;gBACR,IAAI,EAAE,QAAQ,EAAE,wCAAwC;gBACxD,gBAAgB,EAAE,WAAW;gBAC7B,eAAe,EAAE,UAAU;gBAC3B,kBAAkB,EAAE,SAAS;aAC9B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,MAAM,EAAE;gBACN,MAAM,EAAE,gBAAgB;gBACxB,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB;SACF,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,uBAAuB,CACpC,QAA6B;QAE7B,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,QAAQ,CAAC;QAEpD,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/D,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,qCAAqC;QACrC,MAAM,UAAU,GAAG,IAAI,GAAG,EAA8C,CAAC;QAEzE,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClC,8DAA8D;YAC9D,+CAA+C;YAC/C,MAAM,WAAW,GACd,KAAa,CAAC,gBAAgB,IAAI,KAAK,CAAC,MAAM,IAAI,SAAS,CAAC;YAC/D,MAAM,UAAU,GAAI,KAAa,CAAC,eAAe,IAAI,SAAS,CAAC;YAC/D,MAAM,SAAS,GAAI,KAAa,CAAC,kBAAkB,IAAI,KAAK,CAAC,IAAI,CAAC;YAElE,8BAA8B;YAC9B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;gBACjC,UAAU,CAAC,GAAG,CAAC,WAAW,EAAE,IAAI,GAAG,EAAiC,CAAC,CAAC;YACxE,CAAC;YACD,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,CAAC,WAAW,CAAE,CAAC;YAE/C,4CAA4C;YAC5C,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC/B,SAAS,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YAChC,CAAC;YACD,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,UAAU,CAAE,CAAC;YAE1C,iDAAiD;YACjD,MAAM,QAAQ,GAAG,GAAG,WAAW,IAAI,UAAU,IAAI,SAAS,EAAE,CAAC;YAE7D,wDAAwD;YACxD,MAAM,UAAU,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC;YAChC,OAAQ,UAAkB,CAAC,gBAAgB,CAAC;YAC5C,OAAQ,UAAkB,CAAC,eAAe,CAAC;YAC3C,OAAQ,UAAkB,CAAC,kBAAkB,CAAC;YAE9C,mCAAmC;YACnC,MAAM,CAAC,IAAI,CAAC;gBACV,GAAG,UAAU;gBACb,IAAI,EAAE,SAAS,EAAE,sBAAsB;gBACvC,MAAM,EAAE,UAAU,EAAE,uBAAuB;gBAC3C,OAAO,EAAE,WAAW,EAAE,oBAAoB;gBAC1C,QAAQ,EAAE,yCAAyC;aACW,CAAC,CAAC;QACpE,CAAC;QAED,mCAAmC;QACnC,MAAM,QAAQ,GAAwB,EAAE,CAAC;QAEzC,KAAK,MAAM,CAAC,WAAW,EAAE,SAAS,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC;YAC5D,MAAM,OAAO,GAAuB,EAAE,CAAC;YAEvC,KAAK,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,SAAS,CAAC,OAAO,EAAE,EAAE,CAAC;gBACvD,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,UAAU;oBAChB,MAAM;iBACP,CAAC,CAAC;YACL,CAAC;YAED,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,WAAW;gBACjB,OAAO;aACR,CAAC,CAAC;QACL,CAAC;QAED,OAAO;YACL,MAAM,EAAE;gBACN,QAAQ;aACT;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,MAAM,CAAU,KAAK,CAAC,iBAAiB,CACrC,QAAa,EACb,MAAc,EACd,WAAgB;QAEhB,MAAM,EAAE,yBAAyB,EAAE,GAAG,MAAM,MAAM,CAChD,0DAA0D,CAC3D,CAAC;QACF,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,EAAE,CAAC,EAAE,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;QAC7D,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;QAE9D,MAAM,EAAE,MAAM,EAAE,kBAAkB,EAAE,GAAG,yBAAyB,CAC9D,MAAM,EACN,QAAQ,CACT,CAAC;QAEF,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC;QAExD,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;YAC9B,MAAM,EAAE,CAAC;iBACN,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;iBACjB,QAAQ,CAAC,0CAA0C,CAAC;SACxD,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAC3B,gBAAgB,EAChB,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,QAAQ,EACd,KAAK,IAAI,EAAE;YACT,OAAO,MAAM,cAAc,CAAC;gBAC1B,KAAK;gBACL,MAAM,EAAE,cAAc;gBACtB,MAAM,EAAE,kBAAkB;aAC3B,CAAC,CAAC;QACL,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;YAC5B,SAAS,EAAE,CAAC,IAAI,CACd,0CAA0C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CACnE,CAAC;YACF,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,qFAAqF;QACrF,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAClC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CACxD,CAAC;QAEF,yDAAyD;QACzD,wCAAwC;QACxC,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAiB,EAAE,EAAE;YACvE,OAAO,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,OAAO,cAAc,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,MAAM,CAAU,UAAU,CACxB,MAA6B,EAC7B,MAAgB;QAEhB,qDAAqD;QACrD,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;QAElC,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YAC5C,iEAAiE;YACjE,wCAAwC;YACxC,OAAO,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,OAAO,aAAa,CAAC;IACvB,CAAC;CACF"}

package/dist/ai-service/integrations/store.d.ts

@@ -1,19 +1,49 @@
+import { PluginType, type Plugin, type PluginRegistry } from "@superblocksteam/shared";
 import { IntegrationMetadata } from "./metadata/index.js";
 import type { OpenApiSpec } from "../../util/open-api.js";
 import type { LLMProvider } from "../llm/types.js";
 import type { Clark, ClarkStateHandlerParams } from "../state-machine/clark-fsm.js";
-import type { SdkIntegration } from "../types.js";
+import type { SdkIntegration, EditorClient } from "../types.js";
 import type { PromptContextIntegration } from "@superblocksteam/library-shared/types";
+export type IntegrationId = string;
+export type PluginId = string;
+export type Integration = IntegrationHeader & {
+    metadata: IntegrationMetadataRecord;
+};
+export type IntegrationHeader = {
+    id: IntegrationId;
+    name: string;
+    plugin: PluginHeader;
+};
+export type PluginHeader = {
+    id: PluginId;
+    type: PluginType;
+    isBuiltIn: boolean;
+    basePluginId?: PluginId;
+};
+export type IntegrationMetadataRecord = Record<string, any> & {
+    openApiSpecRef?: string;
+};
 export declare class IntegrationStore {
     private openApiSpecs;
+    private pluginRegistry;
+    private integrations;
+    private metadataCache;
+    constructor(pluginRegistry?: PluginRegistry);
+    availableIntegrations(): IntegrationHeader[];
+    getPlugin(pluginId: PluginId): Plugin | undefined;
+    insertFromPromptContext(integration: PromptContextIntegration, editorClient?: EditorClient): void;
+    getHeader(id: IntegrationId): IntegrationHeader | undefined;
+    getMetadata(id: IntegrationId, editorClient?: EditorClient): Promise<IntegrationMetadataRecord | undefined>;
+    private getIntegration;
     getIntegrationForPromptContext(integration: PromptContextIntegration, prompt: string, clark: Clark, params: ClarkStateHandlerParams): Promise<SdkIntegration | null>;
-    getRelevantMetadata({ clark, integration, prompt, llmProvider, }: {
+    getRelevantMetadata({ clark, integrationId, prompt, llmProvider, }: {
         clark: Clark;
-        integration: PromptContextIntegration;
+        integrationId: IntegrationId;
         prompt: string;
         llmProvider: LLMProvider;
     }): Promise<Record<string, any>>;
     getMetadataClassForIntegration(integration: PromptContextIntegration): typeof IntegrationMetadata;
-    getOpenApiSpec(integration: PromptContextIntegration, clark: Clark): Promise<OpenApiSpec | undefined>;
+    getOpenApiSpec(integration: Integration, clark: Clark): Promise<OpenApiSpec | undefined>;
 }
 //# sourceMappingURL=store.d.ts.map

package/dist/ai-service/integrations/store.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../../../src/ai-service/integrations/store.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAG1D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,KAAK,EACV,KAAK,EACL,uBAAuB,EACxB,MAAM,+BAA+B,CAAC;AACvC,OAAO,KAAK,EACV,cAAc,EAGf,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,uCAAuC,CAAC;AAEtF,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,YAAY,CAAmC;IAMjD,8BAA8B,CAClC,WAAW,EAAE,wBAAwB,EACrC,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,uBAAuB,GAC9B,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC;IAiC3B,mBAAmB,CAAC,EACxB,KAAK,EACL,WAAW,EACX,MAAM,EACN,WAAW,GACZ,EAAE;QACD,KAAK,EAAE,KAAK,CAAC;QACb,WAAW,EAAE,wBAAwB,CAAC;QACtC,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,WAAW,CAAC;KAC1B,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAqBhC,8BAA8B,CAC5B,WAAW,EAAE,wBAAwB,GACpC,OAAO,mBAAmB;IAoBvB,cAAc,CAClB,WAAW,EAAE,wBAAwB,EACrC,KAAK,EAAE,KAAK,GACX,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC;CAyBpC"}
+{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../../../src/ai-service/integrations/store.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,UAAU,EAEV,KAAK,MAAM,EACX,KAAK,cAAc,EACpB,MAAM,yBAAyB,CAAC;AASjC,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAG1D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,KAAK,EACV,KAAK,EACL,uBAAuB,EACxB,MAAM,+BAA+B,CAAC;AACvC,OAAO,KAAK,EACV,cAAc,EAGd,YAAY,EACb,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,uCAAuC,CAAC;AAEtF,MAAM,MAAM,aAAa,GAAG,MAAM,CAAC;AACnC,MAAM,MAAM,QAAQ,GAAG,MAAM,CAAC;AAE9B,MAAM,MAAM,WAAW,GAAG,iBAAiB,GAAG;IAC5C,QAAQ,EAAE,yBAAyB,CAAC;CACrC,CAAC;AAaF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,EAAE,EAAE,aAAa,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,YAAY,CAAC;CACtB,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACzB,EAAE,EAAE,QAAQ,CAAC;IACb,IAAI,EAAE,UAAU,CAAC;IACjB,SAAS,EAAE,OAAO,CAAC;IACnB,YAAY,CAAC,EAAE,QAAQ,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG;IAC5D,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB,CAAC;AAEF,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,YAAY,CAAmC;IACvD,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,YAAY,CAA2B;IAC/C,OAAO,CAAC,aAAa,CACT;gBAEA,cAAc,GAAE,cAAyC;IAIrE,qBAAqB,IAAI,iBAAiB,EAAE;IAM5C,SAAS,CAAC,QAAQ,EAAE,QAAQ,GAAG,MAAM,GAAG,SAAS;IAIjD,uBAAuB,CACrB,WAAW,EAAE,wBAAwB,EACrC,YAAY,CAAC,EAAE,YAAY;IA8C7B,SAAS,CAAC,EAAE,EAAE,aAAa,GAAG,iBAAiB,GAAG,SAAS;IAQrD,WAAW,CACf,EAAE,EAAE,aAAa,EACjB,YAAY,CAAC,EAAE,YAAY,GAC1B,OAAO,CAAC,yBAAyB,GAAG,SAAS,CAAC;YAwBnC,cAAc;IAmBtB,8BAA8B,CAClC,WAAW,EAAE,wBAAwB,EACrC,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,uBAAuB,GAC9B,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC;IAiC3B,mBAAmB,CAAC,EACxB,KAAK,EACL,aAAa,EACb,MAAM,EACN,WAAW,GACZ,EAAE;QACD,KAAK,EAAE,KAAK,CAAC;QACb,aAAa,EAAE,aAAa,CAAC;QAC7B,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,WAAW,CAAC;KAC1B,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAuChC,8BAA8B,CAC5B,WAAW,EAAE,wBAAwB,GACpC,OAAO,mBAAmB;IAuBvB,cAAc,CAClB,WAAW,EAAE,WAAW,EACxB,KAAK,EAAE,KAAK,GACX,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC;CAyBpC"}

package/dist/ai-service/integrations/store.js

@@ -32,23 +32,39 @@ var __esDecorate = (this && this.__esDecorate) || function (ctor, descriptorIn,
     if (target) Object.defineProperty(target, contextIn.name, descriptor);
     done = true;
 };
+import { getBasePluginId, PluginType, RegisteredPlugins, } from "@superblocksteam/shared";
 import { isEmpty } from "lodash-es";
+import { getLogger } from "../../util/logger.js";
 import { datasourceSdkClassByType } from "../const.js";
 import { databaseIntegrationTypes } from "../const.js";
 import llmobs from "../llmobs/index.js";
 import { DatabaseIntegrationMetadata } from "./metadata/database.js";
+import { DatabricksIntegrationMetadata } from "./metadata/databricks.js";
 import { GraphQLIntegrationMetadata } from "./metadata/graphql-based.js";
 import { IntegrationMetadata } from "./metadata/index.js";
 import { OpenApiIntegrationMetadata } from "./metadata/open-api.js";
 import { TableBasedIntegrationMetadata } from "./metadata/table-based.js";
+function toPromptContextIntegration(integration) {
+    return {
+        id: integration.id,
+        name: integration.name,
+        type: integration.plugin.id,
+        metadata: integration.metadata,
+    };
+}
 let IntegrationStore = (() => {
     let _instanceExtraInitializers = [];
+    let _getMetadata_decorators;
     let _getIntegrationForPromptContext_decorators;
     let _getRelevantMetadata_decorators;
     let _getOpenApiSpec_decorators;
     return class IntegrationStore {
         static {
             const _metadata = typeof Symbol === "function" && Symbol.metadata ? Object.create(null) : void 0;
+            _getMetadata_decorators = [llmobs.decorate({
+                    kind: "task",
+                    name: "IntegrationStore.getMetadata",
+                })];
             _getIntegrationForPromptContext_decorators = [llmobs.decorate({
                     kind: "task",
                     name: "IntegrationStore.getIntegrationForPromptContext",
@@ -61,12 +77,98 @@ let IntegrationStore = (() => {
                     kind: "task",
                     name: "IntegrationStore.getOpenApiSpec",
                 })];
+            __esDecorate(this, null, _getMetadata_decorators, { kind: "method", name: "getMetadata", static: false, private: false, access: { has: obj => "getMetadata" in obj, get: obj => obj.getMetadata }, metadata: _metadata }, null, _instanceExtraInitializers);
             __esDecorate(this, null, _getIntegrationForPromptContext_decorators, { kind: "method", name: "getIntegrationForPromptContext", static: false, private: false, access: { has: obj => "getIntegrationForPromptContext" in obj, get: obj => obj.getIntegrationForPromptContext }, metadata: _metadata }, null, _instanceExtraInitializers);
             __esDecorate(this, null, _getRelevantMetadata_decorators, { kind: "method", name: "getRelevantMetadata", static: false, private: false, access: { has: obj => "getRelevantMetadata" in obj, get: obj => obj.getRelevantMetadata }, metadata: _metadata }, null, _instanceExtraInitializers);
             __esDecorate(this, null, _getOpenApiSpec_decorators, { kind: "method", name: "getOpenApiSpec", static: false, private: false, access: { has: obj => "getOpenApiSpec" in obj, get: obj => obj.getOpenApiSpec }, metadata: _metadata }, null, _instanceExtraInitializers);
             if (_metadata) Object.defineProperty(this, Symbol.metadata, { enumerable: true, configurable: true, writable: true, value: _metadata });
         }
         openApiSpecs = (__runInitializers(this, _instanceExtraInitializers), {});
+        pluginRegistry;
+        integrations = [];
+        metadataCache = new Map();
+        constructor(pluginRegistry = RegisteredPlugins("foo")) {
+            this.pluginRegistry = pluginRegistry;
+        }
+        availableIntegrations() {
+            return this.integrations.filter((i) => i.plugin.type === PluginType.CODE || !i.plugin.isBuiltIn);
+        }
+        getPlugin(pluginId) {
+            return this.pluginRegistry.get(pluginId);
+        }
+        insertFromPromptContext(integration, editorClient) {
+            const pluginId = integration.type;
+            const plugin = this.pluginRegistry.get(pluginId);
+            const basePluginId = getBasePluginId(pluginId) ?? pluginId;
+            const basePlugin = this.pluginRegistry.get(basePluginId);
+            if (!plugin) {
+                // this should never happen
+                getLogger().error(`No plugin found for integration type ${integration.type}`);
+                return;
+            }
+            const pluginHeader = {
+                id: plugin?.id,
+                type: plugin.type,
+                isBuiltIn: this.pluginRegistry
+                    .defaultPlugins()
+                    .map((p) => p.id)
+                    .includes(pluginId),
+                basePluginId: basePlugin?.id !== pluginId ? basePlugin?.id : undefined,
+            };
+            const header = {
+                id: integration.id,
+                name: integration.name,
+                plugin: pluginHeader,
+            };
+            const existingIndex = this.integrations.findIndex((i) => i.id === header.id);
+            if (existingIndex !== -1) {
+                this.integrations[existingIndex] = header;
+            }
+            else {
+                this.integrations.push(header);
+            }
+            if (editorClient) {
+                // prefetch metadata
+                this.getMetadata(integration.id, editorClient).catch((e) => {
+                    getLogger().error(`Error prefetching metadata for integration ${integration.id}: ${e}`);
+                });
+            }
+        }
+        getHeader(id) {
+            return this.integrations.find((i) => i.id === id);
+        }
+        async getMetadata(id, editorClient) {
+            if (!editorClient) {
+                getLogger().warn(`[integration-store] No editor client provided, cannot fetch metadata for integration ${id}. Returning cached metadata if available.`);
+                return this.metadataCache.get(id);
+            }
+            // stale-while-revalidate strategy
+            const promise = editorClient?.call
+                .aiGetIntegration({ integrationId: id })
+                .then((integration) => {
+                const existing = this.metadataCache.get(id) || {};
+                this.metadataCache.set(id, {
+                    ...existing,
+                    ...integration.metadata,
+                });
+                return this.metadataCache.get(id);
+            });
+            if (this.metadataCache.has(id)) {
+                return this.metadataCache.get(id);
+            }
+            return await promise;
+        }
+        async getIntegration(id, editorClient) {
+            const header = this.getHeader(id);
+            if (!header) {
+                return undefined;
+            }
+            const metadata = await this.getMetadata(id, editorClient);
+            return {
+                ...header,
+                metadata: metadata || {},
+            };
+        }
         async getIntegrationForPromptContext(integration, prompt, clark, params) {
             const { name, id } = integration;
             const type = integration.type;
@@ -79,7 +181,7 @@ let IntegrationStore = (() => {
             }
             const metadata = await this.getRelevantMetadata({
                 clark,
-                integration,
+                integrationId: integration.id,
                 prompt,
                 llmProvider: params.llmProvider,
             });
@@ -90,18 +192,30 @@ let IntegrationStore = (() => {
                 metadata,
             };
         }
-        async getRelevantMetadata({ clark, integration, prompt, llmProvider, }) {
+        async getRelevantMetadata({ clark, integrationId, prompt, llmProvider, }) {
+            getLogger().debug(`[integration-store] Fetching metadata for integration ${integrationId}
+    `);
             // If we're operating on an openapi integration, we need to fetch its full spec.
             // It would be nice to push this down, but the cache lives on this instance as this.openApiSpecs
+            const integration = await this.getIntegration(integrationId, clark.context.peer);
+            if (!integration) {
+                throw new Error(`No integration found for id ${integrationId}`);
+            }
             let metadata = integration.metadata;
-            if (isOpenApiIntegration(integration)) {
+            const promptContextIntegration = toPromptContextIntegration(integration);
+            if (isOpenApiIntegration(promptContextIntegration)) {
                 const openApiSpec = await this.getOpenApiSpec(integration, clark);
                 metadata = {
-                    ...(integration.metadata || {}),
+                    ...(metadata ?? (integration.metadata || {})),
                     openApiSpec,
                 };
             }
-            const klass = this.getMetadataClassForIntegration(integration);
+            if (!metadata) {
+                // we couldn't fetch metadata for some reason, throw error
+                throw new Error(`No metadata found for integration ${integration.id}`);
+            }
+            const klass = this.getMetadataClassForIntegration(promptContextIntegration);
+            getLogger().info(`[integration-store] Using metadata class ${klass.name} for integration ${integration.id} of type ${integration.plugin.id}`);
             return await klass.refineMetadataForPrompt({
                 metadata,
                 prompt,
@@ -112,6 +226,9 @@ let IntegrationStore = (() => {
             if (isOpenApiIntegration(integration)) {
                 return OpenApiIntegrationMetadata;
             }
+            if (isDatabricksIntegration(integration)) {
+                return DatabricksIntegrationMetadata;
+            }
             if (isDatabaseIntegration(integration)) {
                 return DatabaseIntegrationMetadata;
             }
@@ -155,6 +272,9 @@ function isSdkIntegration(integration) {
     return (typeof integration.type === "string" &&
         integration.type in datasourceSdkClassByType);
 }
+function isDatabricksIntegration(integration) {
+    return integration.type === "databricks";
+}
 function isDatabaseIntegration(integration) {
     return (databaseIntegrationTypes.includes(integration.type) || integration.type === "salesforce");
 }

package/dist/ai-service/integrations/store.js.map

@@ -1 +1 @@
-{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../src/ai-service/integrations/store.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,2BAA2B,EAAE,MAAM,wBAAwB,CAAC;AACrE,OAAO,EAAE,0BAA0B,EAAE,MAAM,6BAA6B,CAAC;AACzE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,0BAA0B,EAAE,MAAM,wBAAwB,CAAC;AACpE,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;IAc7D,gBAAgB;;;;;iBAAhB,gBAAgB;;;0DAG1B,MAAM,CAAC,QAAQ,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,iDAAiD;iBACxD,CAAC;+CAmCD,MAAM,CAAC,QAAQ,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,sCAAsC;iBAC7C,CAAC;0CAkDD,MAAM,CAAC,QAAQ,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,iCAAiC;iBACxC,CAAC;YA1FF,6OAAM,8BAA8B,6DAgCnC;YAMD,4MAAM,mBAAmB,6DA6BxB;YAwBD,6LAAM,cAAc,6DA2BnB;;;QA5HO,YAAY,IADT,mDAAgB,EACyB,EAAE,EAAC;QAMvD,KAAK,CAAC,8BAA8B,CAClC,WAAqC,EACrC,MAAc,EACd,KAAY,EACZ,MAA+B;YAE/B,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,WAAW,CAAC;YACjC,MAAM,IAAI,GAAG,WAAW,CAAC,IAA0B,CAAC;YAEpD,IACE,CAAC,IAAI;gBACL,CAAC,EAAE;gBACH,CAAC,gBAAgB,CAAC;oBAChB,IAAI;iBACL,CAAC,EACF,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC;gBAC9C,KAAK;gBACL,WAAW;gBACX,MAAM;gBACN,WAAW,EAAE,MAAM,CAAC,WAAW;aAChC,CAAC,CAAC;YAEH,OAAO;gBACL,IAAI;gBACJ,EAAE;gBACF,IAAI;gBACJ,QAAQ;aACT,CAAC;QACJ,CAAC;QAMD,KAAK,CAAC,mBAAmB,CAAC,EACxB,KAAK,EACL,WAAW,EACX,MAAM,EACN,WAAW,GAMZ;YACC,gFAAgF;YAChF,gGAAgG;YAChG,IAAI,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC;YACpC,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACtC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;gBAElE,QAAQ,GAAG;oBACT,GAAG,CAAC,WAAW,CAAC,QAAQ,IAAI,EAAE,CAAC;oBAC/B,WAAW;iBACZ,CAAC;YACJ,CAAC;YAED,MAAM,KAAK,GAAG,IAAI,CAAC,8BAA8B,CAAC,WAAW,CAAC,CAAC;YAC/D,OAAO,MAAM,KAAK,CAAC,uBAAuB,CAAC;gBACzC,QAAQ;gBACR,MAAM;gBACN,WAAW;aACZ,CAAC,CAAC;QACL,CAAC;QAED,8BAA8B,CAC5B,WAAqC;YAErC,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACtC,OAAO,0BAA0B,CAAC;YACpC,CAAC;YACD,IAAI,qBAAqB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACvC,OAAO,2BAA2B,CAAC;YACrC,CAAC;YACD,IAAI,uBAAuB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACzC,OAAO,6BAA6B,CAAC;YACvC,CAAC;YACD,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACtC,OAAO,0BAA0B,CAAC;YACpC,CAAC;YACD,OAAO,mBAAmB,CAAC;QAC7B,CAAC;QAMD,KAAK,CAAC,cAAc,CAClB,WAAqC,EACrC,KAAY;YAEZ,IAAI,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;gBACtC,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YAC3C,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;YACnC,CAAC;YAED,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;gBACzC,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,oKAAoK;YACpK,iHAAiH;YACjH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBAC5D,aAAa,EAAE,WAAW,CAAC,EAAE;gBAC7B,GAAG,EAAE,WAAW,CAAC,QAAQ,CAAC,cAAc;aACzC,CAAC,CAAC;YACH,IAAI,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBACpB,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,GAAG,MAAqB,CAAC;YAC1D,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC3C,CAAC;;;SA7HU,gBAAgB;AAgI7B,SAAS,oBAAoB,CAAC,WAAqC;IACjE,OAAO,WAAW,CAAC,QAAQ,EAAE,cAAc,KAAK,SAAS,CAAC;AAC5D,CAAC;AAED,SAAS,gBAAgB,CACvB,WAAoC;IAEpC,OAAO,CACL,OAAO,WAAW,CAAC,IAAI,KAAK,QAAQ;QACpC,WAAW,CAAC,IAAI,IAAI,wBAAwB,CAC7C,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,WAAqC;IAClE,OAAO,CACL,wBAAwB,CAAC,QAAQ,CAC/B,WAAW,CAAC,IAAkC,CAC/C,IAAI,WAAW,CAAC,IAAI,KAAK,YAAY,CACvC,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,WAAqC;IACpE,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,IAAI,WAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC9E,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC5D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,oBAAoB,CAAC,WAAqC;IACjE,OAAO,WAAW,CAAC,IAAI,KAAK,oBAAoB,CAAC;AACnD,CAAC"}
+{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../src/ai-service/integrations/store.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EACL,eAAe,EACf,UAAU,EACV,iBAAiB,GAGlB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,2BAA2B,EAAE,MAAM,wBAAwB,CAAC;AACrE,OAAO,EAAE,6BAA6B,EAAE,MAAM,0BAA0B,CAAC;AACzE,OAAO,EAAE,0BAA0B,EAAE,MAAM,6BAA6B,CAAC;AACzE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,0BAA0B,EAAE,MAAM,wBAAwB,CAAC;AACpE,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAsB1E,SAAS,0BAA0B,CACjC,WAAwB;IAExB,OAAO;QACL,EAAE,EAAE,WAAW,CAAC,EAAE;QAClB,IAAI,EAAE,WAAW,CAAC,IAAI;QACtB,IAAI,EAAE,WAAW,CAAC,MAAM,CAAC,EAAwB;QACjD,QAAQ,EAAE,WAAW,CAAC,QAAQ;KAC/B,CAAC;AACJ,CAAC;IAmBY,gBAAgB;;;;;;iBAAhB,gBAAgB;;;uCAyE1B,MAAM,CAAC,QAAQ,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,8BAA8B;iBACrC,CAAC;0DA2CD,MAAM,CAAC,QAAQ,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,iDAAiD;iBACxD,CAAC;+CAmCD,MAAM,CAAC,QAAQ,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,sCAAsC;iBAC7C,CAAC;0CAuED,MAAM,CAAC,QAAQ,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,iCAAiC;iBACxC,CAAC;YA7JF,oLAAM,WAAW,6DAyBhB;YAqBD,6OAAM,8BAA8B,6DAgCnC;YAMD,4MAAM,mBAAmB,6DA+CxB;YA2BD,6LAAM,cAAc,6DA2BnB;;;QArQO,YAAY,IADT,mDAAgB,EACyB,EAAE,EAAC;QAC/C,cAAc,CAAiB;QAC/B,YAAY,GAAwB,EAAE,CAAC;QACvC,aAAa,GACnB,IAAI,GAAG,EAAE,CAAC;QAEZ,YAAY,iBAAiC,iBAAiB,CAAC,KAAK,CAAC;YACnE,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACvC,CAAC;QAED,qBAAqB;YACnB,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAC7B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAChE,CAAC;QACJ,CAAC;QAED,SAAS,CAAC,QAAkB;YAC1B,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC3C,CAAC;QAED,uBAAuB,CACrB,WAAqC,EACrC,YAA2B;YAE3B,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAgB,CAAC;YAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,YAAY,GAAG,eAAe,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC;YAC3D,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;YACzD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,2BAA2B;gBAC3B,SAAS,EAAE,CAAC,KAAK,CACf,wCAAwC,WAAW,CAAC,IAAI,EAAE,CAC3D,CAAC;gBACF,OAAO;YACT,CAAC;YACD,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,MAAM,EAAE,EAAE;gBACd,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,SAAS,EAAE,IAAI,CAAC,cAAc;qBAC3B,cAAc,EAAE;qBAChB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAChB,QAAQ,CAAC,QAAQ,CAAC;gBACrB,YAAY,EAAE,UAAU,EAAE,EAAE,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS;aACvE,CAAC;YACF,MAAM,MAAM,GAAG;gBACb,EAAE,EAAE,WAAW,CAAC,EAAE;gBAClB,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,MAAM,EAAE,YAAY;aACrB,CAAC;YAEF,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,CAC/C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,MAAM,CAAC,EAAE,CAC1B,CAAC;YACF,IAAI,aAAa,KAAK,CAAC,CAAC,EAAE,CAAC;gBACzB,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,GAAG,MAAM,CAAC;YAC5C,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACjC,CAAC;YACD,IAAI,YAAY,EAAE,CAAC;gBACjB,oBAAoB;gBACpB,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;oBACzD,SAAS,EAAE,CAAC,KAAK,CACf,8CAA8C,WAAW,CAAC,EAAE,KAAK,CAAC,EAAE,CACrE,CAAC;gBACJ,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,SAAS,CAAC,EAAiB;YACzB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QACpD,CAAC;QAMD,KAAK,CAAC,WAAW,CACf,EAAiB,EACjB,YAA2B;YAE3B,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,SAAS,EAAE,CAAC,IAAI,CACd,wFAAwF,EAAE,2CAA2C,CACtI,CAAC;gBACF,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACpC,CAAC;YACD,kCAAkC;YAClC,MAAM,OAAO,GAAG,YAAY,EAAE,IAAI;iBAC/B,gBAAgB,CAAC,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC;iBACvC,IAAI,CAAC,CAAC,WAAW,EAAE,EAAE;gBACpB,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;gBAClD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE;oBACzB,GAAG,QAAQ;oBACX,GAAG,WAAW,CAAC,QAAQ;iBACxB,CAAC,CAAC;gBACH,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACpC,CAAC,CAAC,CAAC;YACL,IAAI,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC/B,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACpC,CAAC;YACD,OAAO,MAAM,OAAO,CAAC;QACvB,CAAC;QAEO,KAAK,CAAC,cAAc,CAC1B,EAAiB,EACjB,YAA2B;YAE3B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAClC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC;YAC1D,OAAO;gBACL,GAAG,MAAM;gBACT,QAAQ,EAAE,QAAQ,IAAI,EAAE;aACzB,CAAC;QACJ,CAAC;QAMD,KAAK,CAAC,8BAA8B,CAClC,WAAqC,EACrC,MAAc,EACd,KAAY,EACZ,MAA+B;YAE/B,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,WAAW,CAAC;YACjC,MAAM,IAAI,GAAG,WAAW,CAAC,IAA0B,CAAC;YAEpD,IACE,CAAC,IAAI;gBACL,CAAC,EAAE;gBACH,CAAC,gBAAgB,CAAC;oBAChB,IAAI;iBACL,CAAC,EACF,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC;gBAC9C,KAAK;gBACL,aAAa,EAAE,WAAW,CAAC,EAAE;gBAC7B,MAAM;gBACN,WAAW,EAAE,MAAM,CAAC,WAAW;aAChC,CAAC,CAAC;YAEH,OAAO;gBACL,IAAI;gBACJ,EAAE;gBACF,IAAI;gBACJ,QAAQ;aACT,CAAC;QACJ,CAAC;QAMD,KAAK,CAAC,mBAAmB,CAAC,EACxB,KAAK,EACL,aAAa,EACb,MAAM,EACN,WAAW,GAMZ;YACC,SAAS,EAAE,CAAC,KAAK,CACf,yDAAyD,aAAa;KACvE,CACA,CAAC;YACF,gFAAgF;YAChF,gGAAgG;YAChG,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAC3C,aAAa,EACb,KAAK,CAAC,OAAO,CAAC,IAAI,CACnB,CAAC;YACF,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,+BAA+B,aAAa,EAAE,CAAC,CAAC;YAClE,CAAC;YACD,IAAI,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAC;YACpC,MAAM,wBAAwB,GAAG,0BAA0B,CAAC,WAAW,CAAC,CAAC;YACzE,IAAI,oBAAoB,CAAC,wBAAwB,CAAC,EAAE,CAAC;gBACnD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;gBAElE,QAAQ,GAAG;oBACT,GAAG,CAAC,QAAQ,IAAI,CAAC,WAAW,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;oBAC7C,WAAW;iBACZ,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,0DAA0D;gBAC1D,MAAM,IAAI,KAAK,CAAC,qCAAqC,WAAW,CAAC,EAAE,EAAE,CAAC,CAAC;YACzE,CAAC;YACD,MAAM,KAAK,GAAG,IAAI,CAAC,8BAA8B,CAAC,wBAAwB,CAAC,CAAC;YAC5E,SAAS,EAAE,CAAC,IAAI,CACd,4CAA4C,KAAK,CAAC,IAAI,oBAAoB,WAAW,CAAC,EAAE,YAAY,WAAW,CAAC,MAAM,CAAC,EAAE,EAAE,CAC5H,CAAC;YACF,OAAO,MAAM,KAAK,CAAC,uBAAuB,CAAC;gBACzC,QAAQ;gBACR,MAAM;gBACN,WAAW;aACZ,CAAC,CAAC;QACL,CAAC;QAED,8BAA8B,CAC5B,WAAqC;YAErC,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACtC,OAAO,0BAA0B,CAAC;YACpC,CAAC;YACD,IAAI,uBAAuB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACzC,OAAO,6BAA6B,CAAC;YACvC,CAAC;YACD,IAAI,qBAAqB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACvC,OAAO,2BAA2B,CAAC;YACrC,CAAC;YACD,IAAI,uBAAuB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACzC,OAAO,6BAA6B,CAAC;YACvC,CAAC;YACD,IAAI,oBAAoB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACtC,OAAO,0BAA0B,CAAC;YACpC,CAAC;YACD,OAAO,mBAAmB,CAAC;QAC7B,CAAC;QAMD,KAAK,CAAC,cAAc,CAClB,WAAwB,EACxB,KAAY;YAEZ,IAAI,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;gBACtC,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YAC3C,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;YACnC,CAAC;YAED,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;gBACzC,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,oKAAoK;YACpK,iHAAiH;YACjH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBAC5D,aAAa,EAAE,WAAW,CAAC,EAAE;gBAC7B,GAAG,EAAE,WAAW,CAAC,QAAQ,CAAC,cAAc;aACzC,CAAC,CAAC;YACH,IAAI,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBACpB,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,GAAG,MAAqB,CAAC;YAC1D,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC3C,CAAC;;;SAtQU,gBAAgB;AAyQ7B,SAAS,oBAAoB,CAAC,WAAqC;IACjE,OAAO,WAAW,CAAC,QAAQ,EAAE,cAAc,KAAK,SAAS,CAAC;AAC5D,CAAC;AAED,SAAS,gBAAgB,CACvB,WAAoC;IAEpC,OAAO,CACL,OAAO,WAAW,CAAC,IAAI,KAAK,QAAQ;QACpC,WAAW,CAAC,IAAI,IAAI,wBAAwB,CAC7C,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,WAAqC;IACpE,OAAO,WAAW,CAAC,IAAI,KAAK,YAAY,CAAC;AAC3C,CAAC;AAED,SAAS,qBAAqB,CAAC,WAAqC;IAClE,OAAO,CACL,wBAAwB,CAAC,QAAQ,CAC/B,WAAW,CAAC,IAAkC,CAC/C,IAAI,WAAW,CAAC,IAAI,KAAK,YAAY,CACvC,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,WAAqC;IACpE,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,IAAI,WAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC9E,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC5D,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,oBAAoB,CAAC,WAAqC;IACjE,OAAO,WAAW,CAAC,IAAI,KAAK,oBAAoB,CAAC;AACnD,CAAC"}

package/dist/ai-service/llm/context/caching/constants.d.ts

@@ -0,0 +1,27 @@
+/**
+ * Provider cache configuration constants.
+ *
+ * These are the parameters for Anthropic's recent models
+ * https://docs.claude.com/en/docs/build-with-claude/prompt-caching#cache-limitations
+ */
+export declare const PROVIDER_CACHE_MAX_BREAKPOINTS = 4;
+export declare const PROVIDER_CACHE_MIN_CACHEABLE_TOKENS = 1024;
+/**
+ * Provider-specific cache control options.
+ *
+ * Different LLM providers use different formats for cache control.
+ * This maps to the appropriate format for each provider.
+ */
+export declare const CACHE_BREAKPOINT_PROVIDER_OPTIONS: {
+    bedrock: {
+        cachePoint: {
+            type: string;
+        };
+    };
+    anthropic: {
+        cacheControl: {
+            type: string;
+        };
+    };
+};
+//# sourceMappingURL=constants.d.ts.map

package/dist/ai-service/llm/context/caching/constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/caching/constants.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,eAAO,MAAM,8BAA8B,IAAI,CAAC;AAChD,eAAO,MAAM,mCAAmC,OAAO,CAAC;AAExD;;;;;GAKG;AACH,eAAO,MAAM,iCAAiC;;;;;;;;;;;CAI7C,CAAC"}

package/dist/ai-service/llm/context/caching/constants.js

@@ -0,0 +1,20 @@
+/**
+ * Provider cache configuration constants.
+ *
+ * These are the parameters for Anthropic's recent models
+ * https://docs.claude.com/en/docs/build-with-claude/prompt-caching#cache-limitations
+ */
+export const PROVIDER_CACHE_MAX_BREAKPOINTS = 4;
+export const PROVIDER_CACHE_MIN_CACHEABLE_TOKENS = 1024;
+/**
+ * Provider-specific cache control options.
+ *
+ * Different LLM providers use different formats for cache control.
+ * This maps to the appropriate format for each provider.
+ */
+export const CACHE_BREAKPOINT_PROVIDER_OPTIONS = {
+    bedrock: { cachePoint: { type: "default" } },
+    // Vertex provider uses anthropic key
+    anthropic: { cacheControl: { type: "ephemeral" } },
+};
+//# sourceMappingURL=constants.js.map

package/dist/ai-service/llm/context/caching/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/caching/constants.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAAC,CAAC;AAChD,MAAM,CAAC,MAAM,mCAAmC,GAAG,IAAI,CAAC;AAExD;;;;;GAKG;AACH,MAAM,CAAC,MAAM,iCAAiC,GAAG;IAC/C,OAAO,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;IAC5C,qCAAqC;IACrC,SAAS,EAAE,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE;CACnD,CAAC"}

package/dist/ai-service/llm/context/caching/factory.d.ts

@@ -0,0 +1,12 @@
+/**
+ * Factory for creating cache strategy instances.
+ */
+import type { CacheStrategy } from "./strategy.js";
+import type { CacheStrategyType } from "./types.js";
+/**
+ * Factory for creating cache strategy instances.
+ */
+export declare class CacheStrategyFactory {
+    static create(type?: CacheStrategyType): CacheStrategy;
+}
+//# sourceMappingURL=factory.d.ts.map

package/dist/ai-service/llm/context/caching/factory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/caching/factory.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAEpD;;GAEG;AACH,qBAAa,oBAAoB;IAC/B,MAAM,CAAC,MAAM,CACX,IAAI,GAAE,iBAA0C,GAC/C,aAAa;CASjB"}

package/dist/ai-service/llm/context/caching/factory.js

@@ -0,0 +1,28 @@
+/**
+ * Factory for creating cache strategy instances.
+ */
+import { QuantizedIntervalCacheStrategy } from "./quantized-strategy.js";
+import { DEFAULT_CACHE_STRATEGY } from "./types.js";
+/**
+ * Factory for creating cache strategy instances.
+ */
+export class CacheStrategyFactory {
+    static create(type = DEFAULT_CACHE_STRATEGY) {
+        switch (type) {
+            case "none":
+                return new NoOpCacheStrategy();
+            case "quantized":
+            default:
+                return new QuantizedIntervalCacheStrategy();
+        }
+    }
+}
+/**
+ * No-op cache strategy (disables caching).
+ */
+class NoOpCacheStrategy {
+    findBreakpoints() {
+        return [];
+    }
+}
+//# sourceMappingURL=factory.js.map

package/dist/ai-service/llm/context/caching/factory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.js","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/caching/factory.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,8BAA8B,EAAE,MAAM,yBAAyB,CAAC;AACzE,OAAO,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAIpD;;GAEG;AACH,MAAM,OAAO,oBAAoB;IAC/B,MAAM,CAAC,MAAM,CACX,OAA0B,sBAAsB;QAEhD,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,MAAM;gBACT,OAAO,IAAI,iBAAiB,EAAE,CAAC;YACjC,KAAK,WAAW,CAAC;YACjB;gBACE,OAAO,IAAI,8BAA8B,EAAE,CAAC;QAChD,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,iBAAiB;IACrB,eAAe;QACb,OAAO,EAAE,CAAC;IACZ,CAAC;CACF"}

package/dist/ai-service/llm/context/caching/index.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Public API exports for caching module.
+ */
+export type { CacheInput, TokenCounts, CacheStrategyType } from "./types.js";
+export { DEFAULT_CACHE_STRATEGY } from "./types.js";
+export { type CacheStrategy } from "./strategy.js";
+export { QuantizedIntervalCacheStrategy } from "./quantized-strategy.js";
+export { CacheStrategyFactory } from "./factory.js";
+export { CACHE_BREAKPOINT_PROVIDER_OPTIONS, PROVIDER_CACHE_MAX_BREAKPOINTS, PROVIDER_CACHE_MIN_CACHEABLE_TOKENS, } from "./constants.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/ai-service/llm/context/caching/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/caching/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,YAAY,EAAE,UAAU,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAC7E,OAAO,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AACpD,OAAO,EAAE,KAAK,aAAa,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,8BAA8B,EAAE,MAAM,yBAAyB,CAAC;AACzE,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AACpD,OAAO,EACL,iCAAiC,EACjC,8BAA8B,EAC9B,mCAAmC,GACpC,MAAM,gBAAgB,CAAC"}

package/dist/ai-service/llm/context/caching/index.js

@@ -0,0 +1,9 @@
+/**
+ * Public API exports for caching module.
+ */
+export { DEFAULT_CACHE_STRATEGY } from "./types.js";
+export {} from "./strategy.js";
+export { QuantizedIntervalCacheStrategy } from "./quantized-strategy.js";
+export { CacheStrategyFactory } from "./factory.js";
+export { CACHE_BREAKPOINT_PROVIDER_OPTIONS, PROVIDER_CACHE_MAX_BREAKPOINTS, PROVIDER_CACHE_MIN_CACHEABLE_TOKENS, } from "./constants.js";
+//# sourceMappingURL=index.js.map

package/dist/ai-service/llm/context/caching/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/ai-service/llm/context/caching/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AACpD,OAAO,EAAsB,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,8BAA8B,EAAE,MAAM,yBAAyB,CAAC;AACzE,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AACpD,OAAO,EACL,iCAAiC,EACjC,8BAA8B,EAC9B,mCAAmC,GACpC,MAAM,gBAAgB,CAAC"}

package/dist/ai-service/llm/context/caching/quantized-strategy.d.ts

@@ -0,0 +1,22 @@
+/**
+ * Quantized interval cache strategy implementation.
+ *
+ * Places cache breakpoints at regular intervals to maximize cache hits
+ * while respecting provider limits.
+ */
+import type { CacheStrategy } from "./strategy.js";
+import type { CacheInput } from "./types.js";
+/**
+ * Quantized interval cache strategy.
+ *
+ * Finds optimal positions for cache breakpoints in message array.
+ *
+ * Strategy:
+ * - Always place breakpoint at system prompt (index 0) if present
+ * - Divide total tokens into intervals of minCacheableTokens
+ * - Space breakpoints evenly across intervals, respecting maxBreakpoints
+ */
+export declare class QuantizedIntervalCacheStrategy implements CacheStrategy {
+    findBreakpoints(input: CacheInput): number[];
+}
+//# sourceMappingURL=quantized-strategy.d.ts.map