Create app.py

app.py

@@ -0,0 +1,65 @@
+from flask import Flask, request, Response, jsonify
+import requests
+
+app = Flask(__name__)
+
+@app.route('/')
+def index():
+    return "Hello, this is the root page of your Flask application!"
+
+@app.route('/hf/v1/chat/completions', methods=['POST'])
+def chat():
+    try:
+        # 从请求头中获取 API 密钥和基础 URL
+        auth_header = request.headers.get('Authorization')
+        if not auth_header or not auth_header.startswith('Bearer '):
+            return Response("Unauthorized", status=401)
+        
+        api_key, base_url = auth_header.split(" ")[1:]
+
+        # 构建目标 URL
+        target_url = f"{base_url}/v1/chat/completions"
+
+        # 准备请求头
+        headers = {
+            'Authorization': f'Bearer {api_key}',
+            'Content-Type': 'application/json'
+        }
+
+        # 检查是否为流式请求
+        is_stream = request.json.get('stream', False)
+
+        # 转发请求
+        response = requests.post(
+            target_url,
+            headers=headers,
+            json=request.json,
+            stream=is_stream
+        )
+
+        if is_stream:
+            # 流式响应
+            def generate():
+                for chunk in response.iter_content(chunk_size=1024):
+                    yield chunk
+
+            return Response(
+                generate(),
+                content_type=response.headers['Content-Type'],
+                status=response.status_code
+            )
+        else:
+            # 非流式响应
+            return Response(
+                response.content,
+                content_type=response.headers['Content-Type'],
+                status=response.status_code
+            )
+
+    except Exception as e:
+        print("Exception:", e)
+        return jsonify({"error": str(e)}), 500
+
+if __name__ == "__main__":
+    app.run(host='0.0.0.0', port=7860, threaded=True)
+