@elizaos/plugin-sql 2.0.0-alpha.18 → 2.0.0-alpha.19

package/dist/cjs/index.node.cjs

@@ -5217,11 +5217,14 @@ class BaseDrizzleAdapter extends import_core10.DatabaseAdapter {
   async addParticipant(entityId, roomId) {
     return this.withDatabase(async () => {
       try {
-        await this.db.insert(participantTable).values({
-          entityId,
-          roomId,
-          agentId: this.agentId
-        }).onConflictDoNothing();
+        const existing = await this.db.select({ id: participantTable.id }).from(participantTable).where(import_drizzle_orm28.and(import_drizzle_orm28.eq(participantTable.entityId, entityId), import_drizzle_orm28.eq(participantTable.roomId, roomId), import_drizzle_orm28.eq(participantTable.agentId, this.agentId))).limit(1);
+        if (existing.length === 0) {
+          await this.db.insert(participantTable).values({
+            entityId,
+            roomId,
+            agentId: this.agentId
+          });
+        }
         return true;
       } catch (error) {
         import_core10.logger.error({
@@ -5238,12 +5241,16 @@ class BaseDrizzleAdapter extends import_core10.DatabaseAdapter {
   async addParticipantsRoom(entityIds, roomId) {
     return this.withDatabase(async () => {
       try {
-        const values = entityIds.map((id) => ({
-          entityId: id,
-          roomId,
-          agentId: this.agentId
-        }));
-        await this.db.insert(participantTable).values(values).onConflictDoNothing().execute();
+        for (const id of entityIds) {
+          const existing = await this.db.select({ id: participantTable.id }).from(participantTable).where(import_drizzle_orm28.and(import_drizzle_orm28.eq(participantTable.entityId, id), import_drizzle_orm28.eq(participantTable.roomId, roomId), import_drizzle_orm28.eq(participantTable.agentId, this.agentId))).limit(1);
+          if (existing.length === 0) {
+            await this.db.insert(participantTable).values({
+              entityId: id,
+              roomId,
+              agentId: this.agentId
+            });
+          }
+        }
         return true;
       } catch (error) {
         import_core10.logger.error({
@@ -5393,19 +5400,27 @@ class BaseDrizzleAdapter extends import_core10.DatabaseAdapter {
   }
   async getRelationships(params) {
     return this.withDatabase(async () => {
-      const { entityId, tags } = params;
-      let query;
+      const { entityIds: rawEntityIds, entityId, tags, limit, offset } = params;
+      const entityIds = (rawEntityIds && rawEntityIds.length > 0 ? rawEntityIds : entityId ? [entityId] : []).filter((id) => typeof id === "string" && id.trim().length > 0);
+      if (entityIds.length === 0) {
+        return [];
+      }
+      const entityFilter = import_drizzle_orm28.sql.join(entityIds.map((id) => import_drizzle_orm28.sql`(${relationshipTable.sourceEntityId} = ${id} OR ${relationshipTable.targetEntityId} = ${id})`), import_drizzle_orm28.sql` OR `);
+      let query = import_drizzle_orm28.sql`
+        SELECT * FROM ${relationshipTable}
+        WHERE (${entityFilter})
+      `;
       if (tags && tags.length > 0) {
         query = import_drizzle_orm28.sql`
-          SELECT * FROM ${relationshipTable}
-          WHERE (${relationshipTable.sourceEntityId} = ${entityId} OR ${relationshipTable.targetEntityId} = ${entityId})
+          ${query}
           AND ${relationshipTable.tags} && CAST(ARRAY[${import_drizzle_orm28.sql.join(tags, import_drizzle_orm28.sql`, `)}] AS text[])
         `;
-      } else {
-        query = import_drizzle_orm28.sql`
-          SELECT * FROM ${relationshipTable}
-          WHERE ${relationshipTable.sourceEntityId} = ${entityId} OR ${relationshipTable.targetEntityId} = ${entityId}
-        `;
+      }
+      if (typeof limit === "number") {
+        query = import_drizzle_orm28.sql`${query} LIMIT ${limit}`;
+      }
+      if (typeof offset === "number" && offset > 0) {
+        query = import_drizzle_orm28.sql`${query} OFFSET ${offset}`;
       }
       const result = await this.db.execute(query);
       return result.rows.map((relationship) => ({
@@ -7130,5 +7145,5 @@ var plugin = {
 };
 var typescript_default = plugin;
 
-//# debugId=30BB64705897BABA64756E2164756E21
+//# debugId=B2FA58A515BC4FB464756E2164756E21
 //# sourceMappingURL=index.node.cjs.map