Improve failure devex (#2247)

skyvern/forge/sdk/db/client.py

@@ -674,6 +674,18 @@ class AgentDB:
             LOG.error("UnexpectedError", exc_info=True)
             raise
 
+    async def get_all_organizations(self) -> list[Organization]:
+        try:
+            async with self.Session() as session:
+                organizations = (await session.scalars(select(OrganizationModel))).all()
+                return [convert_to_organization(organization) for organization in organizations]
+        except SQLAlchemyError:
+            LOG.error("SQLAlchemyError", exc_info=True)
+            raise
+        except Exception:
+            LOG.error("UnexpectedError", exc_info=True)
+            raise
+
     async def get_organization(self, organization_id: str) -> Organization | None:
         try:
             async with self.Session() as session:

skyvern/forge/sdk/services/org_auth_service.py

@@ -1,6 +1,7 @@
 import time
 from typing import Annotated
 
+import structlog
 from asyncache import cached
 from cachetools import TTLCache
 from fastapi import Header, HTTPException, status
@@ -15,6 +16,8 @@ from skyvern.forge.sdk.db.client import AgentDB
 from skyvern.forge.sdk.models import TokenPayload
 from skyvern.forge.sdk.schemas.organizations import Organization, OrganizationAuthTokenType
 
+LOG = structlog.get_logger()
+
 AUTHENTICATION_TTL = 60 * 60  # one hour
 CACHE_SIZE = 128
 ALGORITHM = "HS256"
@@ -91,6 +94,7 @@ async def _get_current_org_cached(x_api_key: str, db: AgentDB) -> Organization:
         )
         api_key_data = TokenPayload(**payload)
     except (JWTError, ValidationError):
+        LOG.error("Error decoding JWT", exc_info=True)
         raise HTTPException(
             status_code=status.HTTP_403_FORBIDDEN,
             detail="Could not validate credentials",