captify 0.1.18 → 0.1.19

package/dist/model-budget/services/usage-tracking.service.js

@@ -0,0 +1,203 @@
+/**
+ * Usage Tracking Service
+ *
+ * Centralized service for tracking LLM usage across all operations
+ * (chat, agents, config generation, etc.). Provides a unified interface
+ * for checking limits before requests and recording usage after completion.
+ */
+import { LoggerFactory } from "../../shared/services/logger";
+import { getModelBudgetService, } from "./model-budget.service";
+import { BudgetExceededError, TokenLimitExceededError, } from "../model-budget.errors";
+import { calculateTokenUsage } from "../../chat/chat.utilities";
+import { USER_CONTEXT } from "../../shared/shared.constants";
+// ============================================================================
+// Service Implementation
+// ============================================================================
+/**
+ * Service for tracking LLM usage across all operations.
+ *
+ * Provides methods to:
+ * - Check limits before making LLM calls
+ * - Record usage after LLM calls complete
+ * - Handle streaming responses with async usage tracking
+ *
+ * @example
+ * ```typescript
+ * const tracker = getUsageTrackingService();
+ *
+ * // Before making an LLM call
+ * await tracker.checkLimitsOrThrow({ tenantId, providerId, modelId, userCredentials });
+ *
+ * // After LLM call completes
+ * await tracker.recordUsage(
+ *   { tenantId, providerId, modelId, userCredentials },
+ *   model,
+ *   result.usage,
+ * );
+ * ```
+ */
+export class UsageTrackingService {
+    constructor(deps) {
+        var _a, _b;
+        this.logger =
+            (_a = deps === null || deps === void 0 ? void 0 : deps.logger) !== null && _a !== void 0 ? _a : LoggerFactory.createLogger({
+                serviceName: "UsageTrackingService",
+                userContext: USER_CONTEXT,
+            });
+        this.budgetService = (_b = deps === null || deps === void 0 ? void 0 : deps.budgetService) !== null && _b !== void 0 ? _b : getModelBudgetService();
+    }
+    /**
+     * Check budget and rate limits before making an LLM call.
+     * Uses atomic rate limit check to prevent race conditions.
+     * Throws appropriate error if limits are exceeded.
+     *
+     * @param params - Tracking parameters
+     * @throws BudgetExceededError if budget limit is exceeded
+     * @throws RateLimitExceededError if rate limit is exceeded
+     * @throws TokenLimitExceededError if token limit is exceeded
+     */
+    async checkLimitsOrThrow(params) {
+        var _a, _b, _c, _d, _e, _f;
+        // First, atomically check and record rate limit
+        // This throws RateLimitExceededError if limit is exceeded
+        await this.budgetService.recordRequest({
+            tenantId: params.tenantId,
+            providerId: params.providerId,
+            modelId: params.modelId,
+            userCredentials: params.userCredentials,
+        });
+        // Then check budget limits (these don't have race condition issues)
+        const limitCheck = await this.budgetService.checkLimits({
+            tenantId: params.tenantId,
+            providerId: params.providerId,
+            modelId: params.modelId,
+            userCredentials: params.userCredentials,
+        });
+        if (!limitCheck.allowed) {
+            if (limitCheck.reason === "budget_exceeded" && limitCheck.budget) {
+                throw new BudgetExceededError({
+                    modelId: params.modelId,
+                    providerId: params.providerId,
+                    spent: (_a = limitCheck.budget.spent) !== null && _a !== void 0 ? _a : 0,
+                    limit: (_b = limitCheck.budget.limit) !== null && _b !== void 0 ? _b : 0,
+                    periodEnds: (_c = limitCheck.budget.periodEnd) !== null && _c !== void 0 ? _c : Date.now(),
+                });
+            }
+            if (limitCheck.reason === "token_limit_exceeded" && limitCheck.budget) {
+                throw new TokenLimitExceededError({
+                    modelId: params.modelId,
+                    providerId: params.providerId,
+                    used: (_d = limitCheck.budget.tokenSpent) !== null && _d !== void 0 ? _d : 0,
+                    limit: (_e = limitCheck.budget.tokenLimit) !== null && _e !== void 0 ? _e : 0,
+                    periodEnds: (_f = limitCheck.budget.periodEnd) !== null && _f !== void 0 ? _f : Date.now(),
+                });
+            }
+        }
+    }
+    /**
+     * Record usage after an LLM call completes.
+     * Only records the spend (request count is recorded in checkLimitsOrThrow).
+     *
+     * @param params - Tracking parameters
+     * @param model - Model configuration (for pricing)
+     * @param usage - Usage data from AI SDK
+     * @returns Result with tokens and cost recorded
+     */
+    async recordUsage(params, model, usage) {
+        var _a, _b, _c;
+        if (!usage) {
+            this.logger.debug("No usage data to record", {
+                tenantId: params.tenantId,
+                modelId: params.modelId,
+            });
+            return { tokens: 0, cost: 0, recorded: false };
+        }
+        const normalizedUsage = {
+            inputTokens: (_a = usage.inputTokens) !== null && _a !== void 0 ? _a : 0,
+            outputTokens: (_b = usage.outputTokens) !== null && _b !== void 0 ? _b : 0,
+            totalTokens: (_c = usage.totalTokens) !== null && _c !== void 0 ? _c : 0,
+        };
+        const tokenUsage = calculateTokenUsage(normalizedUsage, model);
+        // Record spend for budget tracking
+        // Note: Request count is already recorded in checkLimitsOrThrow (before the call)
+        await this.budgetService.recordSpend({
+            tenantId: params.tenantId,
+            providerId: params.providerId,
+            modelId: params.modelId,
+            amount: tokenUsage.estimatedCost,
+            tokens: tokenUsage.totalTokens,
+            userCredentials: params.userCredentials,
+        });
+        this.logger.debug("Recorded usage", {
+            tenantId: params.tenantId,
+            providerId: params.providerId,
+            modelId: params.modelId,
+            tokens: tokenUsage.totalTokens,
+            cost: tokenUsage.estimatedCost,
+        });
+        return {
+            tokens: tokenUsage.totalTokens,
+            cost: tokenUsage.estimatedCost,
+            recorded: true,
+        };
+    }
+    /**
+     * Record usage from a streaming response.
+     * This is a fire-and-forget operation that handles the async usage promise.
+     *
+     * @param params - Tracking parameters
+     * @param model - Model configuration (for pricing)
+     * @param usagePromise - Promise that resolves to usage data when stream completes
+     */
+    recordStreamUsage(params, model, usagePromise) {
+        usagePromise
+            .then(async (usage) => {
+            if (usage) {
+                await this.recordUsage(params, model, usage);
+            }
+        })
+            .catch((error) => {
+            this.logger.error("Failed to record stream usage", {
+                tenantId: params.tenantId,
+                providerId: params.providerId,
+                modelId: params.modelId,
+                error: error instanceof Error ? error.message : String(error),
+            });
+        });
+    }
+    /**
+     * Wrapper for executing an LLM operation with automatic usage tracking.
+     * Checks limits before, executes the operation, and records usage after.
+     *
+     * @param params - Tracking parameters
+     * @param model - Model configuration (for pricing)
+     * @param operation - The LLM operation to execute
+     * @returns The result of the operation
+     */
+    async withTracking(params, model, operation) {
+        // Check limits before operation
+        await this.checkLimitsOrThrow(params);
+        // Execute the operation
+        const result = await operation();
+        // Record usage after operation
+        await this.recordUsage(params, model, result.usage);
+        return result;
+    }
+}
+// ============================================================================
+// Singleton
+// ============================================================================
+let usageTrackingServiceInstance = null;
+/**
+ * Get or create the singleton UsageTrackingService instance.
+ */
+export function getUsageTrackingService(deps) {
+    if ((deps === null || deps === void 0 ? void 0 : deps.logger) || (deps === null || deps === void 0 ? void 0 : deps.budgetService)) {
+        return new UsageTrackingService(deps);
+    }
+    if (!usageTrackingServiceInstance) {
+        usageTrackingServiceInstance = new UsageTrackingService();
+    }
+    return usageTrackingServiceInstance;
+}
+//# sourceMappingURL=usage-tracking.service.js.map

