style: ruff check fix
This commit is contained in:
parent
eeb53f4774
commit
286e06f25b
@ -43,7 +43,7 @@ dependencies = [
|
||||
"zhipuai>=2.1.5.20250421",
|
||||
]
|
||||
[tool.ruff]
|
||||
line-length = 140 # 代码最大行宽
|
||||
line-length = 210 # 代码最大行宽
|
||||
lint.select = [ # 选择的规则
|
||||
"F",
|
||||
"E",
|
||||
|
||||
@ -24,7 +24,7 @@ def load_info_config():
|
||||
return get_default_info_config()
|
||||
|
||||
# 读取配置文件
|
||||
with open(config_path, 'r', encoding='utf-8') as file:
|
||||
with open(config_path, encoding='utf-8') as file:
|
||||
config = yaml.safe_load(file)
|
||||
|
||||
# logger.info(f"成功加载信息配置文件: {config_path}")
|
||||
@ -38,7 +38,7 @@ def get_default_info_config():
|
||||
"""获取默认信息配置"""
|
||||
return {
|
||||
"organization": {
|
||||
"name": "江南大学",
|
||||
"name": "江南语析",
|
||||
"short_name": "语析",
|
||||
"logo": "/favicon.svg",
|
||||
"avatar": "/avatar.jpg"
|
||||
@ -54,7 +54,7 @@ def get_default_info_config():
|
||||
"🤖 多模型支持"
|
||||
],
|
||||
"footer": {
|
||||
"copyright": "© 江南大学 2025 [WIP] v0.12.138"
|
||||
"copyright": "© 江南语析 2025 [WIP] v0.12.138"
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@ -582,7 +582,7 @@ class KnowledgeBase:
|
||||
for field in ["text", "file_id", "hash", "start_char_idx", "end_char_idx"]:
|
||||
try:
|
||||
item_dict["entity"][field] = getattr(entity_data, field, None)
|
||||
except:
|
||||
except Exception:
|
||||
continue
|
||||
|
||||
all_db_result_dicts.append(item_dict)
|
||||
|
||||
@ -45,7 +45,9 @@ def do_parse(
|
||||
new_pdf_bytes = convert_pdf_bytes_to_bytes_by_pypdfium2(pdf_bytes, start_page_id, end_page_id)
|
||||
pdf_bytes_list[idx] = new_pdf_bytes
|
||||
|
||||
infer_results, all_image_lists, all_pdf_docs, lang_list, ocr_enabled_list = pipeline_doc_analyze(pdf_bytes_list, p_lang_list, parse_method=parse_method, formula_enable=p_formula_enable,table_enable=p_table_enable)
|
||||
|
||||
result = pipeline_doc_analyze(pdf_bytes_list, p_lang_list, parse_method=parse_method, formula_enable=p_formula_enable,table_enable=p_table_enable)
|
||||
infer_results, all_image_lists, all_pdf_docs, lang_list, ocr_enabled_list = result
|
||||
|
||||
|
||||
md_results = []
|
||||
@ -244,4 +246,4 @@ if __name__ == "__main__":
|
||||
if doc_path.suffix in pdf_suffixes + image_suffixes:
|
||||
doc_path_list.append(doc_path)
|
||||
|
||||
parse_doc(doc_path_list, output_dir, backend="vlm-sglang-client", server_url="http://172.19.13.5:30000") # faster(client).
|
||||
parse_doc(doc_path_list, output_dir, backend="vlm-sglang-client", server_url="http://172.19.13.5:30000") # faster(client).
|
||||
|
||||
@ -1,10 +1,9 @@
|
||||
import logging
|
||||
import os
|
||||
from datetime import datetime
|
||||
|
||||
import pytz
|
||||
|
||||
from loguru import logger
|
||||
from loguru import logger as loguru_logger
|
||||
|
||||
DATETIME = datetime.now(pytz.timezone('Asia/Shanghai')).strftime('%Y-%m-%d-%H%M%S')
|
||||
# DATETIME = "debug" # 为了方便,调试的时候输出到 debug.log 文件
|
||||
@ -12,12 +11,12 @@ LOG_FILE = f'saves/log/project-{DATETIME}.log'
|
||||
def setup_logger(name, level="DEBUG", console=True):
|
||||
"""使用 loguru 设置日志记录器"""
|
||||
os.makedirs("saves/log", exist_ok=True)
|
||||
|
||||
|
||||
# 移除默认的 handler
|
||||
logger.remove()
|
||||
|
||||
loguru_logger.remove()
|
||||
|
||||
# 添加文件日志(无颜色)
|
||||
logger.add(
|
||||
loguru_logger.add(
|
||||
LOG_FILE,
|
||||
level=level,
|
||||
format="{time:YYYY-MM-DD HH:mm:ss} - {level} - {name} - {message}",
|
||||
@ -26,21 +25,23 @@ def setup_logger(name, level="DEBUG", console=True):
|
||||
retention="30 days", # 保留30天的日志
|
||||
compression="zip" # 压缩旧日志文件
|
||||
)
|
||||
|
||||
|
||||
# 添加控制台日志(有颜色)
|
||||
if console:
|
||||
logger.add(
|
||||
loguru_logger.add(
|
||||
lambda msg: print(msg, end=""),
|
||||
level=level,
|
||||
format="<green>{time:YYYY-MM-DD HH:mm:ss}</green> - <level>{level}</level> - <cyan>{name}</cyan> - <level>{message}</level>",
|
||||
colorize=True
|
||||
)
|
||||
|
||||
return logger
|
||||
|
||||
return loguru_logger
|
||||
|
||||
|
||||
# 设置根日志记录器
|
||||
logger = setup_logger('Yuxi')
|
||||
|
||||
__all__ = ['logger']
|
||||
|
||||
# If you want to disable logging from external libraries
|
||||
# logging.getLogger('some_external_library').setLevel(logging.CRITICAL)
|
||||
|
||||
@ -32,7 +32,7 @@ export const useInfoStore = defineStore('info', () => {
|
||||
|
||||
// 计算属性 - 页脚信息
|
||||
const footer = computed(() => infoConfig.value.footer || {
|
||||
copyright: "© 江南大学 2025 [WIP] v0.12.138"
|
||||
copyright: "© 江南语析 2025 [WIP] v0.12.138"
|
||||
})
|
||||
|
||||
// 动作方法
|
||||
|
||||
Loading…
Reference in New Issue
Block a user