From b86b2f16afb9e0c20025bb44aeffd0d8373393fd Mon Sep 17 00:00:00 2001 From: Rohit Malhotra Date: Wed, 4 Mar 2026 11:06:26 -0500 Subject: [PATCH] Fix mypy type errors in enterprise/server/routes/billing.py and api_keys.py (#13196) Co-authored-by: openhands --- enterprise/server/routes/api_keys.py | 4 ++-- enterprise/server/routes/billing.py | 10 ++++++++++ 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/enterprise/server/routes/api_keys.py b/enterprise/server/routes/api_keys.py index 1e3f8a0d51..a543d333e3 100644 --- a/enterprise/server/routes/api_keys.py +++ b/enterprise/server/routes/api_keys.py @@ -22,7 +22,7 @@ async def get_byor_key_from_db(user_id: str) -> str | None: return None current_org_id = user.current_org_id - current_org_member: OrgMember = None + current_org_member: OrgMember | None = None for org_member in user.org_members: if org_member.org_id == current_org_id: current_org_member = org_member @@ -41,7 +41,7 @@ async def store_byor_key_in_db(user_id: str, key: str) -> None: return None current_org_id = user.current_org_id - current_org_member: OrgMember = None + current_org_member: OrgMember | None = None for org_member in user.org_members: if org_member.org_id == current_org_id: current_org_member = org_member diff --git a/enterprise/server/routes/billing.py b/enterprise/server/routes/billing.py index 51e5ee3fb1..cf8b72b689 100644 --- a/enterprise/server/routes/billing.py +++ b/enterprise/server/routes/billing.py @@ -146,6 +146,11 @@ async def create_customer_setup_session( ) -> CreateBillingSessionResponse: await validate_billing_enabled() customer_info = await stripe_service.find_or_create_customer_by_user_id(user_id) + if not customer_info: + raise HTTPException( + status_code=status.HTTP_400_BAD_REQUEST, + detail='Could not find or create customer for user', + ) base_url = _get_base_url(request) checkout_session = await stripe.checkout.Session.create_async( customer=customer_info['customer_id'], @@ -167,6 +172,11 @@ async def create_checkout_session( await validate_billing_enabled() base_url = _get_base_url(request) customer_info = await stripe_service.find_or_create_customer_by_user_id(user_id) + if not customer_info: + raise HTTPException( + status_code=status.HTTP_400_BAD_REQUEST, + detail='Could not find or create customer for user', + ) checkout_session = await stripe.checkout.Session.create_async( customer=customer_info['customer_id'], line_items=[