Update app.py

app.py

@@ -12,9 +12,47 @@ logger = logging.getLogger(__name__)
 # 加载环境变量
 load_dotenv()
 
+
+
 # API配置
-API_URL = os.getenv("API_URL")
-API_KEY = os.getenv("API_KEY")
+BASE_URL = os.getenv("API_URL", "").rstrip('/')  # 移除末尾的斜杠
+API_KEY = os.getenv("API_KEY", "")
+
+# 构建完整的API URL
+API_URL = f"{BASE_URL}/v1/chat/completions"
+
+# 验证环境变量
+if not BASE_URL or not API_KEY:
+    raise ValueError("""
+    请确保设置了必要的环境变量：
+    - API_URL: API基础地址 (例如: https://api.example.com)
+    - API_KEY: API密钥
+    可以在Hugging Face Space的Settings -> Repository Secrets中设置这些变量
+    """)
+
+class ChatBot:
+    def __init__(self):
+        # 修正 headers 的设置
+        self.headers = {
+            "Authorization": f"Bearer {API_KEY}",  # 正确使用 API_KEY
+            "Content-Type": "application/json",
+            "Accept": "text/event-stream"
+        }
+        self.verify_api_config()
+    
+    def verify_api_config(self):
+        try:
+            # 使用 OPTIONS 请求来验证API端点
+            response = requests.options(API_URL, timeout=5)
+            logger.info(f"API endpoint: {API_URL}")
+            logger.info(f"API headers: {self.headers}")
+            
+            if response.status_code >= 400:
+                logger.error(f"API配置可能有误: {response.status_code}")
+                logger.error(f"API响应: {response.text[:200]}")
+        except Exception as e:
+            logger.error(f"API连接测试失败: {str(e)}")
+
 
 def format_message(role, content):
     return {"role": role, "content": content}