style: apply ruff formatting for v0.1.0 release

Auto-formatted 10 files with ruff format
Minor formatting adjustments for consistency

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: jwesleye

src/basic_agent_chat_loop/components/template_manager.py

@@ -33,7 +33,9 @@ def _initialize_templates(self):
             self.prompts_dir.mkdir(parents=True, exist_ok=True)
 
             # Create sample templates
-            self._create_sample_template("explain", """# Explain this code
+            self._create_sample_template(
+                "explain",
+                """# Explain this code
 
 Please explain the following code in detail:
 
@@ -44,9 +46,12 @@ def _initialize_templates(self):
 - How it works
 - Any potential issues or improvements
 - Best practices being followed or violated
-""")
+""",
+            )
 
-            self._create_sample_template("review", """# Code Review
+            self._create_sample_template(
+                "review",
+                """# Code Review
 
 Please review the following code:
 
@@ -58,9 +63,12 @@ def _initialize_templates(self):
 - Performance considerations
 - Security concerns
 - Suggestions for improvement
-""")
+""",
+            )
 
-            self._create_sample_template("debug", """# Debug Help
+            self._create_sample_template(
+                "debug",
+                """# Debug Help
 
 I'm having trouble with this code:
 
@@ -71,9 +79,12 @@ def _initialize_templates(self):
 2. Explain why it's happening
 3. Suggest a fix
 4. Provide the corrected code
-""")
+""",
+            )
 
-            self._create_sample_template("optimize", """# Optimize This Code
+            self._create_sample_template(
+                "optimize",
+                """# Optimize This Code
 
 Please optimize the following code:
 
@@ -84,9 +95,12 @@ def _initialize_templates(self):
 - Memory efficiency
 - Code simplicity
 - Best practices
-""")
+""",
+            )
 
-            self._create_sample_template("test", """# Write Tests
+            self._create_sample_template(
+                "test",
+                """# Write Tests
 
 Please write comprehensive tests for the following code:
 
@@ -96,9 +110,12 @@ def _initialize_templates(self):
 - Unit tests for all functions
 - Edge cases and error handling
 - Test descriptions explaining what each test validates
-""")
+""",
+            )
 
-            self._create_sample_template("document", """# Add Documentation
+            self._create_sample_template(
+                "document",
+                """# Add Documentation
 
 Please add comprehensive documentation to this code:
 
@@ -109,7 +126,8 @@ def _initialize_templates(self):
 - Inline comments for complex logic
 - Usage examples
 - Type hints if missing
-""")
+""",
+            )
 
         except Exception as e:
             logger.debug(f"Could not initialize templates: {e}")

src/basic_agent_chat_loop/components/ui_components.py

@@ -124,9 +124,13 @@ def render(self) -> str:
         # Add tokens if enabled and available
         if self.show_tokens and self.total_tokens > 0:
             if self.total_tokens >= self.TOKEN_MILLIONS_THRESHOLD:
-                token_str = f"{self.total_tokens / self.TOKEN_MILLIONS_THRESHOLD:.1f}M tokens"
+                token_str = (
+                    f"{self.total_tokens / self.TOKEN_MILLIONS_THRESHOLD:.1f}M tokens"
+                )
             elif self.total_tokens >= self.TOKEN_THOUSANDS_THRESHOLD:
-                token_str = f"{self.total_tokens / self.TOKEN_THOUSANDS_THRESHOLD:.1f}K tokens"
+                token_str = (
+                    f"{self.total_tokens / self.TOKEN_THOUSANDS_THRESHOLD:.1f}K tokens"
+                )
             else:
                 token_str = f"{self.total_tokens} tokens"
             parts.append(token_str)

tests/__init__.py

@@ -1 +1 @@
-"""Tests for basic_agent_chat_loop."""
+"""Tests for basic_agent_chat_loop."""

tests/conftest.py

@@ -10,6 +10,7 @@
 if str(src_path) not in sys.path:
     sys.path.insert(0, str(src_path))
 
+
 @pytest.fixture
 def temp_config_file(tmp_path):
     """Create a temporary config file."""
