fix a bug

Author: misrasaurabh1

codeflash/context/code_context_extractor.py

@@ -322,16 +322,14 @@ def extract_code_markdown_context_from_files(
             continue
         if code_context.strip():
             if code_context_type != CodeContextType.HASHING:
-                code_context = (
-                    add_needed_imports_from_module(
-                        src_module_code=original_code,
-                        dst_module_code=code_context,
-                        src_path=file_path,
-                        dst_path=file_path,
-                        project_root=project_root_path,
-                        helper_functions=list(
-                            helpers_of_fto.get(file_path, set()) | helpers_of_helpers.get(file_path, set())
-                        ),
+                code_context = add_needed_imports_from_module(
+                    src_module_code=original_code,
+                    dst_module_code=code_context,
+                    src_path=file_path,
+                    dst_path=file_path,
+                    project_root=project_root_path,
+                    helper_functions=list(
+                        helpers_of_fto.get(file_path, set()) | helpers_of_helpers.get(file_path, set())
                     ),
                 )
             code_string_context = CodeString(code=code_context, file_path=file_path.relative_to(project_root_path))
@@ -357,15 +355,13 @@ def extract_code_markdown_context_from_files(
 
         if code_context.strip():
             if code_context_type != CodeContextType.HASHING:
-                code_context = (
-                    add_needed_imports_from_module(
-                        src_module_code=original_code,
-                        dst_module_code=code_context,
-                        src_path=file_path,
-                        dst_path=file_path,
-                        project_root=project_root_path,
-                        helper_functions=list(helpers_of_helpers_no_overlap.get(file_path, set())),
-                    ),
+                code_context = add_needed_imports_from_module(
+                    src_module_code=original_code,
+                    dst_module_code=code_context,
+                    src_path=file_path,
+                    dst_path=file_path,
+                    project_root=project_root_path,
+                    helper_functions=list(helpers_of_helpers_no_overlap.get(file_path, set())),
                 )
             code_string_context = CodeString(code=code_context, file_path=file_path.relative_to(project_root_path))
             code_context_markdown.code_strings.append(code_string_context)

tests/test_git_utils.py

@@ -11,30 +11,35 @@ class TestGitUtils(unittest.TestCase):
     def test_test_get_repo_owner_and_name(self, mock_get_remote_url):
         # Test with a standard GitHub HTTPS URL
         mock_get_remote_url.return_value = "https://github.com/owner/repo.git"
+        get_repo_owner_and_name.cache_clear()
         owner, repo_name = get_repo_owner_and_name()
         assert owner == "owner"
         assert repo_name == "repo"
 
         # Test with a GitHub SSH URL
         mock_get_remote_url.return_value = "[email protected]:owner/repo.git"
+        get_repo_owner_and_name.cache_clear()
         owner, repo_name = get_repo_owner_and_name()
         assert owner == "owner"
         assert repo_name == "repo"
 
         # Test with another GitHub SSH URL
         mock_get_remote_url.return_value = "[email protected]:codeflash-ai/posthog.git"
+        get_repo_owner_and_name.cache_clear()
         owner, repo_name = get_repo_owner_and_name()
         assert owner == "codeflash-ai"
         assert repo_name == "posthog"
 
         # Test with a URL without the .git suffix
         mock_get_remote_url.return_value = "https://github.com/owner/repo"
+        get_repo_owner_and_name.cache_clear()
         owner, repo_name = get_repo_owner_and_name()
         assert owner == "owner"
         assert repo_name == "repo"
 
         # Test with another GitHub SSH URL
         mock_get_remote_url.return_value = "[email protected]:codeflash-ai/posthog/"
+        get_repo_owner_and_name.cache_clear()
         owner, repo_name = get_repo_owner_and_name()
         assert owner == "codeflash-ai"
         assert repo_name == "posthog"