删除 src/prompt_manager.py

src/prompt_manager.py

@@ -1,111 +0,0 @@
-# src/prompt_manager.py
-
-from dataclasses import dataclass
-from langchain.prompts import ChatPromptTemplate
-from langchain.schema import SystemMessage, HumanMessage
-
-@dataclass
-class CodeExecutorPrompts:
-    understand_prompt: ChatPromptTemplate
-    code_gen_prompt: ChatPromptTemplate
-    code_fix_prompt: ChatPromptTemplate
-    rewrite_prompt_template: ChatPromptTemplate
-    cypher_conversion_prompt: ChatPromptTemplate  # 新增Cypher转换提示模板
-
-class PromptManager:
-    def __init__(self):
-        self.prompts = self._init_prompts()
-
-    def _init_prompts(self) -> CodeExecutorPrompts:
-        understand_prompt = ChatPromptTemplate.from_template(
-"""
-你是一名电力造价业务专家，请基于以下工程文件业务结构，将用户自然语言问题改写成专业查询语句：
-
-**工程文件业务结构**：
-{business_structure}
-
-**改写规则**：
-1. **定位目标对象**：从业务结构中识别核心对象（如 `ProjectDivisionTree`→项目划分树、`FeeScheduleItem`→费用表）。
-2. **提取条件**：从用户输入中解析关键条件（如名称、量、类型），用【】标注变量。
-3. **构建专业语句**：格式为：`在[目标对象]中查找【条件】的项。`
-   - 使用业务术语（如“项目划分项”而非“项目”）。
-   - 条件需明确属性（如【名称】、【量】、【类型】）。
-4. **精确映射结构**：若用户查询层级（如“叶节点”），需在条件中体现。
-
-**用户输入**：{user_input}
-**改写输出**：（仅输出改写后的语句）
-""")
-
-        code_gen_prompt = ChatPromptTemplate.from_template("""
-你是一个专业的Python工程师。我会给你一个用户问题，你需要将其转换为对应的Python代码
-
-用户问题:
-{user_request}
-
-上下文信息:
-{context}
-
-工程数据访问库:
-{bowei_api_docs}
-
-# 工作流程
-1. 从用户问题中提取关键信息（节点路径、节点类型、节点名称等）
-2. 根据"用户问题"和"上下文信息"选择最匹配的"工程数据访问库"中的方法
-3. 生成可直接执行的Python代码
-
-# 代码模板（必须严格遵循）
-def neo4j_find_function():
-    project = ProjectBuilder.build()
-    status, data, error, helper_info = project.[SELECTED_METHOD]([PARAMETERS])
-    return status, data, error, helper_info
-
-# 执行规则
-- 参数必须从用户问题或上下文信息中提取
-- 必须确保生成的代码可以直接执行
-- 禁止修改代码模板结构
-- 禁止添加任何注释或解释
-
-# 输出格式
-def neo4j_find_function():
-    project = ProjectBuilder.build()
-    status, data, error, helper_info = project.[SELECTED_METHOD]([PARAMETERS])
-    return status, data, error, helper_info
-""")
-
-        code_fix_prompt = ChatPromptTemplate.from_messages([
-            SystemMessage(content="你是一个专业的Python开发者。下面是之前生成的代码和执行时出现的错误，请修复代码中的错误，确保代码语法正确且能完成访问博微造价工程数据的任务。只输出修复后的Python代码，不要添加解释。"),
-            HumanMessage(content="原始代码：\n{code}\n\n错误信息：\n{error}\n")
-        ])
-
-        rewrite_prompt_template = ChatPromptTemplate.from_template(
-            """你是一个专业的工程业务助理，结合以下工程业务结构信息和相关知识：
-{business_structure}
-
-相关知识内容：
-{context}
-
-请根据用户的问题，结合上述信息，理解并改写成一个针对工程数据的访问请求（简洁明了的描述）。
-请只输出改写后的访问请求文本，不要多余解释。"""
-        )
-
-        cypher_conversion_prompt = ChatPromptTemplate.from_template(
-            """
-你是一个Neo4j专家。请将用户的自然语言问题转换成一个有效的Cypher查询语句，查询知识图谱中相关信息。
-只返回Cypher语句，不要任何解释，最多返回5条。
-
-业务结构信息：
-{business_structure}
-
-用户问题：
-{user_input}
-Cypher查询语句：
-"""
-        )
-
-        return CodeExecutorPrompts(
-            understand_prompt=understand_prompt,
-            code_gen_prompt=code_gen_prompt,
-            code_fix_prompt=code_fix_prompt,
-            rewrite_prompt_template=rewrite_prompt_template,
-            cypher_conversion_prompt=cypher_conversion_prompt,
-        )