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

fix:domain param bug

Browse Source
This commit is contained in:
aries_ckt
2024-08-20 10:28:50 +08:00
parent 3573d8a935
commit d4842ed635
2 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -714,7 +714,6 @@ async def chat_with_domain_flow(dialogue: ConversationVo, domain_type: str):
raise ValueError(f"Cant find the DAG for domain type {domain_type}") raise ValueError(f"Cant find the DAG for domain type {domain_type}")
end_task = cast(BaseOperator, dags[0].leaf_nodes[0]) end_task = cast(BaseOperator, dags[0].leaf_nodes[0])
space = dialogue.select_param space = dialogue.select_param
connector_manager = CFG.local_db_manager connector_manager = CFG.local_db_manager
# TODO: Some flow maybe not connector # TODO: Some flow maybe not connector

6
dbgpt/serve/agent/db/gpts_app.py
View File

@@ -1382,12 +1382,10 @@ def adapt_native_app_model(dialogue: ConversationVo):
ks_service = KnowledgeService() ks_service = KnowledgeService()
knowledge_spaces = ks_service.get_knowledge_space( knowledge_spaces = ks_service.get_knowledge_space(
KnowledgeSpaceRequest( KnowledgeSpaceRequest(name=dialogue.select_param)
name=dialogue.select_param, user_id=dialogue.user_name
)
) )
if len(knowledge_spaces) == 1: if len(knowledge_spaces) == 1:
dialogue.select_param = knowledge_spaces[0].id dialogue.select_param = knowledge_spaces[0].name
return dialogue return dialogue
except Exception as e: except Exception as e:
logger.info(f"adapt_native_app_model error: {e}") logger.info(f"adapt_native_app_model error: {e}")