From 58500569ba1992abd415f41529889d7be31cbed8 Mon Sep 17 00:00:00 2001 From: Shrenil Patel Date: Wed, 26 Feb 2025 02:12:22 -0500 Subject: [PATCH 1/2] [Cursor] Fix import errors and add missing dependencies - Fix relative imports in llm_api.py and plan_exec_llm.py - Add html5lib dependency to requirements.txt - Add __init__.py files to make tools and tests proper Python packages --- requirements.txt | 1 + tools/__init__.py | 0 tools/llm_api.py | 2 +- tools/plan_exec_llm.py | 2 +- 4 files changed, 3 insertions(+), 2 deletions(-) create mode 100644 tools/__init__.py diff --git a/requirements.txt b/requirements.txt index 108b74c..807c9a3 100644 --- a/requirements.txt +++ b/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 diff --git a/tools/__init__.py b/tools/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tools/llm_api.py b/tools/llm_api.py index 03c0178..fee4954 100644 --- a/tools/llm_api.py +++ b/tools/llm_api.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""" diff --git a/tools/plan_exec_llm.py b/tools/plan_exec_llm.py index c0c2f61..9861924 100644 --- a/tools/plan_exec_llm.py +++ b/tools/plan_exec_llm.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' From c6dd35fbb589f723e316dfbdf7c8e5e751c38782 Mon Sep 17 00:00:00 2001 From: Shrenil Patel Date: Wed, 26 Feb 2025 02:13:18 -0500 Subject: [PATCH 2/2] [Cursor] Update imports in test_plan_exec_llm.py --- tests/test_plan_exec_llm.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/test_plan_exec_llm.py b/tests/test_plan_exec_llm.py index 07f2a8c..bf54828 100644 --- a/tests/test_plan_exec_llm.py +++ b/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):