From 3b1bf7da6771bd1d6852d3dcc1f906d5ae5c95d1 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 12 三月 2025 13:49:12 +0800
Subject: [PATCH] 增加内容优化
---
app/service/v2/app_driver/chat_agent.py | 21 +++++++++++++++++++--
1 files changed, 19 insertions(+), 2 deletions(-)
diff --git a/app/service/v2/app_driver/chat_agent.py b/app/service/v2/app_driver/chat_agent.py
index 45f804e..a9d1790 100644
--- a/app/service/v2/app_driver/chat_agent.py
+++ b/app/service/v2/app_driver/chat_agent.py
@@ -1,5 +1,6 @@
import json
+from Log import logger
# from Log import logger
from app.models.v2.session_model import ChatData
from app.service.v2.app_driver.chat_base import ChatBase
@@ -9,6 +10,7 @@
async def chat_completions(self, url, data, headers):
complete_response = ""
+ # print(data)
async for line in self.http_stream(url, data, headers):
# logger.error(line)
if line.startswith("data:"):
@@ -25,8 +27,8 @@
yield json_data
except json.JSONDecodeError as e:
- # logger.info("Invalid JSON data------------------")
- print(e)
+ logger.info("Invalid JSON data------------------")
+ # print(e)
@staticmethod
async def request_data(query: str, conversation_id: str, user:str, chat_data: ChatData) -> dict:
@@ -46,6 +48,21 @@
"files": files
}
+ @staticmethod
+ async def complex_request_data(query: str, conversation_id: str, user: str, files: list=None, inputs: dict=None) -> dict:
+ if not files:
+ files = []
+ if not inputs:
+ inputs = {}
+ return {
+ "inputs": inputs,
+ "query": query,
+ "response_mode": "streaming",
+ "conversation_id": conversation_id,
+ "user": user,
+ "files": files
+ }
+
if __name__ == "__main__":
async def aa():
--
Gitblit v1.8.0