chore: coderrabit suggested fix

Author: smokeyScraper

backend/app/api/v1/integrations.py

@@ -7,7 +7,7 @@
     IntegrationListResponse,
     IntegrationStatusResponse
 )
-from app.services.integration_service import integration_service
+from app.services.integration_service import integration_service, IntegrationNotFoundError
 from app.core.dependencies import get_current_user
 
 router = APIRouter()
@@ -76,8 +76,10 @@ async def update_integration(
     """Update an existing integration."""
     try:
         return await integration_service.update_integration(user_id, integration_id, request)
-    except ValueError as e:
+    except IntegrationNotFoundError as e:
         raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=str(e)) from e
+    except ValueError as e:
+        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(e)) from e
     except HTTPException:
         raise
     except Exception as e:
@@ -95,6 +97,7 @@ async def delete_integration(
     try:
         await integration_service.delete_integration(user_id, integration_id)
     except ValueError as e:
+        # For delete operation, ValueError means "not found"
         raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=str(e)) from e
     except HTTPException:
         raise

backend/app/services/integration_service.py

@@ -13,6 +13,11 @@
 logger = logging.getLogger(__name__)
 
 
+class IntegrationNotFoundError(Exception):
+    """Raised when an integration is not found."""
+    pass
+
+
 class IntegrationService:
     """Service for registering organizations (stores org info only)."""
 
@@ -142,7 +147,7 @@ async def update_integration(
 
             existing = await self.get_integration(user_id, integration_id)
             if not existing:
-                raise ValueError("Integration not found")
+                raise IntegrationNotFoundError("Integration not found")
 
             if request.organization_link is not None:
                 base_config = (update_data.get("config") or existing.config or {}).copy()
@@ -169,6 +174,11 @@ async def update_integration(
     async def delete_integration(self, user_id: UUID, integration_id: UUID) -> bool:
         """Delete an integration."""
         try:
+            # Check if integration exists before deleting
+            existing = await self.get_integration(user_id, integration_id)
+            if not existing:
+                raise ValueError("Integration not found")
+
             await self.supabase.table("organization_integrations")\
                 .delete()\
                 .eq("id", str(integration_id))\

backend/requirements.txt

@@ -224,7 +224,7 @@ watchfiles==1.0.5
 wcwidth==0.2.13
 weaviate-client==4.15.4
 websocket-client==1.8.0
-websockets>=14.2,<15.0
+websockets>=15.0.1,<16.0.0
 wrapt==1.17.2
 xxhash==3.5.0
 yarl==1.20.1