pre commit run

Author: avirajsingh7

backend/app/api/routes/api_keys.py

@@ -28,13 +28,17 @@ def create_key(
     """
     Generate a new API key for the user's organization.
     """
-    logger.info(f"[apikey.create] Received create API key request | project_id={project_id}, user_id={user_id}")
+    logger.info(
+        f"[apikey.create] Received create API key request | project_id={project_id}, user_id={user_id}"
+    )
 
     project = validate_project(session, project_id)
 
     existing_api_key = get_api_key_by_project_user(session, project_id, user_id)
     if existing_api_key:
-        logger.warning(f"[apikey.create] API key already exists | project_id={project_id}, user_id={user_id}")
+        logger.warning(
+            f"[apikey.create] API key already exists | project_id={project_id}, user_id={user_id}"
+        )
         raise HTTPException(
             status_code=400,
             detail="API Key already exists for this user and project.",
@@ -47,7 +51,9 @@ def create_key(
         project_id=project_id,
     )
 
-    logger.info(f"[apikey.create] Created API key | api_key_id={api_key.id}, project_id={project_id}, user_id={user_id}")
+    logger.info(
+        f"[apikey.create] Created API key | api_key_id={api_key.id}, project_id={project_id}, user_id={user_id}"
+    )
     return APIResponse.success_response(api_key)
 
 
@@ -61,13 +67,17 @@ def list_keys(
     Retrieve all API keys for the given project. Superusers get all keys;
     regular users get only their own.
     """
-    logger.info(f"[apikey.list] Fetching API keys | project_id={project_id}, requested_by_user_id={current_user.id}")
+    logger.info(
+        f"[apikey.list] Fetching API keys | project_id={project_id}, requested_by_user_id={current_user.id}"
+    )
 
     project = validate_project(session=session, project_id=project_id)
 
     if current_user.is_superuser:
         api_keys = get_api_keys_by_project(session=session, project_id=project_id)
-        logger.info(f"[apikey.list] Superuser access - {len(api_keys)} key(s) found | project_id={project_id}")
+        logger.info(
+            f"[apikey.list] Superuser access - {len(api_keys)} key(s) found | project_id={project_id}"
+        )
     else:
         user_api_key = get_api_key_by_project_user(
             session=session, project_id=project_id, user_id=current_user.id
@@ -81,7 +91,9 @@ def list_keys(
             detail="No API keys found for this project.",
         )
 
-    logger.info(f"[apikey.list] Found API key | project_id={project_id}, requested_by_user_id={current_user.id}")
+    logger.info(
+        f"[apikey.list] Found API key | project_id={project_id}, requested_by_user_id={current_user.id}"
+    )
     return APIResponse.success_response(api_keys)
 
 
@@ -94,7 +106,9 @@ def get_key(
     """
     Retrieve an API key by ID.
     """
-    logger.info(f"[apikey.get] Fetching API key | api_key_id={api_key_id}, requested_by_user_id={current_user.id}")
+    logger.info(
+        f"[apikey.get] Fetching API key | api_key_id={api_key_id}, requested_by_user_id={current_user.id}"
+    )
 
     api_key = get_api_key(session, api_key_id)
     if not api_key:
@@ -114,11 +128,15 @@ def revoke_key(
     """
     Soft delete an API key (revoke access).
     """
-    logger.info(f"[apikey.revoke] Revoking API key | api_key_id={api_key_id}, requested_by_user_id={current_user.id}")
+    logger.info(
+        f"[apikey.revoke] Revoking API key | api_key_id={api_key_id}, requested_by_user_id={current_user.id}"
+    )
 
     api_key = get_api_key(session, api_key_id)
     if not api_key:
-        logger.warning(f"[apikey.revoke] API key not found or already deleted | api_key_id={api_key_id}")
+        logger.warning(
+            f"[apikey.revoke] API key not found or already deleted | api_key_id={api_key_id}"
+        )
         raise HTTPException(404, "API key not found or already deleted")
 
     delete_api_key(session, api_key_id)

backend/app/api/routes/collections.py

@@ -173,9 +173,7 @@ def __call__(self, response: APIResponse, status: str):
         post_callback(self.url, response)
 
     def fail(self, body):
-        logger.error(
-            f"[WebHookCallback.fail] Callback failed | {{'body': '{body}'}}"
-        )
+        logger.error(f"[WebHookCallback.fail] Callback failed | {{'body': '{body}'}}")
         self(APIResponse.failure_response(body), "incomplete")
 
     def success(self, body):
@@ -459,4 +457,4 @@ def collection_documents(
     logger.info(
         f"[collection_documents] Documents retrieved successfully | {{'collection_id': '{collection_id}', 'document_count': {len(data)}}}"
     )
-    return APIResponse.success_response(data)
+    return APIResponse.success_response(data)

backend/app/api/routes/credentials.py

@@ -32,16 +32,20 @@
     description="Creates new credentials for a specific organization and project combination. This endpoint requires superuser privileges. Each organization can have different credentials for different providers and projects. Only one credential per provider is allowed per organization-project combination.",
 )
 def create_new_credential(*, session: SessionDep, creds_in: CredsCreate):
-    logger.info(f"[credential.create] Creating credentials | org_id={creds_in.organization_id}, project_id={creds_in.project_id}")
-    
+    logger.info(
+        f"[credential.create] Creating credentials | org_id={creds_in.organization_id}, project_id={creds_in.project_id}"
+    )
+
     # Validate organization
     validate_organization(session, creds_in.organization_id)
 
     # Validate project if provided
     if creds_in.project_id:
         project = validate_project(session, creds_in.project_id)
         if project.organization_id != creds_in.organization_id:
-            logger.warning(f"[credential.create] Project does not belong to organization | org_id={creds_in.organization_id}, project_id={creds_in.project_id}")
+            logger.warning(
+                f"[credential.create] Project does not belong to organization | org_id={creds_in.organization_id}, project_id={creds_in.project_id}"
+            )
             raise HTTPException(
                 status_code=400,
                 detail="Project does not belong to the specified organization",
@@ -56,7 +60,9 @@ def create_new_credential(*, session: SessionDep, creds_in: CredsCreate):
             project_id=creds_in.project_id,
         )
         if existing_cred:
-            logger.warning(f"[credential.create] Credential exists | provider={provider}, org_id={creds_in.organization_id}, project_id={creds_in.project_id}")
+            logger.warning(
+                f"[credential.create] Credential exists | provider={provider}, org_id={creds_in.organization_id}, project_id={creds_in.project_id}"
+            )
             raise HTTPException(
                 status_code=400,
                 detail=(
@@ -68,10 +74,14 @@ def create_new_credential(*, session: SessionDep, creds_in: CredsCreate):
     # Create credentials
     new_creds = set_creds_for_org(session=session, creds_add=creds_in)
     if not new_creds:
-        logger.error(f"[credential.create] Failed to create credentials | org_id={creds_in.organization_id}")
+        logger.error(
+            f"[credential.create] Failed to create credentials | org_id={creds_in.organization_id}"
+        )
         raise Exception(status_code=500, detail="Failed to create credentials")
 
-    logger.info(f"[credential.create] Credentials created | count={len(new_creds)}, org_id={creds_in.organization_id}")
+    logger.info(
+        f"[credential.create] Credentials created | count={len(new_creds)}, org_id={creds_in.organization_id}"
+    )
     return APIResponse.success_response([cred.to_public() for cred in new_creds])
 
 
@@ -83,13 +93,19 @@ def create_new_credential(*, session: SessionDep, creds_in: CredsCreate):
     description="Retrieves all provider credentials associated with a specific organization and project combination. If project_id is not provided, returns credentials for the organization level. This endpoint requires superuser privileges.",
 )
 def read_credential(*, session: SessionDep, org_id: int, project_id: int | None = None):
-    logger.info(f"[credential.read] Fetching credentials | org_id={org_id}, project_id={project_id}")
+    logger.info(
+        f"[credential.read] Fetching credentials | org_id={org_id}, project_id={project_id}"
+    )
     creds = get_creds_by_org(session=session, org_id=org_id, project_id=project_id)
     if not creds:
-        logger.warning(f"[credential.read] No credentials found | org_id={org_id}, project_id={project_id}")
+        logger.warning(
+            f"[credential.read] No credentials found | org_id={org_id}, project_id={project_id}"
+        )
         raise HTTPException(status_code=404, detail="Credentials not found")
 
-    logger.info(f"[credential.read] Retrieved {len(creds)} credentials | org_id={org_id}, project_id={project_id}")
+    logger.info(
+        f"[credential.read] Retrieved {len(creds)} credentials | org_id={org_id}, project_id={project_id}"
+    )
     return APIResponse.success_response([cred.to_public() for cred in creds])
 
 
@@ -103,7 +119,9 @@ def read_credential(*, session: SessionDep, org_id: int, project_id: int | None
 def read_provider_credential(
     *, session: SessionDep, org_id: int, provider: str, project_id: int | None = None
 ):
-    logger.info(f"[credential.read_provider] Fetching provider credential | org_id={org_id}, provider={provider}, project_id={project_id}")
+    logger.info(
+        f"[credential.read_provider] Fetching provider credential | org_id={org_id}, provider={provider}, project_id={project_id}"
+    )
     provider_enum = validate_provider(provider)
     provider_creds = get_provider_credential(
         session=session,
@@ -112,10 +130,14 @@ def read_provider_credential(
         project_id=project_id,
     )
     if provider_creds is None:
-        logger.warning(f"[credential.read_provider] Credential not found | org_id={org_id}, provider={provider}, project_id={project_id}")
+        logger.warning(
+            f"[credential.read_provider] Credential not found | org_id={org_id}, provider={provider}, project_id={project_id}"
+        )
         raise HTTPException(status_code=404, detail="Provider credentials not found")
 
-    logger.info(f"[credential.read_provider] Credential found | org_id={org_id}, provider={provider}, project_id={project_id}")
+    logger.info(
+        f"[credential.read_provider] Credential found | org_id={org_id}, provider={provider}, project_id={project_id}"
+    )
     return APIResponse.success_response(provider_creds)
 
 
@@ -127,7 +149,9 @@ def read_provider_credential(
     description="Updates credentials for a specific organization and project combination. Can update specific provider credentials or add new providers. If project_id is provided in the update, credentials will be moved to that project. This endpoint requires superuser privileges.",
 )
 def update_credential(*, session: SessionDep, org_id: int, creds_in: CredsUpdate):
-    logger.info(f"[credential.update] Updating credentials | org_id={org_id}, provider={creds_in.provider}, project_id={creds_in.project_id}")
+    logger.info(
+        f"[credential.update] Updating credentials | org_id={org_id}, provider={creds_in.provider}, project_id={creds_in.project_id}"
+    )
     validate_organization(session, org_id)
     if not creds_in or not creds_in.provider or not creds_in.credential:
         logger.warning(f"[credential.update] Invalid update payload | org_id={org_id}")
@@ -139,7 +163,9 @@ def update_credential(*, session: SessionDep, org_id: int, creds_in: CredsUpdate
         session=session, org_id=org_id, creds_in=creds_in
     )
 
-    logger.info(f"[credential.update] Credentials updated | count={len(updated_creds)}, org_id={org_id}, provider={creds_in.provider}")
+    logger.info(
+        f"[credential.update] Credentials updated | count={len(updated_creds)}, org_id={org_id}, provider={creds_in.provider}"
+    )
     return APIResponse.success_response([cred.to_public() for cred in updated_creds])
 
 
@@ -152,10 +178,14 @@ def update_credential(*, session: SessionDep, org_id: int, creds_in: CredsUpdate
 def delete_provider_credential(
     *, session: SessionDep, org_id: int, provider: str, project_id: int | None = None
 ):
-    logger.info(f"[credential.delete_provider] Deleting provider credential | org_id={org_id}, provider={provider}, project_id={project_id}")
+    logger.info(
+        f"[credential.delete_provider] Deleting provider credential | org_id={org_id}, provider={provider}, project_id={project_id}"
+    )
     provider_enum = validate_provider(provider)
     if not provider_enum:
-        logger.warning(f"[credential.delete_provider] Invalid provider | provider={provider}")
+        logger.warning(
+            f"[credential.delete_provider] Invalid provider | provider={provider}"
+        )
         raise HTTPException(status_code=400, detail="Invalid provider")
 
     provider_creds = get_provider_credential(
@@ -165,14 +195,18 @@ def delete_provider_credential(
         project_id=project_id,
     )
     if provider_creds is None:
-        logger.warning(f"[credential.delete_provider] Credential not found | org_id={org_id}, provider={provider}, project_id={project_id}")
+        logger.warning(
+            f"[credential.delete_provider] Credential not found | org_id={org_id}, provider={provider}, project_id={project_id}"
+        )
         raise HTTPException(status_code=404, detail="Provider credentials not found")
 
     updated_creds = remove_provider_credential(
         session=session, org_id=org_id, provider=provider_enum, project_id=project_id
     )
 
-    logger.info(f"[credential.delete_provider] Credential deleted | org_id={org_id}, provider={provider}, project_id={project_id}")
+    logger.info(
+        f"[credential.delete_provider] Credential deleted | org_id={org_id}, provider={provider}, project_id={project_id}"
+    )
     return APIResponse.success_response(
         {"message": "Provider credentials removed successfully"}
     )
@@ -188,13 +222,19 @@ def delete_provider_credential(
 def delete_all_credentials(
     *, session: SessionDep, org_id: int, project_id: int | None = None
 ):
-    logger.info(f"[credential.delete_all] Deleting all credentials | org_id={org_id}, project_id={project_id}")
+    logger.info(
+        f"[credential.delete_all] Deleting all credentials | org_id={org_id}, project_id={project_id}"
+    )
     creds = remove_creds_for_org(session=session, org_id=org_id, project_id=project_id)
     if not creds:
-        logger.warning(f"[credential.delete_all] No credentials found to delete | org_id={org_id}, project_id={project_id}")
+        logger.warning(
+            f"[credential.delete_all] No credentials found to delete | org_id={org_id}, project_id={project_id}"
+        )
         raise HTTPException(
             status_code=404, detail="Credentials for organization not found"
         )
 
-    logger.info(f"[credential.delete_all] All credentials deleted | org_id={org_id}, project_id={project_id}")
+    logger.info(
+        f"[credential.delete_all] All credentials deleted | org_id={org_id}, project_id={project_id}"
+    )
     return APIResponse.success_response({"message": "Credentials deleted successfully"})

backend/app/api/routes/documents.py

@@ -148,4 +148,4 @@ def doc_info(
     logger.info(
         f"[doc_info] Document retrieved successfully | {{'user_id': '{current_user.id}', 'document_id': '{doc_id}'}}"
     )
-    return APIResponse.success_response(data)
+    return APIResponse.success_response(data)

backend/app/api/routes/login.py

@@ -42,10 +42,14 @@ def login_access_token(
         session=session, email=form_data.username, password=form_data.password
     )
     if not user:
-        logger.warning(f"[login.access_token] Invalid credentials | email={form_data.username}")
+        logger.warning(
+            f"[login.access_token] Invalid credentials | email={form_data.username}"
+        )
         raise HTTPException(status_code=400, detail="Incorrect email or password")
     elif not user.is_active:
-        logger.warning(f"[login.access_token] Inactive user login attempt | user_id={user.id}")
+        logger.warning(
+            f"[login.access_token] Inactive user login attempt | user_id={user.id}"
+        )
         raise HTTPException(status_code=400, detail="Inactive user")
 
     access_token_expires = timedelta(minutes=token_expiry_minutes)
@@ -137,7 +141,9 @@ def recover_password_html_content(email: str, session: SessionDep) -> Any:
     """
     HTML Content for Password Recovery
     """
-    logger.info(f"[login.recover_password_html] HTML recovery content requested | email={email}")
+    logger.info(
+        f"[login.recover_password_html] HTML recovery content requested | email={email}"
+    )
 
     user = get_user_by_email(session=session, email=email)
     if not user:
@@ -152,7 +158,9 @@ def recover_password_html_content(email: str, session: SessionDep) -> Any:
         email_to=user.email, email=email, token=password_reset_token
     )
 
-    logger.info(f"[login.recover_password_html] HTML content generated | user_id={user.id}")
+    logger.info(
+        f"[login.recover_password_html] HTML content generated | user_id={user.id}"
+    )
     return HTMLResponse(
         content=email_data.html_content, headers={"subject:": email_data.subject}
     )