Skip to content

Code hierarchy

CodeHierarchyAgentPack #

Bases: BaseLlamaPack

代码层级代理包。

Source code in llama_index/packs/code_hierarchy/base.py 
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
37
38
39
40
41
42
43
class CodeHierarchyAgentPack(BaseLlamaPack):
    """代码层级代理包。"""

    def __init__(self, split_nodes: List[BaseNode], llm: OpenAI, verbose: bool = True):
        """初始化代码层级代理包。"""
        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 = OpenAIAgent.from_tools(
            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:
        """在用户消息上运行代理程序。"""
        return str(self.agent.chat(user_message))

run #

run(user_message: str) -> str

在用户消息上运行代理程序。

Source code in llama_index/packs/code_hierarchy/base.py 
41
42
43
def run(self, user_message: str) -> str:
    """在用户消息上运行代理程序。"""
    return str(self.agent.chat(user_message))