fix: resolve flake8 linting errors

Author: Eigenwise

atomic-agents/atomic_agents/agents/atomic_agent.py

@@ -2,15 +2,15 @@
 from instructor import Mode
 from instructor.processing.multimodal import Image, Audio, PDF
 from pydantic import BaseModel, Field
-from typing import Optional, Type, Generator, AsyncGenerator, get_args, get_origin, Dict, List, Callable, Any, Union
+from typing import Optional, Type, Generator, AsyncGenerator, get_args, get_origin, Dict, List, Callable, Any
 import logging
 from atomic_agents.context.chat_history import ChatHistory
 from atomic_agents.context.system_prompt_generator import (
     BaseDynamicContextProvider,
     SystemPromptGenerator,
 )
 from atomic_agents.base.base_io_schema import BaseIOSchema
-from atomic_agents.utils.token_counter import get_token_counter, TokenCountResult, TokenCountError
+from atomic_agents.utils.token_counter import get_token_counter, TokenCountResult
 import json
 
 from instructor.dsl.partial import PartialBase
@@ -690,7 +690,6 @@ def hooks_enabled(self) -> bool:
     import instructor
     import asyncio
     from rich.live import Live
-    import json
 
     def _create_schema_table(title: str, schema: Type[BaseModel]) -> Table:
         """Create a table displaying schema information.

atomic-agents/tests/agents/test_atomic_agent.py

@@ -827,7 +827,7 @@ def test_get_context_token_count_includes_schema_overhead(
     )
     agent = AtomicAgent[BasicChatInputSchema, BasicChatOutputSchema](config)
 
-    result = agent.get_context_token_count()
+    agent.get_context_token_count()
 
     # Verify count_context was called with system message that includes schema
     call_args = mock_counter_instance.count_context.call_args
@@ -860,7 +860,7 @@ def test_get_context_token_count_no_system_prompt(mock_get_token_counter, mock_i
     )
     agent = AtomicAgent[BasicChatInputSchema, BasicChatOutputSchema](config)
 
-    result = agent.get_context_token_count()
+    agent.get_context_token_count()
 
     # Even without system prompt, schema should be included (for JSON mode)
     call_args = mock_counter_instance.count_context.call_args
@@ -932,7 +932,7 @@ def hook_handler(result):
 
     agent.register_hook("token:counted", hook_handler)
 
-    result = agent.get_context_token_count()
+    agent.get_context_token_count()
 
     assert len(hook_called) == 1
     assert hook_called[0] == expected_result
@@ -965,7 +965,6 @@ def hook_handler(result):
 def test_get_context_token_count_multimodal_content(mock_get_token_counter, mock_instructor):
     """Test that multimodal content is properly serialized for token counting."""
     from instructor.processing.multimodal import Image
-    from instructor import Mode
 
     mock_counter_instance = Mock()
     mock_get_token_counter.return_value = mock_counter_instance
@@ -993,7 +992,7 @@ class MultimodalInputSchema(BaseIOSchema):
     agent.history.add_message("user", multimodal_input)
 
     # Get token count
-    result = agent.get_context_token_count()
+    agent.get_context_token_count()
 
     # Verify count_context was called
     assert mock_counter_instance.count_context.called

atomic-agents/tests/utils/test_token_counter.py

@@ -1,5 +1,5 @@
 import pytest
-from unittest.mock import Mock, patch
+from unittest.mock import patch
 from atomic_agents.utils.token_counter import (
     TokenCounter,
     TokenCountResult,