agents.rag.multi_agent_rag.enhanced_multi_rag

Enhanced Multi-Agent RAG with Built-in Compatibility.

This module demonstrates RAG systems using the compatibility-enhanced multi-agent base, providing automatic compatibility checking and adaptation.

Classes

EnhancedRAGConditionalAgent

RAG conditional agent with built-in compatibility checking and smart routing.

EnhancedRAGParallelAgent

RAG parallel agent with built-in compatibility checking for consensus building.

EnhancedRAGSequentialAgent

RAG sequential agent with built-in compatibility checking.

SmartRAGFactory

Factory for creating RAG systems with automatic compatibility management.

Functions

demonstrate_enhanced_rag_compatibility()

Demonstrate the enhanced RAG system with built-in compatibility checking.

Module Contents

class agents.rag.multi_agent_rag.enhanced_multi_rag.EnhancedRAGConditionalAgent(retrieval_agent=None, grading_agent=None, answer_agent=None, compatibility_mode=CompatibilityMode.AUTO_FIX, **kwargs)

Bases: haive.agents.multi.compatibility_enhanced_base.CompatibilityEnhancedConditionalAgent

RAG conditional agent with built-in compatibility checking and smart routing.

This system checks compatibility at each routing decision and can adapt agents on-the-fly if compatibility issues are detected.

Init .

Parameters:

  • retrieval_agent (haive.agents.rag.multi_agent_rag.agents.SimpleRAGAgent | None) – [TODO: Add description]

  • grading_agent (haive.agents.rag.multi_agent_rag.agents.DocumentGradingAgent | None) – [TODO: Add description]

  • answer_agent (haive.agents.rag.multi_agent_rag.agents.SimpleRAGAnswerAgent | None) – [TODO: Add description]

  • compatibility_mode (haive.agents.multi.compatibility_enhanced_base.CompatibilityMode) – [TODO: Add description]

class agents.rag.multi_agent_rag.enhanced_multi_rag.EnhancedRAGParallelAgent(rag_variants=None, compatibility_mode=CompatibilityMode.ADAPTIVE, **kwargs)

Bases: haive.agents.multi.compatibility_enhanced_base.CompatibilityEnhancedParallelAgent

RAG parallel agent with built-in compatibility checking for consensus building.

This system runs multiple RAG workflows in parallel and ensures they can all work with the same state schema.

Init .

Parameters:

  • rag_variants (list[EnhancedRAGSequentialAgent] | None) – [TODO: Add description]

  • compatibility_mode (haive.agents.multi.compatibility_enhanced_base.CompatibilityMode) – [TODO: Add description]

class agents.rag.multi_agent_rag.enhanced_multi_rag.EnhancedRAGSequentialAgent(retrieval_agent=None, grading_agent=None, answer_agent=None, compatibility_mode=CompatibilityMode.ADAPTIVE, **kwargs)

Bases: haive.agents.multi.compatibility_enhanced_base.CompatibilityEnhancedSequentialAgent

RAG sequential agent with built-in compatibility checking.

This system automatically validates that retrieval -> grading -> generation agents are compatible and applies adapters if needed.

Init .

Parameters:

  • retrieval_agent (haive.agents.rag.multi_agent_rag.agents.SimpleRAGAgent | None) – [TODO: Add description]

  • grading_agent (haive.agents.rag.multi_agent_rag.agents.DocumentGradingAgent | None) – [TODO: Add description]

  • answer_agent (haive.agents.rag.multi_agent_rag.agents.SimpleRAGAnswerAgent | None) – [TODO: Add description]

  • compatibility_mode (haive.agents.multi.compatibility_enhanced_base.CompatibilityMode) – [TODO: Add description]

class agents.rag.multi_agent_rag.enhanced_multi_rag.SmartRAGFactory

Factory for creating RAG systems with automatic compatibility management.

This factory analyzes the provided agents and creates the most appropriate multi-agent structure with optimal compatibility settings.

static create_optimal_rag_system(agents, documents=None, preferred_mode=None, compatibility_mode=CompatibilityMode.ADAPTIVE)

Create an optimal RAG system based on provided agents.

Parameters:

  • agents (list[Any]) – List of agents to include

  • documents (list[langchain_core.documents.Document] | None) – Optional documents for RAG agents

  • preferred_mode (str | None) – Preferred execution mode (“sequential”, “conditional”, “parallel”)

  • compatibility_mode (haive.agents.multi.compatibility_enhanced_base.CompatibilityMode) – How to handle compatibility issues

Returns:

Optimally configured RAG system with compatibility checking

Return type:

haive.agents.multi.compatibility_enhanced_base.CompatibilityEnhancedMultiAgent

static create_safe_rag_system(documents=None, include_grading=True, use_iterative_grading=False, compatibility_mode=CompatibilityMode.STRICT)

Create a safe RAG system with strict compatibility checking.

This method creates a RAG system that is guaranteed to be compatible or will fail with clear error messages.

Parameters:

  • documents (list[langchain_core.documents.Document] | None)

  • include_grading (bool)

  • use_iterative_grading (bool)

  • compatibility_mode (haive.agents.multi.compatibility_enhanced_base.CompatibilityMode)

Return type:

EnhancedRAGSequentialAgent

agents.rag.multi_agent_rag.enhanced_multi_rag.demonstrate_enhanced_rag_compatibility()

Demonstrate the enhanced RAG system with built-in compatibility checking.

Return type:

None