@neverinfamous/postgres-mcp 1.3.0 → 2.1.0

package/dist/adapters/postgresql/tools/vector/search-advanced.js

@@ -0,0 +1,626 @@
+/**
+ * PostgreSQL pgvector - Search & Analysis
+ *
+ * High-level search and analysis: cluster, hybridSearch, performance.
+ */
+import { z } from "zod";
+import { readOnly } from "../../../../utils/annotations.js";
+import { getToolIcons } from "../../../../utils/icons.js";
+import { formatPostgresError } from "../core/error-helpers.js";
+import { sanitizeIdentifier, sanitizeTableName, } from "../../../../utils/identifiers.js";
+import { checkTableAndColumn, truncateVector } from "./data.js";
+import { VectorClusterOutputSchema, HybridSearchOutputSchema, VectorPerformanceOutputSchema, } from "../../schemas/index.js";
+/**
+ * Parse a PostgreSQL vector string to a number array.
+ */
+function parseVector(vecStr) {
+    if (typeof vecStr !== "string")
+        return null;
+    try {
+        const cleaned = vecStr.replace(/[[\]()]/g, "");
+        return cleaned.split(",").map(Number);
+    }
+    catch {
+        return null;
+    }
+}
+export function createVectorClusterTool(adapter) {
+    // Schema with parameter smoothing
+    const ClusterSchemaBase = z.object({
+        table: z.string().optional().describe("Table name"),
+        tableName: z.string().optional().describe("Alias for table"),
+        column: z.string().optional().describe("Vector column"),
+        col: z.string().optional().describe("Alias for column"),
+        k: z.any().optional().describe("Number of clusters"),
+        clusters: z.any().optional().describe("Alias for k (number of clusters)"),
+        iterations: z.any().optional().describe("Max iterations (default: 10)"),
+        sampleSize: z.any().optional().describe("Sample size for large tables"),
+        schema: z.string().optional().describe("Database schema (default: public)"),
+    });
+    const ClusterSchema = ClusterSchemaBase.transform((data) => {
+        const rawK = (data.k ?? data.clusters);
+        const rawIterations = data.iterations;
+        const rawSampleSize = data.sampleSize;
+        return {
+            table: data.table ?? data.tableName ?? "",
+            column: data.column ?? data.col ?? "",
+            k: rawK != null ? Number(rawK) : undefined,
+            iterations: rawIterations != null ? Number(rawIterations) : undefined,
+            sampleSize: rawSampleSize != null ? Number(rawSampleSize) : undefined,
+            schema: data.schema,
+        };
+    }).refine((data) => data.k !== undefined, {
+        message: "k (or clusters alias) is required",
+    });
+    return {
+        name: "pg_vector_cluster",
+        description: "Perform K-means clustering on vectors. Returns cluster centroids only (not row assignments). To assign rows to clusters, compare row vectors to centroids using pg_vector_distance.",
+        group: "vector",
+        inputSchema: ClusterSchemaBase,
+        outputSchema: VectorClusterOutputSchema,
+        annotations: readOnly("Vector Cluster"),
+        icons: getToolIcons("vector", readOnly("Vector Cluster")),
+        handler: async (params, _context) => {
+            try {
+                const parsed = ClusterSchema.parse(params);
+                // Refine guarantees k is defined, but add explicit check for TypeScript
+                const k = parsed.k;
+                if (k === undefined) {
+                    throw new Error("k (or clusters alias) is required");
+                }
+                if (isNaN(k)) {
+                    return {
+                        success: false,
+                        error: `Validation error: k must be a valid number, received "${String(parsed.k)}"`,
+                        suggestion: "Provide a numeric value for k (e.g., 3, 5, 10)",
+                    };
+                }
+                if (k < 1) {
+                    return {
+                        success: false,
+                        error: "k must be at least 1 (number of clusters)",
+                        suggestion: "Provide k >= 1, typically between 2 and 20",
+                    };
+                }
+                const maxIter = parsed.iterations ?? 10;
+                const sample = parsed.sampleSize ?? 10000;
+                const schemaName = parsed.schema ?? "public";
+                const tableName = sanitizeTableName(parsed.table, parsed.schema);
+                const columnName = sanitizeIdentifier(parsed.column);
+                // Two-step existence check: table first, then column
+                const existenceCheck = await checkTableAndColumn(adapter, parsed.table, parsed.column, schemaName);
+                if (existenceCheck) {
+                    return { success: false, ...existenceCheck };
+                }
+                // Validate column is actually a vector type
+                const typeCheckSql = `
+        SELECT udt_name FROM information_schema.columns
+        WHERE table_schema = $1 AND table_name = $2 AND column_name = $3
+      `;
+                const typeResult = await adapter.executeQuery(typeCheckSql, [
+                    schemaName,
+                    parsed.table,
+                    parsed.column,
+                ]);
+                const udtName = typeResult.rows?.[0]?.["udt_name"];
+                if (udtName !== "vector") {
+                    return {
+                        success: false,
+                        error: `Column '${parsed.column}' is not a vector column (type: ${udtName ?? "unknown"})`,
+                        suggestion: "Use a column with vector type for clustering",
+                    };
+                }
+                const sampleSql = `
+                SELECT ${columnName} as vec
+                FROM ${tableName}
+                WHERE ${columnName} IS NOT NULL
+                ORDER BY RANDOM()
+                LIMIT ${String(sample)}
+            `;
+                const sampleResult = await adapter.executeQuery(sampleSql);
+                const vectors = (sampleResult.rows ?? []);
+                if (vectors.length < k) {
+                    return {
+                        success: false,
+                        error: `Cannot create ${String(k)} clusters with only ${String(vectors.length)} data points. Reduce k to at most ${String(vectors.length)} or increase sampleSize.`,
+                        k: k,
+                        availableDataPoints: vectors.length,
+                        sampleSize: sample,
+                    };
+                }
+                const initialCentroids = vectors.slice(0, k).map((v) => v.vec);
+                const clusterSql = `
+                WITH sample_vectors AS (
+                    SELECT ROW_NUMBER() OVER () as id, ${columnName} as vec
+                    FROM ${tableName}
+                    WHERE ${columnName} IS NOT NULL
+                    LIMIT ${String(sample)}
+                ),
+                centroids AS (
+                    SELECT unnest($1::vector[]) as centroid
+                )
+                SELECT
+                    c.centroid,
+                    COUNT(*) as cluster_size,
+                    AVG(s.vec) as new_centroid
+                FROM sample_vectors s
+                CROSS JOIN LATERAL (
+                    SELECT centroid, ROW_NUMBER() OVER (ORDER BY s.vec <-> centroid) as rn
+                    FROM centroids
+                ) c
+                WHERE c.rn = 1
+                GROUP BY c.centroid
+            `;
+                let centroids = initialCentroids;
+                for (let i = 0; i < maxIter; i++) {
+                    try {
+                        const result = await adapter.executeQuery(clusterSql, [centroids]);
+                        centroids = (result.rows ?? []).map((r) => r["new_centroid"]);
+                    }
+                    catch {
+                        break;
+                    }
+                }
+                // Truncate large centroids for display (like pg_vector_aggregate does)
+                const parsedCentroids = centroids.map((c) => {
+                    const parsed = parseVector(c);
+                    if (parsed === null) {
+                        return { vector: c };
+                    }
+                    // For large vectors, use preview format (first 10 dimensions)
+                    if (parsed.length > 10) {
+                        const truncated = truncateVector(parsed, 10);
+                        return {
+                            preview: truncated.preview,
+                            dimensions: truncated.dimensions,
+                            truncated: truncated.truncated,
+                        };
+                    }
+                    return { vector: parsed };
+                });
+                return {
+                    k: k,
+                    iterations: maxIter,
+                    sampleSize: vectors.length,
+                    centroids: parsedCentroids,
+                    note: "For production clustering, consider using specialized libraries",
+                };
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: formatPostgresError(error, { tool: "pg_vector_cluster" }),
+                };
+            }
+        },
+    };
+}
+export function createHybridSearchTool(adapter) {
+    // Schema with parameter smoothing
+    const HybridSearchSchemaBase = z.object({
+        table: z.string().optional().describe("Table name"),
+        tableName: z.string().optional().describe("Alias for table"),
+        vectorColumn: z.string().optional().describe("Vector column"),
+        vectorCol: z.string().optional().describe("Alias for vectorColumn"),
+        textColumn: z.string().describe("Text column for FTS"),
+        vector: z.array(z.number()).describe("Query vector"),
+        textQuery: z.string().describe("Text search query"),
+        vectorWeight: z.coerce
+            .number()
+            .optional()
+            .describe("Weight for vector score (0-1, default: 0.5)"),
+        limit: z.coerce.number().optional().describe("Max results"),
+        select: z
+            .array(z.string())
+            .optional()
+            .describe("Columns to return (defaults to non-vector columns)"),
+    });
+    const HybridSearchSchema = HybridSearchSchemaBase.transform((data) => ({
+        table: data.table ?? data.tableName ?? "",
+        vectorColumn: data.vectorColumn ?? data.vectorCol ?? "",
+        textColumn: data.textColumn,
+        vector: data.vector,
+        textQuery: data.textQuery,
+        vectorWeight: data.vectorWeight,
+        limit: data.limit,
+        select: data.select,
+    }));
+    return {
+        name: "pg_hybrid_search",
+        description: "Combined vector similarity and full-text search with weighted scoring.",
+        group: "vector",
+        inputSchema: HybridSearchSchemaBase,
+        outputSchema: HybridSearchOutputSchema,
+        annotations: readOnly("Hybrid Search"),
+        icons: getToolIcons("vector", readOnly("Hybrid Search")),
+        handler: async (params, _context) => {
+            try {
+                const parsed = HybridSearchSchema.parse(params);
+                // Validate required parameters before using them
+                if (parsed.table === "") {
+                    return {
+                        success: false,
+                        error: "table (or tableName) parameter is required",
+                        requiredParams: [
+                            "table",
+                            "vectorColumn",
+                            "textColumn",
+                            "vector",
+                            "textQuery",
+                        ],
+                    };
+                }
+                if (parsed.vectorColumn === "") {
+                    return {
+                        success: false,
+                        error: "vectorColumn (or vectorCol) parameter is required",
+                        requiredParams: [
+                            "table",
+                            "vectorColumn",
+                            "textColumn",
+                            "vector",
+                            "textQuery",
+                        ],
+                    };
+                }
+                // Parse schema.table format (embedded schema takes priority)
+                let resolvedTable = parsed.table;
+                let resolvedSchema;
+                if (parsed.table.includes(".")) {
+                    const parts = parsed.table.split(".");
+                    resolvedSchema = parts[0];
+                    resolvedTable = parts[1] ?? parsed.table;
+                }
+                const schemaName = resolvedSchema ?? "public";
+                const tableName = sanitizeTableName(resolvedTable, schemaName);
+                // P154: Verify table and vectorColumn exist before querying
+                const existenceError = await checkTableAndColumn(adapter, resolvedTable, parsed.vectorColumn, schemaName);
+                if (existenceError !== null) {
+                    return { success: false, ...existenceError };
+                }
+                // Check column type - reject if it's a tsvector
+                const colTypeSql = `
+                SELECT data_type, udt_name
+                FROM information_schema.columns
+                WHERE table_schema = $1 AND table_name = $2 AND column_name = $3
+            `;
+                const colTypeResult = await adapter.executeQuery(colTypeSql, [
+                    schemaName,
+                    resolvedTable,
+                    parsed.vectorColumn,
+                ]);
+                const colType = colTypeResult.rows?.[0];
+                if (colType?.udt_name === "tsvector" ||
+                    colType?.data_type === "tsvector") {
+                    return {
+                        success: false,
+                        error: `Column '${parsed.vectorColumn}' is tsvector, not vector. For hybrid search, vectorColumn must be a pgvector column (type 'vector'). Use textColumn for text search.`,
+                        suggestion: `Specify a different vector column, or check your table structure with pg_describe_table`,
+                    };
+                }
+                if (colType?.udt_name !== "vector" && colType !== undefined) {
+                    const actualType = colType.udt_name ?? colType.data_type ?? "unknown";
+                    return {
+                        success: false,
+                        error: `Column '${parsed.vectorColumn}' has type '${actualType}', not 'vector'. Hybrid search requires a pgvector column.`,
+                        columnType: actualType,
+                    };
+                }
+                // Check textColumn type to determine if we need to_tsvector() wrapping
+                const textColTypeResult = await adapter.executeQuery(colTypeSql, [
+                    schemaName,
+                    resolvedTable,
+                    parsed.textColumn,
+                ]);
+                const textColType = textColTypeResult.rows?.[0];
+                const isTextColumnTsvector = textColType?.udt_name === "tsvector" ||
+                    textColType?.data_type === "tsvector";
+                // Use tsvector column directly, otherwise wrap with to_tsvector()
+                const textExpr = isTextColumnTsvector
+                    ? `"${parsed.textColumn}"`
+                    : `to_tsvector('english', "${parsed.textColumn}")`;
+                const vectorWeight = parsed.vectorWeight ?? 0.5;
+                // Fix floating point precision (e.g., 0.30000000000000004 -> 0.3)
+                const textWeight = Math.round((1 - vectorWeight) * 1000) / 1000;
+                const limitVal = parsed.limit ?? 10;
+                const vectorStr = `[${parsed.vector.join(",")}]`;
+                // Build select clause - use specified columns, excluding vector column if using t.*
+                let selectCols;
+                if (parsed.select !== undefined && parsed.select.length > 0) {
+                    // Use only the explicitly selected columns
+                    selectCols = parsed.select.map((c) => `t."${c}"`).join(", ");
+                }
+                else {
+                    // Get all columns except vector columns to avoid token waste
+                    const colsSql = `
+                    SELECT column_name
+                    FROM information_schema.columns
+                    WHERE table_schema = $1 AND table_name = $2
+                    AND udt_name != 'vector'
+                    ORDER BY ordinal_position
+                `;
+                    const colsResult = await adapter.executeQuery(colsSql, [
+                        schemaName,
+                        resolvedTable,
+                    ]);
+                    const cols = (colsResult.rows ?? []).map((r) => r["column_name"]);
+                    selectCols =
+                        cols.length > 0 ? cols.map((c) => `t."${c}"`).join(", ") : "t.*";
+                }
+                const sql = `
+                WITH vector_scores AS (
+                    SELECT
+                        ctid,
+                        1 - ("${parsed.vectorColumn}" <=> '${vectorStr}'::vector) as vector_score
+                    FROM ${tableName}
+                    WHERE "${parsed.vectorColumn}" IS NOT NULL
+                    ORDER BY "${parsed.vectorColumn}" <=> '${vectorStr}'::vector
+                    LIMIT ${String(limitVal * 3)}
+                ),
+                text_scores AS (
+                    SELECT
+                        ctid,
+                        ts_rank(${textExpr}, plainto_tsquery($1)) as text_score
+                    FROM ${tableName}
+                    WHERE ${textExpr} @@ plainto_tsquery($1)
+                )
+                SELECT
+                    ${selectCols},
+                    COALESCE(v.vector_score, 0) * ${String(vectorWeight)} +
+                    COALESCE(ts.text_score, 0) * ${String(textWeight)} as combined_score,
+                    COALESCE(v.vector_score, 0) as vector_score,
+                    COALESCE(ts.text_score, 0) as text_score
+                FROM ${tableName} t
+                LEFT JOIN vector_scores v ON t.ctid = v.ctid
+                LEFT JOIN text_scores ts ON t.ctid = ts.ctid
+                WHERE v.ctid IS NOT NULL OR ts.ctid IS NOT NULL
+                ORDER BY combined_score DESC
+                LIMIT ${String(limitVal)}
+            `;
+                try {
+                    const result = await adapter.executeQuery(sql, [parsed.textQuery]);
+                    return {
+                        results: result.rows,
+                        count: result.rows?.length ?? 0,
+                        vectorWeight,
+                        textWeight,
+                    };
+                }
+                catch (error) {
+                    if (error instanceof Error) {
+                        // Parse column not found errors
+                        const colMatch = /column "([^"]+)" does not exist/.exec(error.message);
+                        if (colMatch) {
+                            const missingCol = colMatch[1] ?? "";
+                            // Determine which parameter has the issue
+                            let paramName = "column";
+                            if (missingCol === parsed.textColumn) {
+                                paramName = "textColumn";
+                            }
+                            else if (missingCol === parsed.vectorColumn) {
+                                paramName = "vectorColumn";
+                            }
+                            return {
+                                success: false,
+                                error: `Column '${missingCol}' does not exist in table '${resolvedTable}'`,
+                                parameterWithIssue: paramName,
+                                suggestion: "Use pg_describe_table to find available columns",
+                            };
+                        }
+                        // Parse dimension mismatch errors
+                        const dimMatch = /different vector dimensions (\d+) and (\d+)/.exec(error.message);
+                        if (dimMatch) {
+                            const expectedDim = dimMatch[1] ?? "0";
+                            const providedDim = dimMatch[2] ?? "0";
+                            return {
+                                success: false,
+                                error: `Vector dimension mismatch: column expects ${expectedDim} dimensions, but you provided ${providedDim} dimensions.`,
+                                expectedDimensions: parseInt(expectedDim, 10),
+                                providedDimensions: parseInt(providedDim, 10),
+                                suggestion: "Ensure your query vector has the same dimensions as the column.",
+                            };
+                        }
+                        // Parse relation not found errors
+                        const relationMatch = /relation "([^"]+)" does not exist/.exec(error.message);
+                        if (relationMatch) {
+                            const missingRelation = relationMatch[1] ?? "";
+                            return {
+                                success: false,
+                                error: `Table '${missingRelation}' does not exist in schema '${schemaName}'`,
+                                suggestion: "Use pg_list_tables to find available tables",
+                            };
+                        }
+                        // Return generic database error as {success: false} instead of throwing
+                        return {
+                            success: false,
+                            error: error.message,
+                            suggestion: "Check your query parameters and table structure",
+                        };
+                    }
+                    // For non-Error exceptions, return generic error
+                    return {
+                        success: false,
+                        error: "An unexpected error occurred",
+                        details: String(error),
+                    };
+                }
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: formatPostgresError(error, { tool: "pg_hybrid_search" }),
+                };
+            }
+        },
+    };
+}
+export function createVectorPerformanceTool(adapter) {
+    // Schema with parameter smoothing
+    const PerformanceSchemaBase = z.object({
+        table: z.string().optional().describe("Table name"),
+        tableName: z.string().optional().describe("Alias for table"),
+        column: z.string().optional().describe("Vector column"),
+        col: z.string().optional().describe("Alias for column"),
+        testVector: z
+            .array(z.number())
+            .optional()
+            .describe("Test vector for benchmarking"),
+        schema: z.string().optional().describe("Database schema (default: public)"),
+    });
+    const PerformanceSchema = PerformanceSchemaBase.transform((data) => ({
+        table: data.table ?? data.tableName ?? "",
+        column: data.column ?? data.col ?? "",
+        testVector: data.testVector,
+        schema: data.schema,
+    }));
+    return {
+        name: "pg_vector_performance",
+        description: "Analyze vector search performance and index effectiveness. Provide testVector for benchmarking (recommended).",
+        group: "vector",
+        inputSchema: PerformanceSchemaBase,
+        outputSchema: VectorPerformanceOutputSchema,
+        annotations: readOnly("Vector Performance"),
+        icons: getToolIcons("vector", readOnly("Vector Performance")),
+        handler: async (params, _context) => {
+            try {
+                const parsed = PerformanceSchema.parse(params);
+                // Validate required params
+                if (parsed.table === "") {
+                    return {
+                        success: false,
+                        error: "table (or tableName) parameter is required",
+                        requiredParams: ["table", "column"],
+                    };
+                }
+                if (parsed.column === "") {
+                    return {
+                        success: false,
+                        error: "column (or col) parameter is required for the vector column name",
+                        requiredParams: ["table", "column"],
+                    };
+                }
+                const tableName = sanitizeTableName(parsed.table, parsed.schema);
+                const columnName = sanitizeIdentifier(parsed.column);
+                const schemaName = parsed.schema ?? "public";
+                // Two-step existence check: table first, then column
+                const existenceCheck = await checkTableAndColumn(adapter, parsed.table, parsed.column, schemaName);
+                if (existenceCheck) {
+                    return { success: false, ...existenceCheck };
+                }
+                const indexSql = `
+                SELECT
+                    i.indexname,
+                    i.indexdef,
+                    pg_size_pretty(pg_relation_size((i.schemaname || '.' || i.indexname)::regclass)) as index_size,
+                    s.idx_scan,
+                    s.idx_tup_read
+                FROM pg_indexes i
+                LEFT JOIN pg_stat_user_indexes s ON s.indexrelname = i.indexname AND s.schemaname = i.schemaname
+                WHERE i.tablename = $1 AND i.schemaname = $2
+                AND i.indexdef LIKE '%vector%'
+            `;
+                const indexResult = await adapter.executeQuery(indexSql, [
+                    parsed.table,
+                    schemaName,
+                ]);
+                const statsSql = `
+                SELECT
+                    reltuples::bigint as estimated_rows,
+                    pg_size_pretty(pg_relation_size('${tableName}'::regclass)) as table_size
+                FROM pg_class c
+                JOIN pg_namespace n ON c.relnamespace = n.oid
+                WHERE c.relname = $1 AND n.nspname = $2
+            `;
+                const statsResult = await adapter.executeQuery(statsSql, [
+                    parsed.table,
+                    schemaName,
+                ]);
+                // PostgreSQL returns bigint as string, cast as needed
+                const stats = (statsResult.rows?.[0] ?? {});
+                let benchmark = null;
+                let testVectorSource;
+                let testVector = parsed.testVector;
+                // Auto-generate test vector from first row if not provided
+                if (testVector === undefined) {
+                    try {
+                        const sampleSql = `SELECT ${columnName}::text as vec FROM ${tableName} WHERE ${columnName} IS NOT NULL LIMIT 1`;
+                        const sampleResult = await adapter.executeQuery(sampleSql);
+                        const sampleRow = sampleResult.rows?.[0];
+                        if (sampleRow?.vec !== undefined) {
+                            // Parse vector string like "[0.1,0.2,0.3]" to array
+                            const vecStr = sampleRow.vec.replace(/[[\]]/g, "");
+                            testVector = vecStr.split(",").map(Number);
+                            testVectorSource = "auto-generated from first row";
+                        }
+                    }
+                    catch {
+                        // Silently ignore - benchmark just won't be available
+                    }
+                }
+                else {
+                    testVectorSource = "user-provided";
+                }
+                if (testVector !== undefined && testVector.length > 0) {
+                    const vectorStr = `[${testVector.join(",")}]`;
+                    const benchSql = `
+                    EXPLAIN ANALYZE
+                    SELECT * FROM ${tableName}
+                    ORDER BY ${columnName} <-> '${vectorStr}'::vector
+                    LIMIT 10
+                `;
+                    const benchResult = await adapter.executeQuery(benchSql);
+                    // Truncate large vectors in EXPLAIN output to reduce payload size
+                    // Pattern matches vector literals like '[0.1,0.2,...,0.9]'::vector
+                    const vectorPattern = /\[[\d.,\s-e]+\]'::vector/g;
+                    const truncatedRows = (benchResult.rows ?? []).map((row) => {
+                        const planLine = row["QUERY PLAN"];
+                        if (planLine && planLine.length > 200) {
+                            // Truncate long vector literals in query plan
+                            const truncated = planLine.replace(vectorPattern, `[...${String(testVector.length)} dims]'::vector`);
+                            return { "QUERY PLAN": truncated };
+                        }
+                        return row;
+                    });
+                    benchmark = truncatedRows;
+                }
+                // Convert PostgreSQL bigint strings to numbers for output schema compliance
+                const estimatedRows = Number(stats.estimated_rows ?? 0);
+                // Map indexes to convert bigint stats to numbers (idx_scan, idx_tup_read)
+                const indexes = (indexResult.rows ?? []).map((row) => ({
+                    ...row,
+                    idx_scan: row["idx_scan"] != null ? Number(row["idx_scan"]) : null,
+                    idx_tup_read: row["idx_tup_read"] != null ? Number(row["idx_tup_read"]) : null,
+                }));
+                const response = {
+                    table: parsed.table,
+                    column: parsed.column,
+                    tableSize: stats.table_size,
+                    // PostgreSQL returns -1 for tables that haven't been analyzed; normalize to 0
+                    estimatedRows: estimatedRows < 0 ? 0 : estimatedRows,
+                    indexes,
+                    benchmark,
+                    recommendations: (indexResult.rows?.length ?? 0) === 0
+                        ? [
+                            "No vector index found - consider creating one for better performance",
+                        ]
+                        : [],
+                };
+                if (testVectorSource !== undefined) {
+                    response["testVectorSource"] = testVectorSource;
+                }
+                if (benchmark === null) {
+                    response["hint"] =
+                        "No vectors in table to auto-generate test. Provide testVector param for benchmarking.";
+                }
+                return response;
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: formatPostgresError(error, { tool: "pg_vector_performance" }),
+                };
+            }
+        },
+    };
+}
+//# sourceMappingURL=search-advanced.js.map

