Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Sweep: Add tests for context agent #3492

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

sweep-nightly[bot]
Copy link
Contributor

@sweep-nightly sweep-nightly bot commented Apr 8, 2024

Description

This pull request introduces unit tests for the context agent within the SweepAI project, specifically targeting the context_pruning.py module. It aims to ensure the reliability and correctness of the context agent's functionality, particularly in handling function calls and retrieving relevant context for given queries.

Summary

  • Added MagicMock and patch from unittest.mock to mock dependencies and function calls during testing.
  • Implemented a test for handle_function_call to verify that it correctly processes a function call, modifies the repository context manager as expected, and asserts the success of the operation.
  • Implemented a test for get_relevant_context to ensure it accurately retrieves and returns the relevant context snippets for a given query by mocking the context_dfs function and asserting the correct snippets are returned.
  • These tests are crucial for maintaining the integrity of the context agent's operations, especially when handling complex queries and function calls within the sweepai/core/context_pruning.py module.

Fixes #3491.


🎉 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.

Copy link
Contributor Author

sweep-nightly bot commented Apr 8, 2024

Rollback Files For Sweep

  • Rollback changes to sweepai/core/context_pruning.py

This is an automated message generated by Sweep AI.

Copy link
Contributor Author

sweep-nightly bot commented Apr 8, 2024

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.

Copy link

vercel bot commented Apr 8, 2024

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 8, 2024 5:56am

@sweep-nightly sweep-nightly bot added the sweep Assigns Sweep to an issue or pull request. label Apr 8, 2024
@sweep-nightly sweep-nightly bot mentioned this pull request Apr 8, 2024
2 tasks
@kevinlu1248 kevinlu1248 marked this pull request as ready for review April 8, 2024 05:58
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
sweep Assigns Sweep to an issue or pull request.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Add tests for context agent
0 participants