auto-coder 0.1.398__py3-none-any.whl → 0.1.400__py3-none-any.whl

autocoder/sdk/utils/validators.py

@@ -0,0 +1,287 @@
+
+
+
+
+
+"""
+Auto-Coder SDK 验证工具
+
+提供各种数据验证功能。
+"""
+
+import re
+import uuid
+from typing import Any, List, Dict
+from pathlib import Path
+
+from ..models.options import AutoCodeOptions
+from ..exceptions import ValidationError
+
+
+def validate_session_id(session_id: str) -> bool:
+    """
+    验证会话ID格式
+    
+    Args:
+        session_id: 会话ID
+        
+    Returns:
+        bool: 是否有效
+        
+    Raises:
+        ValidationError: 验证失败时抛出
+    """
+    if not session_id:
+        raise ValidationError("session_id", "cannot be empty")
+    
+    # 检查是否为有效的UUID格式
+    try:
+        uuid.UUID(session_id)
+        return True
+    except ValueError:
+        pass
+    
+    # 检查是否为有效的自定义格式（字母数字和连字符）
+    if re.match(r'^[a-zA-Z0-9\-_]+$', session_id):
+        return True
+    
+    raise ValidationError(
+        "session_id", 
+        "must be a valid UUID or alphanumeric string with hyphens/underscores"
+    )
+
+
+def validate_options(options: AutoCodeOptions) -> bool:
+    """
+    验证AutoCodeOptions对象
+    
+    Args:
+        options: 选项对象
+        
+    Returns:
+        bool: 是否有效
+        
+    Raises:
+        ValidationError: 验证失败时抛出
+    """
+    if not isinstance(options, AutoCodeOptions):
+        raise ValidationError("options", "must be an AutoCodeOptions instance")
+    
+    # AutoCodeOptions内部已经有验证，这里调用即可
+    options.validate()
+    return True
+
+
+def validate_file_path(file_path: str, must_exist: bool = False) -> bool:
+    """
+    验证文件路径
+    
+    Args:
+        file_path: 文件路径
+        must_exist: 是否必须存在
+        
+    Returns:
+        bool: 是否有效
+        
+    Raises:
+        ValidationError: 验证失败时抛出
+    """
+    if not file_path:
+        raise ValidationError("file_path", "cannot be empty")
+    
+    try:
+        path = Path(file_path)
+        
+        if must_exist and not path.exists():
+            raise ValidationError("file_path", f"file does not exist: {file_path}")
+        
+        # 检查路径是否合法（不包含非法字符）
+        if any(char in str(path) for char in ['<', '>', ':', '"', '|', '?', '*']):
+            raise ValidationError("file_path", f"contains invalid characters: {file_path}")
+        
+        return True
+    except OSError as e:
+        raise ValidationError("file_path", f"invalid path: {str(e)}")
+
+
+def validate_directory_path(dir_path: str, must_exist: bool = False, create_if_missing: bool = False) -> bool:
+    """
+    验证目录路径
+    
+    Args:
+        dir_path: 目录路径
+        must_exist: 是否必须存在
+        create_if_missing: 如果不存在是否创建
+        
+    Returns:
+        bool: 是否有效
+        
+    Raises:
+        ValidationError: 验证失败时抛出
+    """
+    if not dir_path:
+        raise ValidationError("dir_path", "cannot be empty")
+    
+    try:
+        path = Path(dir_path)
+        
+        if not path.exists():
+            if must_exist and not create_if_missing:
+                raise ValidationError("dir_path", f"directory does not exist: {dir_path}")
+            elif create_if_missing:
+                path.mkdir(parents=True, exist_ok=True)
+        elif path.exists() and not path.is_dir():
+            raise ValidationError("dir_path", f"path exists but is not a directory: {dir_path}")
+        
+        return True
+    except OSError as e:
+        raise ValidationError("dir_path", f"invalid directory path: {str(e)}")
+
+
+def validate_prompt(prompt: str, max_length: int = 10000) -> bool:
+    """
+    验证提示内容
+    
+    Args:
+        prompt: 提示内容
+        max_length: 最大长度
+        
+    Returns:
+        bool: 是否有效
+        
+    Raises:
+        ValidationError: 验证失败时抛出
+    """
+    if not prompt:
+        raise ValidationError("prompt", "cannot be empty")
+    
+    if not isinstance(prompt, str):
+        raise ValidationError("prompt", "must be a string")
+    
+    if len(prompt.strip()) == 0:
+        raise ValidationError("prompt", "cannot be only whitespace")
+    
+    if len(prompt) > max_length:
+        raise ValidationError("prompt", f"exceeds maximum length of {max_length} characters")
+    
+    return True
+
+
+def validate_output_format(output_format: str) -> bool:
+    """
+    验证输出格式
+    
+    Args:
+        output_format: 输出格式
+        
+    Returns:
+        bool: 是否有效
+        
+    Raises:
+        ValidationError: 验证失败时抛出
+    """
+    from ..constants import OUTPUT_FORMATS
+    
+    if not output_format:
+        raise ValidationError("output_format", "cannot be empty")
+    
+    if output_format not in OUTPUT_FORMATS:
+        raise ValidationError(
+            "output_format",
+            f"must be one of: {', '.join(OUTPUT_FORMATS.keys())}"
+        )
+    
+    return True
+
+
+def validate_permission_mode(permission_mode: str) -> bool:
+    """
+    验证权限模式
+    
+    Args:
+        permission_mode: 权限模式
+        
+    Returns:
+        bool: 是否有效
+        
+    Raises:
+        ValidationError: 验证失败时抛出
+    """
+    from ..constants import PERMISSION_MODES
+    
+    if not permission_mode:
+        raise ValidationError("permission_mode", "cannot be empty")
+    
+    if permission_mode not in PERMISSION_MODES:
+        raise ValidationError(
+            "permission_mode",
+            f"must be one of: {', '.join(PERMISSION_MODES.keys())}"
+        )
+    
+    return True
+
+
+def validate_allowed_tools(allowed_tools: List[str]) -> bool:
+    """
+    验证允许的工具列表
+    
+    Args:
+        allowed_tools: 工具列表
+        
+    Returns:
+        bool: 是否有效
+        
+    Raises:
+        ValidationError: 验证失败时抛出
+    """
+    from ..constants import ALLOWED_TOOLS
+    
+    if not isinstance(allowed_tools, list):
+        raise ValidationError("allowed_tools", "must be a list")
+    
+    if not allowed_tools:
+        return True  # 空列表表示使用默认工具
+    
+    invalid_tools = set(allowed_tools) - set(ALLOWED_TOOLS)
+    if invalid_tools:
+        raise ValidationError(
+            "allowed_tools",
+            f"invalid tools: {', '.join(invalid_tools)}. "
+            f"Valid tools: {', '.join(ALLOWED_TOOLS)}"
+        )
+    
+    return True
+
+
+def validate_metadata(metadata: Dict[str, Any], max_size: int = 1000) -> bool:
+    """
+    验证元数据
+    
+    Args:
+        metadata: 元数据字典
+        max_size: 最大键值对数量
+        
+    Returns:
+        bool: 是否有效
+        
+    Raises:
+        ValidationError: 验证失败时抛出
+    """
+    if not isinstance(metadata, dict):
+        raise ValidationError("metadata", "must be a dictionary")
+    
+    if len(metadata) > max_size:
+        raise ValidationError("metadata", f"cannot exceed {max_size} key-value pairs")
+    
+    # 检查键的有效性
+    for key in metadata.keys():
+        if not isinstance(key, str):
+            raise ValidationError("metadata", "all keys must be strings")
+        if len(key) > 100:
+            raise ValidationError("metadata", f"key '{key}' exceeds maximum length of 100 characters")
+    
+    return True
+
+
+
+

autocoder/version.py

@@ -1 +1,2 @@
-__version__ = "0.1.398"
+__version__ = "0.1.400"
+

auto_coder-0.1.398.dist-info/METADATA

@@ -1,111 +0,0 @@
-Metadata-Version: 2.1
-Name: auto-coder
-Version: 0.1.398
-Summary: AutoCoder: AutoCoder
-Author: allwefantasy
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: contextlib2
-Requires-Dist: ninja
-Requires-Dist: jinja2
-Requires-Dist: rich
-Requires-Dist: paramiko
-Requires-Dist: tqdm
-Requires-Dist: loguru
-Requires-Dist: pyjava >=0.6.21
-Requires-Dist: fastapi
-Requires-Dist: uvicorn
-Requires-Dist: retrying
-Requires-Dist: zhipuai
-Requires-Dist: dashscope
-Requires-Dist: tiktoken
-Requires-Dist: tabulate
-Requires-Dist: jupyter-client
-Requires-Dist: prompt-toolkit
-Requires-Dist: tokenizers
-Requires-Dist: aiofiles
-Requires-Dist: readerwriterlock
-Requires-Dist: byzerllm[saas] >=0.1.190
-Requires-Dist: patch
-Requires-Dist: diff-match-patch
-Requires-Dist: GitPython
-Requires-Dist: openai >=1.14.3
-Requires-Dist: anthropic
-Requires-Dist: google-generativeai
-Requires-Dist: protobuf
-Requires-Dist: azure-cognitiveservices-speech
-Requires-Dist: real-agent
-Requires-Dist: duckdb
-Requires-Dist: python-docx
-Requires-Dist: docx2txt
-Requires-Dist: pdf2image
-Requires-Dist: docx2pdf
-Requires-Dist: pypdf
-Requires-Dist: pyperclip
-Requires-Dist: colorama
-Requires-Dist: pylint
-Requires-Dist: reportlab
-Requires-Dist: pathspec
-Requires-Dist: openpyxl
-Requires-Dist: python-pptx
-Requires-Dist: watchfiles
-Requires-Dist: cairosvg
-Requires-Dist: matplotlib
-Requires-Dist: mammoth
-Requires-Dist: markdownify
-Requires-Dist: pdfminer.six
-Requires-Dist: puremagic
-Requires-Dist: pydub
-Requires-Dist: youtube-transcript-api
-Requires-Dist: SpeechRecognition
-Requires-Dist: pathvalidate
-Requires-Dist: pexpect
-Requires-Dist: setuptools
-Requires-Dist: filelock
-Requires-Dist: mcp ; python_version >= "3.10"
-
-<p align="center">
-  <picture>    
-    <img alt="auto-coder" src="./logo/auto-coder.jpeg" width=55%>
-  </picture>
-</p>
-
-<h3 align="center">
-Auto-Coder (powered by Byzer-LLM)
-</h3>
-
-<p align="center">
-<a href="https://uelng8wukz.feishu.cn/wiki/QIpkwpQo2iSdkwk9nP6cNSPlnPc"><b>中文</b></a> |
-
-</p>
-
----
-
-*Latest News* 🔥
-- [2025/01] Release Auto-Coder 0.1.208
-- [2024/09] Release Auto-Coder 0.1.163
-- [2024/08] Release Auto-Coder 0.1.143
-- [2024/07] Release Auto-Coder 0.1.115
-- [2024/06] Release Auto-Coder 0.1.82
-- [2024/05] Release Auto-Coder 0.1.73
-- [2024/04] Release Auto-Coder 0.1.46
-- [2024/03] Release Auto-Coder 0.1.25
-- [2024/03] Release Auto-Coder 0.1.24
-
----
-
-## Installation
-
-```shell
-conda create --name autocoder python=3.10.11
-conda activate autocoder
-pip install -U auto-coder
-auto-coder.chat
-```
-
-## Tutorial
-
-0. [Auto-Coder.Chat: 通向智能编程之路](https://uelng8wukz.feishu.cn/wiki/QIpkwpQo2iSdkwk9nP6cNSPlnPc)

autocoder/common/conversations/compatibility.py

@@ -1,303 +0,0 @@
-"""
-Compatibility layer for existing conversation implementations.
-
-This module provides adapter functions and classes to help
-transition from the existing conversation implementations to 
-the new unified conversation manager.
-"""
-
-from typing import Dict, List, Any, Optional, Callable, Union
-import time
-from pydantic import BaseModel
-
-from autocoder.common import AutoCoderArgs
-from autocoder.common.conversations.conversation_manager import (
-    ConversationManager, 
-    ConversationType, 
-    get_conversation_manager,
-    Message
-)
-
-
-# ======= CommandMessage (auto_command.py) Compatibility =======
-
-class CommandMessage(BaseModel):
-    """Compatible with auto_command.py CommandMessage."""
-    role: str
-    content: str
-
-
-class ExtendedCommandMessage(BaseModel):
-    """Compatible with auto_command.py ExtendedCommandMessage."""
-    message: CommandMessage
-    timestamp: str
-
-
-class CommandConversation(BaseModel):
-    """Compatible with auto_command.py CommandConversation."""
-    history: Dict[str, ExtendedCommandMessage]
-    current_conversation: List[ExtendedCommandMessage]
-
-
-def command_message_to_message(cmd_msg: Union[CommandMessage, ExtendedCommandMessage]) -> Message:
-    """Convert a CommandMessage or ExtendedCommandMessage to a Message."""
-    if isinstance(cmd_msg, ExtendedCommandMessage):
-        return Message(
-            role=cmd_msg.message.role,
-            content=cmd_msg.message.content,
-            timestamp=int(cmd_msg.timestamp)
-        )
-    else:
-        return Message(
-            role=cmd_msg.role,
-            content=cmd_msg.content
-        )
-
-
-def message_to_command_message(msg: Message) -> ExtendedCommandMessage:
-    """Convert a Message to an ExtendedCommandMessage."""
-    return ExtendedCommandMessage(
-        message=CommandMessage(
-            role=msg.role,
-            content=msg.content or ""
-        ),
-        timestamp=str(msg.timestamp or int(time.time()))
-    )
-
-
-def load_command_conversation(args: AutoCoderArgs, conv_id: str = "command") -> CommandConversation:
-    """
-    Load a command conversation in the old format.
-    
-    Args:
-        args: AutoCoderArgs object
-        conv_id: Conversation ID (defaults to "command")
-        
-    Returns:
-        A CommandConversation object compatible with auto_command.py
-    """
-    manager = get_conversation_manager(args)
-    conversation = manager.get_conversation(
-        conv_id=conv_id, 
-        conv_type=ConversationType.COMMAND,
-        create_if_not_exists=True
-    )
-    
-    # Convert to old format
-    current_conversation = [
-        message_to_command_message(msg) 
-        for msg in conversation.messages
-    ]
-    
-    # Convert archived conversations
-    history = {}
-    if conversation.archived_conversations:
-        for timestamp, messages in conversation.archived_conversations.items():
-            history[timestamp] = [
-                message_to_command_message(msg) 
-                for msg in messages
-            ]
-    
-    return CommandConversation(
-        history=history,
-        current_conversation=current_conversation
-    )
-
-
-def save_command_conversation(
-    args: AutoCoderArgs, 
-    conv: CommandConversation, 
-    conv_id: str = "command"
-) -> None:
-    """
-    Save a command conversation from the old format.
-    
-    Args:
-        args: AutoCoderArgs object
-        conv: CommandConversation object
-        conv_id: Conversation ID (defaults to "command")
-    """
-    manager = get_conversation_manager(args)
-    conversation = manager.get_conversation(
-        conv_id=conv_id, 
-        conv_type=ConversationType.COMMAND,
-        create_if_not_exists=True
-    )
-    
-    # Convert messages
-    conversation.messages = [
-        command_message_to_message(msg) 
-        for msg in conv.current_conversation
-    ]
-    
-    # Convert archived conversations
-    conversation.archived_conversations = {}
-    for timestamp, messages in conv.history.items():
-        conversation.archived_conversations[timestamp] = [
-            command_message_to_message(msg) 
-            for msg in messages
-        ]
-    
-    # Save
-    manager._save_conversation(conversation)
-
-
-def save_to_command_memory_file(
-    args: AutoCoderArgs, 
-    query: str, 
-    response: str, 
-    conv_id: str = "command"
-) -> None:
-    """
-    Save a command conversation message pair.
-    
-    Args:
-        args: AutoCoderArgs object
-        query: User query
-        response: Assistant response
-        conv_id: Conversation ID
-    """
-    manager = get_conversation_manager(args)
-    conversation = manager.get_conversation(
-        conv_id=conv_id, 
-        conv_type=ConversationType.COMMAND,
-        create_if_not_exists=True
-    )
-    
-    # Add the messages
-    manager.add_user_message(conv_id, query)
-    manager.add_assistant_message(conv_id, response)
-
-
-# ======= AgenticConversation (agentic_edit_conversation.py) Compatibility =======
-
-def get_agentic_conversation(
-    args: AutoCoderArgs, 
-    conversation_name: Optional[str] = None,
-    event_id: Optional[str] = None
-) -> 'AgenticConversationWrapper':
-    """
-    Get an AgenticConversation wrapper that mimics the original class.
-    
-    Args:
-        args: AutoCoderArgs object
-        conversation_name: Optional conversation name
-        event_id: Optional event ID
-        
-    Returns:
-        An AgenticConversationWrapper object compatible with agentic_edit_conversation.py
-    """
-    return AgenticConversationWrapper(args, conversation_name, event_id)
-
-
-class AgenticConversationWrapper:
-    """Wrapper that mimics the AgenticConversation class from agentic_edit_conversation.py."""
-    
-    def __init__(
-        self, 
-        args: AutoCoderArgs, 
-        conversation_name: Optional[str] = None,
-        event_id: Optional[str] = None,
-        initial_history: Optional[List[Dict[str, Any]]] = None
-    ):
-        """
-        Initialize the AgenticConversation wrapper.
-        
-        Args:
-            args: AutoCoderArgs object
-            conversation_name: Optional conversation name, used as conv_id
-            event_id: Optional event ID to associate with this conversation
-            initial_history: Optional initial conversation history
-        """
-        self.args = args
-        self.project_path = args.source_dir
-        self.conversation_name = conversation_name or "agentic_edit"
-        self.event_id = event_id
-        
-        # Get the conversation manager
-        self.manager = get_conversation_manager(args)
-        
-        # Get or create the conversation
-        self.conversation = self.manager.get_conversation(
-            conv_id=self.conversation_name,
-            event_id=self.event_id,
-            conv_type=ConversationType.AGENTIC_EDIT,
-            create_if_not_exists=True
-        )
-        
-        # Add initial history if provided
-        if initial_history:
-            self._add_initial_history(initial_history)
-    
-    def _add_initial_history(self, history: List[Dict[str, Any]]) -> None:
-        """Add initial history to the conversation."""
-        for msg in history:
-            role = msg.get("role", "")
-            content = msg.get("content", "")
-            tool_calls = msg.get("tool_calls", None)
-            tool_call_id = msg.get("tool_call_id", None)
-            
-            self.manager.add_message(
-                self.conversation.id,
-                role,
-                content,
-                tool_calls=tool_calls,
-                tool_call_id=tool_call_id
-            )
-    
-    def add_message(self, role: str, content: Any, **kwargs) -> None:
-        """Add a message to the conversation."""
-        self.manager.add_message(
-            self.conversation.id,
-            role,
-            content,
-            **kwargs
-        )
-    
-    def add_user_message(self, content: str) -> None:
-        """Add a user message to the conversation."""
-        self.manager.add_user_message(self.conversation.id, content)
-    
-    def add_assistant_message(self, content: str) -> None:
-        """Add an assistant message to the conversation."""
-        self.manager.add_assistant_message(self.conversation.id, content)
-    
-    def append_to_last_message(self, content: str, role: str = "assistant") -> None:
-        """Append content to the last message."""
-        self.manager.append_to_last_message(
-            self.conversation.id,
-            content,
-            role
-        )
-    
-    def add_assistant_tool_call_message(self, tool_calls: List[Dict[str, Any]], content: Optional[str] = None) -> None:
-        """Add a tool call message from the assistant."""
-        self.manager.add_tool_call_message(
-            self.conversation.id,
-            tool_calls,
-            content
-        )
-    
-    def add_tool_result_message(self, tool_call_id: str, content: Any) -> None:
-        """Add a tool result message."""
-        self.manager.add_tool_result_message(
-            self.conversation.id,
-            tool_call_id,
-            content
-        )
-    
-    def get_history(self) -> List[Dict[str, Any]]:
-        """Get the conversation history in the format expected by agentic_edit."""
-        return self.manager.get_history_as_dict_list(self.conversation.id)
-    
-    def clear_history(self) -> None:
-        """Clear the conversation history."""
-        self.manager.clear_conversation(self.conversation.id)
-    
-    def __len__(self) -> int:
-        """Return the number of messages in the conversation."""
-        return len(self.conversation.messages)
-    
-    def __str__(self) -> str:
-        """Return a string representation of the conversation."""
-        return str(self.get_history())