调试后整体可以使用的版本。

main.py

@@ -56,11 +56,11 @@ from src.code_executor import CodeExecutor
 from src.dialog_manager import DialogManager
 from src.neo4j_raw_retriever import Neo4jRawRetriever
 from src.embedding_client import EmbeddingClient
-
-from project import ProjectBuilder
+from project import ProjectBuilder, ProjectToolkit
 from project_implementation import ProjectToolkitNeo4j
 
 
+
 def main():
     config = Config()
     # 根据配置设置环境变量
@@ -87,7 +87,9 @@ def main():
     # 创建Neo4j检索器
     knowledge_retriever = Neo4jRawRetriever(neo4j_conf)
 
-    ProjectBuilder.register(ProjectToolkitNeo4j, neo4j_conf)
+
+    ProjectBuilder.register(ProjectToolkitNeo4j, knowledge_retriever.driver)
+
 
     code_executor = CodeExecutor(prompt_manager.prompts, llm_client_coder)