Benedict King commited on
Commit
bead939
·
1 Parent(s): 81035a5

fix: Generate a random API key using only alphanumeric characters

Browse files
Files changed (2) hide show
  1. .gitignore +2 -1
  2. 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
- print(json.dumps(provider, indent=4, ensure_ascii=False, default=circular_list_encoder))
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
- api_key = "sk-" + secrets.token_urlsafe(36)
 
 
 
 
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 路由中返回成功和失败百分比