feat/onboarding-new-user

Author: GitHub CI

api/ee/databases/postgres/migrations/core/data_migrations/workspaces.py

@@ -56,8 +56,7 @@ def create_default_project_for_workspaces(session: Connection):
             for workspace in workspaces:
                 # Create a new default project for each workspace
                 get_or_create_workspace_default_project(
-                    session=session,
-                    workspace=workspace,  # type: ignore
+                    session=session, workspace=workspace  # type: ignore
                 )
 
             # Commit the changes for the current batch

api/ee/databases/postgres/migrations/core/utils.py

@@ -73,9 +73,7 @@ async def get_current_migration_head_from_db(engine: AsyncEngine):
 
     async with engine.connect() as connection:
         try:
-            result = await connection.execute(
-                text("SELECT version_num FROM alembic_version")
-            )  # type: ignore
+            result = await connection.execute(text("SELECT version_num FROM alembic_version"))  # type: ignore
         except (asyncpg.exceptions.UndefinedTableError, ProgrammingError):
             # Note: If the alembic_version table does not exist, it will result in raising an UndefinedTableError exception.
             # We need to suppress the error and return a list with the alembic_version table name to inform the user that there is a pending migration \
@@ -85,9 +83,9 @@ async def get_current_migration_head_from_db(engine: AsyncEngine):
             return "alembic_version"
 
         migration_heads = [row[0] for row in result.fetchall()]
-        assert len(migration_heads) == 1, (
-            "There can only be one migration head stored in the database."
-        )
+        assert (
+            len(migration_heads) == 1
+        ), "There can only be one migration head stored in the database."
         return migration_heads[0]
 
 

api/ee/databases/postgres/migrations/tracing/utils.py

@@ -66,9 +66,7 @@ async def get_current_migration_head_from_db(engine: AsyncEngine):
 
     async with engine.connect() as connection:
         try:
-            result = await connection.execute(
-                text("SELECT version_num FROM alembic_version")
-            )  # type: ignore
+            result = await connection.execute(text("SELECT version_num FROM alembic_version"))  # type: ignore
         except (asyncpg.exceptions.UndefinedTableError, ProgrammingError):
             # Note: If the alembic_version table does not exist, it will result in raising an UndefinedTableError exception.
             # We need to suppress the error and return a list with the alembic_version table name to inform the user that there is a pending migration \
@@ -78,9 +76,9 @@ async def get_current_migration_head_from_db(engine: AsyncEngine):
             return "alembic_version"
 
         migration_heads = [row[0] for row in result.fetchall()]
-        assert len(migration_heads) == 1, (
-            "There can only be one migration head stored in the database."
-        )
+        assert (
+            len(migration_heads) == 1
+        ), "There can only be one migration head stored in the database."
         return migration_heads[0]
 
 

api/ee/src/apis/fastapi/billing/router.py

@@ -35,7 +35,7 @@
 
 stripe.api_key = environ.get("STRIPE_API_KEY")
 
-MAC_ADDRESS = ":".join(f"{(getnode() >> ele) & 0xFF:02x}" for ele in range(40, -1, -8))
+MAC_ADDRESS = ":".join(f"{(getnode() >> ele) & 0xff:02x}" for ele in range(40, -1, -8))
 STRIPE_WEBHOOK_SECRET = environ.get("STRIPE_WEBHOOK_SECRET")
 STRIPE_TARGET = environ.get("STRIPE_TARGET") or MAC_ADDRESS
 AGENTA_PRICING = loads(environ.get("AGENTA_PRICING") or "{}")

api/ee/src/core/subscriptions/service.py

@@ -25,7 +25,7 @@
 
 stripe.api_key = environ.get("STRIPE_SECRET_KEY")
 
-MAC_ADDRESS = ":".join(f"{(getnode() >> ele) & 0xFF:02x}" for ele in range(40, -1, -8))
+MAC_ADDRESS = ":".join(f"{(getnode() >> ele) & 0xff:02x}" for ele in range(40, -1, -8))
 STRIPE_TARGET = environ.get("STRIPE_TARGET") or MAC_ADDRESS
 AGENTA_PRICING = loads(environ.get("AGENTA_PRICING") or "{}")
 

api/ee/src/services/db_manager_ee.py