package/dist/model-budget/services/usage-tracking.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-tracking.service.js","sourceRoot":"","sources":["../../../lib/model-budget/services/usage-tracking.service.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAIH,OAAO,EAAE,aAAa,EAAe,MAAM,8BAA8B,CAAC;AAC1E,OAAO,EACL,qBAAqB,GAEtB,MAAM,kDAAkD,CAAC;AAC1D,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,wCAAwC,CAAC;AAChD,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAOhE,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAE7D,+EAA+E;AAC/E,yBAAyB;AACzB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAM,OAAO,oBAAoB;IAI/B,YAAY,IAA+B;;QACzC,IAAI,CAAC,MAAM;YACT,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,MAAM,mCACZ,aAAa,CAAC,YAAY,CAAC;gBACzB,WAAW,EAAE,sBAAsB;gBACnC,WAAW,EAAE,YAAY;aAC1B,CAAC,CAAC;QACL,IAAI,CAAC,aAAa,GAAG,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,aAAa,mCAAI,qBAAqB,EAAE,CAAC;IACtE,CAAC;IAED;;;;;;;;;OASG;IACH,KAAK,CAAC,kBAAkB,CAAC,MAA2B;;QAClD,gDAAgD;QAChD,0DAA0D;QAC1D,MAAM,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC;YACrC,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,eAAe,EAAE,MAAM,CAAC,eAAe;SACxC,CAAC,CAAC;QAEH,oEAAoE;QACpE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC;YACtD,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,eAAe,EAAE,MAAM,CAAC,eAAe;SACxC,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,IAAI,UAAU,CAAC,MAAM,KAAK,iBAAiB,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;gBACjE,MAAM,IAAI,mBAAmB,CAAC;oBAC5B,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,KAAK,EAAE,MAAA,UAAU,CAAC,MAAM,CAAC,KAAK,mCAAI,CAAC;oBACnC,KAAK,EAAE,MAAA,UAAU,CAAC,MAAM,CAAC,KAAK,mCAAI,CAAC;oBACnC,UAAU,EAAE,MAAA,UAAU,CAAC,MAAM,CAAC,SAAS,mCAAI,IAAI,CAAC,GAAG,EAAE;iBACtD,CAAC,CAAC;YACL,CAAC;YACD,IAAI,UAAU,CAAC,MAAM,KAAK,sBAAsB,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;gBACtE,MAAM,IAAI,uBAAuB,CAAC;oBAChC,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,IAAI,EAAE,MAAA,UAAU,CAAC,MAAM,CAAC,UAAU,mCAAI,CAAC;oBACvC,KAAK,EAAE,MAAA,UAAU,CAAC,MAAM,CAAC,UAAU,mCAAI,CAAC;oBACxC,UAAU,EAAE,MAAA,UAAU,CAAC,MAAM,CAAC,SAAS,mCAAI,IAAI,CAAC,GAAG,EAAE;iBACtD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,WAAW,CACf,MAA2B,EAC3B,KAAY,EACZ,KAA6C;;QAE7C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE;gBAC3C,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAC,CAAC;YACH,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QACjD,CAAC;QAED,MAAM,eAAe,GAAG;YACtB,WAAW,EAAE,MAAA,KAAK,CAAC,WAAW,mCAAI,CAAC;YACnC,YAAY,EAAE,MAAA,KAAK,CAAC,YAAY,mCAAI,CAAC;YACrC,WAAW,EAAE,MAAA,KAAK,CAAC,WAAW,mCAAI,CAAC;SACpC,CAAC;QAEF,MAAM,UAAU,GAAG,mBAAmB,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;QAE/D,mCAAmC;QACnC,kFAAkF;QAClF,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC;YACnC,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,MAAM,EAAE,UAAU,CAAC,aAAa;YAChC,MAAM,EAAE,UAAU,CAAC,WAAW;YAC9B,eAAe,EAAE,MAAM,CAAC,eAAe;SACxC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE;YAClC,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,MAAM,EAAE,UAAU,CAAC,WAAW;YAC9B,IAAI,EAAE,UAAU,CAAC,aAAa;SAC/B,CAAC,CAAC;QAEH,OAAO;YACL,MAAM,EAAE,UAAU,CAAC,WAAW;YAC9B,IAAI,EAAE,UAAU,CAAC,aAAa;YAC9B,QAAQ,EAAE,IAAI;SACf,CAAC;IACJ,CAAC;IAED;;;;;;;OAOG;IACH,iBAAiB,CACf,MAA2B,EAC3B,KAAY,EACZ,YAAiE;QAEjE,YAAY;aACT,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACpB,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE;gBACjD,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACP,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,YAAY,CAChB,MAA2B,EAC3B,KAAY,EACZ,SAA2B;QAE3B,gCAAgC;QAChC,MAAM,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAEtC,wBAAwB;QACxB,MAAM,MAAM,GAAG,MAAM,SAAS,EAAE,CAAC;QAEjC,+BAA+B;QAC/B,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAEpD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAE/E,IAAI,4BAA4B,GAAgC,IAAI,CAAC;AAErE;;GAEG;AACH,MAAM,UAAU,uBAAuB,CAAC,IAA+B;IACrE,IAAI,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,MAAM,MAAI,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,aAAa,CAAA,EAAE,CAAC;QACxC,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC;IAED,IAAI,CAAC,4BAA4B,EAAE,CAAC;QAClC,4BAA4B,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC5D,CAAC;IAED,OAAO,4BAA4B,CAAC;AACtC,CAAC"}

package/dist/ontology/api/index.d.ts

