@nicia-ai/typegraph 0.21.0 → 0.22.0

package/dist/{chunk-UQVBVBVG.js → chunk-5ZMMHUWR.js}

@@ -1,6 +1,6 @@
 import { DatabaseOperationError, UniquenessError } from './chunk-WFMTAEEN.js';
-import { buildPostgresNodeIndexBuilders, buildPostgresEdgeIndexBuilders, buildSqliteNodeIndexBuilders, buildSqliteEdgeIndexBuilders } from './chunk-SMMOAVJ5.js';
-import { tsvectorStrategy, fts5Strategy, MAX_PG_IDENTIFIER_LENGTH, quotedTableName, quotedColumn, edgeColumnList, sqlNull, nodeColumnList } from './chunk-4XNE2JG3.js';
+import { buildPostgresNodeIndexBuilders, buildPostgresEdgeIndexBuilders, buildSqliteNodeIndexBuilders, buildSqliteEdgeIndexBuilders } from './chunk-3PPEHLGK.js';
+import { tsvectorStrategy, fts5Strategy, MAX_PG_IDENTIFIER_LENGTH, quotedTableName, quotedColumn, edgeColumnList, sqlNull, nodeColumnList } from './chunk-753F33CV.js';
 import { customType, pgTable, timestamp, integer, jsonb, text, primaryKey, index, boolean, getTableConfig as getTableConfig$1 } from 'drizzle-orm/pg-core';
 import { sqliteTable, text as text$1, integer as integer$1, primaryKey as primaryKey$1, index as index$1, blob, getTableConfig } from 'drizzle-orm/sqlite-core';
 import { sql, getTableName } from 'drizzle-orm';
@@ -503,8 +503,17 @@ function flattenSqlChunk(chunk) {
     return chunk;
   }
   if (typeof chunk === "object" && chunk !== null) {
-    if ("value" in chunk && Array.isArray(chunk.value)) {
-      return chunk.value.map((part) => flattenSqlChunk(part)).join("");
+    if ("name" in chunk && typeof chunk.name === "string") {
+      return `"${chunk.name}"`;
+    }
+    if ("value" in chunk) {
+      const value = chunk.value;
+      if (typeof value === "string") {
+        return value;
+      }
+      if (Array.isArray(value)) {
+        return value.map((part) => flattenSqlChunk(part)).join("");
+      }
     }
     if ("queryChunks" in chunk && Array.isArray(chunk.queryChunks)) {
       return chunk.queryChunks.map((part) => flattenSqlChunk(part)).join("");
@@ -1835,6 +1844,30 @@ function buildUpsertEmbeddingPostgres(tables3, params, timestamp2) {
       ${column(embeddings3.updatedAt)} = ${timestamp2}
   `;
 }
+function buildUpsertEmbeddingSqlite(tables3, params, timestamp2) {
+  const { embeddings: embeddings3 } = tables3;
+  const embeddingJson = JSON.stringify(params.embedding);
+  assertFiniteNumberArray(params.embedding, "embedding");
+  const columns = sql.raw(
+    `"${embeddings3.graphId.name}", "${embeddings3.nodeKind.name}", "${embeddings3.nodeId.name}", "${embeddings3.fieldPath.name}", "${embeddings3.embedding.name}", "${embeddings3.dimensions.name}", "${embeddings3.createdAt.name}", "${embeddings3.updatedAt.name}"`
+  );
+  const conflictColumns = sql.raw(
+    `"${embeddings3.graphId.name}", "${embeddings3.nodeKind.name}", "${embeddings3.nodeId.name}", "${embeddings3.fieldPath.name}"`
+  );
+  const column = (target) => sql.raw(`"${target.name}"`);
+  return sql`
+    INSERT INTO ${embeddings3} (${columns})
+    VALUES (
+      ${params.graphId}, ${params.nodeKind}, ${params.nodeId}, ${params.fieldPath},
+      vec_f32(${embeddingJson}), ${params.dimensions}, ${timestamp2}, ${timestamp2}
+    )
+    ON CONFLICT (${conflictColumns})
+    DO UPDATE SET
+      ${column(embeddings3.embedding)} = vec_f32(${embeddingJson}),
+      ${column(embeddings3.dimensions)} = ${params.dimensions},
+      ${column(embeddings3.updatedAt)} = ${timestamp2}
+  `;
+}
 function buildDeleteEmbedding(tables3, params) {
   const { embeddings: embeddings3 } = tables3;
   return sql`
@@ -2037,7 +2070,20 @@ function createCommonOperationStrategy(tables3, dialect, fulltextStrategy) {
   };
 }
 function createSqliteOperationStrategy(tables3, fulltextStrategy) {
-  return createCommonOperationStrategy(tables3, "sqlite", fulltextStrategy);
+  const common = createCommonOperationStrategy(
+    tables3,
+    "sqlite",
+    fulltextStrategy
+  );
+  return {
+    ...common,
+    buildUpsertEmbedding(params, timestamp2) {
+      return buildUpsertEmbeddingSqlite(tables3, params, timestamp2);
+    },
+    buildDeleteEmbedding(params) {
+      return buildDeleteEmbedding(tables3, params);
+    }
+  };
 }
 function createPostgresOperationStrategy(tables3, fulltextStrategy) {
   const common = createCommonOperationStrategy(
@@ -2248,5 +2294,5 @@ function compileQueryWithDialect(db, query, backendName) {
 }
 
 export { POSTGRES_CAPABILITIES, POSTGRES_ROW_MAPPER_CONFIG, SQLITE_CAPABILITIES, 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, nodes, nodes2, nowIso, schemaVersions, schemaVersions2, tables, tables2, uniques, uniques2, wrapWithManagedClose };
-//# sourceMappingURL=chunk-UQVBVBVG.js.map
-//# sourceMappingURL=chunk-UQVBVBVG.js.map
+//# sourceMappingURL=chunk-5ZMMHUWR.js.map
+//# sourceMappingURL=chunk-5ZMMHUWR.js.map