github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-06 19:40:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: native app bug (#1988)

Browse Source
This commit is contained in:
Aries-ckt
2024-09-09 21:20:35 +08:00
committed by GitHub
parent 0e512bcddd
commit fe29f977f3
2 changed files with 166 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
dbgpt/app/openapi/api_v1/api_v1.py
View File

@@ -705,12 +705,17 @@ def message2Vo(message: dict, order, model_name) -> MessageVo:
def _parse_domain_type(dialogue: ConversationVo) -> Optional[str]:
if dialogue.chat_mode == ChatScene.ChatKnowledge.value():
# Supported in the knowledge chat
space_name = dialogue.select_param
spaces = knowledge_service.get_knowledge_space(
KnowledgeSpaceRequest(name=space_name)
)
if dialogue.app_code == "" or dialogue.app_code == "chat_knowledge":
spaces = knowledge_service.get_knowledge_space(
KnowledgeSpaceRequest(name=dialogue.select_param)
)
else:
spaces = knowledge_service.get_knowledge_space(
KnowledgeSpaceRequest(id=dialogue.select_param)
)
if len(spaces) == 0:
raise ValueError(f"Knowledge space {space_name} not found")
raise ValueError(f"Knowledge space {dialogue.select_param} not found")
dialogue.select_param = spaces[0].name
if spaces[0].domain_type:
return spaces[0].domain_type
else: