Create app.py

app.py

@@ -0,0 +1,86 @@
+from flask import Flask, request, Response
+from flask_cors import CORS
+import requests
+import json
+from datetime import datetime, timedelta
+import os
+
+app = Flask(__name__)
+CORS(app)
+
+CHATANYWHERE_ENDPOINT = 'https://api.chatanywhere.tech'
+
+class APIKeyManager:
+    def __init__(self):
+        self.api_keys = os.environ.get('API_KEYS', '').split(',')
+        self.daily_uses = {key: 0 for key in self.api_keys}
+        self.last_reset = datetime.now().date()
+
+    def reset_daily_uses(self):
+        today = datetime.now().date()
+        if today > self.last_reset:
+            self.daily_uses = {key: 0 for key in self.api_keys}
+            self.last_reset = today
+
+    def get_available_key(self):
+        self.reset_daily_uses()
+        for key in self.api_keys:
+            if self.daily_uses[key] < 200:  # 假设每个密钥每天可用200次
+                self.daily_uses[key] += 1
+                return key
+        return None
+
+    def get_key_info(self):
+        self.reset_daily_uses()
+        return [{"key": key, "uses": uses} for key, uses in self.daily_uses.items()]
+
+key_manager = APIKeyManager()
+
+@app.route('/v1/models', methods=['GET'])
+def list_models():
+    models = [
+        {
+            "id": "gpt-3.5-turbo",
+            "object": "model",
+            "created": 1677610602,
+            "owned_by": "openai",
+            "permission": [{
+                "id": "modelperm-M56FXnG1AsIr3SXq8BYPvXJA",
+                "object": "model_permission",
+                "created": 1679602088,
+                "allow_create_engine": False,
+                "allow_sampling": True,
+                "allow_logprobs": True,
+                "allow_search_indices": False,
+                "allow_view": True,
+                "allow_fine_tuning": False,
+                "organization": "*",
+                "group": None,
+                "is_blocking": False
+            }],
+            "root": "gpt-3.5-turbo",
+            "parent": None
+        }
+    ]
+    return Response(json.dumps({"data": models, "object": "list"}), mimetype='application/json')
+
+@app.route('/v1/chat/completions', methods=['POST'])
+def proxy_request():
+    api_key = key_manager.get_available_key()
+    if not api_key:
+        return Response(json.dumps({"error": "No available API key"}), status=429, mimetype='application/json')
+
+    headers = {
+        'Authorization': f'Bearer {api_key}',
+        'Content-Type': 'application/json'
+    }
+
+    response = requests.post(f'{CHATANYWHERE_ENDPOINT}/v1/chat/completions', headers=headers, data=request.data)
+    return Response(response.content, status=response.status_code, mimetype='application/json')
+
+@app.route('/key_info', methods=['GET'])
+def key_info():
+    return Response(json.dumps(key_manager.get_key_info()), mimetype='application/json')
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)