@nicia-ai/typegraph 0.23.0 → 0.24.0

package/dist/{chunk-H2CAJ4ZV.js → chunk-EY4LBX3F.js}

@@ -1903,9 +1903,7 @@ function buildVectorSearchMinScoreCondition(distanceExpression, metric, minScore
     }
   }
 }
-function buildVectorSearchPostgres(tables3, params) {
-  const { embeddings: embeddings3 } = tables3;
-  const queryLiteral = formatEmbeddingLiteral(params.queryEmbedding);
+function assertVectorSearchBounds(params) {
   if (params.minScore !== void 0 && !Number.isFinite(params.minScore)) {
     throw new TypeError(
       `minScore must be a finite number, got: ${params.minScore}`
@@ -1914,6 +1912,11 @@ function buildVectorSearchPostgres(tables3, params) {
   if (!Number.isInteger(params.limit) || params.limit <= 0) {
     throw new Error(`limit must be a positive integer, got: ${params.limit}`);
   }
+}
+function buildVectorSearchPostgres(tables3, params) {
+  const { embeddings: embeddings3 } = tables3;
+  const queryLiteral = formatEmbeddingLiteral(params.queryEmbedding);
+  assertVectorSearchBounds(params);
   const embeddingColumn = sql`${embeddings3.embedding}`;
   const distanceExpression = buildDistanceExpression(
     embeddingColumn,
@@ -1949,6 +1952,65 @@ function buildVectorSearchPostgres(tables3, params) {
     LIMIT ${params.limit}
   `;
 }
+function buildSqliteDistanceExpression(embeddingColumn, embeddingJson, metric) {
+  switch (metric) {
+    case "cosine": {
+      return sql`vec_distance_cosine(${embeddingColumn}, vec_f32(${embeddingJson}))`;
+    }
+    case "l2": {
+      return sql`vec_distance_l2(${embeddingColumn}, vec_f32(${embeddingJson}))`;
+    }
+    case "inner_product": {
+      throw new Error(
+        "Inner product distance is not supported by sqlite-vec. Use 'cosine' or 'l2' metrics instead."
+      );
+    }
+    default: {
+      const _exhaustive = metric;
+      throw new Error(`Unsupported vector metric: ${String(_exhaustive)}`);
+    }
+  }
+}
+function buildVectorSearchSqlite(tables3, params) {
+  const { embeddings: embeddings3 } = tables3;
+  assertFiniteNumberArray(params.queryEmbedding, "queryEmbedding");
+  const embeddingJson = JSON.stringify(params.queryEmbedding);
+  assertVectorSearchBounds(params);
+  const embeddingColumn = sql`${embeddings3.embedding}`;
+  const distanceExpression = buildSqliteDistanceExpression(
+    embeddingColumn,
+    embeddingJson,
+    params.metric
+  );
+  const conditions = [
+    sql`${embeddings3.graphId} = ${params.graphId}`,
+    sql`${embeddings3.nodeKind} = ${params.nodeKind}`,
+    sql`${embeddings3.fieldPath} = ${params.fieldPath}`
+  ];
+  if (params.minScore !== void 0) {
+    conditions.push(
+      buildVectorSearchMinScoreCondition(
+        distanceExpression,
+        params.metric,
+        params.minScore
+      )
+    );
+  }
+  const whereClause = sql.join(conditions, sql` AND `);
+  const scoreExpression = buildVectorSearchScoreExpression(
+    distanceExpression,
+    params.metric
+  );
+  return sql`
+    SELECT
+      ${embeddings3.nodeId} as node_id,
+      ${scoreExpression} as score
+    FROM ${embeddings3}
+    WHERE ${whereClause}
+    ORDER BY ${distanceExpression} ASC
+    LIMIT ${params.limit}
+  `;
+}
 
 // src/backend/drizzle/operations/strategy.ts
 function bindTableOperationBuilders(tables3, builders) {
@@ -2045,6 +2107,9 @@ function createSqliteOperationStrategy(tables3, fulltextStrategy) {
     },
     buildDeleteEmbedding(params) {
       return buildDeleteEmbedding(tables3, params);
+    },
+    buildVectorSearch(params) {
+      return buildVectorSearchSqlite(tables3, params);
     }
   };
 }
@@ -2274,5 +2339,5 @@ function compileQueryWithDialect(db, query, backendName) {
 }
 
 export { POSTGRES_ROW_MAPPER_CONFIG, SQLITE_ROW_MAPPER_CONFIG, compileQueryWithDialect, createCommonOperationBackend, createEdgeRowMapper, createNodeRowMapper, createPostgresOperationStrategy, createPostgresTables, createPostgresVectorIndex, createSchemaVersionRowMapper, createSqliteOperationStrategy, createSqliteTables, createSqliteVectorIndex, createUniqueRowMapper, dropPostgresVectorIndex, dropSqliteVectorIndex, edges, edges2, embeddings, embeddings2, formatPostgresTimestamp, generatePostgresDDL, generatePostgresMigrationSQL, generateSqliteDDL, generateSqliteMigrationSQL, generateVectorIndexName, getOrCreateLru, nodes, nodes2, nowIso, schemaVersions, schemaVersions2, tables, tables2, uniques, uniques2 };
-//# sourceMappingURL=chunk-H2CAJ4ZV.js.map
-//# sourceMappingURL=chunk-H2CAJ4ZV.js.map
+//# sourceMappingURL=chunk-EY4LBX3F.js.map
+//# sourceMappingURL=chunk-EY4LBX3F.js.map