fix update_agent_version_in_library

Significant-Gravitas · Jan 13, 2025 · 85180e1 · 85180e1
1 parent 700f2a3
commit 85180e1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/autogpt_platform/backend/backend/server/v2/library/db.py b/autogpt_platform/backend/backend/server/v2/library/db.py
@@ -105,12 +105,15 @@ async def update_agent_version_in_library(
     Updates the agent version in the library
     """
     try:
-        await prisma.models.LibraryAgent.prisma().update_many(
+        library_agent = await prisma.models.LibraryAgent.prisma().find_first_or_raise(
             where={
                 "userId": user_id,
                 "agentId": agent_id,
                 "useGraphIsActiveVersion": True,
             },
+        )
+        await prisma.models.LibraryAgent.prisma().update(
+            where={"id": library_agent.id},
             data=prisma.types.LibraryAgentUpdateInput(
                 Agent=prisma.types.AgentGraphUpdateOneWithoutRelationsInput(
                     connect=prisma.types._AgentGraphCompoundPrimaryKey(