style: lint after rebase

Author: KNedelec

packages/gg_api_core/src/gg_api_core/client.py

@@ -129,7 +129,7 @@ def _init_urls(self, gitguardian_url: str | None = None):
 
     def _init_personal_access_token(self, personal_access_token: str | None = None):
         """Initialize authentication token based on transport mode.
-        
+
         Authentication architecture:
         - stdio mode: OAuth (interactive) OR PAT from env var (non-interactive)
         - HTTP mode: Per-request Authorization header ONLY (multi-tenant capable)
@@ -958,7 +958,9 @@ async def get_incidents(self, incident_ids: list[str]) -> list[dict[str, Any]]:
 
         return incidents
 
-    async def update_incident(self, incident_id: str, status: str | None = None, custom_tags: list | None = None) -> dict[str, Any]:
+    async def update_incident(
+        self, incident_id: str, status: str | None = None, custom_tags: list | None = None
+    ) -> dict[str, Any]:
         """Update a secret incident.
 
         Args:
@@ -1392,9 +1394,7 @@ async def update_incident_note(self, incident_id: str, note_id: str, content: st
             Updated note details
         """
         logger.info(f"Updating note {note_id} for incident {incident_id}")
-        return await self._request_patch(
-            f"/incidents/secrets/{incident_id}/notes/{note_id}", json={"content": content}
-        )
+        return await self._request_patch(f"/incidents/secrets/{incident_id}/notes/{note_id}", json={"content": content})
 
     async def delete_incident_note(self, incident_id: str, note_id: str) -> dict[str, Any]:
         """Delete a note from a secret incident.

packages/gg_api_core/src/gg_api_core/mcp_server.py

@@ -43,7 +43,7 @@ def __init__(self, *args, **kwargs):
         kwargs["lifespan"] = self._create_token_scope_lifespan(original_lifespan)
         # Call parent __init__ in the MRO chain
         super().__init__(*args, **kwargs)
-    
+
     def clear_cache(self) -> None:
         """Clear cached token information and scopes."""
         self._token_scopes = set()
@@ -106,8 +106,8 @@ def __init__(self, *args, default_scopes: list[str] | None = None, **kwargs):
         # Map each tool to its required scopes (instance attribute)
         self._tool_scopes: dict[str, set[str]] = {}
 
-        self.add_middleware(self._scope_filtering_middleware)  
-    
+        self.add_middleware(self._scope_filtering_middleware)
+
     def clear_cache(self) -> None:
         """Clear cached data. Override in subclasses that cache."""
         pass

packages/gg_api_core/src/gg_api_core/tools/assign_incident.py

@@ -1,6 +1,5 @@
 import logging
 from typing import Any
-from typing_extensions import cast
 
 from fastmcp.exceptions import ToolError
 from pydantic import BaseModel, Field, model_validator

packages/gg_api_core/src/gg_api_core/tools/find_current_source_id.py

@@ -123,7 +123,9 @@ async def find_current_source_id(repository_path: str = ".") -> FindCurrentSourc
         logger.info(f"Detected repository name: {repository_name} (method: {detection_method})")
 
         # Search for the source in GitGuardian with robust non-exact matching
-        source_result: dict[str, Any] | list[dict[str, Any]] | None = await client.get_source_by_name(repository_name, return_all_on_no_match=True)
+        source_result: dict[str, Any] | list[dict[str, Any]] | None = await client.get_source_by_name(
+            repository_name, return_all_on_no_match=True
+        )
 
         # Handle exact match (single dict result)
         if isinstance(source_result, dict):
@@ -194,7 +196,7 @@ async def find_current_source_id(repository_path: str = ".") -> FindCurrentSourc
                     message = f"No exact match for '{repository_name}', but found {len(fallback_result)} potential matches using repo name '{repo_only}'."
                     if detection_method == "directory name":
                         message += f" (repository name inferred from {detection_method})"
-                    
+
                     return FindCurrentSourceIdResult(
                         repository_name=repository_name,
                         message=message,
@@ -207,7 +209,8 @@ async def find_current_source_id(repository_path: str = ".") -> FindCurrentSourc
                                 monitored=source.get("monitored"),
                                 deleted_at=source.get("deleted_at"),
                             )
-                            for source in fallback_result if source.get("id") is not None
+                            for source in fallback_result
+                            if source.get("id") is not None
                         ],
                     )
 

