Sweep: create a new agent to be used in ticket_utils.py

[wwzeng1]

Details

The agent should filter unnecessary terms out of the search query to be sent into lexical search. Use a prompt to do this, using name_agent.py as a reference

[sweep-nightly[bot]]

✨ Track Sweep's progress on our progress dashboard!

💎 Sweep Pro: I'm using GPT-4. You have unlimited GPT-4 tickets. (tracking ID: 7bc8e3e248)

---

[!TIP] I'll email you at william@sweep.dev when I complete this pull request!

---

Actions (click)

• [x] ↻ Restart Sweep

GitHub Actions✓

Here are the GitHub Actions logs prior to making any changes:

Sandbox logs for 0db6b3e

Checking sweepai/utils/ticket_utils.py for syntax errors... ✅ sweepai/utils/ticket_utils.py has no syntax errors! 1/1 ✓

Checking sweepai/utils/ticket_utils.py for syntax errors...
✅ sweepai/utils/ticket_utils.py has no syntax errors!

Sandbox passed on the latest main, so sandbox checks will be enabled for this issue.

---

Step 1: 🔎 Searching

I found the following snippets in your repository. I will now analyze these snippets and come up with a plan.

Some code snippets I think are relevant in decreasing order of relevance (click to expand). If some file is missing from here, you can mention the path in the ticket description.

https://github.com/sweepai/sweep/blob/0db6b3ed20bc3c50be0fff20bb33516d7fa8811a/sweepai/core/prompts.py#L853-L861 https://github.com/sweepai/sweep/blob/0db6b3ed20bc3c50be0fff20bb33516d7fa8811a/sweepai/utils/ticket_utils.py#L92-L113 https://github.com/sweepai/sweep/blob/0db6b3ed20bc3c50be0fff20bb33516d7fa8811a/sweepai/agents/name_agent.py#L5-L55

---

Step 2: ⌨️ Coding

• [X] Create sweepai/agents/query_filter_agent.py ✓ https://github.com/sweepai/sweep/commit/1b24e73308bdcba66960dde23ceff6ae2811eccf Edit

  Create sweepai/agents/query_filter_agent.py with contents:
  • Create a new Python file named `query_filter_agent.py` in the `sweepai/agents/` directory.
  • Import the `ChatGPT` class from `sweepai/core/chat.py`.
  • Define a new class `QueryFilterAgent` that inherits from `ChatGPT`.
  • Inside the `QueryFilterAgent` class, define a method `filter_search_query` that takes parameters such as `search_query`, `title`, `summary`, and `replies_text`.
  • Construct a prompt within `filter_search_query` that instructs the AI to filter out unnecessary terms from the search query. Use the `doc_query_rewriter_prompt` from `sweepai/core/prompts.py` as a reference for structuring the prompt.
  • Call the `chat` method of the `ChatGPT` class with the constructed prompt to get the AI's response.
  • Process the AI's response to extract the filtered search query.
  • Return the filtered search query from the `filter_search_query` method.

• [X] Running GitHub Actions for sweepai/agents/query_filter_agent.py ✓ Edit

  Check sweepai/agents/query_filter_agent.py with contents:

Ran GitHub Actions for 1b24e73308bdcba66960dde23ceff6ae2811eccf:
• Vercel Preview Comments: ✓

• [ ] Modify sweepai/utils/ticket_utils.py ▶ Edit

  Modify sweepai/utils/ticket_utils.py with contents:
  • Import the `QueryFilterAgent` class from the newly created `sweepai/agents/query_filter_agent.py`.
  • Inside the `fetch_relevant_files` function, after constructing the `search_query` in line 110, instantiate a `QueryFilterAgent` object.
  • Call the `filter_search_query` method of the `QueryFilterAgent` object, passing the `title`, `summary`, and `replies_text` as arguments.
  • Replace the original `search_query` with the filtered search query returned by the `filter_search_query` method.
  • Ensure that the filtered `search_query` is used in subsequent operations within the `fetch_relevant_files` function.

• [ ] Running GitHub Actions for sweepai/utils/ticket_utils.py ▶ Edit

  Check sweepai/utils/ticket_utils.py with contents:

---

Step 3: 🔁 Code Review

Working on it...

---

🎉 Latest improvements to Sweep:

• We just released a dashboard to track Sweep's progress on your issue in real-time, showing every stage of the process – from search to planning and coding.
• Sweep uses OpenAI's latest Assistant API to plan code changes and modify code! This is 3x faster and significantly more reliable as it allows Sweep to edit code and validate the changes in tight iterations, the same way as a human would.
• Try using the GitHub issues extension to create Sweep issues directly from your editor! GitHub Issues and Pull Requests.

---

💡 To recreate the pull request edit the issue title or description. To tweak the pull request, leave a comment on the pull request. ^{Join Our Discord}