From d4842ed6354096eb88870cc400cc73ec348f6b2f Mon Sep 17 00:00:00 2001 From: aries_ckt <916701291@qq.com> Date: Tue, 20 Aug 2024 10:28:50 +0800 Subject: [PATCH] fix:domain param bug --- dbgpt/app/openapi/api_v1/api_v1.py | 1 - dbgpt/serve/agent/db/gpts_app.py | 6 ++---- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/dbgpt/app/openapi/api_v1/api_v1.py b/dbgpt/app/openapi/api_v1/api_v1.py index 8ec962c3e..05e052bed 100644 --- a/dbgpt/app/openapi/api_v1/api_v1.py +++ b/dbgpt/app/openapi/api_v1/api_v1.py @@ -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}") end_task = cast(BaseOperator, dags[0].leaf_nodes[0]) - space = dialogue.select_param connector_manager = CFG.local_db_manager # TODO: Some flow maybe not connector diff --git a/dbgpt/serve/agent/db/gpts_app.py b/dbgpt/serve/agent/db/gpts_app.py index 5634daba8..a744c2247 100644 --- a/dbgpt/serve/agent/db/gpts_app.py +++ b/dbgpt/serve/agent/db/gpts_app.py @@ -1382,12 +1382,10 @@ def adapt_native_app_model(dialogue: ConversationVo): ks_service = KnowledgeService() knowledge_spaces = ks_service.get_knowledge_space( - KnowledgeSpaceRequest( - name=dialogue.select_param, user_id=dialogue.user_name - ) + KnowledgeSpaceRequest(name=dialogue.select_param) ) if len(knowledge_spaces) == 1: - dialogue.select_param = knowledge_spaces[0].id + dialogue.select_param = knowledge_spaces[0].name return dialogue except Exception as e: logger.info(f"adapt_native_app_model error: {e}")