将事件名称转义成中文及优化关键字检索

This commit is contained in:
wanyaokun
2024-09-05 09:24:35 +08:00
parent ae19725d72
commit aba6475c5a
2 changed files with 42 additions and 4 deletions
+20 -2
View File
@@ -32,6 +32,24 @@ v1_router = v = APIRouter()
gEvent_handler = None
CH_Event_map={
'CHUNKING':'文本切片',
'NODE_PARSING':'节点解析',
'EMBEDDING':'生成向量',
'LLM':'知识问答',
'QUERY':'查询',
'RETRIEVE':'检索',
'SYNTHESIZE':'答案合成',
'TREE':'总结',
'SUB_QUESTION':'问题分解',
'TEMPLATING':'生成提示词模板',
'FUNCTION_CALL':'函数调用',
'RERANKING':'节点重排',
'EXCEPTION':'执行异常',
'AGENT_STEP':'单步执行'
}
class ChatCallbackEvent(BaseModel):
event_type: ChatEventType
payload: Optional[Dict[str, Any]] = None
@@ -99,7 +117,7 @@ class ChatCallbackEvent(BaseModel):
"id": self.payload.get('nodeid'),
"node_id": self.payload.get('nodeid'),
"node_type": "http-request",
"title": f"正在执行事件:{self.payload.get('title')}",
"title": CH_Event_map[self.payload.get('title')],
"index": self.payload.get('index'),
"predecessor_node_id": self.payload.get('predecessor_node_id'),
"inputs": '',
@@ -117,7 +135,7 @@ class ChatCallbackEvent(BaseModel):
"id": self.payload.get('nodeid'),
"node_id": self.payload.get('nodeid'),
"node_type": "http-request",
"title": f"事件执行结束:{self.payload.get('title')}",
"title": CH_Event_map[self.payload.get('title')],
"index": self.payload.get('index'),
"predecessor_node_id": self.payload.get('predecessor_node_id'),
"inputs": '',