Merge pull request #9 from blopit/fix-import-errors

Fix import errors

Author: blopit

requirements.txt

@@ -7,6 +7,7 @@ pytz>=2024.1  # Added for timezone support
 aiohttp>=3.9.3
 requests>=2.32.3
 playwright>=1.50.0
+html5lib>=1.1  # Added for HTML parsing
 
 # Search Engine
 duckduckgo-search>=7.4.3

tests/test_plan_exec_llm.py

@@ -6,10 +6,8 @@
 from pathlib import Path
 import sys
 
-# Add the parent directory to the Python path so we can import the module
-sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 from tools.plan_exec_llm import load_environment, read_plan_status, read_file_content, create_llm_client, query_llm
-from tools.plan_exec_llm import TokenUsage
+from tools.token_tracker import TokenUsage
 
 class TestPlanExecLLM(unittest.TestCase):
     def setUp(self):

tools/__init__.py

@@ -12,7 +12,7 @@
 from typing import Optional, Union, List
 import mimetypes
 import time
-from token_tracker import TokenUsage, APIResponse, get_token_tracker
+from .token_tracker import TokenUsage, APIResponse, get_token_tracker
 
 def load_environment():
     """Load environment variables from .env files in order of precedence"""

tools/llm_api.py

@@ -7,7 +7,7 @@
 from dotenv import load_dotenv
 import sys
 import time
-from token_tracker import TokenUsage, APIResponse, get_token_tracker
+from .token_tracker import TokenUsage, APIResponse, get_token_tracker
 
 STATUS_FILE = '.cursorrules'