aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/core/agent/__init__.py
blob: bd6dda79b8a689a050aed3ad47066c449bf9e7f6 (about) (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
# FIXME: Once the agent is properly type annotated, remove the type: ignore comments
from .base import (  # type: ignore
    R2RAgent,
    R2RStreamingAgent,
    R2RXMLStreamingAgent,
)
from .rag import (  # type: ignore
    R2RRAGAgent,
    R2RStreamingRAGAgent,
    R2RXMLToolsRAGAgent,
    R2RXMLToolsStreamingRAGAgent,
)

# Import the concrete implementations
from .research import (
    R2RResearchAgent,
    R2RStreamingResearchAgent,
    R2RXMLToolsResearchAgent,
    R2RXMLToolsStreamingResearchAgent,
)

__all__ = [
    # Base
    "R2RAgent",
    "R2RStreamingAgent",
    "R2RXMLStreamingAgent",
    # RAG Agents
    "R2RRAGAgent",
    "R2RXMLToolsRAGAgent",
    "R2RStreamingRAGAgent",
    "R2RXMLToolsStreamingRAGAgent",
    "R2RResearchAgent",
    "R2RStreamingResearchAgent",
    "R2RXMLToolsResearchAgent",
    "R2RXMLToolsStreamingResearchAgent",
]