@utaba/deep-memory-storage-sqlserver 0.5.0 → 0.7.0

package/dist/index.cjs

@@ -878,18 +878,20 @@ var SqlServerStorageProvider = class {
     }
     const updated = {
       ...existing,
+      entityType: updates.entityType ?? existing.entityType,
       label: updates.label ?? existing.label,
       slug: updates.slug ?? existing.slug,
-      summary: updates.summary !== void 0 ? updates.summary : existing.summary,
+      summary: updates.summary === void 0 ? existing.summary : updates.summary ?? void 0,
       properties: updates.properties ?? existing.properties,
-      data: updates.data !== void 0 ? updates.data : existing.data,
-      dataFormat: updates.dataFormat !== void 0 ? updates.dataFormat : existing.dataFormat,
+      data: updates.data === void 0 ? existing.data : updates.data ?? void 0,
+      dataFormat: updates.dataFormat === void 0 ? existing.dataFormat : updates.dataFormat ?? void 0,
       provenance: updates.provenance,
       embedding: updates.embedding ?? existing.embedding
     };
-    const req = pool.request().input("repoId", import_mssql.default.UniqueIdentifier, repositoryId).input("entityId", import_mssql.default.NVarChar, entityId).input("slug", import_mssql.default.NVarChar, updated.slug).input("label", import_mssql.default.NVarChar, updated.label).input("summary", import_mssql.default.NVarChar, updated.summary ?? null).input("properties", import_mssql.default.NVarChar, JSON.stringify(updated.properties)).input("data", import_mssql.default.NVarChar, updated.data ?? null).input("dataFormat", import_mssql.default.NVarChar, updated.dataFormat ?? null).input("embedding", import_mssql.default.NVarChar, updated.embedding ? JSON.stringify(updated.embedding) : null).input("modifiedBy", import_mssql.default.NVarChar, updates.provenance.modifiedBy).input("modifiedByType", import_mssql.default.NVarChar, updates.provenance.modifiedByType).input("modifiedAt", import_mssql.default.NVarChar, updates.provenance.modifiedAt).input("modifiedInConversation", import_mssql.default.NVarChar, updates.provenance.modifiedInConversation ?? null).input("modifiedFromMessage", import_mssql.default.NVarChar, updates.provenance.modifiedFromMessage ?? null);
+    const req = pool.request().input("repoId", import_mssql.default.UniqueIdentifier, repositoryId).input("entityId", import_mssql.default.NVarChar, entityId).input("entityType", import_mssql.default.NVarChar, updated.entityType).input("slug", import_mssql.default.NVarChar, updated.slug).input("label", import_mssql.default.NVarChar, updated.label).input("summary", import_mssql.default.NVarChar, updated.summary ?? null).input("properties", import_mssql.default.NVarChar, JSON.stringify(updated.properties)).input("data", import_mssql.default.NVarChar, updated.data ?? null).input("dataFormat", import_mssql.default.NVarChar, updated.dataFormat ?? null).input("embedding", import_mssql.default.NVarChar, updated.embedding ? JSON.stringify(updated.embedding) : null).input("modifiedBy", import_mssql.default.NVarChar, updates.provenance.modifiedBy).input("modifiedByType", import_mssql.default.NVarChar, updates.provenance.modifiedByType).input("modifiedAt", import_mssql.default.NVarChar, updates.provenance.modifiedAt).input("modifiedInConversation", import_mssql.default.NVarChar, updates.provenance.modifiedInConversation ?? null).input("modifiedFromMessage", import_mssql.default.NVarChar, updates.provenance.modifiedFromMessage ?? null);
     await req.query(`
       UPDATE ${this.t("dm_entities")} SET
+        [entity_type] = @entityType,
         [slug] = @slug,
         [label] = @label,
         [summary] = @summary,
@@ -923,6 +925,28 @@ var SqlServerStorageProvider = class {
       throw new import_deep_memory.EntityNotFoundError(entityId);
     }
   }
+  async deleteEntities(repositoryId, ids) {
+    if (ids.length === 0) return { deleted: [], notFound: [] };
+    const pool = this.getPool();
+    const tvp1 = this.createIdListTvp(ids);
+    const cascadeReq = pool.request().input("repoId", import_mssql.default.UniqueIdentifier, repositoryId).input("entityIds", tvp1);
+    await cascadeReq.query(`
+      DELETE FROM ${this.t("dm_relationships")}
+        WHERE [repository_id] = @repoId AND [source_entity_id] IN (SELECT [id] FROM @entityIds);
+      DELETE FROM ${this.t("dm_relationships")}
+        WHERE [repository_id] = @repoId AND [target_entity_id] IN (SELECT [id] FROM @entityIds);
+    `);
+    const tvp2 = this.createIdListTvp(ids);
+    const result = await pool.request().input("repoId", import_mssql.default.UniqueIdentifier, repositoryId).input("entityIds", tvp2).query(
+      `DELETE FROM ${this.t("dm_entities")}
+         OUTPUT DELETED.[entity_id]
+         WHERE [repository_id] = @repoId
+           AND [entity_id] IN (SELECT [id] FROM @entityIds)`
+    );
+    const deleted = result.recordset.map((row) => row.entity_id);
+    const deletedSet = new Set(deleted);
+    return { deleted, notFound: ids.filter((id) => !deletedSet.has(id)) };
+  }
   async deleteEntitiesByType(repositoryId, entityType) {
     await this.assertRepository(repositoryId);
     const pool = this.getPool();
@@ -1148,6 +1172,20 @@ var SqlServerStorageProvider = class {
       throw new import_deep_memory.RelationshipNotFoundError(relationshipId);
     }
   }
+  async deleteRelationships(repositoryId, ids) {
+    if (ids.length === 0) return { deleted: [], notFound: [] };
+    const pool = this.getPool();
+    const tvp = this.createIdListTvp(ids);
+    const result = await pool.request().input("repoId", import_mssql.default.UniqueIdentifier, repositoryId).input("relIds", tvp).query(
+      `DELETE FROM ${this.t("dm_relationships")}
+         OUTPUT DELETED.[relationship_id]
+         WHERE [repository_id] = @repoId
+           AND [relationship_id] IN (SELECT [id] FROM @relIds)`
+    );
+    const deleted = result.recordset.map((row) => row.relationship_id);
+    const deletedSet = new Set(deleted);
+    return { deleted, notFound: ids.filter((id) => !deletedSet.has(id)) };
+  }
   async deleteRelationshipsByType(repositoryId, relationshipType) {
     await this.assertRepository(repositoryId);
     const pool = this.getPool();