```python import os LLM_API_CONFIG = { "api_url": os.getenv("LLM_API_URL", "http://aigateway.his-beta.huawei.com/v1"), "api_key": os.getenv("LLM_API_KEY", "sk-2a0c9fe3140a471caa7d652ae655e91a"), "model": os.getenv("LLM_MODEL", "qwen-coder"), "timeout": int(os.getenv("LLM_API_TIMEOUT", "200")), # 超时时间（秒） "max_retries": int(os.getenv("LLM_API_MAX_RETRIES", "3")), # 最大重试次数 } EXCEL_FILE_PATH = os.getenv("EXCEL_FILE_PATH", "表语义.xlsx") SQL_GET_NUM_CONFIG = { "url": os.getenv("SQL_GET_NUM_URL", "http://sql-service/api/query"), "timeout": int(os.getenv("SQL_GET_NUM_TIMEOUT", "30")), "max_retries": int(os.getenv("SQL_GET_NUM_MAX_RETRIES", "2")), } FIRST_QUERY_PROMPT_TEMPLATE = """ 你是一个数据分析专家和SQL专家。以下是所有可用的数据库表的详细字段信息。 【表字段详情】 {table_fields_detail} 【表标签（源系统映射）】 {table_tags} 用户查询：{user_query} 请根据以上信息完成以下任务： 1. 从用户查询中提取关键要素，包括： - contract_number: 合同号（如'0006082500070T'），必须准确提取。 - metric: 需要查询的指标描述（如净额、Net Amount、不含税金额等），用中文或英文描述，保持原意。 - group_by: 分组字段描述（如“按合同版本分组”），如果没有则null。 - source_system_hint: 如果用户提到了源系统（如isales），则填写该源系统；否则为null。 2. 推断可能包含该合同号数据的候选表列表（candidate_tables），并针对每张候选表生成一条用于检测该合同号是否存在的SQL语句。规则： - 如果没有指定源系统，则需要为所有表格生成对应的SQL语句，即候选表为所有表格。 - 如果用户指定了源系统（source_system_hint非空），则候选表应从该源系统对应的表中选取（参考表标签）； - 对于每张候选表，合同号字段名可能在不同表中有不同名称，你需要根据该表的所有字段信息（'字段名'和'中文名称'等），推断出最可能是合同号字段名的那个字段名（如'hw_contract_num','合同num','华为合同号'等），然后生成一条检测SQL（注意，你不需要生成提取SQL）：SELECT 1 FROM 表名 WHERE 合同号字段名 = 'contract_number' LIMIT 1。 - 注意：合同号字段名不可能是表名。 请以JSON格式输出，格式如下： {{ "contract_number": "提取的合同号字符串", "metric": "指标描述字符串", "group_by": "分组描述字符串或null", "source_system_hint": "源系统字符串或null", "detection_queries": [ {{"table": "表名1", "sql": "SELECT 1 FROM 表名1 WHERE 合同号字段名1 = 'contract_number' LIMIT 1"}}, {{"table": "表名2", "sql": "SELECT 1 FROM 表名2 WHERE 合同号字段名2 = 'contract_number' LIMIT 1"}}, ... ] }} 要求：只输出JSON，不要添加任何其他文字。确保JSON合法。 """ SECOND_QUERY_PROMPT_TEMPLATE = """ 你是一个SQL专家。现在我们已经确定了用户查询中合同号实际存在于以下表中，并提取了关键要素。 用户原始查询：{user_query} 提取的要素： - 合同号: {contract_number} - 查询指标: {metric} - 分组要求: {group_by} 以下是实际包含该合同号的表的详细字段信息（包括字段名、中文名称、同义词）： {table_fields_detail} 请根据以上信息生成两个SQL查询语句，从这些表中选择最合适的一张表，查询该合同号对应的记录。 第一个SQL（指标查询）：根据用户查询的指标({metric})和分组要求({group_by})生成查询语句。 - 如果分组要求为空或未指定（即{group_by}为"无"或空），则直接查询该合同号对应的指标字段（可能需根据指标描述选择具体字段），如果指标涉及多个字段（如不同币种），请选择最合理的字段，无需聚合。 - 如果分组要求不为空（例如"按合同版本分组"），则按分组字段进行分组，并根据查询指标选择合适的字段进行聚合（如SUM），SELECT分组字段和聚合后的结果。 第二个SQL（全量查询）：查询该合同号对应表的所有字段，即 SELECT * FROM 表名 WHERE 合同号字段 = '{contract_number}'。 要求： - 两个SQL都针对同一张最合适的表。 - 字段名使用英文名。 - 以如下格式输出两个SQL，中间用分隔符 "---" 隔开： 第一个SQL语句 --- 第二个SQL语句 只输出这两个SQL，不要添加任何解释或其他内容。如果无法生成SQL，输出空字符串。 """ ```