47 lines
1.4 KiB
Rust
47 lines
1.4 KiB
Rust
use crate::models::DocumentFormat;
|
||
use serde::{Deserialize, Serialize};
|
||
use utoipa::ToSchema;
|
||
|
||
/// 解析引擎枚举
|
||
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, ToSchema)]
|
||
pub enum ParserEngine {
|
||
MinerU, // PDF专用
|
||
MarkItDown, // 其他格式
|
||
}
|
||
|
||
impl ParserEngine {
|
||
/// 根据文档格式选择解析引擎
|
||
pub fn select_for_format(format: &DocumentFormat) -> Self {
|
||
match format {
|
||
DocumentFormat::PDF => ParserEngine::MinerU,
|
||
_ => ParserEngine::MarkItDown, // 其他所有格式使用MarkItDown
|
||
}
|
||
}
|
||
|
||
/// 获取引擎名称
|
||
pub fn get_name(&self) -> &'static str {
|
||
match self {
|
||
ParserEngine::MinerU => "MinerU",
|
||
ParserEngine::MarkItDown => "MarkItDown",
|
||
}
|
||
}
|
||
|
||
/// 获取引擎描述
|
||
pub fn get_description(&self) -> &'static str {
|
||
match self {
|
||
ParserEngine::MinerU => "专业PDF解析引擎，支持图片提取、表格识别、布局保持",
|
||
ParserEngine::MarkItDown => {
|
||
"多格式文档解析引擎，支持Word、Excel、PowerPoint、图片、音频等"
|
||
}
|
||
}
|
||
}
|
||
|
||
/// 检查是否支持指定格式
|
||
pub fn supports_format(&self, format: &DocumentFormat) -> bool {
|
||
match self {
|
||
ParserEngine::MinerU => matches!(format, DocumentFormat::PDF),
|
||
ParserEngine::MarkItDown => !matches!(format, DocumentFormat::PDF),
|
||
}
|
||
}
|
||
}
|