@@ -645,7 +645,9 @@ async def remove_user_from_workspace(
     project = await db_manager.get_project_by_id(project_id=project_id)
 
     async with engine.core_session() as session:
-        if not user:  # User is an invited user who has not yet created an account and therefore does not have a user object
+        if (
+            not user
+        ):  # User is an invited user who has not yet created an account and therefore does not have a user object
             pass
         else:
             # Ensure that a user can not remove the owner of the workspace

api/ee/src/services/workspace_manager.py

@@ -317,9 +317,9 @@ async def accept_workspace_invitation(
 
         invitation = await check_valid_invitation(project_id, user.email, token)
         if invitation is not None:
-            assert invitation.role is not None, (
-                "Invitation does not have any workspace role"
-            )
+            assert (
+                invitation.role is not None
+            ), "Invitation does not have any workspace role"
             await db_manager_ee.add_user_to_workspace_and_org(
                 organization, workspace, user, project_id, invitation.role
             )

api/ee/src/utils/entitlements.py

@@ -36,25 +36,25 @@ class EntitlementsException(Exception):
     pass
 
 
-NOT_ENTITLED_RESPONSE: Callable[[Tracker], JSONResponse] = (
-    lambda tracker=None: JSONResponse(
-        status_code=403,
-        content={
-            "detail": (
-                "You have reached your monthly quota limit. Please upgrade your plan to continue."
-                if tracker == Tracker.COUNTERS
+NOT_ENTITLED_RESPONSE: Callable[
+    [Tracker], JSONResponse
+] = lambda tracker=None: JSONResponse(
+    status_code=403,
+    content={
+        "detail": (
+            "You have reached your monthly quota limit. Please upgrade your plan to continue."
+            if tracker == Tracker.COUNTERS
+            else (
+                "You have reached your quota limit. Please upgrade your plan to continue."
+                if tracker == Tracker.GAUGES
                 else (
-                    "You have reached your quota limit. Please upgrade your plan to continue."
-                    if tracker == Tracker.GAUGES
-                    else (
-                        "You do not have access to this feature. Please upgrade your plan to continue."
-                        if tracker == Tracker.FLAGS
-                        else "You do not have access to this feature."
-                    )
+                    "You do not have access to this feature. Please upgrade your plan to continue."
+                    if tracker == Tracker.FLAGS
+                    else "You do not have access to this feature."
                 )
-            ),
-        },
-    )
+            )
+        ),
+    },
 )
 
 
@@ -163,7 +163,7 @@ async def check_entitlements(
 
     # TODO: remove this line
     log.info(
-        f"adjusting: {organization_id} | {(('0' if (meter.month != 0 and meter.month < 10) else '') + str(meter.month)) if meter.month != 0 else '  '}.{meter.year if meter.year else '    '} | {'allow' if check else 'deny '} | {meter.key}: {meter.value - meter.synced} [{meter.value}]"
+        f"adjusting: {organization_id} | {(('0' if (meter.month != 0 and meter.month < 10) else '') + str(meter.month)) if meter.month != 0 else '  '}.{meter.year if meter.year else '    '} | {'allow' if check else 'deny '} | {meter.key}: {meter.value-meter.synced} [{meter.value}]"
     )
 
     return check is True, meter, _

api/ee/src/utils/permissions.py

@@ -218,17 +218,17 @@ async def check_rbac_permission(
         bool: True if the user belongs to the workspace and has the specified permission, False otherwise.
     """
 
-    assert project_id is not None, (
-        "Project_ID is required to check object-level permissions"
-    )
+    assert (
+        project_id is not None
+    ), "Project_ID is required to check object-level permissions"
 
     # Assert that either permission or role is provided, but not both
-    assert (permission is not None) or (role is not None), (
-        "Either 'permission' or 'role' must be provided, but neither is provided"
-    )
-    assert not ((permission is not None) and (role is not None)), (
-        "'permission' and 'role' cannot both be provided at the same time"
-    )
+    assert (permission is not None) or (
+        role is not None
+    ), "Either 'permission' or 'role' must be provided, but neither is provided"
+    assert not (
+        (permission is not None) and (role is not None)
+    ), "'permission' and 'role' cannot both be provided at the same time"
 
     if project_id is not None:
         project = await db_manager.get_project_by_id(project_id)
@@ -281,9 +281,9 @@ async def check_project_has_role_or_permission(
     if not check:
         return True
 
-    assert role is not None or permission is not None, (
-        "Either role or permission must be provided"
-    )
+    assert (
+        role is not None or permission is not None
+    ), "Either role or permission must be provided"
 
     project_members = await db_manager_ee.get_project_members(
         project_id=str(project.id)

api/oss/databases/postgres/migrations/core/data_migrations/projects.py

@@ -131,13 +131,13 @@ def add_default_evaluators_to_project(session: Session, project_id: str):
             }
 
             for setting_name, default_value in settings_values.items():
-                assert default_value != "", (
-                    f"Default value for ground truth key '{setting_name}' in Evaluator is empty"
-                )
+                assert (
+                    default_value != ""
+                ), f"Default value for ground truth key '{setting_name}' in Evaluator is empty"
 
-            assert hasattr(evaluator, "name") and hasattr(evaluator, "key"), (
-                f"'name' and 'key' does not exist in the evaluator: {evaluator}"
-            )
+            assert hasattr(evaluator, "name") and hasattr(
+                evaluator, "key"
+            ), f"'name' and 'key' does not exist in the evaluator: {evaluator}"
 
             evaluator_config = EvaluatorConfigDB(
                 project_id=uuid.UUID(project_id),