From 0d35e37216ddfd94665b44f95a1ba4f4d870afe1 Mon Sep 17 00:00:00 2001
From: Vineeth Voruganti <13438633+VVoruganti@users.noreply.github.com>
Date: Mon, 9 Feb 2026 15:30:17 -0500
Subject: [PATCH] fix: PR nitpicks

---
 src/crud/peer_card.py | 8 ++++++--
 src/routers/peers.py  | 6 +-----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/crud/peer_card.py b/src/crud/peer_card.py
index 79e64bb0..37c2749c 100644
--- a/src/crud/peer_card.py
+++ b/src/crud/peer_card.py
@@ -6,7 +6,7 @@ from typing import cast
 from sqlalchemy import update
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from src import models, schemas
+from src import exceptions, models, schemas
 from src.cache.client import cache
 from src.crud.peer import get_or_create_peers, get_peer, peer_cache_key
 
@@ -84,7 +84,11 @@ async def set_peer_card(
             )
         )
     )
-    await db.execute(stmt)
+    result = await db.execute(stmt)
+    if result.rowcount == 0:
+        raise exceptions.ResourceNotFoundException(
+            f"Peer {observer} not found in workspace {workspace_name}"
+        )
     await db.commit()
 
     # Invalidate cache - read-through pattern
diff --git a/src/routers/peers.py b/src/routers/peers.py
index 3bff98a3..e9eea331 100644
--- a/src/routers/peers.py
+++ b/src/routers/peers.py
@@ -338,11 +338,7 @@ async def set_peer_card(
         observed=observed,
     )
 
-    # Return the updated peer card
-    peer_card = await crud.get_peer_card(
-        db, workspace_id, observer=peer_id, observed=observed
-    )
-    return schemas.PeerCardResponse(peer_card=peer_card)
+    return schemas.PeerCardResponse(peer_card=peer_card_data.peer_card)
 
 
 @router.get(