Benedict King commited on
Commit ·
bead939
1
Parent(s): 81035a5
fix: Generate a random API key using only alphanumeric characters
Browse files- .gitignore +2 -1
- main.py +9 -3
.gitignore
CHANGED
|
@@ -9,4 +9,5 @@ node_modules
|
|
| 9 |
*.jpg
|
| 10 |
*.json
|
| 11 |
*.png
|
| 12 |
-
*.db
|
|
|
|
|
|
| 9 |
*.jpg
|
| 10 |
*.json
|
| 11 |
*.png
|
| 12 |
+
*.db
|
| 13 |
+
.aider*
|
main.py
CHANGED
|
@@ -22,6 +22,9 @@ from typing import List, Dict, Union
|
|
| 22 |
from urllib.parse import urlparse
|
| 23 |
|
| 24 |
import os
|
|
|
|
|
|
|
|
|
|
| 25 |
is_debug = bool(os.getenv("DEBUG", False))
|
| 26 |
|
| 27 |
from sqlalchemy import inspect, text
|
|
@@ -463,9 +466,8 @@ class ModelRequestHandler:
|
|
| 463 |
# if model_name in provider['model'].keys():
|
| 464 |
# provider_list.append(provider)
|
| 465 |
if is_debug:
|
| 466 |
-
import json
|
| 467 |
for provider in provider_list:
|
| 468 |
-
|
| 469 |
return provider_list
|
| 470 |
|
| 471 |
async def request_model(self, request: Union[RequestModel, ImageGenerationRequest, AudioTranscriptionRequest, ModerationRequest], token: str, endpoint=None):
|
|
@@ -698,7 +700,11 @@ async def audio_transcriptions(
|
|
| 698 |
|
| 699 |
@app.get("/generate-api-key", dependencies=[Depends(rate_limit_dependency)])
|
| 700 |
def generate_api_key():
|
| 701 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 702 |
return JSONResponse(content={"api_key": api_key})
|
| 703 |
|
| 704 |
# 在 /stats 路由中返回成功和失败百分比
|
|
|
|
| 22 |
from urllib.parse import urlparse
|
| 23 |
|
| 24 |
import os
|
| 25 |
+
import string
|
| 26 |
+
import json
|
| 27 |
+
|
| 28 |
is_debug = bool(os.getenv("DEBUG", False))
|
| 29 |
|
| 30 |
from sqlalchemy import inspect, text
|
|
|
|
| 466 |
# if model_name in provider['model'].keys():
|
| 467 |
# provider_list.append(provider)
|
| 468 |
if is_debug:
|
|
|
|
| 469 |
for provider in provider_list:
|
| 470 |
+
logger.info("available provider: %s", json.dumps(provider, indent=4, ensure_ascii=False, default=circular_list_encoder))
|
| 471 |
return provider_list
|
| 472 |
|
| 473 |
async def request_model(self, request: Union[RequestModel, ImageGenerationRequest, AudioTranscriptionRequest, ModerationRequest], token: str, endpoint=None):
|
|
|
|
| 700 |
|
| 701 |
@app.get("/generate-api-key", dependencies=[Depends(rate_limit_dependency)])
|
| 702 |
def generate_api_key():
|
| 703 |
+
# Define the character set (only alphanumeric)
|
| 704 |
+
chars = string.ascii_letters + string.digits
|
| 705 |
+
# Generate a random string of 36 characters
|
| 706 |
+
random_string = ''.join(secrets.choice(chars) for _ in range(36))
|
| 707 |
+
api_key = "sk-" + random_string
|
| 708 |
return JSONResponse(content={"api_key": api_key})
|
| 709 |
|
| 710 |
# 在 /stats 路由中返回成功和失败百分比
|