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

package/dist/node/index.node.js

@@ -5177,11 +5177,14 @@ class BaseDrizzleAdapter extends 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(and(eq2(participantTable.entityId, entityId), eq2(participantTable.roomId, roomId), eq2(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) {
         logger9.error({
@@ -5198,12 +5201,16 @@ class BaseDrizzleAdapter extends 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(and(eq2(participantTable.entityId, id), eq2(participantTable.roomId, roomId), eq2(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) {
         logger9.error({
@@ -5353,19 +5360,27 @@ class BaseDrizzleAdapter extends 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 = sql27.join(entityIds.map((id) => sql27`(${relationshipTable.sourceEntityId} = ${id} OR ${relationshipTable.targetEntityId} = ${id})`), sql27` OR `);
+      let query = sql27`
+        SELECT * FROM ${relationshipTable}
+        WHERE (${entityFilter})
+      `;
       if (tags && tags.length > 0) {
         query = sql27`
-          SELECT * FROM ${relationshipTable}
-          WHERE (${relationshipTable.sourceEntityId} = ${entityId} OR ${relationshipTable.targetEntityId} = ${entityId})
+          ${query}
           AND ${relationshipTable.tags} && CAST(ARRAY[${sql27.join(tags, sql27`, `)}] AS text[])
         `;
-      } else {
-        query = sql27`
-          SELECT * FROM ${relationshipTable}
-          WHERE ${relationshipTable.sourceEntityId} = ${entityId} OR ${relationshipTable.targetEntityId} = ${entityId}
-        `;
+      }
+      if (typeof limit === "number") {
+        query = sql27`${query} LIMIT ${limit}`;
+      }
+      if (typeof offset === "number" && offset > 0) {
+        query = sql27`${query} OFFSET ${offset}`;
       }
       const result = await this.db.execute(query);
       return result.rows.map((relationship) => ({
@@ -7103,5 +7118,5 @@ export {
   DatabaseMigrationService
 };
 
-//# debugId=11DF3BC6739C08E664756E2164756E21
+//# debugId=606A9AD08C7E6C2364756E2164756E21
 //# sourceMappingURL=index.node.js.map