@@ -25,20 +26,23 @@ def temp_config_file(tmp_path):
 """)
     return config_file
 
+
 @pytest.fixture
 def temp_aliases_file(tmp_path):
     """Create a temporary aliases file."""
     aliases_file = tmp_path / ".chat_aliases"
     return aliases_file
 
+
 @pytest.fixture
 def mock_agent():
     """Create a mock agent for testing."""
+
     class MockAgent:
         def __init__(self):
             self.name = "Test Agent"
             self.description = "A test agent"
-            self.model = type('obj', (object,), {'model_id': 'test-model'})()
+            self.model = type("obj", (object,), {"model_id": "test-model"})()
 
         def __call__(self, query):
             return {"message": "Test response"}

tests/unit/test_agent_loader.py

@@ -123,7 +123,9 @@ def test_load_agent_without_root_agent(self, tmp_path):
         agent_file = tmp_path / "no_root.py"
         agent_file.write_text("# No root_agent here")
 
-        with pytest.raises(AttributeError, match="must expose a 'root_agent' attribute"):
+        with pytest.raises(
+            AttributeError, match="must expose a 'root_agent' attribute"
+        ):
             load_agent_module(str(agent_file))
 
     def test_agent_directory_added_to_sys_path(self, mock_agent_file):

tests/unit/test_alias_manager.py

@@ -86,7 +86,9 @@ def test_add_alias_nonexistent_path(self, alias_manager):
         assert success is False
         assert "not found" in message
 
-    def test_add_alias_duplicate_without_overwrite(self, populated_alias_manager, tmp_path):
+    def test_add_alias_duplicate_without_overwrite(
+        self, populated_alias_manager, tmp_path
+    ):
         """Test adding duplicate alias without overwrite flag."""
         agent_path = tmp_path / "new_agent.py"
         agent_path.write_text("# new agent")
@@ -98,7 +100,9 @@ def test_add_alias_duplicate_without_overwrite(self, populated_alias_manager, tm
         assert "already exists" in message
         assert "--overwrite" in message
 
-    def test_add_alias_duplicate_with_overwrite(self, populated_alias_manager, tmp_path):
+    def test_add_alias_duplicate_with_overwrite(
+        self, populated_alias_manager, tmp_path
+    ):
         """Test adding duplicate alias with overwrite flag."""
         agent_path = tmp_path / "new_agent.py"
         agent_path.write_text("# new agent")

tests/unit/test_chat_config.py

@@ -318,7 +318,9 @@ def test_get_config_reload(self, temp_config_file, tmp_path, monkeypatch):
         config2 = get_config(temp_config_file, reload=True)
 
         # Should be different configuration
-        assert config1.get("features.show_tokens") != config2.get("features.show_tokens")
+        assert config1.get("features.show_tokens") != config2.get(
+            "features.show_tokens"
+        )
 
 
 class TestEdgeCases:

tests/unit/test_chat_loop_utils.py

@@ -187,7 +187,9 @@ def test_extract_none_response(self):
     def test_extract_usage_from_dict_result_bedrock(self):
         """Test extraction from AWS Bedrock style response."""
         agent = Mock()
-        with patch("basic_agent_chat_loop.chat_loop.extract_agent_metadata") as mock_extract:
+        with patch(
+            "basic_agent_chat_loop.chat_loop.extract_agent_metadata"
+        ) as mock_extract:
             mock_extract.return_value = {"model_id": "test", "tool_count": 0}
             chat_loop = ChatLoop(agent, "Test", "Desc")
 
@@ -208,7 +210,9 @@ def test_extract_usage_from_dict_result_bedrock(self):
     def test_extract_usage_from_object_anthropic(self):
         """Test extraction from Anthropic style response."""
         agent = Mock()
-        with patch("basic_agent_chat_loop.chat_loop.extract_agent_metadata") as mock_extract:
+        with patch(
+            "basic_agent_chat_loop.chat_loop.extract_agent_metadata"
+        ) as mock_extract:
             mock_extract.return_value = {"model_id": "test", "tool_count": 0}
             chat_loop = ChatLoop(agent, "Test", "Desc")
 
@@ -224,7 +228,9 @@ def test_extract_usage_from_object_anthropic(self):
     def test_extract_usage_from_dict_usage(self):
         """Test extraction from dict with usage key."""
         agent = Mock()
-        with patch("basic_agent_chat_loop.chat_loop.extract_agent_metadata") as mock_extract:
+        with patch(
+            "basic_agent_chat_loop.chat_loop.extract_agent_metadata"
+        ) as mock_extract:
             mock_extract.return_value = {"model_id": "test", "tool_count": 0}
             chat_loop = ChatLoop(agent, "Test", "Desc")
 
@@ -239,7 +245,9 @@ def test_extract_usage_from_dict_usage(self):
     def test_extract_usage_prompt_completion_tokens(self):
         """Test extraction with prompt_tokens/completion_tokens names."""
         agent = Mock()
-        with patch("basic_agent_chat_loop.chat_loop.extract_agent_metadata") as mock_extract:
+        with patch(
+            "basic_agent_chat_loop.chat_loop.extract_agent_metadata"
+        ) as mock_extract:
             mock_extract.return_value = {"model_id": "test", "tool_count": 0}
             chat_loop = ChatLoop(agent, "Test", "Desc")
 
@@ -257,7 +265,9 @@ def test_extract_usage_prompt_completion_tokens(self):
     def test_extract_usage_zero_tokens(self):
         """Test with zero token usage."""
         agent = Mock()
-        with patch("basic_agent_chat_loop.chat_loop.extract_agent_metadata") as mock_extract:
+        with patch(
+            "basic_agent_chat_loop.chat_loop.extract_agent_metadata"
+        ) as mock_extract:
             mock_extract.return_value = {"model_id": "test", "tool_count": 0}
             chat_loop = ChatLoop(agent, "Test", "Desc")
 
@@ -271,7 +281,9 @@ def test_extract_usage_zero_tokens(self):
     def test_extract_usage_partial_tokens(self):
         """Test with only input tokens."""
         agent = Mock()
-        with patch("basic_agent_chat_loop.chat_loop.extract_agent_metadata") as mock_extract:
+        with patch(
+            "basic_agent_chat_loop.chat_loop.extract_agent_metadata"
+        ) as mock_extract:
             mock_extract.return_value = {"model_id": "test", "tool_count": 0}
             chat_loop = ChatLoop(agent, "Test", "Desc")
 

tests/unit/test_template_manager.py

@@ -1,6 +1,5 @@
 """Tests for TemplateManager component."""
 
-from pathlib import Path
 
 import pytest
 
@@ -34,10 +33,7 @@ def populated_prompts_dir(prompts_dir):
 
     # Template with different heading
     (prompts_dir / "complex.md").write_text(
-        "# Complex Template\n"
-        "\n"
-        "Content: {input}\n"
-        "Additional: {extra}"
+        "# Complex Template\n\nContent: {input}\nAdditional: {extra}"
     )
 
     return prompts_dir
@@ -104,7 +100,9 @@ def test_list_with_descriptions_no_heading(self, prompts_dir):
         assert len(templates) == 1
         assert templates[0] == ("simple", "simple")  # Uses name as description
 
-    def test_list_with_descriptions_with_heading(self, prompts_dir, populated_prompts_dir):
+    def test_list_with_descriptions_with_heading(
+        self, prompts_dir, populated_prompts_dir
+    ):
         """Test listing templates with markdown headings."""
         manager = TemplateManager(prompts_dir)
         templates = manager.list_templates_with_descriptions()
@@ -143,7 +141,9 @@ def test_load_simple_template(self, prompts_dir, populated_prompts_dir):
 
         assert content == "This is a simple template."
 
-    def test_load_template_with_variable_substitution(self, prompts_dir, populated_prompts_dir):
+    def test_load_template_with_variable_substitution(
+        self, prompts_dir, populated_prompts_dir
+    ):
         """Test loading template and substituting variables."""
         manager = TemplateManager(prompts_dir)
         content = manager.load_template("review", "my_code.py")
@@ -198,7 +198,9 @@ def test_empty_template_file(self, prompts_dir):
 
     def test_template_without_input_placeholder(self, prompts_dir):
         """Test loading template without {input} placeholder."""
-        (prompts_dir / "no_placeholder.md").write_text("# Title\n\nContent without placeholder")
+        (prompts_dir / "no_placeholder.md").write_text(
+            "# Title\n\nContent without placeholder"
+        )
 
         manager = TemplateManager(prompts_dir)
         content = manager.load_template("no_placeholder", "extra text")

tests/unit/test_token_tracker.py

@@ -1,5 +1,5 @@
 """Tests for TokenTracker component."""
-import pytest
+
 from basic_agent_chat_loop.components.token_tracker import TokenTracker