packages/gg_api_core/src/gg_api_core/tools/list_honeytokens.py

@@ -1,4 +1,3 @@
-import json
 import logging
 from typing import Any
 
@@ -34,15 +33,17 @@ class ListHoneytokensResult(BaseModel):
     """Result from listing honeytokens."""
 
     honeytokens: list[dict[str, Any]] = Field(description="List of honeytoken objects")
-    next_cursor: str | None = Field(default=None, description="Cursor for fetching the next page (null if no more results)")
+    next_cursor: str | None = Field(
+        default=None, description="Cursor for fetching the next page (null if no more results)"
+    )
 
 
 async def list_honeytokens(params: ListHoneytokensParams) -> ListHoneytokensResult:
     """
     List honeytokens from the GitGuardian dashboard with filtering options.
 
-    IMPORTANT: When the user asks for "my honeytokens", "my tokens", "honeytokens I created", 
-    "honeytokens created by me", or similar possessive/personal references, you MUST set mine=True 
+    IMPORTANT: When the user asks for "my honeytokens", "my tokens", "honeytokens I created",
+    "honeytokens created by me", or similar possessive/personal references, you MUST set mine=True
     to filter to only the current user's honeytokens.
 
     Args:
@@ -75,7 +76,6 @@ async def list_honeytokens(params: ListHoneytokensParams) -> ListHoneytokensResu
         except Exception as e:
             logger.warning(f"Failed to get current user info for 'mine' filter: {str(e)}")
 
-
     try:
         response = await client.list_honeytokens(
             status=params.status,
@@ -91,7 +91,7 @@ async def list_honeytokens(params: ListHoneytokensParams) -> ListHoneytokensResu
 
         honeytokens_data = response["data"]
         next_cursor = response["cursor"]
-        
+
         logger.debug(f"Found {len(honeytokens_data)} honeytokens")
         return ListHoneytokensResult(honeytokens=honeytokens_data, next_cursor=next_cursor)
     except Exception as e:

packages/gg_api_core/src/gg_api_core/tools/list_incidents.py

@@ -121,16 +121,21 @@ class ListIncidentsParams(BaseModel):
     exclude_tags: list[str | TagNames] | None = Field(
         default=cast(list[str | TagNames], DEFAULT_EXCLUDED_TAGS), description="Exclude incidents with these tag names."
     )
-    status: list[str | IncidentStatus] | None = Field(default=cast(list[str | IncidentStatus], DEFAULT_STATUSES), description="Filter by status (list of status names)")
+    status: list[str | IncidentStatus] | None = Field(
+        default=cast(list[str | IncidentStatus], DEFAULT_STATUSES),
+        description="Filter by status (list of status names)",
+    )
     mine: bool = Field(
         default=False,
         description="If True, fetch only incidents assigned to the current user. Set to False to get all incidents.",
     )
     severity: list[str | IncidentSeverity] | None = Field(
-        default=cast(list[str | IncidentSeverity], DEFAULT_SEVERITIES), description="Filter by severity (list of severity names)"
+        default=cast(list[str | IncidentSeverity], DEFAULT_SEVERITIES),
+        description="Filter by severity (list of severity names)",
     )
     validity: list[str | IncidentValidity] | None = Field(
-        default=cast(list[str | IncidentValidity], DEFAULT_VALIDITIES), description="Filter by validity (list of validity names)"
+        default=cast(list[str | IncidentValidity], DEFAULT_VALIDITIES),
+        description="Filter by validity (list of validity names)",
     )
 
 
@@ -220,7 +225,7 @@ async def list_incidents(params: ListIncidentsParams) -> ListIncidentsResult | L
         response = await client.list_incidents(**api_params)
         incidents_data = response["data"]
         next_cursor = response["cursor"]
-        
+
         count = len(incidents_data)
         return ListIncidentsResult(
             source_id=params.source_id,

packages/gg_api_core/src/gg_api_core/tools/list_repo_occurrences.py

@@ -49,7 +49,9 @@ class ListRepoOccurrencesFilters(BaseModel):
         default=DEFAULT_EXCLUDED_TAGS,
         description="Exclude occurrences with these tag names. Pass empty list to disable filtering.",
     )
-    status: list[IncidentStatus] | None = Field(default=DEFAULT_STATUSES, description="Filter by status (list of status names)")
+    status: list[IncidentStatus] | None = Field(
+        default=DEFAULT_STATUSES, description="Filter by status (list of status names)"
+    )
     severity: list[IncidentSeverity] | None = Field(
         default=DEFAULT_SEVERITIES, description="Filter by severity (list of severity names)"
     )
@@ -246,7 +248,7 @@ async def list_repo_occurrences(
         occurrences_data = result["data"]
         next_cursor = result["cursor"]
         has_more = result["has_more"]
-        
+
         count = len(occurrences_data)
         return ListRepoOccurrencesResult(
             repository=params.repository_name,

packages/gg_api_core/src/gg_api_core/tools/list_users.py

@@ -2,7 +2,6 @@
 import logging
 from typing import Any
 
-from fastmcp.exceptions import ToolError
 from pydantic import BaseModel, Field
 
 from gg_api_core.utils import get_client
@@ -90,6 +89,4 @@ async def list_users(params: ListUsersParams) -> ListUsersResult:
         # Single page request
         result = await client.list_members(params=query_params)
         logger.debug(f"Found {len(result['data'])} members")
-        return ListUsersResult(
-            members=result["data"], total_count=len(result["data"]), next_cursor=result["cursor"]
-        )
+        return ListUsersResult(members=result["data"], total_count=len(result["data"]), next_cursor=result["cursor"])

packages/gg_api_core/src/gg_api_core/tools/remediate_secret_incidents.py

@@ -107,10 +107,8 @@ async def remediate_secret_incidents(
     try:
         # Use the list_repo_occurrences_params and update with parent-level repository info
         if params.list_repo_occurrences_params is None:
-            return RemediateSecretIncidentsError(
-                error="list_repo_occurrences_params is required", sub_tools_results={}
-            )
-        
+            return RemediateSecretIncidentsError(error="list_repo_occurrences_params is required", sub_tools_results={})
+
         occurrences_params = params.list_repo_occurrences_params.model_copy(
             update={
                 "repository_name": params.repository_name or params.list_repo_occurrences_params.repository_name,

packages/gg_api_core/src/gg_api_core/tools/write_custom_tags.py

@@ -28,11 +28,11 @@ class WriteCustomTagsParams(BaseModel):
 async def write_custom_tags(params: WriteCustomTagsParams):
     """
     Create or delete custom tags in the GitGuardian dashboard.
