提交qiming-mcp-proxy

qiming-mcp-proxy/document-parser/src/models/parser_engine.rs

@@ -0,0 +1,46 @@
+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),
+        }
+    }
+}