新增自定义日志模块,替换全局logging实现;优化日志输出格式与颜色支持

This commit is contained in:
yinpeng
2024-12-16 16:20:53 +08:00
parent 829c1b02ec
commit 56a069dbac
10 changed files with 131 additions and 24 deletions

View File

@@ -1,6 +1,5 @@
from http.client import HTTPException
from fastapi import APIRouter, Depends, Header
import logging
from fastapi.responses import StreamingResponse
from app.core.security import SecurityService
@@ -10,9 +9,10 @@ from app.services.chat_service import ChatService
from app.services.embedding_service import EmbeddingService
from app.schemas.request_model import ChatRequest, EmbeddingRequest
from app.core.config import settings
from app.core.logger import get_api_logger
router = APIRouter()
logger = logging.getLogger(__name__)
logger = get_api_logger()
# 初始化服务
security_service = SecurityService(settings.ALLOWED_TOKENS)
@@ -59,7 +59,6 @@ async def chat_completion(
tools=request.tools,
tool_choice=request.tool_choice,
)
# 处理流式响应
if request.stream: