re-style: 格式化代码
This commit is contained in:
@@ -1,14 +1,15 @@
|
||||
import base64
|
||||
import io
|
||||
|
||||
from PIL import Image
|
||||
from datetime import datetime
|
||||
|
||||
from src.common.logger import get_logger
|
||||
from PIL import Image
|
||||
|
||||
from src.common.database.sqlalchemy_models import LLMUsage, get_db_session
|
||||
from src.common.logger import get_logger
|
||||
from src.config.api_ada_configs import ModelInfo
|
||||
from .payload_content.message import Message, MessageBuilder
|
||||
|
||||
from .model_client.base_client import UsageRecord
|
||||
from .payload_content.message import Message, MessageBuilder
|
||||
|
||||
logger = get_logger("消息压缩工具")
|
||||
|
||||
@@ -38,7 +39,7 @@ def compress_messages(messages: list[Message], img_target_size: int = 1 * 1024 *
|
||||
|
||||
return image_data
|
||||
except Exception as e:
|
||||
logger.error(f"图片转换格式失败: {str(e)}")
|
||||
logger.error(f"图片转换格式失败: {e!s}")
|
||||
return image_data
|
||||
|
||||
def rescale_image(image_data: bytes, scale: float) -> tuple[bytes, tuple[int, int] | None, tuple[int, int] | None]:
|
||||
@@ -87,7 +88,7 @@ def compress_messages(messages: list[Message], img_target_size: int = 1 * 1024 *
|
||||
return output_buffer.getvalue(), original_size, new_size
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"图片缩放失败: {str(e)}")
|
||||
logger.error(f"图片缩放失败: {e!s}")
|
||||
import traceback
|
||||
|
||||
logger.error(traceback.format_exc())
|
||||
@@ -188,7 +189,7 @@ class LLMUsageRecorder:
|
||||
f"总计: {model_usage.total_tokens}"
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"记录token使用情况失败: {str(e)}")
|
||||
logger.error(f"记录token使用情况失败: {e!s}")
|
||||
|
||||
|
||||
llm_usage_recorder = LLMUsageRecorder()
|
||||
|
||||
Reference in New Issue
Block a user