GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-09-05 07:26:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

为每个API端点添加日志分隔符

Browse Source
This commit is contained in:
yinpeng
2024-12-16 19:48:28 +08:00
parent 09c2993210
commit 16a2f64cd3
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/api/routes.py
+4 -1
View File
@@ -28,6 +28,7 @@ async def list_models(
authorization: str = Header(None), authorization: str = Header(None),
token: str = Depends(security_service.verify_authorization), token: str = Depends(security_service.verify_authorization),
): ):
logger.info("-" * 50 + "list_models" + "-" * 50)
logger.info("Handling models list request") logger.info("Handling models list request")
api_key = await key_manager.get_next_working_key() api_key = await key_manager.get_next_working_key()
logger.info(f"Using API key: {api_key}") logger.info(f"Using API key: {api_key}")
@@ -41,7 +42,7 @@ async def chat_completion(
authorization: str = Header(None), authorization: str = Header(None),
token: str = Depends(security_service.verify_authorization), token: str = Depends(security_service.verify_authorization),
): ):
logger.info("-" * 50) logger.info("-" * 50 + "chat_completion" + "-" * 50)
logger.info(f"Handling chat completion request for model: {request.model}") logger.info(f"Handling chat completion request for model: {request.model}")
logger.info(f"Request: \n{request.model_dump_json(indent=2)}") logger.info(f"Request: \n{request.model_dump_json(indent=2)}")
api_key = await key_manager.get_next_working_key() api_key = await key_manager.get_next_working_key()
@@ -85,6 +86,7 @@ async def embedding(
authorization: str = Header(None), authorization: str = Header(None),
token: str = Depends(security_service.verify_authorization), token: str = Depends(security_service.verify_authorization),
): ):
logger.info("-" * 50 + "embedding" + "-" * 50)
logger.info(f"Handling embedding request for model: {request.model}") logger.info(f"Handling embedding request for model: {request.model}")
api_key = await key_manager.get_next_working_key() api_key = await key_manager.get_next_working_key()
logger.info(f"Using API key: {api_key}") logger.info(f"Using API key: {api_key}")
@@ -106,6 +108,7 @@ async def get_keys_list(
token: str = Depends(security_service.verify_authorization), token: str = Depends(security_service.verify_authorization),
): ):
"""获取有效和无效的API key列表""" """获取有效和无效的API key列表"""
logger.info("-" * 50 + "get_keys_list" + "-" * 50)
logger.info("Handling keys list request") logger.info("Handling keys list request")
try: try:
keys_status = await key_manager.get_keys_by_status() keys_status = await key_manager.get_keys_by_status()