Skip to content

Commit

Permalink
nit: typing
Browse files Browse the repository at this point in the history
  • Loading branch information
pablodanswer committed Oct 3, 2024
1 parent 45b1ff7 commit c35251b
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 6 deletions.
10 changes: 5 additions & 5 deletions backend/danswer/server/manage/users.py
Original file line number Diff line number Diff line change
Expand Up @@ -220,9 +220,9 @@ def bulk_invite_users(
for email in all_emails:
send_user_email_invite(email, current_user)

invited_users = write_invited_users(all_emails)
number_of_invited_users = write_invited_users(all_emails)
if not MULTI_TENANT:
return invited_users
return number_of_invited_users
try:
logger.info("Registering tenant users")
register_tenant_users(current_tenant_id.get(), get_total_users(db_session))
Expand All @@ -233,7 +233,7 @@ def bulk_invite_users(
except Exception as e:
logger.error(f"Error sending email invite to invited users: {e}")

return invited_users
return number_of_invited_users
except Exception as e:
print("exception is ")
print(e)
Expand All @@ -257,7 +257,7 @@ def remove_invited_user(

tenant_id = current_tenant_id.get()
remove_users_from_tenant([user_email.user_email], tenant_id)
remaining_users = write_invited_users(remaining_users)
number_of_invited_users = write_invited_users(remaining_users)

try:
if MULTI_TENANT:
Expand All @@ -269,7 +269,7 @@ def remove_invited_user(
)
raise

return remaining_users
return number_of_invited_users


@router.patch("/manage/admin/deactivate-user")
Expand Down
4 changes: 3 additions & 1 deletion backend/ee/danswer/server/tenants/api.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
from typing import cast

import stripe
from fastapi import APIRouter
from fastapi import Depends
Expand Down Expand Up @@ -84,7 +86,7 @@ async def create_checkout_session(
tenant_id = current_tenant_id.get()
response = fetch_tenant_stripe_information(tenant_id)
response.get("stripe_customer_id")
stripe_subscription_id = response.get("stripe_subscription_id")
stripe_subscription_id = cast(str, response.get("stripe_subscription_id"))

# Modified code to update existing subscription
subscription = stripe.Subscription.retrieve(stripe_subscription_id)
Expand Down

0 comments on commit c35251b

Please sign in to comment.