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

package/dist/index.d.cts

@@ -48,6 +48,10 @@ declare class SqlServerStorageProvider implements StorageProvider {
     getEntities(repositoryId: string, entityIds: string[]): Promise<Map<string, StoredEntity>>;
     updateEntity(repositoryId: string, entityId: string, updates: StoredEntityUpdate): Promise<StoredEntity>;
     deleteEntity(repositoryId: string, entityId: string): Promise<void>;
+    deleteEntities(repositoryId: string, ids: string[]): Promise<{
+        deleted: string[];
+        notFound: string[];
+    }>;
     deleteEntitiesByType(repositoryId: string, entityType: string): Promise<{
         deletedEntities: number;
         deletedRelationships: number;
@@ -62,6 +66,10 @@ declare class SqlServerStorageProvider implements StorageProvider {
      */
     private buildRelationshipUnion;
     deleteRelationship(repositoryId: string, relationshipId: string): Promise<void>;
+    deleteRelationships(repositoryId: string, ids: string[]): Promise<{
+        deleted: string[];
+        notFound: string[];
+    }>;
     deleteRelationshipsByType(repositoryId: string, relationshipType: string): Promise<{
         deletedRelationships: number;
     }>;

package/dist/index.d.ts

@@ -48,6 +48,10 @@ declare class SqlServerStorageProvider implements StorageProvider {
     getEntities(repositoryId: string, entityIds: string[]): Promise<Map<string, StoredEntity>>;
     updateEntity(repositoryId: string, entityId: string, updates: StoredEntityUpdate): Promise<StoredEntity>;
     deleteEntity(repositoryId: string, entityId: string): Promise<void>;
+    deleteEntities(repositoryId: string, ids: string[]): Promise<{
+        deleted: string[];
+        notFound: string[];
+    }>;
     deleteEntitiesByType(repositoryId: string, entityType: string): Promise<{
         deletedEntities: number;
         deletedRelationships: number;
@@ -62,6 +66,10 @@ declare class SqlServerStorageProvider implements StorageProvider {
      */
     private buildRelationshipUnion;
     deleteRelationship(repositoryId: string, relationshipId: string): Promise<void>;
+    deleteRelationships(repositoryId: string, ids: string[]): Promise<{
+        deleted: string[];
+        notFound: string[];
+    }>;
     deleteRelationshipsByType(repositoryId: string, relationshipType: string): Promise<{
         deletedRelationships: number;
     }>;

package/dist/index.js

@@ -847,18 +847,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", sql.UniqueIdentifier, repositoryId).input("entityId", sql.NVarChar, entityId).input("slug", sql.NVarChar, updated.slug).input("label", sql.NVarChar, updated.label).input("summary", sql.NVarChar, updated.summary ?? null).input("properties", sql.NVarChar, JSON.stringify(updated.properties)).input("data", sql.NVarChar, updated.data ?? null).input("dataFormat", sql.NVarChar, updated.dataFormat ?? null).input("embedding", sql.NVarChar, updated.embedding ? JSON.stringify(updated.embedding) : null).input("modifiedBy", sql.NVarChar, updates.provenance.modifiedBy).input("modifiedByType", sql.NVarChar, updates.provenance.modifiedByType).input("modifiedAt", sql.NVarChar, updates.provenance.modifiedAt).input("modifiedInConversation", sql.NVarChar, updates.provenance.modifiedInConversation ?? null).input("modifiedFromMessage", sql.NVarChar, updates.provenance.modifiedFromMessage ?? null);
+    const req = pool.request().input("repoId", sql.UniqueIdentifier, repositoryId).input("entityId", sql.NVarChar, entityId).input("entityType", sql.NVarChar, updated.entityType).input("slug", sql.NVarChar, updated.slug).input("label", sql.NVarChar, updated.label).input("summary", sql.NVarChar, updated.summary ?? null).input("properties", sql.NVarChar, JSON.stringify(updated.properties)).input("data", sql.NVarChar, updated.data ?? null).input("dataFormat", sql.NVarChar, updated.dataFormat ?? null).input("embedding", sql.NVarChar, updated.embedding ? JSON.stringify(updated.embedding) : null).input("modifiedBy", sql.NVarChar, updates.provenance.modifiedBy).input("modifiedByType", sql.NVarChar, updates.provenance.modifiedByType).input("modifiedAt", sql.NVarChar, updates.provenance.modifiedAt).input("modifiedInConversation", sql.NVarChar, updates.provenance.modifiedInConversation ?? null).input("modifiedFromMessage", sql.NVarChar, updates.provenance.modifiedFromMessage ?? null);
     await req.query(`
       UPDATE ${this.t("dm_entities")} SET
+        [entity_type] = @entityType,
         [slug] = @slug,
         [label] = @label,
         [summary] = @summary,
@@ -892,6 +894,28 @@ var SqlServerStorageProvider = class {
       throw new 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", sql.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", sql.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();
@@ -1117,6 +1141,20 @@ var SqlServerStorageProvider = class {
       throw new 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", sql.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();