From 1ab90de913405b45050e0f732a03004087134fda Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期一, 25 十一月 2024 11:11:04 +0800
Subject: [PATCH] 上传dify文件

---
 app/api/files.py |   30 +++++++++++++++---------------
 1 files changed, 15 insertions(+), 15 deletions(-)

diff --git a/app/api/files.py b/app/api/files.py
index dd9166a..22c2216 100644
--- a/app/api/files.py
+++ b/app/api/files.py
@@ -94,21 +94,21 @@
             service = BasicService(base_url=settings.basic_paper_url)
             result = await service.paper_file_upload(chat_id, file.filename, file_content)
 
-        elif agent.agent_type == AgentType.DIFY:
-            file = file[0]
-            # 璇诲彇涓婁紶鐨勬枃浠跺唴瀹�
-            try:
-                file_content = await file.read()
-            except Exception as e:
-                return Response(code=400, msg=str(e))
-            dify_service = DifyService(base_url=settings.dify_base_url)
-            try:
-                token = get_bisheng_token(db, current_user.id)
-                result = await dify_service.upload(token, file.filename, file_content)
-            except Exception as e:
-                raise HTTPException(status_code=500, detail=str(e))
-            result["file_name"] = file.filename
-            return Response(code=200, msg="", data=result)
+    elif agent.agent_type == AgentType.DIFY:
+        file = file[0]
+        # 璇诲彇涓婁紶鐨勬枃浠跺唴瀹�
+        try:
+            file_content = await file.read()
+        except Exception as e:
+            return Response(code=400, msg=str(e))
+        dify_service = DifyService(base_url=settings.dify_base_url)
+        try:
+            token = get_bisheng_token(db, current_user.id)
+            result = await dify_service.upload(token, file.filename, file_content, current_user.id)
+        except Exception as e:
+            raise HTTPException(status_code=500, detail=str(e))
+        result["file_name"] = file.filename
+        return Response(code=200, msg="", data=result)
 
         return Response(code=200, msg="", data=result)
 

--
Gitblit v1.8.0