feat. Integrated agent_rag_handler with the agent class #903
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
- Introduced a new example demonstrating RAG integration with Swarms framework. - Added SimpleMemoryStore class for in-memory storage and querying. - Enhanced Agent class with RAG configuration and methods for RAG operations. - Updated RAGConfig to include default relevance keywords for insights extraction.
harshalmore31
changed the title
![github-advanced-security[bot]](https://avatars.githubusercontent.com/in/57789?s=60&v=4){kind=small}
| def __init__(self): | ||
| self.memories = [] | ||
|
|
||
| def add(self, content: str, metadata: dict = None) -> bool: |
Check failure
Code scanning / Pyre
Incompatible variable type Error
|
|
||
| return self.rag_handler.query_memory(query, context_type) | ||
|
|
||
| def tool_execution_retry(self, response: any, loop_count: int): |
Check failure
Code scanning / Pyre
Parsing failure Error
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
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 comprehensive integration of Retrieval-Augmented Generation (RAG) functionality into the Swarms framework. It includes a new example script (
simple_rag_example.py), updates to theAgentclass to support RAG operations, and enhancements to theAgentRAGHandler. These changes aim to improve the framework's ability to retrieve and utilize external knowledge dynamically.RAG Integration Enhancements:
simple_rag_example.py, demonstrating the usage of RAG with a simple memory store and Swarms Agent. The example includes memory querying, manual saves, and integration of RAG into agent workflows.Agentclass with methods for enabling, disabling, and managing RAG functionality, including memory queries, saving to memory, and retrieving statistics. Legacy methods were replaced with new RAG-specific handlers. [1] [2] [3]AgentRAGHandlerto ensure default relevance keywords are used when none are provided in the RAG configuration.Code Refactoring:
Agentinitialization to support RAG configuration directly and ensure proper setup of the RAG handler. Legacy methods for memory querying and handling were deprecated. [1] [2]_runand related methods inAgentto incorporate RAG operations seamlessly during task execution and final memory consolidation. [1] [2]Framework Updates:
RAGConfigto the Swarmsstructsmodule to facilitate RAG configuration management. [1] [2]_save_additional_componentsand_reinitialize_after_loadmethods to support saving and reinitializing RAG handler statistics. [1] [2] [FEAT] [Integrate AgentRAGHandler] #880📚 Documentation preview 📚: https://swarms--903.org.readthedocs.build/en/903/