Skip to content

Code hierarchy

CodeHierarchyAgentPack #

Bases: BaseLlamaPack

Code hierarchy agent pack.

Source code in llama-index-packs/llama-index-packs-code-hierarchy/llama_index/packs/code_hierarchy/base.py 
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
37
38
39
40
41
42
43
44
45
46
47
48
class CodeHierarchyAgentPack(BaseLlamaPack):
    """Code hierarchy agent pack."""

    def __init__(self, split_nodes: List[BaseNode], llm: OpenAI, verbose: bool = True):
        """Initialize the code hierarchy agent pack."""
        from llama_index.packs.code_hierarchy import CodeHierarchyKeywordQueryEngine

        self.query_engine = CodeHierarchyKeywordQueryEngine(
            nodes=split_nodes,
        )

        self.tool = QueryEngineTool.from_defaults(
            query_engine=self.query_engine,
            name="code_search",
            description="Search the code hierarchy for a specific code element, using keywords or IDs.",
        )

        self.agent = FunctionAgent(
            tools=[self.tool],
            llm=llm,
            system_prompt=self.query_engine.get_tool_instructions(),
            verbose=verbose,
        )

    def get_modules(self) -> Dict[str, Any]:
        return {
            "query_engine": self.query_engine,
            "tool": self.tool,
            "agent": self.agent,
        }

    def run(self, user_message: str) -> str:
        """Run the agent on the user message."""
        return asyncio_run(self.arun(user_message))

    async def arun(self, user_message: str) -> str:
        """Run the agent on the user message."""
        return str(await self.agent.run(user_message))

run #

run(user_message: str) -> str

Run the agent on the user message.

Source code in llama-index-packs/llama-index-packs-code-hierarchy/llama_index/packs/code_hierarchy/base.py 
42
43
44
def run(self, user_message: str) -> str:
    """Run the agent on the user message."""
    return asyncio_run(self.arun(user_message))

arun async #

arun(user_message: str) -> str

Run the agent on the user message.

Source code in llama-index-packs/llama-index-packs-code-hierarchy/llama_index/packs/code_hierarchy/base.py 
46
47
48
async def arun(self, user_message: str) -> str:
    """Run the agent on the user message."""
    return str(await self.agent.run(user_message))