package/dist/adapters/postgresql/tools/vector/search-advanced.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"search-advanced.js","sourceRoot":"","sources":["../../../../../src/adapters/postgresql/tools/vector/search-advanced.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,QAAQ,EAAE,MAAM,kCAAkC,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAC/D,OAAO,EACL,kBAAkB,EAClB,iBAAiB,GAClB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAChE,OAAO,EACL,yBAAyB,EACzB,wBAAwB,EACxB,6BAA6B,GAC9B,MAAM,wBAAwB,CAAC;AAEhC;;GAEG;AACH,SAAS,WAAW,CAAC,MAAe;IAClC,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC5C,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;QAC/C,OAAO,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,OAAwB;IAExB,kCAAkC;IAClC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;QACjC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;QACnD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;QAC5D,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC;QACvD,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACvD,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC;QACpD,QAAQ,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kCAAkC,CAAC;QACzE,UAAU,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;QACvE,UAAU,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;QACvE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;KAC5E,CAAC,CAAC;IAEH,MAAM,aAAa,GAAG,iBAAiB,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE;QACzD,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAY,CAAC;QAClD,MAAM,aAAa,GAAG,IAAI,CAAC,UAAqB,CAAC;QACjD,MAAM,aAAa,GAAG,IAAI,CAAC,UAAqB,CAAC;QACjD,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,SAAS,IAAI,EAAE;YACzC,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,GAAG,IAAI,EAAE;YACrC,CAAC,EAAE,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;YAC1C,UAAU,EAAE,aAAa,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS;YACrE,UAAU,EAAE,aAAa,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS;YACrE,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC;IACJ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,KAAK,SAAS,EAAE;QACxC,OAAO,EAAE,mCAAmC;KAC7C,CAAC,CAAC;IAEH,OAAO;QACL,IAAI,EAAE,mBAAmB;QACzB,WAAW,EACT,qLAAqL;QACvL,KAAK,EAAE,QAAQ;QACf,WAAW,EAAE,iBAAiB;QAC9B,YAAY,EAAE,yBAAyB;QACvC,WAAW,EAAE,QAAQ,CAAC,gBAAgB,CAAC;QACvC,KAAK,EAAE,YAAY,CAAC,QAAQ,EAAE,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QACzD,OAAO,EAAE,KAAK,EAAE,MAAe,EAAE,QAAwB,EAAE,EAAE;YAC3D,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC3C,wEAAwE;gBACxE,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC;gBACnB,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;oBACpB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;gBACvD,CAAC;gBACD,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;oBACb,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,yDAAyD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG;wBACnF,UAAU,EAAE,gDAAgD;qBAC7D,CAAC;gBACJ,CAAC;gBACD,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;oBACV,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,2CAA2C;wBAClD,UAAU,EAAE,4CAA4C;qBACzD,CAAC;gBACJ,CAAC;gBACD,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC;gBACxC,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,IAAI,KAAK,CAAC;gBAC1C,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,IAAI,QAAQ,CAAC;gBAC7C,MAAM,SAAS,GAAG,iBAAiB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;gBACjE,MAAM,UAAU,GAAG,kBAAkB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBAErD,qDAAqD;gBACrD,MAAM,cAAc,GAAG,MAAM,mBAAmB,CAC9C,OAAO,EACP,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,MAAM,EACb,UAAU,CACX,CAAC;gBACF,IAAI,cAAc,EAAE,CAAC;oBACnB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,cAAc,EAAE,CAAC;gBAC/C,CAAC;gBAED,4CAA4C;gBAC5C,MAAM,YAAY,GAAG;;;OAGtB,CAAC;gBACA,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,YAAY,EAAE;oBAC1D,UAAU;oBACV,MAAM,CAAC,KAAK;oBACZ,MAAM,CAAC,MAAM;iBACd,CAAC,CAAC;gBACH,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAEpC,CAAC;gBACd,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;oBACzB,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,WAAW,MAAM,CAAC,MAAM,mCAAmC,OAAO,IAAI,SAAS,GAAG;wBACzF,UAAU,EAAE,8CAA8C;qBAC3D,CAAC;gBACJ,CAAC;gBAED,MAAM,SAAS,GAAG;yBACD,UAAU;uBACZ,SAAS;wBACR,UAAU;;wBAEV,MAAM,CAAC,MAAM,CAAC;aACzB,CAAC;gBACN,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;gBAC3D,MAAM,OAAO,GAAG,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE,CAAsB,CAAC;gBAE/D,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACvB,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,iBAAiB,MAAM,CAAC,CAAC,CAAC,uBAAuB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,qCAAqC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,0BAA0B;wBACnK,CAAC,EAAE,CAAC;wBACJ,mBAAmB,EAAE,OAAO,CAAC,MAAM;wBACnC,UAAU,EAAE,MAAM;qBACnB,CAAC;gBACJ,CAAC;gBAED,MAAM,gBAAgB,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBAE/D,MAAM,UAAU,GAAG;;yDAE8B,UAAU;2BACxC,SAAS;4BACR,UAAU;4BACV,MAAM,CAAC,MAAM,CAAC;;;;;;;;;;;;;;;;aAgB7B,CAAC;gBAEN,IAAI,SAAS,GAAG,gBAAgB,CAAC;gBACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;oBACjC,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;wBACnE,SAAS,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CACjC,CAAC,CAA0B,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,CAAW,CAC5D,CAAC;oBACJ,CAAC;oBAAC,MAAM,CAAC;wBACP,MAAM;oBACR,CAAC;gBACH,CAAC;gBAED,uEAAuE;gBACvE,MAAM,eAAe,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;oBAC1C,MAAM,MAAM,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;oBAC9B,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;wBACpB,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;oBACvB,CAAC;oBACD,8DAA8D;oBAC9D,IAAI,MAAM,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;wBACvB,MAAM,SAAS,GAAG,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;wBAC7C,OAAO;4BACL,OAAO,EAAE,SAAS,CAAC,OAAO;4BAC1B,UAAU,EAAE,SAAS,CAAC,UAAU;4BAChC,SAAS,EAAE,SAAS,CAAC,SAAS;yBAC/B,CAAC;oBACJ,CAAC;oBACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;gBAC5B,CAAC,CAAC,CAAC;gBAEH,OAAO;oBACL,CAAC,EAAE,CAAC;oBACJ,UAAU,EAAE,OAAO;oBACnB,UAAU,EAAE,OAAO,CAAC,MAAM;oBAC1B,SAAS,EAAE,eAAe;oBAC1B,IAAI,EAAE,iEAAiE;iBACxE,CAAC;YACJ,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,OAAO;oBACL,OAAO,EAAE,KAAc;oBACvB,KAAK,EAAE,mBAAmB,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC;iBACjE,CAAC;YACJ,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,OAAwB;IAExB,kCAAkC;IAClC,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;QACtC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;QACnD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;QAC5D,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC;QAC7D,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;QACnE,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;QACtD,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC;QACpD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;QACnD,YAAY,EAAE,CAAC,CAAC,MAAM;aACnB,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CAAC,6CAA6C,CAAC;QAC1D,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC;QAC3D,MAAM,EAAE,CAAC;aACN,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;aACjB,QAAQ,EAAE;aACV,QAAQ,CAAC,oDAAoD,CAAC;KAClE,CAAC,CAAC;IAEH,MAAM,kBAAkB,GAAG,sBAAsB,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACrE,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,SAAS,IAAI,EAAE;QACzC,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,SAAS,IAAI,EAAE;QACvD,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,YAAY,EAAE,IAAI,CAAC,YAAY;QAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAC,CAAC,CAAC;IAEJ,OAAO;QACL,IAAI,EAAE,kBAAkB;QACxB,WAAW,EACT,wEAAwE;QAC1E,KAAK,EAAE,QAAQ;QACf,WAAW,EAAE,sBAAsB;QACnC,YAAY,EAAE,wBAAwB;QACtC,WAAW,EAAE,QAAQ,CAAC,eAAe,CAAC;QACtC,KAAK,EAAE,YAAY,CAAC,QAAQ,EAAE,QAAQ,CAAC,eAAe,CAAC,CAAC;QACxD,OAAO,EAAE,KAAK,EAAE,MAAe,EAAE,QAAwB,EAAE,EAAE;YAC3D,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEhD,iDAAiD;gBACjD,IAAI,MAAM,CAAC,KAAK,KAAK,EAAE,EAAE,CAAC;oBACxB,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,4CAA4C;wBACnD,cAAc,EAAE;4BACd,OAAO;4BACP,cAAc;4BACd,YAAY;4BACZ,QAAQ;4BACR,WAAW;yBACZ;qBACF,CAAC;gBACJ,CAAC;gBACD,IAAI,MAAM,CAAC,YAAY,KAAK,EAAE,EAAE,CAAC;oBAC/B,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,mDAAmD;wBAC1D,cAAc,EAAE;4BACd,OAAO;4BACP,cAAc;4BACd,YAAY;4BACZ,QAAQ;4BACR,WAAW;yBACZ;qBACF,CAAC;gBACJ,CAAC;gBAED,6DAA6D;gBAC7D,IAAI,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC;gBACjC,IAAI,cAAkC,CAAC;gBACvC,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC/B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;oBACtC,cAAc,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;oBAC1B,aAAa,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC;gBAC3C,CAAC;gBACD,MAAM,UAAU,GAAG,cAAc,IAAI,QAAQ,CAAC;gBAC9C,MAAM,SAAS,GAAG,iBAAiB,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC;gBAE/D,4DAA4D;gBAC5D,MAAM,cAAc,GAAG,MAAM,mBAAmB,CAC9C,OAAO,EACP,aAAa,EACb,MAAM,CAAC,YAAY,EACnB,UAAU,CACX,CAAC;gBACF,IAAI,cAAc,KAAK,IAAI,EAAE,CAAC;oBAC5B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,cAAc,EAAE,CAAC;gBAC/C,CAAC;gBAED,gDAAgD;gBAChD,MAAM,UAAU,GAAG;;;;aAId,CAAC;gBACN,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,UAAU,EAAE;oBAC3D,UAAU;oBACV,aAAa;oBACb,MAAM,CAAC,YAAY;iBACpB,CAAC,CAAC;gBACH,MAAM,OAAO,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC,CAEzB,CAAC;gBAEd,IACE,OAAO,EAAE,QAAQ,KAAK,UAAU;oBAChC,OAAO,EAAE,SAAS,KAAK,UAAU,EACjC,CAAC;oBACD,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,WAAW,MAAM,CAAC,YAAY,uIAAuI;wBAC5K,UAAU,EAAE,yFAAyF;qBACtG,CAAC;gBACJ,CAAC;gBAED,IAAI,OAAO,EAAE,QAAQ,KAAK,QAAQ,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;oBAC5D,MAAM,UAAU,GAAG,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,SAAS,IAAI,SAAS,CAAC;oBACtE,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,WAAW,MAAM,CAAC,YAAY,eAAe,UAAU,4DAA4D;wBAC1H,UAAU,EAAE,UAAU;qBACvB,CAAC;gBACJ,CAAC;gBAED,uEAAuE;gBACvE,MAAM,iBAAiB,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,UAAU,EAAE;oBAC/D,UAAU;oBACV,aAAa;oBACb,MAAM,CAAC,UAAU;iBAClB,CAAC,CAAC;gBACH,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,EAAE,CAAC,CAAC,CAEjC,CAAC;gBACd,MAAM,oBAAoB,GACxB,WAAW,EAAE,QAAQ,KAAK,UAAU;oBACpC,WAAW,EAAE,SAAS,KAAK,UAAU,CAAC;gBAExC,kEAAkE;gBAClE,MAAM,QAAQ,GAAG,oBAAoB;oBACnC,CAAC,CAAC,IAAI,MAAM,CAAC,UAAU,GAAG;oBAC1B,CAAC,CAAC,2BAA2B,MAAM,CAAC,UAAU,IAAI,CAAC;gBAErD,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,GAAG,CAAC;gBAChD,kEAAkE;gBAClE,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,YAAY,CAAC,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC;gBAChE,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;gBACpC,MAAM,SAAS,GAAG,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;gBAEjD,oFAAoF;gBACpF,IAAI,UAAkB,CAAC;gBACvB,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC5D,2CAA2C;oBAC3C,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC/D,CAAC;qBAAM,CAAC;oBACN,6DAA6D;oBAC7D,MAAM,OAAO,GAAG;;;;;;iBAMT,CAAC;oBACR,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,OAAO,EAAE;wBACrD,UAAU;wBACV,aAAa;qBACd,CAAC,CAAC;oBACH,MAAM,IAAI,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CACtC,CAAC,CAA0B,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAW,CAC3D,CAAC;oBACF,UAAU;wBACR,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;gBACrE,CAAC;gBAED,MAAM,GAAG,GAAG;;;;gCAIY,MAAM,CAAC,YAAY,UAAU,SAAS;2BAC3C,SAAS;6BACP,MAAM,CAAC,YAAY;gCAChB,MAAM,CAAC,YAAY,UAAU,SAAS;4BAC1C,MAAM,CAAC,QAAQ,GAAG,CAAC,CAAC;;;;;kCAKd,QAAQ;2BACf,SAAS;4BACR,QAAQ;;;sBAGd,UAAU;oDACoB,MAAM,CAAC,YAAY,CAAC;mDACrB,MAAM,CAAC,UAAU,CAAC;;;uBAG9C,SAAS;;;;;wBAKR,MAAM,CAAC,QAAQ,CAAC;aAC3B,CAAC;gBAEN,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;oBACnE,OAAO;wBACL,OAAO,EAAE,MAAM,CAAC,IAAI;wBACpB,KAAK,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC;wBAC/B,YAAY;wBACZ,UAAU;qBACX,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAc,EAAE,CAAC;oBACxB,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;wBAC3B,gCAAgC;wBAChC,MAAM,QAAQ,GAAG,iCAAiC,CAAC,IAAI,CACrD,KAAK,CAAC,OAAO,CACd,CAAC;wBACF,IAAI,QAAQ,EAAE,CAAC;4BACb,MAAM,UAAU,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;4BACrC,0CAA0C;4BAC1C,IAAI,SAAS,GAAG,QAAQ,CAAC;4BACzB,IAAI,UAAU,KAAK,MAAM,CAAC,UAAU,EAAE,CAAC;gCACrC,SAAS,GAAG,YAAY,CAAC;4BAC3B,CAAC;iCAAM,IAAI,UAAU,KAAK,MAAM,CAAC,YAAY,EAAE,CAAC;gCAC9C,SAAS,GAAG,cAAc,CAAC;4BAC7B,CAAC;4BACD,OAAO;gCACL,OAAO,EAAE,KAAK;gCACd,KAAK,EAAE,WAAW,UAAU,8BAA8B,aAAa,GAAG;gCAC1E,kBAAkB,EAAE,SAAS;gCAC7B,UAAU,EAAE,iDAAiD;6BAC9D,CAAC;wBACJ,CAAC;wBAED,kCAAkC;wBAClC,MAAM,QAAQ,GAAG,6CAA6C,CAAC,IAAI,CACjE,KAAK,CAAC,OAAO,CACd,CAAC;wBACF,IAAI,QAAQ,EAAE,CAAC;4BACb,MAAM,WAAW,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC;4BACvC,MAAM,WAAW,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC;4BACvC,OAAO;gCACL,OAAO,EAAE,KAAK;gCACd,KAAK,EAAE,6CAA6C,WAAW,iCAAiC,WAAW,cAAc;gCACzH,kBAAkB,EAAE,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC;gCAC7C,kBAAkB,EAAE,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC;gCAC7C,UAAU,EACR,iEAAiE;6BACpE,CAAC;wBACJ,CAAC;wBAED,kCAAkC;wBAClC,MAAM,aAAa,GAAG,mCAAmC,CAAC,IAAI,CAC5D,KAAK,CAAC,OAAO,CACd,CAAC;wBACF,IAAI,aAAa,EAAE,CAAC;4BAClB,MAAM,eAAe,GAAG,aAAa,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;4BAC/C,OAAO;gCACL,OAAO,EAAE,KAAK;gCACd,KAAK,EAAE,UAAU,eAAe,+BAA+B,UAAU,GAAG;gCAC5E,UAAU,EAAE,6CAA6C;6BAC1D,CAAC;wBACJ,CAAC;wBAED,wEAAwE;wBACxE,OAAO;4BACL,OAAO,EAAE,KAAK;4BACd,KAAK,EAAE,KAAK,CAAC,OAAO;4BACpB,UAAU,EAAE,iDAAiD;yBAC9D,CAAC;oBACJ,CAAC;oBACD,iDAAiD;oBACjD,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,8BAA8B;wBACrC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC;qBACvB,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,OAAO;oBACL,OAAO,EAAE,KAAc;oBACvB,KAAK,EAAE,mBAAmB,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;iBAChE,CAAC;YACJ,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,OAAwB;IAExB,kCAAkC;IAClC,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;QACrC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC;QACnD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;QAC5D,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC;QACvD,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACvD,UAAU,EAAE,CAAC;aACV,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;aACjB,QAAQ,EAAE;aACV,QAAQ,CAAC,8BAA8B,CAAC;QAC3C,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;KAC5E,CAAC,CAAC;IAEH,MAAM,iBAAiB,GAAG,qBAAqB,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACnE,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,SAAS,IAAI,EAAE;QACzC,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,GAAG,IAAI,EAAE;QACrC,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAC,CAAC,CAAC;IAEJ,OAAO;QACL,IAAI,EAAE,uBAAuB;QAC7B,WAAW,EACT,+GAA+G;QACjH,KAAK,EAAE,QAAQ;QACf,WAAW,EAAE,qBAAqB;QAClC,YAAY,EAAE,6BAA6B;QAC3C,WAAW,EAAE,QAAQ,CAAC,oBAAoB,CAAC;QAC3C,KAAK,EAAE,YAAY,CAAC,QAAQ,EAAE,QAAQ,CAAC,oBAAoB,CAAC,CAAC;QAC7D,OAAO,EAAE,KAAK,EAAE,MAAe,EAAE,QAAwB,EAAE,EAAE;YAC3D,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAE/C,2BAA2B;gBAC3B,IAAI,MAAM,CAAC,KAAK,KAAK,EAAE,EAAE,CAAC;oBACxB,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,4CAA4C;wBACnD,cAAc,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC;qBACpC,CAAC;gBACJ,CAAC;gBACD,IAAI,MAAM,CAAC,MAAM,KAAK,EAAE,EAAE,CAAC;oBACzB,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EACH,kEAAkE;wBACpE,cAAc,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC;qBACpC,CAAC;gBACJ,CAAC;gBAED,MAAM,SAAS,GAAG,iBAAiB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;gBACjE,MAAM,UAAU,GAAG,kBAAkB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACrD,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,IAAI,QAAQ,CAAC;gBAE7C,qDAAqD;gBACrD,MAAM,cAAc,GAAG,MAAM,mBAAmB,CAC9C,OAAO,EACP,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,MAAM,EACb,UAAU,CACX,CAAC;gBACF,IAAI,cAAc,EAAE,CAAC;oBACnB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,cAAc,EAAE,CAAC;gBAC/C,CAAC;gBAED,MAAM,QAAQ,GAAG;;;;;;;;;;;aAWZ,CAAC;gBACN,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,QAAQ,EAAE;oBACvD,MAAM,CAAC,KAAK;oBACZ,UAAU;iBACX,CAAC,CAAC;gBAEH,MAAM,QAAQ,GAAG;;;uDAG8B,SAAS;;;;aAInD,CAAC;gBACN,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,QAAQ,EAAE;oBACvD,MAAM,CAAC,KAAK;oBACZ,UAAU;iBACX,CAAC,CAAC;gBACH,sDAAsD;gBACtD,MAAM,KAAK,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAGzC,CAAC;gBAEF,IAAI,SAAS,GAAG,IAAI,CAAC;gBACrB,IAAI,gBAAoC,CAAC;gBACzC,IAAI,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;gBAEnC,2DAA2D;gBAC3D,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;oBAC7B,IAAI,CAAC;wBACH,MAAM,SAAS,GAAG,UAAU,UAAU,sBAAsB,SAAS,UAAU,UAAU,sBAAsB,CAAC;wBAChH,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;wBAC3D,MAAM,SAAS,GAAG,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC,CAE1B,CAAC;wBACd,IAAI,SAAS,EAAE,GAAG,KAAK,SAAS,EAAE,CAAC;4BACjC,oDAAoD;4BACpD,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;4BACnD,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;4BAC3C,gBAAgB,GAAG,+BAA+B,CAAC;wBACrD,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,sDAAsD;oBACxD,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,gBAAgB,GAAG,eAAe,CAAC;gBACrC,CAAC;gBAED,IAAI,UAAU,KAAK,SAAS,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACtD,MAAM,SAAS,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;oBAC9C,MAAM,QAAQ,GAAG;;oCAES,SAAS;+BACd,UAAU,SAAS,SAAS;;iBAE1C,CAAC;oBACR,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;oBAEzD,kEAAkE;oBAClE,mEAAmE;oBACnE,MAAM,aAAa,GAAG,2BAA2B,CAAC;oBAClD,MAAM,aAAa,GAAG,CAAC,WAAW,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAChD,CAAC,GAA4B,EAAE,EAAE;wBAC/B,MAAM,QAAQ,GAAG,GAAG,CAAC,YAAY,CAAuB,CAAC;wBACzD,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;4BACtC,8CAA8C;4BAC9C,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAChC,aAAa,EACb,OAAO,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAClD,CAAC;4BACF,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC;wBACrC,CAAC;wBACD,OAAO,GAAG,CAAC;oBACb,CAAC,CACF,CAAC;oBACF,SAAS,GAAG,aAAa,CAAC;gBAC5B,CAAC;gBAED,4EAA4E;gBAC5E,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC,cAAc,IAAI,CAAC,CAAC,CAAC;gBACxD,0EAA0E;gBAC1E,MAAM,OAAO,GAAG,CAAC,WAAW,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAC1C,CAAC,GAA4B,EAAE,EAAE,CAAC,CAAC;oBACjC,GAAG,GAAG;oBACN,QAAQ,EAAE,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI;oBAClE,YAAY,EACV,GAAG,CAAC,cAAc,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI;iBACnE,CAAC,CACH,CAAC;gBAEF,MAAM,QAAQ,GAA4B;oBACxC,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,SAAS,EAAE,KAAK,CAAC,UAAU;oBAC3B,8EAA8E;oBAC9E,aAAa,EAAE,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa;oBACpD,OAAO;oBACP,SAAS;oBACT,eAAe,EACb,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC,CAAC,KAAK,CAAC;wBACnC,CAAC,CAAC;4BACE,sEAAsE;yBACvE;wBACH,CAAC,CAAC,EAAE;iBACT,CAAC;gBAEF,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;oBACnC,QAAQ,CAAC,kBAAkB,CAAC,GAAG,gBAAgB,CAAC;gBAClD,CAAC;gBACD,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;oBACvB,QAAQ,CAAC,MAAM,CAAC;wBACd,uFAAuF,CAAC;gBAC5F,CAAC;gBAED,OAAO,QAAQ,CAAC;YAClB,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,OAAO;oBACL,OAAO,EAAE,KAAc;oBACvB,KAAK,EAAE,mBAAmB,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,uBAAuB,EAAE,CAAC;iBACrE,CAAC;YACJ,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC"}