Iragent
This project is about using agents as simple as possible with small and large language models
Loading...
Searching...
No Matches
iragent.models.SimpleAgenticRAG Class Reference

Public Member Functions

None __init__ (self, KnowledgeGraphBuilder kg, AgentFactory agent_factory)
 
 ask (self, query)
 

Public Attributes

Callable[[str, int], List[str]] kg_search = kg.search
 
 retriever_agent
 
 generator_agent
 
 manager
 

Constructor & Destructor Documentation

◆ __init__()

None iragent.models.SimpleAgenticRAG.__init__ ( self,
KnowledgeGraphBuilder kg,
AgentFactory agent_factory )

Member Function Documentation

◆ ask()

iragent.models.SimpleAgenticRAG.ask ( self,
query )

Member Data Documentation

◆ generator_agent

iragent.models.SimpleAgenticRAG.generator_agent
Initial value:
= agent_factory.create_agent(
name="generator",
system_prompt=GENERATOR_PROMPT,
max_token = 1024
)

◆ kg_search

Callable[[str, int], List[str]] iragent.models.SimpleAgenticRAG.kg_search = kg.search

◆ manager

iragent.models.SimpleAgenticRAG.manager
Initial value:
= AutoAgentManager(
init_message=None,
agents=[self.retriever_agent, self.generator_agent],
first_agent=self.retriever_agent,
max_round=10,
termination_fn=simple_termination,
termination_word= "[#finish#]"
)

◆ retriever_agent

iragent.models.SimpleAgenticRAG.retriever_agent
Initial value:
= agent_factory.create_agent(
name="retirever",
system_prompt=RETRIEVER_PROMPT,
max_token = 4096,
fn=[self.kg_search]
)

The documentation for this class was generated from the following file: