yym68686 commited on
Commit
c2775d8
·
1 Parent(s): bc7d0c8

Fix a possible bug that occurs when attempting to encode a dictionary (dict) object as a string in a FastAPI application.

Browse files
Files changed (1) hide show
  1. main.py +5 -5
main.py CHANGED
@@ -174,23 +174,23 @@ async def options_handler():
174
  @app.post("/v1/models")
175
  async def list_models(token: str = Depends(verify_api_key)):
176
  models = post_all_models(token)
177
- return {
178
  "object": "list",
179
  "data": models
180
- }
181
 
182
  @app.get("/v1/models")
183
  async def list_models():
184
  models = get_all_models()
185
- return {
186
  "object": "list",
187
  "data": models
188
- }
189
 
190
  @app.get("/generate-api-key")
191
  def generate_api_key():
192
  api_key = "sk-" + secrets.token_urlsafe(32)
193
- return {"api_key": api_key}
194
 
195
  # async def on_fetch(request, env):
196
  # import asgi
 
174
  @app.post("/v1/models")
175
  async def list_models(token: str = Depends(verify_api_key)):
176
  models = post_all_models(token)
177
+ return JSONResponse(content={
178
  "object": "list",
179
  "data": models
180
+ })
181
 
182
  @app.get("/v1/models")
183
  async def list_models():
184
  models = get_all_models()
185
+ return JSONResponse(content={
186
  "object": "list",
187
  "data": models
188
+ })
189
 
190
  @app.get("/generate-api-key")
191
  def generate_api_key():
192
  api_key = "sk-" + secrets.token_urlsafe(32)
193
+ return JSONResponse(content={"api_key": api_key})
194
 
195
  # async def on_fetch(request, env):
196
  # import asgi