Upload 4 files

Dockerfile

@@ -23,5 +23,5 @@ USER user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 
-# Command to run the application
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"] 
+# Command to run the application with multiple workers
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "4"] 

app.py

@@ -98,8 +98,8 @@ async def dify_api(
             )
             return {"answer": response_text}
         else:
-            # 阻塞模式
-            response_data = client.send_request_blocking(
+            # 阻塞模式 - 使用异步方法
+            response_data = await client.send_request_blocking_async(
                 content=query_request.query,
                 conversation_id=query_request.conversation_id,
                 user=query_request.user

dify_api_client.py

@@ -12,6 +12,7 @@ import sys
 import json
 import argparse
 import requests
+import httpx
 import sseclient
 import logging
 import configparser
@@ -167,34 +168,84 @@ class DifyAPIClient:
             logging.error(f"HTTP错误: {e}")
             if hasattr(response, 'text'):
                 logging.error(f"API响应内容: {response.text}")
-            print(f"HTTP错误: {e}")
-            if hasattr(response, 'text') and response.text:
-                print(f"API响应: {response.text}")
-            sys.exit(1)
+            raise Exception(f"HTTP错误: {e}")
         except requests.exceptions.ConnectionError:
             logging.error("连接错误: 无法连接到API服务器")
-            print("连接错误: 无法连接到API服务器")
-            sys.exit(1)
+            raise Exception("连接错误: 无法连接到API服务器")
         except requests.exceptions.Timeout:
             logging.error("超时错误: API请求超时")
-            print("超时错误: API请求超时")
-            sys.exit(1)
+            raise Exception("超时错误: API请求超时")
         except requests.exceptions.RequestException as e:
             logging.error(f"请求错误: {str(e)}", exc_info=True)
-            print(f"请求错误: {e}")
-            sys.exit(1)
+            raise Exception(f"请求错误: {e}")
         except json.JSONDecodeError:
             logging.error("解析错误: API返回的响应不是有效的JSON格式")
             if hasattr(response, 'text'):
                 logging.error(f"API响应内容: {response.text}")
-            print("解析错误: API返回的响应不是有效的JSON格式")
-            if hasattr(response, 'text'):
-                print(f"API响应: {response.text}")
-            sys.exit(1)
+            raise Exception("解析错误: API返回的响应不是有效的JSON格式")
         except Exception as e:
             logging.error(f"未知错误: {str(e)}", exc_info=True)
-            print(f"发送请求时发生未知错误: {str(e)}")
-            sys.exit(1)
+            raise Exception(f"发送请求时发生未知错误: {str(e)}")
+    
+    async def send_request_blocking_async(self, content: str, conversation_id: str = None, user: str = "user") -> Dict[str, Any]:
+        """
+        异步发送阻塞模式请求
+        
+        Args:
+            content: 请求内容
+            conversation_id: 会话ID
+            user: 用户标识
+            
+        Returns:
+            API响应
+        """
+        url = f"{self.base_url}/chat-messages"
+        
+        payload = {
+            "query": content,
+            "inputs": {},
+            "response_mode": "blocking",
+            "user": user
+        }
+        
+        if conversation_id:
+            payload["conversation_id"] = conversation_id
+            
+        logging.info(f"异步发送阻塞模式请求，用户: {user}")
+        if conversation_id:
+            logging.info(f"会话ID: {conversation_id}")
+        
+        logging.debug(f"请求URL: {url}")
+        logging.debug(f"请求负载: {json.dumps(payload, ensure_ascii=False)}")
+        
+        try:
+            async with httpx.AsyncClient() as client:
+                response = await client.post(url, headers=self.headers, json=payload)
+                response.raise_for_status()
+                response_data = response.json()
+                logging.info("API请求成功")
+                logging.debug(f"API响应: {json.dumps(response_data, ensure_ascii=False)}")
+                return response_data
+        except httpx.HTTPStatusError as e:
+            logging.error(f"HTTP错误: {e}")
+            if hasattr(e.response, 'text'):
+                logging.error(f"API响应内容: {e.response.text}")
+            raise Exception(f"HTTP错误: {e}")
+        except httpx.ConnectError:
+            logging.error("连接错误: 无法连接到API服务器")
+            raise Exception("连接错误: 无法连接到API服务器")
+        except httpx.TimeoutException:
+            logging.error("超时错误: API请求超时")
+            raise Exception("超时错误: API请求超时")
+        except httpx.RequestError as e:
+            logging.error(f"请求错误: {str(e)}", exc_info=True)
+            raise Exception(f"请求错误: {e}")
+        except json.JSONDecodeError:
+            logging.error("解析错误: API返回的响应不是有效的JSON格式")
+            raise Exception("解析错误: API返回的响应不是有效的JSON格式")
+        except Exception as e:
+            logging.error(f"未知错误: {str(e)}", exc_info=True)
+            raise Exception(f"发送请求时发生未知错误: {str(e)}")
     
     def send_request_streaming(self, content: str, conversation_id: str = None, user: str = "user") -> str:
         """

requirements.txt

@@ -3,4 +3,6 @@ sseclient-py>=1.7.2
 fastapi>=0.95.0
 uvicorn>=0.21.0
 python-multipart>=0.0.6
-pydantic>=1.10.7 
+pydantic>=1.10.7
+httpx>=0.24.0
+aiofiles>=0.8.0