代码层级智能体包。
源代码位于 llama-index-packs/llama-index-packs-code-hierarchy/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):
"""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 = 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:
"""Run the agent on the user message."""
return str(self.agent.chat(user_message))
|
在用户消息上运行智能体。
run(user_message: str) -> str
回到顶部
运行
| def run(self, user_message: str) -> str:
"""Run the agent on the user message."""
return str(self.agent.chat(user_message))
|
