Fix all lint errors with minimal changes

Author: google-labs-jules[bot]

src/bot/agents/command_center_agent.py

@@ -121,9 +121,15 @@ async def get_agent_response(self, message: discord.Message, thread: discord.Thr
         """
         Send the user's request to Gemini with automatic function‐calling.
 
-        then returns a tuple of:
-          1) the assistant's final text
-          2) a list of all actions taken, e.g. [{"name": ..., "args": ...}, {"tool": ..., "result": ...}, ...]
+        Args:
+            message (discord.Message): The user's message to the agent.
+            thread (discord.Thread): The thread where the conversation is happening.
+
+        Returns:
+            tuple[str, list[dict]]: A tuple of:
+                1) the assistant's final text
+                2) a list of all actions taken, e.g. [{"name": ..., "args": ...}, {"tool": ..., "result": ...}, ...]
+
         """
         async with streamablehttp_client(command_center_config.mcp_server_url, headers=self.headers) as (
             read_stream,
@@ -173,6 +179,9 @@ async def _get_thread_history(self, thread: discord.Thread) -> list[dict]:
         Args:
             thread: The thread to get history for.
 
+        Returns:
+            list[dict]: A list of dictionaries representing the message history.
+
         """
         history = []
         async for msg in thread.history(limit=command_center_config.agent_history_size):

src/bot/agents/tools/restart_service.py

@@ -27,6 +27,9 @@ async def restart_service(service_name: str) -> Dict:
     Args:
         service_name (str): The name of the service to restart.
 
+    Returns:
+        Dict: A dictionary containing the result of the operation.
+
     """
     async with streamablehttp_client(command_center_config.mcp_server_url, headers=headers) as (
         read_stream,

src/bot/agents/tools/trigger_user.py

@@ -27,6 +27,9 @@ async def trigger_user(message: str) -> Dict:
     Args:
         message (str): The message to send to user.
 
+    Returns:
+        Dict: A dictionary containing the result of the operation.
+
     """
     async with streamablehttp_client(command_center_config.mcp_server_url, headers=headers) as (
         read_stream,

src/bot/cogs/command_center.py

@@ -49,6 +49,10 @@ def cog_unload(self):
     async def on_message(self, message: discord.Message):
         """
         Trigger MCP logic when a message is sent in the command center channel and the bot is mentioned.
+
+        Args:
+            message (discord.Message): The message that triggered the event.
+
         """
         # Ignore bot's own messages
         if message.author.bot:

src/bot/core/command_center.py

@@ -21,6 +21,12 @@
 async def handle_webhook_input(bot: commands.Bot, channel: discord.TextChannel, data: dict):
     """
     Entry point for webhook sent to the command center.
+
+    Args:
+        bot (commands.Bot): The Discord bot instance.
+        channel (discord.TextChannel): The channel to send messages to.
+        data (dict): The webhook data.
+
     """
     if data.get("type", "").lower() == "service_down":
         msg = await channel.send(

src/bot/prompt_loaders/mcp.py

@@ -23,6 +23,10 @@ class McpPromptLoader:
     def load_prompts(prompts: List[Prompt]):
         """
         Register all prompt callbacks/views.
+
+        Args:
+            prompts (List[Prompt]): A list of prompts to be loaded.
+
         """
         # register callbacks
         for prompt in prompts:
@@ -32,6 +36,10 @@ def load_prompts(prompts: List[Prompt]):
     def unload_prompts(prompts: List[Prompt]):
         """
         Unregister all prompt callbacks/views.
+
+        Args:
+            prompts (List[Prompt]): A list of prompts to be unloaded.
+
         """
         # unregister callbacks
         for prompt in prompts: