auto-coder 1.0.0__py3-none-any.whl → 2.0.0__py3-none-any.whl

autocoder/common/test_autocoder_args.py

@@ -0,0 +1,232 @@
+import pytest
+from autocoder.common import AutoCoderArgs
+
+
+class TestAutoCoderArgsDefaults:
+    """Test all default values of AutoCoderArgs"""
+
+    def test_all_field_defaults(self):
+        """Test all field default values by creating instance without parameters"""
+        args = AutoCoderArgs()
+        
+        # Test basic fields
+        assert args.request_id is None
+        assert args.source_dir == "."
+        assert args.git_url is None
+        assert args.target_file is None
+        assert args.query is None
+        assert args.template is None
+        assert args.project_type is None
+        assert args.execute is None
+        assert args.package_name == ""
+        assert args.script_path == ""
+        
+        # Test model fields
+        assert args.model == ""
+        assert args.chat_model == ""
+        assert args.model_max_length == 2000
+        assert args.model_max_input_length == 6000
+        assert args.vl_model == ""
+        assert args.sd_model == ""
+        assert args.emb_model == ""
+        assert args.code_model == ""
+        assert args.generate_rerank_model == ""
+        assert args.context_prune_model == ""
+        assert args.conversation_prune_model == ""
+        assert args.inference_model == ""
+        assert args.system_prompt == ""
+        assert args.planner_model == ""
+        assert args.voice2text_model == ""
+        assert args.text2voice_model == ""
+        assert args.commit_model == ""
+        
+        # Test index fields
+        assert args.skip_build_index is False
+        assert args.skip_filter_index is False
+        assert args.index_model == ""
+        assert args.index_filter_model == ""
+        assert args.index_filter_model_max_input_length == 50*1024
+        assert args.index_model_max_length == 0
+        assert args.index_model_max_input_length == 0
+        assert args.index_model_anti_quota_limit == 0
+        
+        # Test agentic fields
+        assert args.enable_agentic_filter is False
+        assert args.enable_agentic_edit is True
+        assert args.enable_agentic_auto_approve is False
+        assert args.enable_agentic_dangerous_command_check is False
+        
+        # Test filter fields
+        assert args.index_filter_level == 0
+        assert args.index_filter_enable_relevance_verification is True
+        assert args.index_filter_workers == 1
+        assert args.index_filter_file_num == 10
+        assert args.index_build_workers == 1
+        
+        # Test other model fields
+        assert args.designer_model == ""
+        assert args.file is None
+        assert args.ray_address == ""
+        assert args.anti_quota_limit == 1
+        assert args.print_request is False
+        assert args.py_packages == ""
+        
+        # Test search fields
+        assert args.search == ""
+        assert args.search_engine == ""
+        assert args.search_engine_token == ""
+        
+        # Test RAG fields
+        assert args.rag_url == ""
+        assert args.rag_token == ""
+        assert args.rag_type == "storage"
+        assert args.rag_storage_type == "duckdb"
+        assert args.rag_params_max_tokens == 500000
+        assert args.rag_doc_filter_relevance == 2
+        assert args.rag_context_window_limit == 120000
+        assert args.rag_duckdb_vector_dim == 1024
+        assert args.rag_duckdb_query_similarity == 0.1
+        assert args.rag_duckdb_query_top_k == 10000
+        assert args.rag_index_build_workers == 10
+        assert args.rag_emb_dim == 1024
+        assert args.rag_emb_text_size == 1024
+        assert args.local_image_host == ""
+        assert args.rag_recall_max_queries == 5
+        assert args.rag_qa_conversation_strategy == "multi_round"
+        
+        # Test relevance and context fields
+        assert args.verify_file_relevance_score == 6
+        assert args.enable_rag_search is False
+        assert args.enable_rag_context is False
+        assert args.collection is None
+        assert args.collections is None
+        
+        # Test merge and human fields
+        assert args.auto_merge is False
+        assert args.human_as_model is False
+        assert args.human_model_num == 1
+        
+        # Test image fields
+        assert args.image_file == ""
+        assert args.image_mode == "direct"
+        assert args.image_max_iter == 1
+        
+        # Test URL fields
+        assert args.urls == ""
+        assert args.urls_use_model is False
+        assert args.command is None
+        assert args.doc_command is None
+        assert args.required_exts is None
+        assert args.hybrid_index_max_output_tokens == 1000000
+        
+        # Test various enable fields
+        assert args.enable_multi_round_generate is False
+        assert args.monitor_mode is False
+        assert args.enable_hybrid_index is False
+        assert args.rag_build_name is None
+        assert args.disable_auto_window is False
+        assert args.filter_batch_size == 5
+        assert args.disable_segment_reorder is False
+        assert args.tokenizer_path is None
+        assert args.skip_confirm is False
+        assert args.silence is False
+        assert args.exclude_files == ""
+        assert args.output == ""
+        assert args.single_file is False
+        assert args.query_prefix is None
+        assert args.query_suffix is None
+        assert args.from_yaml is None
+        assert args.base_dir is None
+        assert args.context is None
+        assert args.editblock_similarity == 0.9
+        assert args.include_project_structure is False
+        assert args.new_session is False
+        
+        # Test prompt and review fields
+        assert args.prompt_review is None
+        assert args.agent_designer_mode == "svg"
+        
+        # Test ratio fields
+        assert args.full_text_ratio == 0.7
+        assert args.segment_ratio == 0.2
+        assert args.buff_ratio == 0.1
+        
+        # Test inference fields
+        assert args.disable_inference_enhance is True
+        assert args.inference_deep_thought is False
+        assert args.inference_slow_without_deep_thought is False
+        assert args.inference_compute_precision == 64
+        assert args.without_contexts is False
+        assert args.skip_events is False
+        assert args.data_cells_max_num == 2000
+        assert args.generate_times_same_model == 1
+        assert args.rank_times_same_model == 1
+        
+        # Test strategy fields
+        assert args.rank_strategy == "file"
+        
+        # Test action and mode fields
+        assert args.action == []
+        assert args.enable_global_memory is False
+        assert args.product_mode == "lite"
+        
+        # Test reasoning content fields
+        assert args.keep_reasoning_content is False
+        assert args.keep_only_reasoning_content is False
+        
+        # Test code apply fields
+        assert args.in_code_apply is False
+        assert args.model_filter_path is None
+        
+        # Test conversation prune fields
+        assert args.conversation_prune_safe_zone_tokens == 50*1024
+        assert args.conversation_prune_group_size == 4
+        assert args.conversation_prune_strategy == "summarize"
+        
+        # Test context prune fields
+        assert args.context_prune_strategy == "extract"
+        assert args.context_prune is True
+        assert args.context_prune_safe_zone_tokens == 24*1024
+        assert args.context_prune_sliding_window_size == 1000
+        assert args.context_prune_sliding_window_overlap == 100
+        
+        # Test command fields
+        assert args.auto_command_max_iterations == 10
+        assert args.use_shell_commands is True
+        assert args.skip_commit is False
+        
+        # Test beta and reproduction fields
+        assert args.enable_beta is False
+        assert args.how_to_reproduce is None
+        
+        # Test URL lists
+        assert args.dynamic_urls == []
+        assert args.add_updated_urls == []
+        
+        # Test history fields
+        assert args.enable_task_history is False
+        assert args.event_file is None
+        
+        # Test active context fields
+        assert args.enable_active_context is False
+        assert args.enable_active_context_in_generate is False
+        
+        # Test generate fields
+        assert args.generate_max_rounds == 5
+        
+        # Test revert fields
+        assert args.revert_commit_id is None
+        
+        # Test auto fix fields
+        assert args.enable_auto_fix_lint is False
+        assert args.enable_auto_fix_compile is False
+        assert args.enable_auto_fix_merge is False
+        assert args.auto_fix_lint_max_attempts == 5
+        assert args.auto_fix_compile_max_attempts == 5
+        assert args.auto_fix_merge_max_attempts == 5
+        
+        # Test rules fields
+        assert args.enable_auto_select_rules is True
+        
+        # Test shadows fields
+        assert args.ignore_clean_shadows is False 

