-
Couldn't load subscription status.
- Fork 36
RAG Agent that calls external retrieval server #206
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
Open
xingdi-eric-yuan
wants to merge
60
commits into
main
Choose a base branch
from
external_retrieval_server
base: main
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
…move it we can do it in post processing
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This pull request introduces a new
RAGAgent(Retrieval-Augmented Generation Agent) to thedebug_gymframework, enabling retrieval-based decision-making by integrating with a standalone retrieval service. The changes include the implementation of the agent, configuration updates, and mock-based tests for validation. Below is a summary of the most important changes:New Feature: RAGAgent Implementation
RAGAgentclass indebug_gym/agents/rag_agent.py, which integrates with a retrieval service to retrieve relevant examples for decision-making. It includes configuration options for indexing methods, retrieval parameters, and caching. The agent conditionally imports the retrieval service client and raises an error if unavailable.debug_gym/agents/__init__.pyto conditionally importRAGAgentonly if the retrieval service is installed, ensuring compatibility with setups without the retrieval service.Configuration Updates
rag_agentinscripts/config_swesmith.yaml, specifying tools, retrieval parameters, and the experience trajectory path for building indexes.Code Refinements
run_agentfunction inscripts/run.pyto refine handling of previous run statuses, ensuring better control over task execution. [1] [2]Testing Enhancements
tests/agents/test_rag_agent_mock_only.pyto provide mock-based tests forRAGAgent, covering import error handling, indexing method parsing, query text extraction, configuration defaults, retrieval service client interface, and prompt-building logic. These tests ensure the agent's logic can be validated without requiring the retrieval service.