rag-gateway大模型2.0
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
zhaoqingang
2025-02-18 f6a0aa15269cae87737d1564ad7a33063f1d94de 
 app/api/chat.py


@@ -323,7 +323,7 @@


        # token = get_dify_token(db, current_user.id)


        try:


            async def forward_to_dify():


                if chat_type == "imageTalk":


                if agent.type == "imageTalk":


                    token = DfTokenDao(db).get_token_by_id(IMAGE_TO_TEXT)


                    if not token:


                        await websocket.send_json({"message": "Invalid token", "type": "error"})


@@ -429,12 +429,13 @@


                                result = {"message": f"内部错误： {e2}", "type": "close"}


                                await websocket.send_json(result)


                                print(f"Error process message of ragflow: {e2}")


                elif chat_type == "reportWorkflow":


                elif agent.type == "reportWorkflow":




                    while True:


                        receive_message = await websocket.receive_json()


                        print(f"Received from client {chat_id}: {receive_message}")


                        upload_files = receive_message.get('upload_files', [])


                        upload_filenames = receive_message.get('upload_filenames', [])


                        title = receive_message.get('title', "")


                        sub_titles = receive_message.get('sub_titles', "")


                        workflow_type = receive_message.get('workflow', 1)


@@ -458,7 +459,7 @@


                                AgentType.DIFY,


                                current_user.id,


                                {"role": "user", "content": title if title else title_query, "type": workflow_type,


                                 "is_clean": is_clean},


                                 "is_clean": is_clean, "upload_filenames":upload_filenames},


                                workflow_type


                            )


                            conversation_id = session.conversation_id


@@ -823,6 +824,7 @@


                        receive_message = await websocket.receive_json()


                        print(f"Received from client {chat_id}: {receive_message}")


                        upload_file_id = receive_message.get('upload_file_id', [])


                        upload_filenames = receive_message.get('upload_filenames', [])


                        question = receive_message.get('message', "")


                        if not question and not image_url:


                            await websocket.send_json({"message": "Invalid request", "type": "error"})


@@ -834,7 +836,7 @@


                                agent_id,


                                AgentType.DIFY,


                                current_user.id,


                                {"role": "user", "content": question}


                                {"role": "user", "content": question, "upload_filenames": upload_filenames}


                            )


                            conversation_id = session.conversation_id


                        except Exception as e: