From e4f1adc90fe9f4d0cc98b1477ca37b44c9cbb81c Mon Sep 17 00:00:00 2001 From: LawyZheng Date: Wed, 10 Sep 2025 14:20:23 +0800 Subject: [PATCH] fix wrong endpoint error (#3402) --- skyvern/forge/sdk/db/client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py index 9073a3e7..6169a7de 100644 --- a/skyvern/forge/sdk/db/client.py +++ b/skyvern/forge/sdk/db/client.py @@ -3387,7 +3387,7 @@ class AgentDB: await session.refresh(credential) return Credential.model_validate(credential) - async def get_credential(self, credential_id: str, organization_id: str) -> Credential: + async def get_credential(self, credential_id: str, organization_id: str) -> Credential | None: async with self.Session() as session: credential = ( await session.scalars( @@ -3399,7 +3399,7 @@ class AgentDB: ).first() if credential: return Credential.model_validate(credential) - raise NotFoundError(f"Credential {credential_id} not found") + return None async def get_credentials(self, organization_id: str, page: int = 1, page_size: int = 10) -> list[Credential]: async with self.Session() as session: