更新对话转工单处理逻辑,增强用户问题和解决方案提取功能,添加槽位填充支持,调整最大工作线程数为10,优化意图识别API,重排序匹配术语,改进数据模型以支持软件名称枚举,提升代码结构和可读性。

This commit is contained in:
2025-06-03 10:35:25 +08:00
parent d4ff7b6fad
commit 38b6f66925
8 changed files with 160 additions and 92 deletions
+30 -7
View File
@@ -9,7 +9,27 @@ Description: 提取和分类的数据模型
from pydantic import BaseModel, Field
from typing import List, Optional, Dict, Tuple
from enum import Enum
class SoftwareName(str, Enum):
"""软件名称枚举类"""
D3 = "配网工程计价通D3软件"
C1 = "新型储能电站建设计价通C1软件"
Z1 = "西藏电力工程计价通Z1软件"
T1 = "技改检修工程计价通T1软件"
T1_LIST = "技改检修清单计价通T1软件"
MAIN = "主网电力建设计价通软件"
UNKNOWN = "" # 未知
# 软件别名映射
ALIASES = {
D3: ["配网D3", "D3软件", "配网工程软件"],
C1: ["储能C1", "C1软件", "储能电站软件", "储能软件"],
Z1: ["西藏Z1", "Z1软件", "西藏电力软件"],
T1: ["技改T1", "T1软件", "技改检修软件"],
T1_LIST: ["技改清单T1", "T1清单软件", "技改检修清单软件"],
MAIN: ["主网软件", "电力建设软件", "主网建设软件", "主网软件"]
}
# 定义输出模型
class Term(BaseModel):
@@ -38,7 +58,7 @@ class QueryRewrite(BaseModel):
# 1. 软件问题
# 1.1 软件功能
class SoftwareFunction(BaseModel):
software_name: str = Field(description="软件名称")
software_name: SoftwareName = Field(description="软件名称")
function_name: str = Field(description="具体功能名称")
operation: str = Field(description="用户操作意图(如何使用功能、功能入口、功能使用场景)")
software_version: Optional[str] = Field(None, description="软件版本")
@@ -57,7 +77,7 @@ class SoftwareFunction(BaseModel):
# 1.2 故障排查
class TroubleShooting(BaseModel):
software_name: str = Field(description="软件名称")
software_name: SoftwareName = Field(description="软件名称")
function_name: str = Field(description="具体功能名称/操作描述")
error_message: str = Field(description="报错信息/异常现象")
software_version: Optional[str] = Field(None, description="软件版本")
@@ -80,7 +100,7 @@ class TroubleShooting(BaseModel):
class ProfessionalConsulting(BaseModel):
scene_subject: str = Field(description="场景主体")
business_scene: str = Field(description="业务场景描述")
software_name: Optional[str] = Field(None, description="软件名称")
software_name: Optional[SoftwareName] = Field(None, description="软件名称")
def check_required_slots(self) -> Tuple[bool, Dict[str, str]]:
"""检查必填槽位是否都存在"""
@@ -95,7 +115,7 @@ class ProfessionalConsulting(BaseModel):
class DataProblem(BaseModel):
expense_type: str = Field(description="费用类型")
operation_purpose: str = Field(description="操作目的")
software_name: Optional[str] = Field(None, description="软件名称")
software_name: Optional[SoftwareName] = Field(None, description="软件名称")
project_type: Optional[str] = Field(None, description="工程类型")
def check_required_slots(self) -> Tuple[bool, Dict[str, str]]:
@@ -141,7 +161,9 @@ class SoftwareLock(BaseModel):
# 3.3 安装下载类
class InstallationDownload(BaseModel):
software_name: str = Field(description="软件/插件名称")
software_name: SoftwareName = Field(description="软件/插件名称,与file_name二选一")
file_name: str = Field(description="文件名,与software_name二选一")
operation_stage: str = Field(description="操作阶段")
os_version: Optional[str] = Field(None, description="操作系统版本")
package_source: Optional[str] = Field(None, description="安装包来源/版本号")
@@ -149,8 +171,9 @@ class InstallationDownload(BaseModel):
def check_required_slots(self) -> Tuple[bool, Dict[str, str]]:
"""检查必填槽位是否都存在"""
missing_slots = {}
if not self.software_name:
if not self.software_name and not self.file_name:
missing_slots["software_name"] = InstallationDownload.model_fields["software_name"].description
missing_slots["file_name"] = InstallationDownload.model_fields["file_name"].description
if not self.operation_stage:
missing_slots["operation_stage"] = InstallationDownload.model_fields["operation_stage"].description
return len(missing_slots) == 0, missing_slots
@@ -158,7 +181,7 @@ class InstallationDownload(BaseModel):
# 3.4 问题排查类
class ProblemDiagnosis(BaseModel):
error_message: str = Field(description="报错信息/异常现象")
software_name: Optional[str] = Field(None, description="软件名称")
software_name: Optional[SoftwareName] = Field(None, description="软件名称")
os_version: Optional[str] = Field(None, description="操作系统版本")
def check_required_slots(self) -> Tuple[bool, Dict[str, str]]: