From ee21fee34afa9daf1c60b2bd5ac4cec05f98b73b Mon Sep 17 00:00:00 2001 From: 13315423919 <13315423919@qq.com> Date: Fri, 7 Nov 2025 09:05:45 +0800 Subject: [PATCH] Add File --- src/summeryanyfile/core/models.py | 231 ++++++++++++++++++++++++++++++ 1 file changed, 231 insertions(+) create mode 100644 src/summeryanyfile/core/models.py diff --git a/src/summeryanyfile/core/models.py b/src/summeryanyfile/core/models.py new file mode 100644 index 0000000..affb45e --- /dev/null +++ b/src/summeryanyfile/core/models.py @@ -0,0 +1,231 @@ +""" +数据模型定义 +""" + +import os +from dataclasses import dataclass +from typing import List, Dict, Any, Literal, TypedDict, Optional +from enum import Enum + + +def _get_default_max_tokens() -> int: + """ + 从环境变量获取默认的 max_tokens 值 + + Returns: + 默认的 max_tokens 值 + """ + try: + # 尝试从环境变量读取 + env_value = os.getenv("MAX_TOKENS") + if env_value: + return int(env_value) + except (ValueError, TypeError): + pass + + # 如果环境变量不存在或无效，返回默认值 + return 4000 + + +def _get_default_temperature() -> float: + """ + 从环境变量获取默认的 temperature 值 + + Returns: + 默认的 temperature 值 + """ + try: + # 尝试从环境变量读取 + env_value = os.getenv("TEMPERATURE") + if env_value: + return float(env_value) + except (ValueError, TypeError): + pass + + # 如果环境变量不存在或无效，返回默认值 + return 0.7 + + +class ChunkStrategy(Enum): + """文档分块策略""" + PARAGRAPH = "paragraph" # 基于段落分块 + SEMANTIC = "semantic" # 语义分块 + RECURSIVE = "recursive" # 递归分块 + HYBRID = "hybrid" # 混合策略 + FAST = "fast" # 快速分块策略 + + +@dataclass +class SlideInfo: + """幻灯片信息数据类""" + page_number: int + title: str + content_points: List[str] + slide_type: Literal["title", "content", "conclusion"] + description: str + chart_config: Optional[Dict[str, Any]] = None # 新增：图表配置 + + def to_dict(self) -> Dict[str, Any]: + """转换为字典格式""" + result = { + "page_number": self.page_number, + "title": self.title, + "content_points": self.content_points, + "slide_type": self.slide_type, + "description": self.description, + } + + # 只有当chart_config存在时才添加到字典中 + if self.chart_config is not None: + result["chart_config"] = self.chart_config + + return result + + @classmethod + def from_dict(cls, data: Dict[str, Any]) -> "SlideInfo": + """从字典创建实例""" + return cls( + page_number=data.get("page_number", 1), + title=data.get("title", ""), + content_points=data.get("content_points", []), + slide_type=data.get("slide_type", "content"), + description=data.get("description", ""), + chart_config=data.get("chart_config"), # 新增：图表配置 + ) + + +class PPTState(TypedDict): + """PPT生成状态""" + document_chunks: List[str] + current_index: int + ppt_title: str + slides: List[Dict[str, Any]] + total_pages: int + page_count_mode: str + document_structure: Dict[str, Any] + accumulated_context: str + # 项目信息参数 + project_topic: str + project_scenario: str + project_requirements: str + target_audience: str + custom_audience: str + ppt_style: str + custom_style_prompt: str + # 页数设置参数 + min_pages: Optional[int] + max_pages: Optional[int] + fixed_pages: Optional[int] + + +@dataclass +class DocumentInfo: + """文档信息""" + title: str + content: str + file_path: str + file_type: str + encoding: str + size: int + + def to_dict(self) -> Dict[str, Any]: + """转换为字典格式""" + return { + "title": self.title, + "content": self.content, + "file_path": self.file_path, + "file_type": self.file_type, + "encoding": self.encoding, + "size": self.size, + } + + +@dataclass +class ProcessingConfig: + """处理配置""" + max_slides: int = 25 + min_slides: int = 5 # 新增：最小页数 + chunk_size: int = 3000 + chunk_overlap: int = 200 + chunk_strategy: ChunkStrategy = ChunkStrategy.PARAGRAPH + llm_model: str = "gpt-4o-mini" + llm_provider: str = "openai" + temperature: float = None # 将在 __post_init__ 中设置默认值 + max_tokens: int = None # 将在 __post_init__ 中设置默认值 + recursion_limit: Optional[int] = None # 工作流递归限制，None表示自动计算 + target_language: str = "zh" # 新增：目标语言，由用户在表单中选择 + + def __post_init__(self): + """后处理验证和默认值设置""" + # 如果 max_tokens 为 None，从环境变量获取默认值 + if self.max_tokens is None: + self.max_tokens = _get_default_max_tokens() + + # 如果 temperature 为 None，从环境变量获取默认值 + if self.temperature is None: + self.temperature = _get_default_temperature() + + # 验证逻辑 + if self.min_slides > self.max_slides: + raise ValueError(f"最小页数({self.min_slides})不能大于最大页数({self.max_slides})") + if self.min_slides < 1: + raise ValueError("最小页数不能小于1") + if self.max_slides > 1000: + raise ValueError("最大页数不能超过1000") + if self.recursion_limit is not None and self.recursion_limit < 10: + raise ValueError("递归限制不能小于10") + + + + @property + def slides_range(self) -> str: + """获取页数范围的字符串表示""" + if self.min_slides == self.max_slides: + return f"{self.min_slides}页" + return f"{self.min_slides}-{self.max_slides}页" + + def to_dict(self) -> Dict[str, Any]: + """转换为字典格式""" + return { + "max_slides": self.max_slides, + "min_slides": self.min_slides, + "slides_range": self.slides_range, + "chunk_size": self.chunk_size, + "chunk_overlap": self.chunk_overlap, + "chunk_strategy": self.chunk_strategy.value, + "llm_model": self.llm_model, + "llm_provider": self.llm_provider, + "temperature": self.temperature, + "max_tokens": self.max_tokens, + "recursion_limit": self.recursion_limit, + "target_language": self.target_language, + } + + +@dataclass +class PPTOutline: + """PPT大纲结果""" + title: str + total_pages: int + page_count_mode: str + slides: List[SlideInfo] + + def to_dict(self) -> Dict[str, Any]: + """转换为字典格式""" + return { + "title": self.title, + "total_pages": self.total_pages, + "page_count_mode": self.page_count_mode, + "slides": [slide.to_dict() for slide in self.slides], + } + + @classmethod + def from_dict(cls, data: Dict[str, Any]) -> "PPTOutline": + """从字典创建实例""" + slides = [SlideInfo.from_dict(slide_data) for slide_data in data.get("slides", [])] + return cls( + title=data.get("title", ""), + total_pages=data.get("total_pages", 0), + page_count_mode=data.get("page_count_mode", "final"), + slides=slides, + )