增加内容优化

zhaoqingang

2025-03-12 3b1bf7da6771bd1d6852d3dcc1f906d5ae5c95d1

 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():