autocoder/common/test_env_manager.py

@@ -0,0 +1,173 @@
+
+#!/usr/bin/env python3
+"""
+环境变量管理模块测试脚本
+
+用于测试 env_manager.py 模块的各项功能
+"""
+
+import os
+import sys
+from pathlib import Path
+
+# 添加项目根目录到路径，以便导入模块
+project_root = Path(__file__).parent.parent.parent.parent
+sys.path.insert(0, str(project_root))
+
+from src.autocoder.common.env_manager import (
+    EnvManager, Environment,
+    get_environment, is_production, is_development,
+    get_env, get_env_bool, get_env_int, get_env_float,
+    set_env, has_env, get_env_info
+)
+
+
+def test_basic_functionality():
+    """测试基本功能"""
+    print("=== 测试基本功能 ===")
+    
+    # 测试默认环境
+    print(f"默认环境: {get_environment()}")
+    print(f"是否为生产环境: {is_production()}")
+    print(f"是否为开发环境: {is_development()}")
+    print()
+
+
+def test_environment_detection():
+    """测试环境检测功能"""
+    print("=== 测试环境检测功能 ===")
+    
+    # 保存原始环境变量
+    original_env = os.environ.get("AUTO_CODER_ENV")
+    
+    # 测试生产环境
+    test_cases = [
+        ("production", Environment.PRODUCTION),
+        ("prod", Environment.PRODUCTION),
+        ("PRODUCTION", Environment.PRODUCTION),
+        ("development", Environment.DEVELOPMENT),
+        ("dev", Environment.DEVELOPMENT),
+        ("DEV", Environment.DEVELOPMENT),
+        ("unknown", Environment.DEVELOPMENT),  # 未知值默认为开发环境
+    ]
+    
+    for env_value, expected in test_cases:
+        set_env("AUTO_CODER_ENV", env_value)
+        result = get_environment()
+        status = "✓" if result == expected else "✗"
+        print(f"{status} 设置 AUTO_CODER_ENV={env_value}, 期望: {expected.value}, 实际: {result.value}")
+    
+    # 恢复原始环境变量
+    if original_env is not None:
+        os.environ["AUTO_CODER_ENV"] = original_env
+    elif "AUTO_CODER_ENV" in os.environ:
+        del os.environ["AUTO_CODER_ENV"]
+    
+    print()
+
+
+def test_type_conversion():
+    """测试类型转换功能"""
+    print("=== 测试类型转换功能 ===")
+    
+    # 测试布尔值转换
+    bool_test_cases = [
+        ("true", True),
+        ("True", True),
+        ("1", True),
+        ("yes", True),
+        ("on", True),
+        ("false", False),
+        ("False", False),
+        ("0", False),
+        ("no", False),
+        ("off", False),
+        ("", False),
+    ]
+    
+    for value, expected in bool_test_cases:
+        set_env("TEST_BOOL", value)
+        result = get_env_bool("TEST_BOOL")
+        status = "✓" if result == expected else "✗"
+        print(f"{status} 布尔值 '{value}' -> {result} (期望: {expected})")
+    
+    # 测试整数转换
+    set_env("TEST_INT", "42")
+    result = get_env_int("TEST_INT")
+    print(f"{'✓' if result == 42 else '✗'} 整数 '42' -> {result}")
+    
+    # 测试浮点数转换
+    set_env("TEST_FLOAT", "3.14")
+    result = get_env_float("TEST_FLOAT")
+    print(f"{'✓' if abs(result - 3.14) < 0.001 else '✗'} 浮点数 '3.14' -> {result}")
+    
+    # 测试无效值的默认值处理
+    set_env("TEST_INVALID_INT", "not_a_number")
+    result = get_env_int("TEST_INVALID_INT", 999)
+    print(f"{'✓' if result == 999 else '✗'} 无效整数使用默认值 -> {result}")
+    
+    # 清理测试环境变量
+    for key in ["TEST_BOOL", "TEST_INT", "TEST_FLOAT", "TEST_INVALID_INT"]:
+        if key in os.environ:
+            del os.environ[key]
+    
+    print()
+
+
+def test_env_management():
+    """测试环境变量管理功能"""
+    print("=== 测试环境变量管理功能 ===")
+    
+    # 测试设置和获取
+    set_env("TEST_VAR", "test_value")
+    result = get_env("TEST_VAR")
+    print(f"{'✓' if result == 'test_value' else '✗'} 设置和获取环境变量: {result}")
+    
+    # 测试环境变量存在性检查
+    exists = has_env("TEST_VAR")
+    print(f"{'✓' if exists else '✗'} 环境变量存在性检查: {exists}")
+    
+    # 测试不存在的环境变量
+    not_exists = has_env("NON_EXISTENT_VAR")
+    print(f"{'✓' if not not_exists else '✗'} 不存在的环境变量检查: {not_exists}")
+    
+    # 测试默认值
+    default_result = get_env("NON_EXISTENT_VAR", "default_value")
+    print(f"{'✓' if default_result == 'default_value' else '✗'} 使用默认值: {default_result}")
+    
+    # 清理测试环境变量
+    if "TEST_VAR" in os.environ:
+        del os.environ["TEST_VAR"]
+    
+    print()
+
+
+def test_env_info():
+    """测试环境信息获取功能"""
+    print("=== 测试环境信息获取功能 ===")
+    
+    env_info = get_env_info()
+    print("环境信息:")
+    for key, value in env_info.items():
+        print(f"  {key}: {value}")
+    
+    print()
+
+
+def main():
+    """主测试函数"""
+    print("环境变量管理模块测试")
+    print("=" * 50)
+    
+    test_basic_functionality()
+    test_environment_detection()
+    test_type_conversion()
+    test_env_management()
+    test_env_info()
+    
+    print("测试完成!")
+
+
+if __name__ == "__main__":
+    main()
+

autocoder/common/test_env_manager_integration.py

@@ -0,0 +1,159 @@
+
+#!/usr/bin/env python3
+"""
+环境变量管理模块集成测试
+
+验证模块在项目中的集成情况
+"""
+
+import sys
+import os
+from pathlib import Path
+
+# 确保可以导入项目模块
+project_root = Path(__file__).parent.parent.parent.parent
+sys.path.insert(0, str(project_root))
+
+def test_import():
+    """测试模块导入"""
+    try:
+        # 测试直接导入
+        from src.autocoder.common.env_manager import (
+            EnvManager, Environment,
+            get_environment, is_production, is_development
+        )
+        print("✓ 模块导入成功")
+        return True
+    except ImportError as e:
+        print(f"✗ 模块导入失败: {e}")
+        return False
+
+def test_basic_functionality():
+    """测试基本功能"""
+    try:
+        from src.autocoder.common.env_manager import (
+            get_environment, is_production, is_development,
+            get_env, set_env
+        )
+        
+        # 测试环境检测
+        env = get_environment()
+        print(f"✓ 当前环境: {env.value}")
+        
+        # 测试环境变量设置和获取
+        test_key = "TEST_INTEGRATION_VAR"
+        test_value = "integration_test_value"
+        
+        set_env(test_key, test_value)
+        retrieved_value = get_env(test_key)
+        
+        if retrieved_value == test_value:
+            print("✓ 环境变量设置和获取功能正常")
+        else:
+            print(f"✗ 环境变量功能异常: 期望 {test_value}, 实际 {retrieved_value}")
+            return False
+        
+        # 清理测试环境变量
+        if test_key in os.environ:
+            del os.environ[test_key]
+        
+        return True
+        
+    except Exception as e:
+        print(f"✗ 基本功能测试失败: {e}")
+        return False
+
+def test_with_existing_project_modules():
+    """测试与现有项目模块的兼容性"""
+    try:
+        # 测试与现有的 detect_env 函数兼容性
+        from src.autocoder.common import detect_env
+        from src.autocoder.common.env_manager import get_env_info
+        
+        # 获取现有的环境信息
+        existing_env_info = detect_env()
+        print(f"✓ 现有环境检测正常: {existing_env_info.os_name}")
+        
+        # 获取新模块的环境信息
+        new_env_info = get_env_info()
+        print(f"✓ 新环境管理模块正常: {new_env_info['environment']}")
+        
+        return True
+        
+    except Exception as e:
+        print(f"✗ 项目模块兼容性测试失败: {e}")
+        return False
+
+def test_environment_switching():
+    """测试环境切换功能"""
+    try:
+        from src.autocoder.common.env_manager import (
+            get_environment, is_production, is_development, set_env
+        )
+        
+        # 保存原始环境
+        original_env = os.environ.get("AUTO_CODER_ENV")
+        
+        # 测试切换到生产环境
+        set_env("AUTO_CODER_ENV", "production")
+        if is_production() and not is_development():
+            print("✓ 生产环境切换成功")
+        else:
+            print("✗ 生产环境切换失败")
+            return False
+        
+        # 测试切换到开发环境
+        set_env("AUTO_CODER_ENV", "development")
+        if is_development() and not is_production():
+            print("✓ 开发环境切换成功")
+        else:
+            print("✗ 开发环境切换失败")
+            return False
+        
+        # 恢复原始环境
+        if original_env:
+            os.environ["AUTO_CODER_ENV"] = original_env
+        elif "AUTO_CODER_ENV" in os.environ:
+            del os.environ["AUTO_CODER_ENV"]
+        
+        return True
+        
+    except Exception as e:
+        print(f"✗ 环境切换测试失败: {e}")
+        return False
+
+def main():
+    """主测试函数"""
+    print("环境变量管理模块集成测试")
+    print("=" * 50)
+    
+    tests = [
+        test_import,
+        test_basic_functionality,
+        test_with_existing_project_modules,
+        test_environment_switching,
+    ]
+    
+    passed = 0
+    total = len(tests)
+    
+    for test in tests:
+        print(f"\n运行测试: {test.__name__}")
+        if test():
+            passed += 1
+        else:
+            print(f"测试 {test.__name__} 失败")
+    
+    print(f"\n测试结果: {passed}/{total} 通过")
+    
+    if passed == total:
+        print("🎉 所有集成测试通过！")
+        return True
+    else:
+        print("❌ 部分集成测试失败")
+        return False
+
+if __name__ == "__main__":
+    success = main()
+    sys.exit(0 if success else 1)
+

autocoder/common/text_similarity/__init__.py

@@ -0,0 +1,9 @@
+"""
+Text Similarity AC Module
+
+This module provides text similarity calculation functionality with sliding window matching.
+"""
+
+from .text_similarity import TextSimilarity
+
+__all__ = ['TextSimilarity']