1515 logger .warning (f"Failed to configure logfire: { e } " )
1616
1717# Local imports
18- from agent_tools .deps import DeepwikiAgentDeps
18+ from agent_tools .deps import CodeWikiDeps
1919from agent_tools .read_code_components import read_code_components_tool
2020from agent_tools .str_replace_editor import str_replace_editor_tool
2121from agent_tools .generate_sub_module_documentations import generate_sub_module_documentation_tool
@@ -48,7 +48,7 @@ def create_agent(self, module_name: str, components: Dict[str, Any],
4848 return Agent (
4949 fallback_models ,
5050 name = module_name ,
51- deps_type = DeepwikiAgentDeps ,
51+ deps_type = CodeWikiDeps ,
5252 tools = [
5353 read_code_components_tool ,
5454 str_replace_editor_tool ,
@@ -60,7 +60,7 @@ def create_agent(self, module_name: str, components: Dict[str, Any],
6060 return Agent (
6161 fallback_models ,
6262 name = module_name ,
63- deps_type = DeepwikiAgentDeps ,
63+ deps_type = CodeWikiDeps ,
6464 tools = [read_code_components_tool , str_replace_editor_tool ],
6565 system_prompt = LEAF_SYSTEM_PROMPT .format (module_name = module_name ),
6666 )
@@ -78,7 +78,7 @@ async def process_module(self, module_name: str, components: Dict[str, Node],
7878 agent = self .create_agent (module_name , components , core_component_ids )
7979
8080 # Create dependencies
81- deps = DeepwikiAgentDeps (
81+ deps = CodeWikiDeps (
8282 absolute_docs_path = working_dir ,
8383 absolute_repo_path = str (os .path .abspath (self .config .repo_path )),
8484 registry = {},
0 commit comments