@@ -1,3 +1,3 @@
-export { ontologyKeys, dynamoDbOntologyKeys, listTables, describeTable, buildOntologyGraph, type ListTablesResponse, type DescribeTableResponse, type BuildGraphResponse, type TableMetadata, type OntologyNodeResponse, type OntologyLinkResponse, type ListTablesOptions, type BuildGraphOptions, } from "./ontology.client";
-export { ontologyHandlers, dynamoDbOntologyHandlers } from "./ontology.handlers";
+export { ontologyKeys, listTables, describeTable, buildOntologyGraph, type ListTablesResponse, type DescribeTableResponse, type BuildGraphResponse, type TableMetadata, type OntologyNodeResponse, type OntologyLinkResponse, type ListTablesOptions, type BuildGraphOptions, } from "./ontology.client";
+export { ontologyHandlers } from "./ontology.handlers";
 //# sourceMappingURL=index.d.ts.map

package/dist/ontology/api/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../lib/ontology/api/index.ts"],"names":[],"mappings":"AACA,OAAO,EACL,YAAY,EACZ,oBAAoB,EACpB,UAAU,EACV,aAAa,EACb,kBAAkB,EAClB,KAAK,kBAAkB,EACvB,KAAK,qBAAqB,EAC1B,KAAK,kBAAkB,EACvB,KAAK,aAAa,EAClB,KAAK,oBAAoB,EACzB,KAAK,oBAAoB,EACzB,KAAK,iBAAiB,EACtB,KAAK,iBAAiB,GACvB,MAAM,oCAAoC,CAAC;AAG5C,OAAO,EAAE,gBAAgB,EAAE,wBAAwB,EAAE,MAAM,sCAAsC,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../lib/ontology/api/index.ts"],"names":[],"mappings":"AACA,OAAO,EACL,YAAY,EACZ,UAAU,EACV,aAAa,EACb,kBAAkB,EAClB,KAAK,kBAAkB,EACvB,KAAK,qBAAqB,EAC1B,KAAK,kBAAkB,EACvB,KAAK,aAAa,EAClB,KAAK,oBAAoB,EACzB,KAAK,oBAAoB,EACzB,KAAK,iBAAiB,EACtB,KAAK,iBAAiB,GACvB,MAAM,oCAAoC,CAAC;AAG5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,sCAAsC,CAAC"}

package/dist/ontology/api/index.js

@@ -1,5 +1,5 @@
 // Re-export client types and functions
-export { ontologyKeys, dynamoDbOntologyKeys, listTables, describeTable, buildOntologyGraph, } from "./ontology.client";
+export { ontologyKeys, listTables, describeTable, buildOntologyGraph, } from "./ontology.client";
 // Re-export handlers
-export { ontologyHandlers, dynamoDbOntologyHandlers } from "./ontology.handlers";
+export { ontologyHandlers } from "./ontology.handlers";
 //# sourceMappingURL=index.js.map

package/dist/ontology/api/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../lib/ontology/api/index.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,OAAO,EACL,YAAY,EACZ,oBAAoB,EACpB,UAAU,EACV,aAAa,EACb,kBAAkB,GASnB,MAAM,oCAAoC,CAAC;AAE5C,qBAAqB;AACrB,OAAO,EAAE,gBAAgB,EAAE,wBAAwB,EAAE,MAAM,sCAAsC,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../lib/ontology/api/index.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,OAAO,EACL,YAAY,EACZ,UAAU,EACV,aAAa,EACb,kBAAkB,GASnB,MAAM,oCAAoC,CAAC;AAE5C,qBAAqB;AACrB,OAAO,EAAE,gBAAgB,EAAE,MAAM,sCAAsC,CAAC"}

package/dist/ontology/api/ontology.client.d.ts

