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 路由中返回成功和失败百分比
|