refactor: clean up code formatting and improve readability across multiple files (#31)

Author: khongtrunght

src/ii_agent/agents/anthropic_fc.py

@@ -105,7 +105,10 @@ async def _process_messages(self):
                         )
 
                     # Only send to websocket if this is not an event from the client and websocket exists
-                    if message.type != EventType.USER_MESSAGE and self.websocket is not None:
+                    if (
+                        message.type != EventType.USER_MESSAGE
+                        and self.websocket is not None
+                    ):
                         try:
                             await self.websocket.send_json(message.model_dump())
                         except Exception as e:
@@ -169,7 +172,9 @@ def run_impl(
                 if ext == "jpg":
                     ext = "jpeg"
                 if ext in ["png", "gif", "jpeg", "webp"]:
-                    base64_image = encode_image(str(self.workspace_manager.workspace_path(file)))
+                    base64_image = encode_image(
+                        str(self.workspace_manager.workspace_path(file))
+                    )
                     image_blocks.append(
                         {
                             "source": {
@@ -215,7 +220,7 @@ def run_impl(
                     tools=all_tool_params,
                     system_prompt=self.system_prompt,
                 )
-                
+
                 if len(model_response) == 0:
                     model_response = [TextResult(text=COMPLETE_MESSAGE)]
 
@@ -226,7 +231,6 @@ def run_impl(
                 pending_tool_calls = self.history.get_pending_tool_calls()
 
                 if len(pending_tool_calls) == 0:
-                    
                     # No tools were called, so assume the task is complete
                     self.logger_for_agent_logs.info("[no tools were called]")
                     self.message_queue.put_nowait(

src/ii_agent/browser/browser.py

@@ -214,18 +214,24 @@ async def _on_page_change(self, page: Page):
 
         self._cdp_session = await self.context.new_cdp_session(page)
 
-        # set viewport size 
-        await self._cdp_session.send("Emulation.setDeviceMetricsOverride", {
-            "width": self.config.viewport_size["width"],
-            "height": self.config.viewport_size["height"],
-            "deviceScaleFactor": 1,
-            "mobile": False,
-        })
+        # set viewport size
+        await self._cdp_session.send(
+            "Emulation.setDeviceMetricsOverride",
+            {
+                "width": self.config.viewport_size["width"],
+                "height": self.config.viewport_size["height"],
+                "deviceScaleFactor": 1,
+                "mobile": False,
+            },
+        )
         # Optional: adjust visible size (for headless rendering)
-        await self._cdp_session.send("Emulation.setVisibleSize", {
-            "width": self.config.viewport_size["width"],
-            "height": self.config.viewport_size["height"],
-        })
+        await self._cdp_session.send(
+            "Emulation.setVisibleSize",
+            {
+                "width": self.config.viewport_size["width"],
+                "height": self.config.viewport_size["height"],
+            },
+        )
 
         self.current_page = page
 
@@ -484,18 +490,24 @@ async def get_cdp_session(self):
         ):
             self._cdp_session = await self.context.new_cdp_session(self.current_page)
 
-            # set viewport size 
-            await self._cdp_session.send("Emulation.setDeviceMetricsOverride", {
-                "width": self.config.viewport_size["width"],
-                "height": self.config.viewport_size["height"],
-                "deviceScaleFactor": 1,
-                "mobile": False,
-            })
+            # set viewport size
+            await self._cdp_session.send(
+                "Emulation.setDeviceMetricsOverride",
+                {
+                    "width": self.config.viewport_size["width"],
+                    "height": self.config.viewport_size["height"],
+                    "deviceScaleFactor": 1,
+                    "mobile": False,
+                },
+            )
             # Optional: adjust visible size (for headless rendering)
-            await self._cdp_session.send("Emulation.setVisibleSize", {
-                "width": self.config.viewport_size["width"],
-                "height": self.config.viewport_size["height"],
-            })
+            await self._cdp_session.send(
+                "Emulation.setVisibleSize",
+                {
+                    "width": self.config.viewport_size["width"],
+                    "height": self.config.viewport_size["height"],
+                },
+            )
 
             # Store reference to the page this session belongs to
             self._cdp_session._page = self.current_page
@@ -547,15 +559,15 @@ async def get_storage_state(self) -> dict[str, Any]:
     async def handle_pdf_url_navigation(self):
         page = await self.get_current_page()
         if is_pdf_url(page.url):
-            await asyncio.sleep(5) # Long sleep to ensure PDF is loaded
+            await asyncio.sleep(5)  # Long sleep to ensure PDF is loaded
             await page.keyboard.press("Escape")
             await asyncio.sleep(0.1)
             await page.keyboard.press("Control+\\")
             await asyncio.sleep(0.1)
             await page.mouse.click(
                 self.config.viewport_size["width"] * 0.75,  # Right side of screen
-                self.config.viewport_size["height"] * 0.25   # Upper portion
+                self.config.viewport_size["height"] * 0.25,  # Upper portion
             )
-            
+
         state = await self.update_state()
-        return state
+        return state

src/ii_agent/llm/context_manager/file_based.py

@@ -23,7 +23,9 @@ class FileBasedContextManager(ContextManager):
     TRUNCATED_TOOL_INPUT_MSG = (
         "[Truncated...re-run tool if you need to see input/output again.]"
     )
-    TRUNCATED_FILE_MSG = "[Truncated...content saved to {relative_path}. You can view it if needed.]"
+    TRUNCATED_FILE_MSG = (
+        "[Truncated...content saved to {relative_path}. You can view it if needed.]"
+    )
 
     def __init__(
         self,
@@ -111,12 +113,18 @@ def apply_truncation_if_needed(
                             content_hash = self._get_content_hash(message.tool_output)
                             if message.tool_name == VisitWebpageTool.name:
                                 # NOTE: assume that the previous message is a tool call
-                                previous_message = truncated_message_lists[turn_idx - 1][0]
+                                previous_message = truncated_message_lists[
+                                    turn_idx - 1
+                                ][0]
                                 if isinstance(previous_message, ToolCall):
-                                    url = previous_message.tool_input.get("url", "unknown_url")
+                                    url = previous_message.tool_input.get(
+                                        "url", "unknown_url"
+                                    )
                                 else:
                                     url = "unknown_url"
-                                    print(f"Previous message is not a tool call: {previous_message}")
+                                    print(
+                                        f"Previous message is not a tool call: {previous_message}"
+                                    )
                                 filename = self._generate_filename_from_url(
                                     url, content_hash
                                 )
@@ -143,7 +151,9 @@ def apply_truncation_if_needed(
 
                             # Update message with reference to file
                             message.tool_output = self.TRUNCATED_FILE_MSG.format(
-                                relative_path=self.workspace_manager.relative_path(filepath),
+                                relative_path=self.workspace_manager.relative_path(
+                                    filepath
+                                ),
                             )
                             self.logger.info(f"Saved {filename} to {filepath}")
                         else:

src/ii_agent/prompts/system_prompt.py

@@ -204,4 +204,4 @@
 </tool_use_rules>
 
 Today is {datetime.now().strftime("%Y-%m-%d")}. The first step of a task is to use sequential thinking module to plan the task. then regularly update the todo.md file to track the progress.
-"""
+"""

src/ii_agent/tools/advanced_tools/gemini/audio_tool.py

@@ -63,7 +63,7 @@ class AudioUnderstandingTool(GeminiTool):
 - Describe, summarize, or answer questions about audio content
 - Analyze specific segments of the audio
 
-Provide one query at a time. Supported formats: {', '.join(SUPPORTED_FORMATS)}
+Provide one query at a time. Supported formats: {", ".join(SUPPORTED_FORMATS)}
 """
 
     input_schema = {

src/ii_agent/tools/advanced_tools/gemini/base.py

@@ -2,10 +2,8 @@
 
 from typing import Optional
 from google import genai
-from google.genai import types
 from ii_agent.tools.base import (
     LLMTool,
-    ToolImplOutput,
 )
 from ii_agent.utils import WorkspaceManager
 

src/ii_agent/tools/bash_tool.py

@@ -429,4 +429,4 @@ def create_docker_bash_tool(
         cwd=cwd,
         command_filters=[docker_filter],
         additional_banned_command_strs=additional_banned_command_strs,
-    )
+    )

src/ii_agent/tools/browser_tools/base.py

@@ -37,4 +37,4 @@ def run_impl(
         message_history: Optional[MessageHistory] = None,
     ) -> ToolImplOutput:
         loop = get_event_loop()
-        return loop.run_until_complete(self._run(tool_input, message_history))
+        return loop.run_until_complete(self._run(tool_input, message_history))

src/ii_agent/tools/browser_tools/click.py

@@ -57,5 +57,5 @@ async def _run(
 
         state = await self.browser.update_state()
         state = await self.browser.handle_pdf_url_navigation()
-        
+
         return utils.format_screenshot_tool_output(state.screenshot, msg)

src/ii_agent/tools/browser_tools/dropdown.py

@@ -83,7 +83,7 @@ async def _run(
         msg = "\n".join(formatted_options)
         msg += "\nIf you decide to use this select element, use the exact option name in select_dropdown_option"
         state = await self.browser.update_state()
-        
+
         return utils.format_screenshot_tool_output(state.screenshot, msg)