chat gpt 代理接口

redis/test_user_redis.py

@@ -5,10 +5,13 @@ import uuid
 import time
 import jwt
 import datetime
+import requests
+import os
 from flask import Flask, request, jsonify, Request
 from redis import Redis
 
 SECERT_KEY = "8U2LL1"
+MY_OPENAI_API_KEY = os.environ.get('MY_OPENAI_API_KEY')
 
 app = Flask(__name__)
 redis = Redis(host='192.168.3.229', port=6379, password='lizhen-redis')
@@ -190,6 +193,47 @@ def validate():
     return jsonify({'code': 0, 'message': 'Chat limit not exceeded'})
 
 
+@app.route('/chat/completions', methods=['POST'])
+def proxy_chat_completions():
+    # token = parse_token(request)
+
+    # # 验证令牌
+    # if not validate_token(token):
+    #     return jsonify({'code': 401, 'message': 'Invalid token'})
+
+    # user_id = get_user_id_from_token(token)
+
+    # if not user_id:
+    #     return jsonify({'code': 400, 'message': 'User not found'})
+
+    # # 获取用户套餐信息
+    # package = get_user_package(user_id)
+    # if not package:
+    #     return jsonify({'code': 400, 'message': 'User has not purchased any package'})
+
+    # # 检查用户聊天次数是否超过限制
+    # if exceeded_chat_limit(user_id, package):
+    #     return jsonify({'code': 400, 'message': 'Chat limit exceeded'})
+
+    # 获取请求数据
+    data = request.get_json()
+    
+    # 设置请求头部信息
+    headers = {
+        'Authorization': f'Bearer {MY_OPENAI_API_KEY}',
+        'Content-Type': 'application/json'
+    }
+    # 将请求转发到 OpenAI API
+    response = requests.post(
+        'https://api.openai.com/v1/chat/completions', json=data, headers=headers)
+
+    # 获取 OpenAI API 的响应数据
+    result = response.json()
+
+    # 返回 OpenAI API 的响应给客户端
+    return result, response.status_code
+
+
 def parse_token(request: Request):
     token_with_bearer = request.headers.get('Authorization')
 
@@ -208,7 +252,7 @@ def generate_token(user_id, username):
         'user_id': user_id,
         'username': username,
         'exp': datetime.datetime.utcnow() + datetime.timedelta(hours=1)
-        }
+    }
 
     # 在这里，您可以使用您的密钥（secret key）来签署令牌
     # 选择适当的签名算法，并设置适当的过期时间等参数