Sweep: Add tests for context agent

[sweep-nightly]

Description

This pull request introduces a significant enhancement to the sweepai project by adding unit tests for the context pruning functionality and refactoring the ripgrep command execution into a separate function. These changes aim to improve the maintainability and testability of the codebase, ensuring that the context pruning logic works as expected and can be easily extended in the future.

Summary

• Refactored the execution of the ripgrep command into a new function run_ripgrep_command in sweepai/core/context_pruning.py to streamline the process of searching code entities within a repository.
• Added a comprehensive suite of unit tests in tests/test_context_pruning.py covering key functionalities such as building the full hierarchy of files, loading a graph from a file, and retrieving relevant context based on a query. These tests ensure the robustness and reliability of the context pruning feature.
• Enhanced code readability and maintainability by removing duplicated ripgrep command execution logic and centralizing it into a single, reusable function.
• The new tests contribute to a safer development environment, allowing for future changes to be made with confidence that the core functionality remains unaffected.

Fixes #3493.

---

🎉 Latest improvements to Sweep:

• New dashboard launched for real-time tracking of Sweep issues, covering all stages from search to coding.
• Integration of OpenAI's latest Assistant API for more efficient and reliable code planning and editing, improving speed by 3x.
• Use the GitHub issues extension for creating Sweep issues directly from your editor.

---

💡 To get Sweep to edit this pull request, you can:

• Comment below, and Sweep can edit the entire PR
• Comment on a file, Sweep will only modify the commented file
• Edit the original issue to get Sweep to recreate the PR from scratch

This is an automated message generated by Sweep AI.

[sweep-nightly]

Rollback Files For Sweep

• Rollback changes to tests/test_context_pruning.py
• Rollback changes to sweepai/core/context_pruning.py

This is an automated message generated by Sweep AI.

[sweep-nightly]

Apply Sweep Rules to your PR?

• Apply: We should use loguru for error logging. If the log is inside an exception, use logger.exception to add tracebacks, where logger is imported from loguru. Use f-strings for string formatting in logger calls (e.g. logger.info(f'Hello {name}') instead of logger.info('Hello {name}', name=name)).
• Apply: There should be no debug log or print statements in production code.
• Apply: All functions should have parameters and output annotated with type hints. Use list, tuple and dict instead of typing.List, typing.Tuple and typing.dict.
• Apply: Leftover TODOs in the code should be handled.
• Apply: All new business logic should have corresponding unit tests in the same directory. For example, sweepai/api_test.py tests sweepai/api.py. Use unittest and unittest.mock as required.
• Apply: Any clearly inefficient or repeated code should be optimized or refactored.
• Apply: Remove any comments before code that are obvious. For example # this prints hello world; print('hello world').

This is an automated message generated by Sweep AI.

[vercel]

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name	Status	Preview	Comments	Updated (UTC)
sweep-docs	✅ Ready (Inspect)	Visit Preview	💬 Add feedback	Apr 30, 2024 1:41am