@@ -1,7 +1,7 @@
 import type { ontologyHandlers } from "./ontology.handlers";
 import type { ApiRouteResponse } from "../../shared/shared.interfaces";
 /**
- * Query key factory for DynamoDB ontology resources.
+ * Query key factory for Data ontology resources.
  * Follows the pattern: ["resource", "operation", params]
  */
 export declare const ontologyKeys: {
@@ -18,20 +18,6 @@ export declare const ontologyKeys: {
     /** Key for graph query with options */
     graph: (options: BuildGraphOptions) => readonly ["ontology", "graph", BuildGraphOptions];
 };
-export declare const dynamoDbOntologyKeys: {
-    /** Base key for all ontology queries */
-    all: readonly ["ontology"];
-    /** Key for table list queries */
-    tables: () => readonly ["ontology", "tables"];
-    /** Key for filtered table list queries */
-    tableList: (options: ListTablesOptions) => readonly ["ontology", "tables", ListTablesOptions];
-    /** Key for single table detail queries */
-    table: (tableName: string) => readonly ["ontology", "table", string];
-    /** Key for graph queries */
-    graphs: () => readonly ["ontology", "graph"];
-    /** Key for graph query with options */
-    graph: (options: BuildGraphOptions) => readonly ["ontology", "graph", BuildGraphOptions];
-};
 /** Response type for listing tables - derived from handler */
 export type ListTablesResponse = ApiRouteResponse<typeof ontologyHandlers.listTables>;
 /** Response type for describing a single table - derived from handler */
@@ -49,14 +35,27 @@ export interface ListTablesOptions {
     exclusiveStartTableName?: string;
     tableNamePrefix?: string;
 }
+/**
+ * Data source for ontology graph generation
+ */
+export type GraphSource = "schemas" | "dynamodb";
 export interface BuildGraphOptions {
+    /**
+     * Data source for graph generation:
+     * - "schemas": Use static schema definitions (default, no AWS required)
+     * - "dynamodb": Query live DynamoDB tables (requires AWS credentials)
+     */
+    source?: GraphSource;
     tableNames?: string[];
+    /** Filter tables by name prefix (dynamodb source only) */
     tableNamePrefix?: string;
     includeAttributes?: boolean;
     includeGlobalSecondaryIndexes?: boolean;
     includeLocalSecondaryIndexes?: boolean;
-    /** Whether to infer relationships between tables based on attribute naming patterns */
+    /** Whether to infer relationships between tables (dynamodb source only) */
     inferRelationships?: boolean;
+    /** Whether to include explicit relationships between tables (schemas source only) */
+    includeRelationships?: boolean;
 }
 export declare function listTables(options?: ListTablesOptions): Promise<ListTablesResponse>;
 export declare function describeTable(tableName: string): Promise<DescribeTableResponse>;

package/dist/ontology/api/ontology.client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ontology.client.d.ts","sourceRoot":"","sources":["../../../lib/ontology/api/ontology.client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,sCAAsC,CAAC;AAE7E,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAQvE;;;GAGG;AACH,eAAO,MAAM,YAAY;IACvB,wCAAwC;;IAExC,iCAAiC;;IAEjC,0CAA0C;yBACrB,iBAAiB;IACtC,0CAA0C;uBACvB,MAAM;IACzB,4BAA4B;;IAE5B,uCAAuC;qBACtB,iBAAiB;CACnC,CAAC;AAGF,eAAO,MAAM,oBAAoB;IAf/B,wCAAwC;;IAExC,iCAAiC;;IAEjC,0CAA0C;yBACrB,iBAAiB;IACtC,0CAA0C;uBACvB,MAAM;IACzB,4BAA4B;;IAE5B,uCAAuC;qBACtB,iBAAiB;CAIY,CAAC;AAMjD,8DAA8D;AAC9D,MAAM,MAAM,kBAAkB,GAAG,gBAAgB,CAAC,OAAO,gBAAgB,CAAC,UAAU,CAAC,CAAC;AAEtF,yEAAyE;AACzE,MAAM,MAAM,qBAAqB,GAAG,gBAAgB,CAAC,OAAO,gBAAgB,CAAC,aAAa,CAAC,CAAC;AAE5F,uEAAuE;AACvE,MAAM,MAAM,kBAAkB,GAAG,gBAAgB,CAAC,OAAO,gBAAgB,CAAC,UAAU,CAAC,CAAC;AAEtF,6DAA6D;AAC7D,MAAM,MAAM,aAAa,GAAG,qBAAqB,CAAC;AAElD,kDAAkD;AAClD,MAAM,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC;AAEvE,kDAAkD;AAClD,MAAM,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC;AAMvE,MAAM,WAAW,iBAAiB;IAChC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,iBAAiB;IAChC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,6BAA6B,CAAC,EAAE,OAAO,CAAC;IACxC,4BAA4B,CAAC,EAAE,OAAO,CAAC;IACvC,uFAAuF;IACvF,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B;AAMD,wBAAsB,UAAU,CAAC,OAAO,GAAE,iBAAsB,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAQ7F;AAED,wBAAsB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAErF;AAED,wBAAsB,kBAAkB,CACtC,OAAO,GAAE,iBAAsB,GAC9B,OAAO,CAAC,kBAAkB,CAAC,CAW7B"}
+{"version":3,"file":"ontology.client.d.ts","sourceRoot":"","sources":["../../../lib/ontology/api/ontology.client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,sCAAsC,CAAC;AAE7E,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAQvE;;;GAGG;AACH,eAAO,MAAM,YAAY;IACvB,wCAAwC;;IAExC,iCAAiC;;IAEjC,0CAA0C;yBACrB,iBAAiB;IACtC,0CAA0C;uBACvB,MAAM;IACzB,4BAA4B;;IAE5B,uCAAuC;qBACtB,iBAAiB;CACnC,CAAC;AAMF,8DAA8D;AAC9D,MAAM,MAAM,kBAAkB,GAAG,gBAAgB,CAAC,OAAO,gBAAgB,CAAC,UAAU,CAAC,CAAC;AAEtF,yEAAyE;AACzE,MAAM,MAAM,qBAAqB,GAAG,gBAAgB,CAAC,OAAO,gBAAgB,CAAC,aAAa,CAAC,CAAC;AAE5F,uEAAuE;AACvE,MAAM,MAAM,kBAAkB,GAAG,gBAAgB,CAAC,OAAO,gBAAgB,CAAC,UAAU,CAAC,CAAC;AAEtF,6DAA6D;AAC7D,MAAM,MAAM,aAAa,GAAG,qBAAqB,CAAC;AAElD,kDAAkD;AAClD,MAAM,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC;AAEvE,kDAAkD;AAClD,MAAM,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC;AAMvE,MAAM,WAAW,iBAAiB;IAChC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED;;GAEG;AACH,MAAM,MAAM,WAAW,GAAG,SAAS,GAAG,UAAU,CAAC;AAEjD,MAAM,WAAW,iBAAiB;IAChC;;;;OAIG;IACH,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,0DAA0D;IAC1D,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,6BAA6B,CAAC,EAAE,OAAO,CAAC;IACxC,4BAA4B,CAAC,EAAE,OAAO,CAAC;IACvC,2EAA2E;IAC3E,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,qFAAqF;IACrF,oBAAoB,CAAC,EAAE,OAAO,CAAC;CAChC;AAMD,wBAAsB,UAAU,CAAC,OAAO,GAAE,iBAAsB,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAQ7F;AAED,wBAAsB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAErF;AAED,wBAAsB,kBAAkB,CACtC,OAAO,GAAE,iBAAsB,GAC9B,OAAO,CAAC,kBAAkB,CAAC,CAa7B"}

package/dist/ontology/api/ontology.client.js

@@ -1,11 +1,11 @@
-import { DYNAMODB_ONTOLOGY_ENDPOINT as BASE_URL } from "../ontology.constants";
+import { DATA_ONTOLOGY_ENDPOINT as BASE_URL } from "../../shared/shared.constants";
 import { buildUrl } from "../../shared/utilities/url";
 import { fetchJson } from "../../shared/utilities/fetch";
 // ============================================================================
 // Query Keys - Consistent patterns for TanStack Query cache management
 // ============================================================================
 /**
- * Query key factory for DynamoDB ontology resources.
+ * Query key factory for Data ontology resources.
  * Follows the pattern: ["resource", "operation", params]
  */
 export const ontologyKeys = {
@@ -22,8 +22,6 @@ export const ontologyKeys = {
     /** Key for graph query with options */
     graph: (options) => [...ontologyKeys.graphs(), options],
 };
-// Keep backward compatibility alias
-export const dynamoDbOntologyKeys = ontologyKeys;
 // ============================================================================
 // API Client Functions - Raw fetch operations
 // ============================================================================
@@ -41,12 +39,14 @@ export async function describeTable(tableName) {
 export async function buildOntologyGraph(options = {}) {
     var _a;
     const url = buildUrl(`${BASE_URL}/graph`, {
+        source: options.source,
         tableNames: (_a = options.tableNames) === null || _a === void 0 ? void 0 : _a.join(","),
         tableNamePrefix: options.tableNamePrefix,
         includeAttributes: options.includeAttributes,
         includeGlobalSecondaryIndexes: options.includeGlobalSecondaryIndexes,
         includeLocalSecondaryIndexes: options.includeLocalSecondaryIndexes,
         inferRelationships: options.inferRelationships,
+        includeRelationships: options.includeRelationships,
     });
     return fetchJson(url.toString());
 }

package/dist/ontology/api/ontology.client.js.map

@@ -1 +1 @@
-{"version":3,"file":"ontology.client.js","sourceRoot":"","sources":["../../../lib/ontology/api/ontology.client.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,0BAA0B,IAAI,QAAQ,EAAE,MAAM,mCAAmC,CAAC;AAE3F,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,MAAM,8BAA8B,CAAC;AAEzD,+EAA+E;AAC/E,uEAAuE;AACvE,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG;IAC1B,wCAAwC;IACxC,GAAG,EAAE,CAAC,UAAU,CAAU;IAC1B,iCAAiC;IACjC,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,YAAY,CAAC,GAAG,EAAE,QAAQ,CAAU;IACtD,0CAA0C;IAC1C,SAAS,EAAE,CAAC,OAA0B,EAAE,EAAE,CAAC,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,EAAE,OAAO,CAAU;IACvF,0CAA0C;IAC1C,KAAK,EAAE,CAAC,SAAiB,EAAE,EAAE,CAAC,CAAC,GAAG,YAAY,CAAC,GAAG,EAAE,OAAO,EAAE,SAAS,CAAU;IAChF,4BAA4B;IAC5B,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,YAAY,CAAC,GAAG,EAAE,OAAO,CAAU;IACrD,uCAAuC;IACvC,KAAK,EAAE,CAAC,OAA0B,EAAE,EAAE,CAAC,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,EAAE,OAAO,CAAU;CACpF,CAAC;AAEF,oCAAoC;AACpC,MAAM,CAAC,MAAM,oBAAoB,GAAG,YAAY,CAAC;AA4CjD,+EAA+E;AAC/E,8CAA8C;AAC9C,+EAA+E;AAE/E,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,UAA6B,EAAE;IAC9D,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,QAAQ,SAAS,EAAE;QACzC,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,uBAAuB,EAAE,OAAO,CAAC,uBAAuB;QACxD,eAAe,EAAE,OAAO,CAAC,eAAe;KACzC,CAAC,CAAC;IAEH,OAAO,SAAS,CAAqB,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;AACvD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,SAAiB;IACnD,OAAO,SAAS,CAAwB,GAAG,QAAQ,WAAW,kBAAkB,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;AACjG,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,UAA6B,EAAE;;IAE/B,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,QAAQ,QAAQ,EAAE;QACxC,UAAU,EAAE,MAAA,OAAO,CAAC,UAAU,0CAAE,IAAI,CAAC,GAAG,CAAC;QACzC,eAAe,EAAE,OAAO,CAAC,eAAe;QACxC,iBAAiB,EAAE,OAAO,CAAC,iBAAiB;QAC5C,6BAA6B,EAAE,OAAO,CAAC,6BAA6B;QACpE,4BAA4B,EAAE,OAAO,CAAC,4BAA4B;QAClE,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;KAC/C,CAAC,CAAC;IAEH,OAAO,SAAS,CAAqB,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;AACvD,CAAC"}
+{"version":3,"file":"ontology.client.js","sourceRoot":"","sources":["../../../lib/ontology/api/ontology.client.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,sBAAsB,IAAI,QAAQ,EAAE,MAAM,+BAA+B,CAAC;AAEnF,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,MAAM,8BAA8B,CAAC;AAEzD,+EAA+E;AAC/E,uEAAuE;AACvE,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG;IAC1B,wCAAwC;IACxC,GAAG,EAAE,CAAC,UAAU,CAAU;IAC1B,iCAAiC;IACjC,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,YAAY,CAAC,GAAG,EAAE,QAAQ,CAAU;IACtD,0CAA0C;IAC1C,SAAS,EAAE,CAAC,OAA0B,EAAE,EAAE,CAAC,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,EAAE,OAAO,CAAU;IACvF,0CAA0C;IAC1C,KAAK,EAAE,CAAC,SAAiB,EAAE,EAAE,CAAC,CAAC,GAAG,YAAY,CAAC,GAAG,EAAE,OAAO,EAAE,SAAS,CAAU;IAChF,4BAA4B;IAC5B,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,YAAY,CAAC,GAAG,EAAE,OAAO,CAAU;IACrD,uCAAuC;IACvC,KAAK,EAAE,CAAC,OAA0B,EAAE,EAAE,CAAC,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,EAAE,OAAO,CAAU;CACpF,CAAC;AA0DF,+EAA+E;AAC/E,8CAA8C;AAC9C,+EAA+E;AAE/E,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,UAA6B,EAAE;IAC9D,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,QAAQ,SAAS,EAAE;QACzC,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,uBAAuB,EAAE,OAAO,CAAC,uBAAuB;QACxD,eAAe,EAAE,OAAO,CAAC,eAAe;KACzC,CAAC,CAAC;IAEH,OAAO,SAAS,CAAqB,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;AACvD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,SAAiB;IACnD,OAAO,SAAS,CAAwB,GAAG,QAAQ,WAAW,kBAAkB,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;AACjG,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,UAA6B,EAAE;;IAE/B,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,QAAQ,QAAQ,EAAE;QACxC,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,UAAU,EAAE,MAAA,OAAO,CAAC,UAAU,0CAAE,IAAI,CAAC,GAAG,CAAC;QACzC,eAAe,EAAE,OAAO,CAAC,eAAe;QACxC,iBAAiB,EAAE,OAAO,CAAC,iBAAiB;QAC5C,6BAA6B,EAAE,OAAO,CAAC,6BAA6B;QACpE,4BAA4B,EAAE,OAAO,CAAC,4BAA4B;QAClE,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;QAC9C,oBAAoB,EAAE,OAAO,CAAC,oBAAoB;KACnD,CAAC,CAAC;IAEH,OAAO,SAAS,CAAqB,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;AACvD,CAAC"}

package/dist/ontology/api/ontology.handlers.d.ts

@@ -42,28 +42,31 @@ export declare const describeTable: (request: import("next/server").NextRequest,
 /**
  * GET /api/ontology/graph
  *
- * Builds an ontology graph from DynamoDB tables for visualization.
+ * Builds an ontology graph for visualization.
  *
+ * @query source - Data source: "schemas" (default) or "dynamodb"
  * @query tableNames - Comma-separated list of specific table names to include
- * @query tableNamePrefix - Filter tables by name prefix
+ * @query tableNamePrefix - Filter tables by name prefix (dynamodb source only)
  * @query includeAttributes - Whether to include attribute nodes (default: true)
  * @query includeGlobalSecondaryIndexes - Whether to include GSI nodes (default: true)
  * @query includeLocalSecondaryIndexes - Whether to include LSI nodes (default: true)
- * @query inferRelationships - Whether to infer relationships between tables (default: true)
+ * @query inferRelationships - Whether to infer relationships (dynamodb source only, default: true)
+ * @query includeRelationships - Whether to include explicit relationships (schemas source only, default: true)
  *
  * @returns OntologyGraphData with nodes and links for visualization
  *
  * @example
- * GET /api/ontology/graph
- * GET /api/ontology/graph?tableNames=Users,Orders
- * GET /api/ontology/graph?tableNamePrefix=prod_&includeAttributes=false
- * GET /api/ontology/graph?inferRelationships=true
+ * GET /api/ontology/graph - Uses schema definitions (no AWS required)
+ * GET /api/ontology/graph?source=schemas - Explicit schema source
+ * GET /api/ontology/graph?source=dynamodb - Query live DynamoDB tables
+ * GET /api/ontology/graph?tableNames=captify_core_agents,captify_core_chat_sessions
+ * GET /api/ontology/graph?source=dynamodb&tableNamePrefix=prod_&includeAttributes=false
  */
 export declare const buildGraph: (request: import("next/server").NextRequest, context: {
     params: Promise<Record<string, unknown>>;
 }) => Promise<import("next/server").NextResponse<{
     error: string;
-} | import("../ontology.interfaces").DynamoDbOntologyGraphData>>;
+} | import("../ontology.interfaces").DataOntologyGraphData>>;
 /**
  * Semantic grouped handlers for Ontology API
  */
@@ -84,25 +87,6 @@ export declare const ontologyHandlers: {
         params: Promise<Record<string, unknown>>;
     }) => Promise<import("next/server").NextResponse<{
         error: string;
-    } | import("../ontology.interfaces").DynamoDbOntologyGraphData>>;
-};
-export declare const dynamoDbOntologyHandlers: {
-    listTables: (request: import("next/server").NextRequest, context: {
-        params: Promise<Record<string, unknown>>;
-    }) => Promise<import("next/server").NextResponse<{
-        error: string;
-    } | import("../ontology.interfaces").ListTablesResult>>;
-    describeTable: (request: import("next/server").NextRequest, context: {
-        params: Promise<{
-            tableName: string;
-        }>;
-    }) => Promise<import("next/server").NextResponse<{
-        error: string;
-    } | import("../ontology.interfaces").DynamoDbTableMetadata>>;
-    buildGraph: (request: import("next/server").NextRequest, context: {
-        params: Promise<Record<string, unknown>>;
-    }) => Promise<import("next/server").NextResponse<{
-        error: string;
-    } | import("../ontology.interfaces").DynamoDbOntologyGraphData>>;
+    } | import("../ontology.interfaces").DataOntologyGraphData>>;
 };
 //# sourceMappingURL=ontology.handlers.d.ts.map

package/dist/ontology/api/ontology.handlers.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ontology.handlers.d.ts","sourceRoot":"","sources":["../../../lib/ontology/api/ontology.handlers.ts"],"names":[],"mappings":"AAQA;;;;;;;;;;;;;;GAcG;AACH,eAAO,MAAM,UAAU;;;;uDAgBtB,CAAC;AAEF;;;;;;;;;;;;;GAaG;AACH,eAAO,MAAM,aAAa;;mBACuD,MAAM;;;;4DAWtF,CAAC;AAEF;;;;;;;;;;;;;;;;;;;GAmBG;AACH,eAAO,MAAM,UAAU;;;;gEAqBtB,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,gBAAgB;;;;;;;;uBA3DoD,MAAM;;;;;;;;;;CA+DtF,CAAC;AAGF,eAAO,MAAM,wBAAwB;;;;;;;;uBAlE4C,MAAM;;;;;;;;;;CAkE/B,CAAC"}
+{"version":3,"file":"ontology.handlers.d.ts","sourceRoot":"","sources":["../../../lib/ontology/api/ontology.handlers.ts"],"names":[],"mappings":"AAWA;;;;;;;;;;;;;;GAcG;AACH,eAAO,MAAM,UAAU;;;;uDA0BtB,CAAC;AAEF;;;;;;;;;;;;;GAaG;AACH,eAAO,MAAM,aAAa;;mBAIqB,MAAM;;;;4DAoBpD,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,eAAO,MAAM,UAAU;;;;4DAgDtB,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,gBAAgB;;;;;;;;uBAlGkB,MAAM;;;;;;;;;;CAsGpD,CAAC"}

package/dist/ontology/api/ontology.handlers.js

@@ -1,5 +1,8 @@
-import { getDynamoDbOntologyService } from "../services";
-import { listTablesQuerySchema, describeTableParamsSchema, buildGraphQuerySchema, } from "../ontology.schemas";
+import { buildGraphQuerySchema, describeTableParamsSchema, listTablesQuerySchema, } from "../ontology.schemas";
+import { buildOntologyGraphFromSchemas } from "../ontology.utilities";
+import { getDataOntologyService } from "../services";
+import { ACTION, RESOURCE } from "../../permissions/permissions.constants";
+import { requirePermission } from "../../permissions/permissions.utilities";
 import { getUserAwsCredentials, handleApi, withAuth } from "../../shared/utilities/api";
 /**
  * GET /api/ontology/tables
@@ -16,16 +19,24 @@ import { getUserAwsCredentials, handleApi, withAuth } from "../../shared/utiliti
  * GET /api/ontology/tables
  * GET /api/ontology/tables?limit=10&tableNamePrefix=prod_
  */
-export const listTables = withAuth(async (req, { idToken, tenantId }) => handleApi(async () => {
+export const listTables = withAuth(async (req, { idToken, tenantId, userId, user }) => handleApi(async () => {
     var _a, _b, _c;
+    const userCredentials = await getUserAwsCredentials(idToken);
+    await requirePermission({
+        userId,
+        tenantId,
+        resource: RESOURCE.ONTOLOGY,
+        action: ACTION.LIST,
+        groups: user.groups,
+        userCredentials,
+    });
     const url = new URL(req.url);
     const query = listTablesQuerySchema.parse({
         limit: (_a = url.searchParams.get("limit")) !== null && _a !== void 0 ? _a : undefined,
         exclusiveStartTableName: (_b = url.searchParams.get("exclusiveStartTableName")) !== null && _b !== void 0 ? _b : undefined,
         tableNamePrefix: (_c = url.searchParams.get("tableNamePrefix")) !== null && _c !== void 0 ? _c : undefined,
     });
-    const userCredentials = await getUserAwsCredentials(idToken);
-    return getDynamoDbOntologyService().listTables(Object.assign(Object.assign({ tenantId }, query), { userCredentials }));
+    return getDataOntologyService().listTables(Object.assign(Object.assign({ tenantId }, query), { userCredentials }));
 }));
 /**
  * GET /api/ontology/tables/:tableName
@@ -41,10 +52,17 @@ export const listTables = withAuth(async (req, { idToken, tenantId }) => handleA
  * @example
  * GET /api/ontology/tables/Users
  */
-export const describeTable = withAuth(async (_req, { idToken, tenantId }, { params }) => handleApi(async () => {
+export const describeTable = withAuth(async (_req, { idToken, tenantId, userId, user }, { params }) => handleApi(async () => {
+    await requirePermission({
+        userId,
+        tenantId,
+        resource: RESOURCE.ONTOLOGY,
+        action: ACTION.READ,
+        groups: user.groups,
+    });
     const { tableName } = describeTableParamsSchema.parse(await params);
     const userCredentials = await getUserAwsCredentials(idToken);
-    return getDynamoDbOntologyService().describeTable({
+    return getDataOntologyService().describeTable({
         tenantId,
         tableName,
         userCredentials,
@@ -53,36 +71,68 @@ export const describeTable = withAuth(async (_req, { idToken, tenantId }, { para
 /**
  * GET /api/ontology/graph
  *
- * Builds an ontology graph from DynamoDB tables for visualization.
+ * Builds an ontology graph for visualization.
  *
+ * @query source - Data source: "schemas" (default) or "dynamodb"
  * @query tableNames - Comma-separated list of specific table names to include
- * @query tableNamePrefix - Filter tables by name prefix
+ * @query tableNamePrefix - Filter tables by name prefix (dynamodb source only)
  * @query includeAttributes - Whether to include attribute nodes (default: true)
  * @query includeGlobalSecondaryIndexes - Whether to include GSI nodes (default: true)
  * @query includeLocalSecondaryIndexes - Whether to include LSI nodes (default: true)
- * @query inferRelationships - Whether to infer relationships between tables (default: true)
+ * @query inferRelationships - Whether to infer relationships (dynamodb source only, default: true)
+ * @query includeRelationships - Whether to include explicit relationships (schemas source only, default: true)
  *
  * @returns OntologyGraphData with nodes and links for visualization
  *
  * @example
- * GET /api/ontology/graph
- * GET /api/ontology/graph?tableNames=Users,Orders
- * GET /api/ontology/graph?tableNamePrefix=prod_&includeAttributes=false
- * GET /api/ontology/graph?inferRelationships=true
+ * GET /api/ontology/graph - Uses schema definitions (no AWS required)
+ * GET /api/ontology/graph?source=schemas - Explicit schema source
+ * GET /api/ontology/graph?source=dynamodb - Query live DynamoDB tables
+ * GET /api/ontology/graph?tableNames=captify_core_agents,captify_core_chat_sessions
+ * GET /api/ontology/graph?source=dynamodb&tableNamePrefix=prod_&includeAttributes=false
  */
-export const buildGraph = withAuth(async (req, { idToken, tenantId }) => handleApi(async () => {
-    var _a, _b, _c, _d, _e, _f;
+export const buildGraph = withAuth(async (req, { idToken, tenantId, userId, user }) => handleApi(async () => {
+    var _a, _b, _c, _d, _e, _f, _g, _h;
+    await requirePermission({
+        userId,
+        tenantId,
+        resource: RESOURCE.ONTOLOGY,
+        action: ACTION.READ,
+        groups: user.groups,
+    });
     const url = new URL(req.url);
     const query = buildGraphQuerySchema.parse({
-        tableNames: (_a = url.searchParams.get("tableNames")) !== null && _a !== void 0 ? _a : undefined,
-        tableNamePrefix: (_b = url.searchParams.get("tableNamePrefix")) !== null && _b !== void 0 ? _b : undefined,
-        includeAttributes: (_c = url.searchParams.get("includeAttributes")) !== null && _c !== void 0 ? _c : undefined,
-        includeGlobalSecondaryIndexes: (_d = url.searchParams.get("includeGlobalSecondaryIndexes")) !== null && _d !== void 0 ? _d : undefined,
-        includeLocalSecondaryIndexes: (_e = url.searchParams.get("includeLocalSecondaryIndexes")) !== null && _e !== void 0 ? _e : undefined,
-        inferRelationships: (_f = url.searchParams.get("inferRelationships")) !== null && _f !== void 0 ? _f : undefined,
+        source: (_a = url.searchParams.get("source")) !== null && _a !== void 0 ? _a : undefined,
+        tableNames: (_b = url.searchParams.get("tableNames")) !== null && _b !== void 0 ? _b : undefined,
+        tableNamePrefix: (_c = url.searchParams.get("tableNamePrefix")) !== null && _c !== void 0 ? _c : undefined,
+        includeAttributes: (_d = url.searchParams.get("includeAttributes")) !== null && _d !== void 0 ? _d : undefined,
+        includeGlobalSecondaryIndexes: (_e = url.searchParams.get("includeGlobalSecondaryIndexes")) !== null && _e !== void 0 ? _e : undefined,
+        includeLocalSecondaryIndexes: (_f = url.searchParams.get("includeLocalSecondaryIndexes")) !== null && _f !== void 0 ? _f : undefined,
+        inferRelationships: (_g = url.searchParams.get("inferRelationships")) !== null && _g !== void 0 ? _g : undefined,
+        includeRelationships: (_h = url.searchParams.get("includeRelationships")) !== null && _h !== void 0 ? _h : undefined,
     });
+    // Use schema-based generation (default) or live DynamoDB query
+    if (query.source === "schemas") {
+        return buildOntologyGraphFromSchemas({
+            tableNames: query.tableNames,
+            includeAttributes: query.includeAttributes,
+            includeGlobalSecondaryIndexes: query.includeGlobalSecondaryIndexes,
+            includeLocalSecondaryIndexes: query.includeLocalSecondaryIndexes,
+            includeRelationships: query.includeRelationships,
+        });
+    }
+    // Legacy DynamoDB-based generation
     const userCredentials = await getUserAwsCredentials(idToken);
-    return getDynamoDbOntologyService().buildOntologyGraph(Object.assign(Object.assign({ tenantId }, query), { userCredentials }));
+    return getDataOntologyService().buildOntologyGraph({
+        tenantId,
+        tableNames: query.tableNames,
+        tableNamePrefix: query.tableNamePrefix,
+        includeAttributes: query.includeAttributes,
+        includeGlobalSecondaryIndexes: query.includeGlobalSecondaryIndexes,
+        includeLocalSecondaryIndexes: query.includeLocalSecondaryIndexes,
+        inferRelationships: query.inferRelationships,
+        userCredentials,
+    });
 }));
 /**
  * Semantic grouped handlers for Ontology API
@@ -92,6 +142,4 @@ export const ontologyHandlers = {
     describeTable,
     buildGraph,
 };
-// Keep backward compatibility alias
-export const dynamoDbOntologyHandlers = ontologyHandlers;
 //# sourceMappingURL=ontology.handlers.js.map

package/dist/ontology/api/ontology.handlers.js.map

@@ -1 +1 @@
-{"version":3,"file":"ontology.handlers.js","sourceRoot":"","sources":["../../../lib/ontology/api/ontology.handlers.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,0BAA0B,EAAE,MAAM,yBAAyB,CAAC;AACrE,OAAO,EACL,qBAAqB,EACrB,yBAAyB,EACzB,qBAAqB,GACtB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,qBAAqB,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAExF;;;;;;;;;;;;;;GAcG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,CACtE,SAAS,CAAC,KAAK,IAAI,EAAE;;IACnB,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,qBAAqB,CAAC,KAAK,CAAC;QACxC,KAAK,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,mCAAI,SAAS;QACjD,uBAAuB,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,yBAAyB,CAAC,mCAAI,SAAS;QACrF,eAAe,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,iBAAiB,CAAC,mCAAI,SAAS;KACtE,CAAC,CAAC;IAEH,MAAM,eAAe,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAC;IAC7D,OAAO,0BAA0B,EAAE,CAAC,UAAU,+BAC5C,QAAQ,IACL,KAAK,KACR,eAAe,IACf,CAAC;AACL,CAAC,CAAC,CACH,CAAC;AAEF;;;;;;;;;;;;;GAaG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,QAAQ,CACnC,KAAK,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAA8C,EAAE,EAAE,CAC5F,SAAS,CAAC,KAAK,IAAI,EAAE;IACnB,MAAM,EAAE,SAAS,EAAE,GAAG,yBAAyB,CAAC,KAAK,CAAC,MAAM,MAAM,CAAC,CAAC;IAEpE,MAAM,eAAe,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAC;IAC7D,OAAO,0BAA0B,EAAE,CAAC,aAAa,CAAC;QAChD,QAAQ;QACR,SAAS;QACT,eAAe;KAChB,CAAC,CAAC;AACL,CAAC,CAAC,CACL,CAAC;AAEF;;;;;;;;;;;;;;;;;;;GAmBG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,CACtE,SAAS,CAAC,KAAK,IAAI,EAAE;;IACnB,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,qBAAqB,CAAC,KAAK,CAAC;QACxC,UAAU,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,mCAAI,SAAS;QAC3D,eAAe,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,iBAAiB,CAAC,mCAAI,SAAS;QACrE,iBAAiB,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,mBAAmB,CAAC,mCAAI,SAAS;QACzE,6BAA6B,EAC3B,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,+BAA+B,CAAC,mCAAI,SAAS;QACpE,4BAA4B,EAC1B,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,8BAA8B,CAAC,mCAAI,SAAS;QACnE,kBAAkB,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,oBAAoB,CAAC,mCAAI,SAAS;KAC5E,CAAC,CAAC;IAEH,MAAM,eAAe,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAC;IAC7D,OAAO,0BAA0B,EAAE,CAAC,kBAAkB,+BACpD,QAAQ,IACL,KAAK,KACR,eAAe,IACf,CAAC;AACL,CAAC,CAAC,CACH,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B,UAAU;IACV,aAAa;IACb,UAAU;CACX,CAAC;AAEF,oCAAoC;AACpC,MAAM,CAAC,MAAM,wBAAwB,GAAG,gBAAgB,CAAC"}
+{"version":3,"file":"ontology.handlers.js","sourceRoot":"","sources":["../../../lib/ontology/api/ontology.handlers.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,qBAAqB,EACrB,yBAAyB,EACzB,qBAAqB,GACtB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,6BAA6B,EAAE,MAAM,mCAAmC,CAAC;AAClF,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,yCAAyC,CAAC;AAC3E,OAAO,EAAE,iBAAiB,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,EAAE,qBAAqB,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAExF;;;;;;;;;;;;;;GAcG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CACpF,SAAS,CAAC,KAAK,IAAI,EAAE;;IACnB,MAAM,eAAe,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAC;IAE7D,MAAM,iBAAiB,CAAC;QACtB,MAAM;QACN,QAAQ;QACR,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,MAAM,EAAE,MAAM,CAAC,IAAI;QACnB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,eAAe;KAChB,CAAC,CAAC;IAEH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,qBAAqB,CAAC,KAAK,CAAC;QACxC,KAAK,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,mCAAI,SAAS;QACjD,uBAAuB,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,yBAAyB,CAAC,mCAAI,SAAS;QACrF,eAAe,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,iBAAiB,CAAC,mCAAI,SAAS;KACtE,CAAC,CAAC;IAEH,OAAO,sBAAsB,EAAE,CAAC,UAAU,+BACxC,QAAQ,IACL,KAAK,KACR,eAAe,IACf,CAAC;AACL,CAAC,CAAC,CACH,CAAC;AAEF;;;;;;;;;;;;;GAaG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,QAAQ,CACnC,KAAK,EACH,IAAI,EACJ,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,EACnC,EAAE,MAAM,EAA8C,EACtD,EAAE,CACF,SAAS,CAAC,KAAK,IAAI,EAAE;IACnB,MAAM,iBAAiB,CAAC;QACtB,MAAM;QACN,QAAQ;QACR,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,MAAM,EAAE,MAAM,CAAC,IAAI;QACnB,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAC,CAAC;IAEH,MAAM,EAAE,SAAS,EAAE,GAAG,yBAAyB,CAAC,KAAK,CAAC,MAAM,MAAM,CAAC,CAAC;IAEpE,MAAM,eAAe,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAC;IAC7D,OAAO,sBAAsB,EAAE,CAAC,aAAa,CAAC;QAC5C,QAAQ;QACR,SAAS;QACT,eAAe;KAChB,CAAC,CAAC;AACL,CAAC,CAAC,CACL,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CACpF,SAAS,CAAC,KAAK,IAAI,EAAE;;IACnB,MAAM,iBAAiB,CAAC;QACtB,MAAM;QACN,QAAQ;QACR,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,MAAM,EAAE,MAAM,CAAC,IAAI;QACnB,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAC,CAAC;IAEH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,qBAAqB,CAAC,KAAK,CAAC;QACxC,MAAM,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,mCAAI,SAAS;QACnD,UAAU,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,mCAAI,SAAS;QAC3D,eAAe,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,iBAAiB,CAAC,mCAAI,SAAS;QACrE,iBAAiB,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,mBAAmB,CAAC,mCAAI,SAAS;QACzE,6BAA6B,EAC3B,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,+BAA+B,CAAC,mCAAI,SAAS;QACpE,4BAA4B,EAC1B,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,8BAA8B,CAAC,mCAAI,SAAS;QACnE,kBAAkB,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,oBAAoB,CAAC,mCAAI,SAAS;QAC3E,oBAAoB,EAAE,MAAA,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,sBAAsB,CAAC,mCAAI,SAAS;KAChF,CAAC,CAAC;IAEH,+DAA+D;IAC/D,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAC/B,OAAO,6BAA6B,CAAC;YACnC,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;YAC1C,6BAA6B,EAAE,KAAK,CAAC,6BAA6B;YAClE,4BAA4B,EAAE,KAAK,CAAC,4BAA4B;YAChE,oBAAoB,EAAE,KAAK,CAAC,oBAAoB;SACjD,CAAC,CAAC;IACL,CAAC;IAED,mCAAmC;IACnC,MAAM,eAAe,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC,CAAC;IAC7D,OAAO,sBAAsB,EAAE,CAAC,kBAAkB,CAAC;QACjD,QAAQ;QACR,UAAU,EAAE,KAAK,CAAC,UAAU;QAC5B,eAAe,EAAE,KAAK,CAAC,eAAe;QACtC,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;QAC1C,6BAA6B,EAAE,KAAK,CAAC,6BAA6B;QAClE,4BAA4B,EAAE,KAAK,CAAC,4BAA4B;QAChE,kBAAkB,EAAE,KAAK,CAAC,kBAAkB;QAC5C,eAAe;KAChB,CAAC,CAAC;AACL,CAAC,CAAC,CACH,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B,UAAU;IACV,aAAa;IACb,UAAU;CACX,CAAC"}

package/dist/ontology/hooks/index.d.ts

@@ -1,3 +1,3 @@
-export * from "./use-dynamodb-ontology";
+export * from "./use-data-ontology";
 export * from "./use-ontology-graph";
 //# sourceMappingURL=index.d.ts.map

package/dist/ontology/hooks/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../lib/ontology/hooks/index.ts"],"names":[],"mappings":"AAAA,cAAc,4CAA4C,CAAC;AAC3D,cAAc,yCAAyC,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../lib/ontology/hooks/index.ts"],"names":[],"mappings":"AAAA,cAAc,wCAAwC,CAAC;AACvD,cAAc,yCAAyC,CAAC"}

package/dist/ontology/hooks/index.js

@@ -1,3 +1,3 @@
-export * from "./use-dynamodb-ontology";
+export * from "./use-data-ontology";
 export * from "./use-ontology-graph";
 //# sourceMappingURL=index.js.map

package/dist/ontology/hooks/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../lib/ontology/hooks/index.ts"],"names":[],"mappings":"AAAA,cAAc,4CAA4C,CAAC;AAC3D,cAAc,yCAAyC,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../lib/ontology/hooks/index.ts"],"names":[],"mappings":"AAAA,cAAc,wCAAwC,CAAC;AACvD,cAAc,yCAAyC,CAAC"}