-    
+
     For creating tags, use the "key" or "key:value" format:
     - "env" creates a label without a value
     - "env:prod" creates a label with key="env" and value="prod"
-    
+
     For deleting tags:
     1. First call read_custom_tags to list all available tags and get their IDs
     2. Then call this function with action="delete_tag" and the specific tag_id
@@ -89,11 +89,11 @@ class UpdateOrCreateIncidentCustomTagsParams(BaseModel):
 async def update_or_create_incident_custom_tags(params: UpdateOrCreateIncidentCustomTagsParams) -> dict[str, Any]:
     """
     Update a secret incident with custom tags, creating tags if they don't exist.
-    
+
     Custom tags can be in two formats:
     - "key" (creates a label without a value)
     - "key:value" (creates a label with a value)
-    
+
     Args:
         params: UpdateOrCreateIncidentCustomTagsParams model containing custom tags configuration
 
@@ -114,18 +114,18 @@ async def update_or_create_incident_custom_tags(params: UpdateOrCreateIncidentCu
                 # Tag is just a key with no value
                 key = tag
                 value = None
-            
+
             # Create the tag if it doesn't exist
             try:
                 await client.create_custom_tag(key, value)
                 logger.debug(f"Created custom tag: {key}={value}")
             except Exception as e:
                 # Tag might already exist, which is fine
                 logger.debug(f"Tag {key}={value} may already exist: {str(e)}")
-            
+
             # Add to parsed tags list in the format expected by update_incident
             parsed_tags.append({"key": key, "value": value})
-        
+
         # Update the incident with the custom tags
         result = await client.update_incident(
             incident_id=str(params.incident_id),