From 6efaa41969ab89c1e1bd468b1b895f6689b7f70d Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期四, 20 二月 2025 10:04:59 +0800 Subject: [PATCH] tmp --- app/service/v2/chat.py | 1 - app/service/v2/app_driver/chat_workflow.py | 27 ++++++++++++--------------- main.py | 4 ++-- 3 files changed, 14 insertions(+), 18 deletions(-) diff --git a/app/service/v2/app_driver/chat_workflow.py b/app/service/v2/app_driver/chat_workflow.py index 64e071d..047cb66 100644 --- a/app/service/v2/app_driver/chat_workflow.py +++ b/app/service/v2/app_driver/chat_workflow.py @@ -28,22 +28,19 @@ print(e) @staticmethod - async def request_data(query: str, conversation_id: str, user:str, chat_data: ChatData) -> dict: - inputs = [] - files = [] - if hasattr(chat_data, "inputs"): + async def request_data(query: str, conversation_id: str, user: str, chat_data: ChatData) -> dict: + data = {"response_mode": "streaming", + "conversation_id": conversation_id, + "user": user} + inputs = {} + # files = [] + if hasattr(chat_data, "inputs"): inputs = chat_data.inputs - if hasattr(chat_data, "files"): + if hasattr(chat_data, "files"): files = chat_data.files - - return { - "inputs":inputs, - "query": query, - "response_mode": "streaming", - "conversation_id": conversation_id, - "user": user, - "files": files - } + data["files"] = files + data["inputs"] = inputs + return data if __name__ == "__main__": @@ -54,7 +51,7 @@ url = f"{base_url}/v1/chat-messages" chat = ChatAgent() data = { - "inputs":{}, + "inputs": {}, "query": "浣犲ソ锛屼綘鑳藉仛浠�涔堬紵", "response_mode": "streaming", "conversation_id": "", diff --git a/app/service/v2/chat.py b/app/service/v2/chat.py index 246b3f1..034235b 100644 --- a/app/service/v2/chat.py +++ b/app/service/v2/chat.py @@ -142,7 +142,6 @@ files = [] node_list = [] token = await get_chat_token(db, chat_id) - chat, url = await get_chat_object(mode) if hasattr(chat_data, "query"): query = chat_data.query diff --git a/main.py b/main.py index bcee8b5..7f504c2 100644 --- a/main.py +++ b/main.py @@ -29,7 +29,7 @@ from app.task.fetch_agent import sync_agents, initialize_agents, sync_agents_v2, sync_knowledge, \ sync_resources_from_json from app.init_config.init_run_data import sync_default_data -from app.task.sync_account_token import sync_token, start_sync_token_task +from app.task.sync_account_token import start_sync_token_task init_db() @@ -47,7 +47,7 @@ sync_agents_v2() sync_knowledge() sync_resources_from_json() - start_sync_token_task() + # start_sync_token_task() yield # 鍦ㄥ簲鐢ㄥ叧闂椂鎵ц娓呯悊鎿嶄綔锛堝鏋滈渶瑕侊級 pass -- Gitblit v1.8.0