|
|
@@ -28,22 +28,19 @@ class ChatWorkflow(ChatBase):
|
|
|
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 @@ if __name__ == "__main__":
|
|
|
url = f"{base_url}/v1/chat-messages"
|
|
|
chat = ChatAgent()
|
|
|
data = {
|
|
|
- "inputs":{},
|
|
|
+ "inputs": {},
|
|
|
"query": "你好,你能做什么?",
|
|
|
"response_mode": "streaming",
|
|
|
"conversation_id": "",
|