diff --git a/.gitignore b/.gitignore index fd12d49..d00e56d 100644 --- a/.gitignore +++ b/.gitignore @@ -9,4 +9,5 @@ node_modules *.jpg *.json *.png -*.db \ No newline at end of file +*.db +.aider* diff --git a/main.py b/main.py index 7b676f1..c9173f9 100644 --- a/main.py +++ b/main.py @@ -22,6 +22,9 @@ from urllib.parse import urlparse import os +import string +import json + is_debug = bool(os.getenv("DEBUG", False)) from sqlalchemy import inspect, text @@ -463,9 +466,8 @@ def get_matching_providers(self, model_name, token): # if model_name in provider['model'].keys(): # provider_list.append(provider) if is_debug: - import json for provider in provider_list: - print(json.dumps(provider, indent=4, ensure_ascii=False, default=circular_list_encoder)) + logger.info("available provider: %s", json.dumps(provider, indent=4, ensure_ascii=False, default=circular_list_encoder)) return provider_list async def request_model(self, request: Union[RequestModel, ImageGenerationRequest, AudioTranscriptionRequest, ModerationRequest], token: str, endpoint=None): @@ -698,7 +700,11 @@ async def audio_transcriptions( @app.get("/generate-api-key", dependencies=[Depends(rate_limit_dependency)]) def generate_api_key(): - api_key = "sk-" + secrets.token_urlsafe(36) + # Define the character set (only alphanumeric) + chars = string.ascii_letters + string.digits + # Generate a random string of 36 characters + random_string = ''.join(secrets.choice(chars) for _ in range(36)) + api_key = "sk-" + random_string return JSONResponse(content={"api_key": api_key}) # 在 /stats 路由中返回成功和失败百分比