From 85180e1c6cb7f48d7dcaac5a3ed6cd22a8851570 Mon Sep 17 00:00:00 2001
From: Reinier van der Leer <pwuts@agpt.co>
Date: Mon, 13 Jan 2025 23:30:47 +0100
Subject: [PATCH] fix `update_agent_version_in_library`

---
 autogpt_platform/backend/backend/server/v2/library/db.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/autogpt_platform/backend/backend/server/v2/library/db.py b/autogpt_platform/backend/backend/server/v2/library/db.py
index 401c78c6a91e..35b15a8cda88 100644
--- 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(