36 lines
947 B
Python
36 lines
947 B
Python
"""核心数据结构定义"""
|
||
|
||
from dataclasses import dataclass
|
||
from datetime import datetime
|
||
from typing import List, Optional
|
||
|
||
|
||
@dataclass
|
||
class Chunk:
|
||
"""语义分块单元"""
|
||
|
||
title: str # AI 生成的摘要标题
|
||
content: str # 分块的文本内容
|
||
tag: str = "" # 业务分类标签（如：产品说明、问答、培训、招商等）
|
||
|
||
|
||
@dataclass
|
||
class ProcessResult:
|
||
"""处理结果"""
|
||
|
||
source_file: str # 源文件路径
|
||
output_file: str # 输出文件路径
|
||
chunks: List[Chunk] # 分块列表
|
||
process_time: datetime # 处理时间
|
||
total_chunks: int # 总分块数
|
||
|
||
|
||
@dataclass
|
||
class CLIArgs:
|
||
"""命令行参数"""
|
||
|
||
input_file: str # 输入文件路径（必需）
|
||
api_key: str # DeepSeek API Key（必需）
|
||
output_file: Optional[str] = None # 输出文件路径（可选，默认同名 .md）
|
||
delimiter: str = "---" # 分块分隔符（可选，默认 ---）
|