|
|
|
|
@ -21,196 +21,170 @@ from typing import Dict, Any |
|
|
|
|
# to access it. |
|
|
|
|
class I18N: |
|
|
|
|
# Define supported languages in current install help script |
|
|
|
|
SUPPORTED_LANGUAGES = ['en', 'zh'] |
|
|
|
|
SUPPORTED_LANGUAGES = ["en", "zh"] |
|
|
|
|
|
|
|
|
|
# The translation dictionary contains a mapping from language code to a dictionary |
|
|
|
|
TRANSLATIONS = { |
|
|
|
|
'en': { |
|
|
|
|
"en": { |
|
|
|
|
# Common |
|
|
|
|
'workspace_not_found': "Workspace root not found.", |
|
|
|
|
'cannot_parse': "Cannot parse {}: {}", |
|
|
|
|
'no_extras_defined': "No extras defined", |
|
|
|
|
'no_extras_found': "No workspace or extras found.", |
|
|
|
|
'operation_canceled': "Operation canceled.", |
|
|
|
|
'available_packages': "Available packages: {}", |
|
|
|
|
'copy_command': "Please copy the above command to execute in terminal. For more help, run:", |
|
|
|
|
'finished': "Finished!", |
|
|
|
|
|
|
|
|
|
"workspace_not_found": "Workspace root not found.", |
|
|
|
|
"cannot_parse": "Cannot parse {}: {}", |
|
|
|
|
"no_extras_defined": "No extras defined", |
|
|
|
|
"no_extras_found": "No workspace or extras found.", |
|
|
|
|
"operation_canceled": "Operation canceled.", |
|
|
|
|
"available_packages": "Available packages: {}", |
|
|
|
|
"copy_command": "Please copy the above command to execute in terminal. For more help, run:", |
|
|
|
|
"finished": "Finished!", |
|
|
|
|
# Description of the CLI command |
|
|
|
|
'cli_description': "UV Workspace Extras Helper - Manage optional dependencies in UV workspace", |
|
|
|
|
'list_cmd_description': "List all extras in the workspace", |
|
|
|
|
'install_cmd_description': "Generate installation commands for extras", |
|
|
|
|
'deploy_cmd_description': "Use predefined deployment templates", |
|
|
|
|
|
|
|
|
|
"cli_description": "UV Workspace Extras Helper - Manage optional dependencies in UV workspace", |
|
|
|
|
"list_cmd_description": "List all extras in the workspace", |
|
|
|
|
"install_cmd_description": "Generate installation commands for extras", |
|
|
|
|
"deploy_cmd_description": "Use predefined deployment templates", |
|
|
|
|
# Option descriptions |
|
|
|
|
'verbose_option': "Show detailed dependency information", |
|
|
|
|
'interactive_option': "Interactive guide to generate installation commands", |
|
|
|
|
'all_option': "Generate command to install all extras", |
|
|
|
|
'china_option': "Use Tsinghua PyPI mirror for faster installation in China", |
|
|
|
|
'preset_option': "Use predefined deployment template", |
|
|
|
|
'list_presets_option': "List all predefined deployment templates", |
|
|
|
|
'language_option': "Specify language (en/zh)", |
|
|
|
|
|
|
|
|
|
"verbose_option": "Show detailed dependency information", |
|
|
|
|
"interactive_option": "Interactive guide to generate installation commands", |
|
|
|
|
"all_option": "Generate command to install all extras", |
|
|
|
|
"china_option": "Use Tsinghua PyPI mirror for faster installation in China", |
|
|
|
|
"preset_option": "Use predefined deployment template", |
|
|
|
|
"list_presets_option": "List all predefined deployment templates", |
|
|
|
|
"language_option": "Specify language (en/zh)", |
|
|
|
|
# List command |
|
|
|
|
'extras_in_workspace': "Extras in workspace:\n", |
|
|
|
|
'available_extras': "Available extras:", |
|
|
|
|
'dependencies': "dependencies", |
|
|
|
|
|
|
|
|
|
"extras_in_workspace": "Extras in workspace:\n", |
|
|
|
|
"available_extras": "Available extras:", |
|
|
|
|
"dependencies": "dependencies", |
|
|
|
|
# Installation command |
|
|
|
|
'install_all_extras': "# Install all optional features:", |
|
|
|
|
'install_extras_for': "# Install {} feature for {}:", |
|
|
|
|
'package_not_in_workspace': "Error: Package '{}' not in workspace or has no extras defined.", |
|
|
|
|
'package_no_extras': "Package '{}' has no extras defined.", |
|
|
|
|
'extra_not_in_package': "Error: Extra '{}' not found in package '{}'.", |
|
|
|
|
'available_extras_in_package': "Available extras: {}", |
|
|
|
|
|
|
|
|
|
"install_all_extras": "# Install all optional features:", |
|
|
|
|
"install_extras_for": "# Install {} feature for {}:", |
|
|
|
|
"package_not_in_workspace": "Error: Package '{}' not in workspace or has no extras defined.", |
|
|
|
|
"package_no_extras": "Package '{}' has no extras defined.", |
|
|
|
|
"extra_not_in_package": "Error: Extra '{}' not found in package '{}'.", |
|
|
|
|
"available_extras_in_package": "Available extras: {}", |
|
|
|
|
# Interactive installation |
|
|
|
|
'welcome': "Welcome to DB-GPT Installation Assistant!", |
|
|
|
|
'help_message': "This tool will help you generate the correct installation commands.\n", |
|
|
|
|
'select_mode': "Please select installation mode", |
|
|
|
|
'select_extras': "Please select extras to install (space to select/deselect, enter to confirm)", |
|
|
|
|
'installation_info': "📋 Installation Information", |
|
|
|
|
'selected_mode': "📦 Selected mode: {}", |
|
|
|
|
'description': "📝 Description: {}", |
|
|
|
|
'note': "ℹ️ Note: {}", |
|
|
|
|
'will_install': "🧩 Will install the following extras: {}", |
|
|
|
|
'config_file': "⚙️ Configuration file: {}", |
|
|
|
|
'generate_command': "Generate installation command?", |
|
|
|
|
'installation_command': "🚀 Installation Command", |
|
|
|
|
'startup_command': "🏃 Startup Command", |
|
|
|
|
'further_configuration': "⚠️ Further Configuration", |
|
|
|
|
'set_api_key': "Please make sure you set the correct API Key in the configuration file {}", |
|
|
|
|
'set_model_path': "Please make sure you set the correct model path in the configuration file {}", |
|
|
|
|
|
|
|
|
|
"welcome": "Welcome to DB-GPT Installation Assistant!", |
|
|
|
|
"help_message": "This tool will help you generate the correct installation commands.\n", |
|
|
|
|
"select_mode": "Please select installation mode", |
|
|
|
|
"select_extras": "Please select extras to install (space to select/deselect, enter to confirm)", |
|
|
|
|
"installation_info": "📋 Installation Information", |
|
|
|
|
"selected_mode": "📦 Selected mode: {}", |
|
|
|
|
"description": "📝 Description: {}", |
|
|
|
|
"note": "ℹ️ Note: {}", |
|
|
|
|
"will_install": "🧩 Will install the following extras: {}", |
|
|
|
|
"config_file": "⚙️ Configuration file: {}", |
|
|
|
|
"generate_command": "Generate installation command?", |
|
|
|
|
"installation_command": "🚀 Installation Command", |
|
|
|
|
"startup_command": "🏃 Startup Command", |
|
|
|
|
"further_configuration": "⚠️ Further Configuration", |
|
|
|
|
"set_api_key": "Please make sure you set the correct API Key in the configuration file {}", |
|
|
|
|
"set_model_path": "Please make sure you set the correct model path in the configuration file {}", |
|
|
|
|
# Deployment command |
|
|
|
|
'available_presets': "Available deployment presets:", |
|
|
|
|
'specify_preset': "Please specify a deployment preset name, or use --list to view all presets", |
|
|
|
|
'preset_not_found': "Error: Preset '{}' not found", |
|
|
|
|
'available_presets_list': "Available presets: {}", |
|
|
|
|
'using_preset': "Using preset '{}' to generate deployment command", |
|
|
|
|
|
|
|
|
|
"available_presets": "Available deployment presets:", |
|
|
|
|
"specify_preset": "Please specify a deployment preset name, or use --list to view all presets", |
|
|
|
|
"preset_not_found": "Error: Preset '{}' not found", |
|
|
|
|
"available_presets_list": "Available presets: {}", |
|
|
|
|
"using_preset": "Using preset '{}' to generate deployment command", |
|
|
|
|
# Preset descriptions |
|
|
|
|
'openai_preset': "OpenAI Proxy Mode", |
|
|
|
|
'openai_desc': "Using OpenAI API as proxy, suitable for environments without GPU", |
|
|
|
|
'openai_note': "Requires OpenAI API Key", |
|
|
|
|
|
|
|
|
|
'deepseek_preset': "DeepSeek Proxy Mode", |
|
|
|
|
'deepseek_desc': "Using DeepSeek API as proxy, suitable for environments without GPU", |
|
|
|
|
'deepseek_note': "Requires DeepSeek API Key", |
|
|
|
|
|
|
|
|
|
'glm4_preset': "GLM4 Local Mode", |
|
|
|
|
'glm4_desc': "Using local GLM4 model, requires GPU environment", |
|
|
|
|
'glm4_note': "Requires local model path configuration", |
|
|
|
|
|
|
|
|
|
'vllm_preset': "VLLM Local Mode", |
|
|
|
|
'vllm_desc': "Using VLLM framework to load local model, requires GPU environment", |
|
|
|
|
'vllm_note': "Requires local model path configuration", |
|
|
|
|
|
|
|
|
|
'llama_cpp_preset': "LLAMA_CPP Local Mode", |
|
|
|
|
'llama_cpp_desc': "Using LLAMA.cpp framework to load local model, can run on CPU but GPU recommended", |
|
|
|
|
'llama_cpp_note': "Requires local model path configuration, for CUDA support set CMAKE_ARGS=\"-DGGML_CUDA=ON\"", |
|
|
|
|
|
|
|
|
|
'ollama_preset': "Ollama Proxy Mode", |
|
|
|
|
'ollama_desc': "Using Ollama as proxy, suitable for environments without GPU", |
|
|
|
|
'ollama_note': "Requires Ollama API Base", |
|
|
|
|
|
|
|
|
|
'custom_preset': "Custom Mode", |
|
|
|
|
'custom_desc': "Manually select needed extras", |
|
|
|
|
'custom_note': "Suitable for advanced users" |
|
|
|
|
"openai_preset": "OpenAI Proxy Mode", |
|
|
|
|
"openai_desc": "Using OpenAI API as proxy, suitable for environments without GPU", |
|
|
|
|
"openai_note": "Requires OpenAI API Key", |
|
|
|
|
"deepseek_preset": "DeepSeek Proxy Mode", |
|
|
|
|
"deepseek_desc": "Using DeepSeek API as proxy, suitable for environments without GPU", |
|
|
|
|
"deepseek_note": "Requires DeepSeek API Key", |
|
|
|
|
"glm4_preset": "GLM4 Local Mode", |
|
|
|
|
"glm4_desc": "Using local GLM4 model, requires GPU environment", |
|
|
|
|
"glm4_note": "Requires local model path configuration", |
|
|
|
|
"vllm_preset": "VLLM Local Mode", |
|
|
|
|
"vllm_desc": "Using VLLM framework to load local model, requires GPU environment", |
|
|
|
|
"vllm_note": "Requires local model path configuration", |
|
|
|
|
"llama_cpp_preset": "LLAMA_CPP Local Mode", |
|
|
|
|
"llama_cpp_desc": "Using LLAMA.cpp framework to load local model, can run on CPU but GPU recommended", |
|
|
|
|
"llama_cpp_note": 'Requires local model path configuration, for CUDA support set CMAKE_ARGS="-DGGML_CUDA=ON"', |
|
|
|
|
"ollama_preset": "Ollama Proxy Mode", |
|
|
|
|
"ollama_desc": "Using Ollama as proxy, suitable for environments without GPU", |
|
|
|
|
"ollama_note": "Requires Ollama API Base", |
|
|
|
|
"custom_preset": "Custom Mode", |
|
|
|
|
"custom_desc": "Manually select needed extras", |
|
|
|
|
"custom_note": "Suitable for advanced users", |
|
|
|
|
}, |
|
|
|
|
'zh': { |
|
|
|
|
"zh": { |
|
|
|
|
# Common |
|
|
|
|
'workspace_not_found': "未找到工作区根目录", |
|
|
|
|
'cannot_parse': "无法解析 {}: {}", |
|
|
|
|
'no_extras_defined': "没有定义 extras", |
|
|
|
|
'no_extras_found': "未找到工作区或没有可选依赖。", |
|
|
|
|
'operation_canceled': "操作已取消。", |
|
|
|
|
'available_packages': "可用的包: {}", |
|
|
|
|
'copy_command': "请复制上面的命令到终端执行。如需更多帮助，请运行：", |
|
|
|
|
'finished': "完成！", |
|
|
|
|
|
|
|
|
|
"workspace_not_found": "未找到工作区根目录", |
|
|
|
|
"cannot_parse": "无法解析 {}: {}", |
|
|
|
|
"no_extras_defined": "没有定义 extras", |
|
|
|
|
"no_extras_found": "未找到工作区或没有可选依赖。", |
|
|
|
|
"operation_canceled": "操作已取消。", |
|
|
|
|
"available_packages": "可用的包: {}", |
|
|
|
|
"copy_command": "请复制上面的命令到终端执行。如需更多帮助，请运行：", |
|
|
|
|
"finished": "完成！", |
|
|
|
|
# Description of the CLI command |
|
|
|
|
'cli_description': "UV Workspace Extras Helper - 管理UV工作区的可选依赖", |
|
|
|
|
'list_cmd_description': "列出工作区中的所有extras", |
|
|
|
|
'install_cmd_description': "生成安装extras的命令", |
|
|
|
|
'deploy_cmd_description': "使用预设的部署方案", |
|
|
|
|
|
|
|
|
|
"cli_description": "UV Workspace Extras Helper - 管理UV工作区的可选依赖", |
|
|
|
|
"list_cmd_description": "列出工作区中的所有extras", |
|
|
|
|
"install_cmd_description": "生成安装extras的命令", |
|
|
|
|
"deploy_cmd_description": "使用预设的部署方案", |
|
|
|
|
# Option descriptions |
|
|
|
|
'verbose_option': "显示详细依赖信息", |
|
|
|
|
'interactive_option': "交互式引导生成安装命令", |
|
|
|
|
'all_option': "生成安装所有extras的命令", |
|
|
|
|
'china_option': "使用清华pip镜像源加速安装", |
|
|
|
|
'preset_option': "使用预设的部署方案", |
|
|
|
|
'list_presets_option': "列出所有预设部署方案", |
|
|
|
|
'language_option': "指定语言 (en/zh)", |
|
|
|
|
|
|
|
|
|
"verbose_option": "显示详细依赖信息", |
|
|
|
|
"interactive_option": "交互式引导生成安装命令", |
|
|
|
|
"all_option": "生成安装所有extras的命令", |
|
|
|
|
"china_option": "使用清华pip镜像源加速安装", |
|
|
|
|
"preset_option": "使用预设的部署方案", |
|
|
|
|
"list_presets_option": "列出所有预设部署方案", |
|
|
|
|
"language_option": "指定语言 (en/zh)", |
|
|
|
|
# List command |
|
|
|
|
'extras_in_workspace': "工作区中的可选依赖 (extras):\n", |
|
|
|
|
'available_extras': "可用的 extras:", |
|
|
|
|
'dependencies': "个依赖", |
|
|
|
|
|
|
|
|
|
"extras_in_workspace": "工作区中的可选依赖 (extras):\n", |
|
|
|
|
"available_extras": "可用的 extras:", |
|
|
|
|
"dependencies": "个依赖", |
|
|
|
|
# Installation command |
|
|
|
|
'install_all_extras': "# 安装所有可选功能:", |
|
|
|
|
'install_extras_for': "# 安装 {} 的 {} 功能:", |
|
|
|
|
'package_not_in_workspace': "错误: 包 '{}' 不在工作区中或没有定义extras。", |
|
|
|
|
'package_no_extras': "包 '{}' 没有定义extras。", |
|
|
|
|
'extra_not_in_package': "错误: 包 '{}' 中没有名为 '{}' 的extra。", |
|
|
|
|
'available_extras_in_package': "可用的extras: {}", |
|
|
|
|
|
|
|
|
|
"install_all_extras": "# 安装所有可选功能:", |
|
|
|
|
"install_extras_for": "# 安装 {} 的 {} 功能:", |
|
|
|
|
"package_not_in_workspace": "错误: 包 '{}' 不在工作区中或没有定义extras。", |
|
|
|
|
"package_no_extras": "包 '{}' 没有定义extras。", |
|
|
|
|
"extra_not_in_package": "错误: 包 '{}' 中没有名为 '{}' 的extra。", |
|
|
|
|
"available_extras_in_package": "可用的extras: {}", |
|
|
|
|
# Interactive installation |
|
|
|
|
'welcome': "欢迎使用 DB-GPT 安装引导助手！", |
|
|
|
|
'help_message': "这个工具将帮助你生成正确的安装命令。\n", |
|
|
|
|
'select_mode': "请选择安装模式", |
|
|
|
|
'select_extras': "请选择需要安装的extras（空格选择/取消，回车确认）", |
|
|
|
|
'installation_info': "📋 安装信息", |
|
|
|
|
'selected_mode': "📦 选择的模式: {}", |
|
|
|
|
'description': "📝 描述: {}", |
|
|
|
|
'note': "ℹ️ 注意事项: {}", |
|
|
|
|
'will_install': "🧩 将安装以下extras: {}", |
|
|
|
|
'config_file': "⚙️ 配置文件: {}", |
|
|
|
|
'generate_command': "是否生成安装命令？", |
|
|
|
|
'installation_command': "🚀 安装命令", |
|
|
|
|
'startup_command': "🏃 启动命令", |
|
|
|
|
'further_configuration': "⚠️ 后续配置", |
|
|
|
|
'set_api_key': "请确保在配置文件 {} 中设置了正确的API Key", |
|
|
|
|
'set_model_path': "请确保在配置文件 {} 中设置了正确的模型路径", |
|
|
|
|
|
|
|
|
|
"welcome": "欢迎使用 DB-GPT 安装引导助手！", |
|
|
|
|
"help_message": "这个工具将帮助你生成正确的安装命令。\n", |
|
|
|
|
"select_mode": "请选择安装模式", |
|
|
|
|
"select_extras": "请选择需要安装的extras（空格选择/取消，回车确认）", |
|
|
|
|
"installation_info": "📋 安装信息", |
|
|
|
|
"selected_mode": "📦 选择的模式: {}", |
|
|
|
|
"description": "📝 描述: {}", |
|
|
|
|
"note": "ℹ️ 注意事项: {}", |
|
|
|
|
"will_install": "🧩 将安装以下extras: {}", |
|
|
|
|
"config_file": "⚙️ 配置文件: {}", |
|
|
|
|
"generate_command": "是否生成安装命令？", |
|
|
|
|
"installation_command": "🚀 安装命令", |
|
|
|
|
"startup_command": "🏃 启动命令", |
|
|
|
|
"further_configuration": "⚠️ 后续配置", |
|
|
|
|
"set_api_key": "请确保在配置文件 {} 中设置了正确的API Key", |
|
|
|
|
"set_model_path": "请确保在配置文件 {} 中设置了正确的模型路径", |
|
|
|
|
# Deployment command |
|
|
|
|
'available_presets': "可用的部署预设:", |
|
|
|
|
'specify_preset': "请指定部署预设名称，或使用 --list 查看所有预设", |
|
|
|
|
'preset_not_found': "错误: 未找到预设 '{}'", |
|
|
|
|
'available_presets_list': "可用的预设: {}", |
|
|
|
|
'using_preset': "使用预设 '{}' 生成部署命令", |
|
|
|
|
|
|
|
|
|
"available_presets": "可用的部署预设:", |
|
|
|
|
"specify_preset": "请指定部署预设名称，或使用 --list 查看所有预设", |
|
|
|
|
"preset_not_found": "错误: 未找到预设 '{}'", |
|
|
|
|
"available_presets_list": "可用的预设: {}", |
|
|
|
|
"using_preset": "使用预设 '{}' 生成部署命令", |
|
|
|
|
# Preset descriptions |
|
|
|
|
'openai_preset': "OpenAI 代理模式", |
|
|
|
|
'openai_desc': "使用OpenAI API作为代理，适合无GPU环境", |
|
|
|
|
'openai_note': "需要提供OpenAI API Key", |
|
|
|
|
|
|
|
|
|
'deepseek_preset': "DeepSeek 代理模式", |
|
|
|
|
'deepseek_desc': "使用DeepSeek API作为代理，适合无GPU环境", |
|
|
|
|
'deepseek_note': "需要提供DeepSeek API Key", |
|
|
|
|
|
|
|
|
|
'glm4_preset': "GLM4 本地模式", |
|
|
|
|
'glm4_desc': "使用本地GLM4模型，需要GPU环境", |
|
|
|
|
'glm4_note': "需要配置本地模型路径", |
|
|
|
|
|
|
|
|
|
'vllm_preset': "VLLM 本地模式", |
|
|
|
|
'vllm_desc': "使用VLLM框架加载本地模型，需要GPU环境", |
|
|
|
|
'vllm_note': "需要配置本地模型路径", |
|
|
|
|
|
|
|
|
|
'llama_cpp_preset': "LLAMA_CPP 本地模式", |
|
|
|
|
'llama_cpp_desc': "使用LLAMA.cpp框架加载本地模型，CPU也可运行但推荐GPU", |
|
|
|
|
'llama_cpp_note': "需要配置本地模型路径，启用CUDA需设置CMAKE_ARGS=\"-DGGML_CUDA=ON\"", |
|
|
|
|
|
|
|
|
|
'ollama_preset': "Ollama 代理模式", |
|
|
|
|
'ollama_desc': "使用Ollama作为代理，适合无GPU环境", |
|
|
|
|
'ollama_note': "需要提供Ollama API Base", |
|
|
|
|
|
|
|
|
|
'custom_preset': "自定义模式", |
|
|
|
|
'custom_desc': "手动选择需要的extras", |
|
|
|
|
'custom_note': "适合高级用户" |
|
|
|
|
} |
|
|
|
|
"openai_preset": "OpenAI 代理模式", |
|
|
|
|
"openai_desc": "使用OpenAI API作为代理，适合无GPU环境", |
|
|
|
|
"openai_note": "需要提供OpenAI API Key", |
|
|
|
|
"deepseek_preset": "DeepSeek 代理模式", |
|
|
|
|
"deepseek_desc": "使用DeepSeek API作为代理，适合无GPU环境", |
|
|
|
|
"deepseek_note": "需要提供DeepSeek API Key", |
|
|
|
|
"glm4_preset": "GLM4 本地模式", |
|
|
|
|
"glm4_desc": "使用本地GLM4模型，需要GPU环境", |
|
|
|
|
"glm4_note": "需要配置本地模型路径", |
|
|
|
|
"vllm_preset": "VLLM 本地模式", |
|
|
|
|
"vllm_desc": "使用VLLM框架加载本地模型，需要GPU环境", |
|
|
|
|
"vllm_note": "需要配置本地模型路径", |
|
|
|
|
"llama_cpp_preset": "LLAMA_CPP 本地模式", |
|
|
|
|
"llama_cpp_desc": "使用LLAMA.cpp框架加载本地模型，CPU也可运行但推荐GPU", |
|
|
|
|
"llama_cpp_note": '需要配置本地模型路径，启用CUDA需设置CMAKE_ARGS="-DGGML_CUDA=ON"', |
|
|
|
|
"ollama_preset": "Ollama 代理模式", |
|
|
|
|
"ollama_desc": "使用Ollama作为代理，适合无GPU环境", |
|
|
|
|
"ollama_note": "需要提供Ollama API Base", |
|
|
|
|
"custom_preset": "自定义模式", |
|
|
|
|
"custom_desc": "手动选择需要的extras", |
|
|
|
|
"custom_note": "适合高级用户", |
|
|
|
|
}, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
def __init__(self, lang=None): |
|
|
|
|
@ -219,6 +193,7 @@ class I18N: |
|
|
|
|
if not lang: |
|
|
|
|
try: |
|
|
|
|
import locale |
|
|
|
|
|
|
|
|
|
try: |
|
|
|
|
# First try to get the locale from the environment |
|
|
|
|
lang = locale.getlocale()[0] |
|
|
|
|
@ -226,18 +201,18 @@ class I18N: |
|
|
|
|
try: |
|
|
|
|
lang = locale.getdefaultlocale()[0] |
|
|
|
|
except (AttributeError, ValueError): |
|
|
|
|
lang = 'en' |
|
|
|
|
lang = "en" |
|
|
|
|
|
|
|
|
|
if lang: |
|
|
|
|
lang = lang.split('_')[0] |
|
|
|
|
lang = lang.split("_")[0] |
|
|
|
|
else: |
|
|
|
|
lang = 'en' |
|
|
|
|
lang = "en" |
|
|
|
|
except (ImportError, AttributeError, ValueError): |
|
|
|
|
lang = 'en' |
|
|
|
|
lang = "en" |
|
|
|
|
|
|
|
|
|
# If the language is not supported, default to English |
|
|
|
|
if lang not in self.SUPPORTED_LANGUAGES: |
|
|
|
|
lang = 'en' |
|
|
|
|
lang = "en" |
|
|
|
|
|
|
|
|
|
self.lang = lang |
|
|
|
|
|
|
|
|
|
@ -246,7 +221,6 @@ class I18N: |
|
|
|
|
return self.TRANSLATIONS.get(self.lang, {}).get(key, key) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
i18n = I18N() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -274,11 +248,11 @@ def extract_workspace_extras(): |
|
|
|
|
workspace_root = current_dir |
|
|
|
|
break |
|
|
|
|
except Exception as e: |
|
|
|
|
print(i18n.get('cannot_parse').format(pyproject_path, e)) |
|
|
|
|
print(i18n.get("cannot_parse").format(pyproject_path, e)) |
|
|
|
|
current_dir = os.path.dirname(current_dir) |
|
|
|
|
|
|
|
|
|
if not workspace_root: |
|
|
|
|
print(i18n.get('workspace_not_found')) |
|
|
|
|
print(i18n.get("workspace_not_found")) |
|
|
|
|
return {} |
|
|
|
|
|
|
|
|
|
# Read the workspace configuration |
|
|
|
|
@ -297,7 +271,9 @@ def extract_workspace_extras(): |
|
|
|
|
matches = glob.glob(full_pattern, recursive=True) |
|
|
|
|
|
|
|
|
|
for match in matches: |
|
|
|
|
if os.path.isdir(match) and os.path.exists(os.path.join(match, "pyproject.toml")): |
|
|
|
|
if os.path.isdir(match) and os.path.exists( |
|
|
|
|
os.path.join(match, "pyproject.toml") |
|
|
|
|
): |
|
|
|
|
# Check if the directory should be excluded |
|
|
|
|
should_exclude = False |
|
|
|
|
for exclude_pattern in exclude_patterns: |
|
|
|
|
@ -320,18 +296,22 @@ def extract_workspace_extras(): |
|
|
|
|
with open(member_path, "rb") as f: |
|
|
|
|
member_data = tomli.load(f) |
|
|
|
|
|
|
|
|
|
project_name = member_data.get("project", {}).get("name", os.path.basename(member_dir)) |
|
|
|
|
optional_deps = member_data.get("project", {}).get("optional-dependencies", {}) |
|
|
|
|
project_name = member_data.get("project", {}).get( |
|
|
|
|
"name", os.path.basename(member_dir) |
|
|
|
|
) |
|
|
|
|
optional_deps = member_data.get("project", {}).get( |
|
|
|
|
"optional-dependencies", {} |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
if optional_deps: |
|
|
|
|
all_extras[project_name] = { |
|
|
|
|
"path": member_dir, |
|
|
|
|
"extras": list(optional_deps.keys()), |
|
|
|
|
"details": optional_deps |
|
|
|
|
"details": optional_deps, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
except Exception as e: |
|
|
|
|
print(i18n.get('cannot_parse').format(member_path, e)) |
|
|
|
|
print(i18n.get("cannot_parse").format(member_path, e)) |
|
|
|
|
|
|
|
|
|
return all_extras |
|
|
|
|
|
|
|
|
|
@ -340,84 +320,120 @@ def extract_workspace_extras(): |
|
|
|
|
def get_deployment_presets(): |
|
|
|
|
"""Get localized deployment presets""" |
|
|
|
|
return { |
|
|
|
|
i18n.get('openai_preset'): { |
|
|
|
|
i18n.get("openai_preset"): { |
|
|
|
|
"extras": ["base", "proxy_openai", "rag", "storage_chromadb", "dbgpts"], |
|
|
|
|
"config": "configs/dbgpt-proxy-openai.toml", |
|
|
|
|
"description": i18n.get('openai_desc'), |
|
|
|
|
"note": i18n.get('openai_note') |
|
|
|
|
"description": i18n.get("openai_desc"), |
|
|
|
|
"note": i18n.get("openai_note"), |
|
|
|
|
}, |
|
|
|
|
i18n.get('deepseek_preset'): { |
|
|
|
|
i18n.get("deepseek_preset"): { |
|
|
|
|
"extras": ["base", "proxy_openai", "rag", "storage_chromadb", "dbgpts"], |
|
|
|
|
"config": "configs/dbgpt-proxy-deepseek.toml", |
|
|
|
|
"description": i18n.get('deepseek_desc'), |
|
|
|
|
"note": i18n.get('deepseek_note') |
|
|
|
|
"description": i18n.get("deepseek_desc"), |
|
|
|
|
"note": i18n.get("deepseek_note"), |
|
|
|
|
}, |
|
|
|
|
i18n.get('glm4_preset'): { |
|
|
|
|
"extras": ["base", "hf", "cuda121", "rag", "storage_chromadb", "quant_bnb", "dbgpts"], |
|
|
|
|
i18n.get("glm4_preset"): { |
|
|
|
|
"extras": [ |
|
|
|
|
"base", |
|
|
|
|
"hf", |
|
|
|
|
"cuda121", |
|
|
|
|
"rag", |
|
|
|
|
"storage_chromadb", |
|
|
|
|
"quant_bnb", |
|
|
|
|
"dbgpts", |
|
|
|
|
], |
|
|
|
|
"config": "configs/dbgpt-local-glm.toml", |
|
|
|
|
"description": i18n.get('glm4_desc'), |
|
|
|
|
"note": i18n.get('glm4_note') |
|
|
|
|
"description": i18n.get("glm4_desc"), |
|
|
|
|
"note": i18n.get("glm4_note"), |
|
|
|
|
}, |
|
|
|
|
i18n.get('vllm_preset'): { |
|
|
|
|
"extras": ["base", "hf", "cuda121", "vllm", "rag", "storage_chromadb", "quant_bnb", "dbgpts"], |
|
|
|
|
i18n.get("vllm_preset"): { |
|
|
|
|
"extras": [ |
|
|
|
|
"base", |
|
|
|
|
"hf", |
|
|
|
|
"cuda121", |
|
|
|
|
"vllm", |
|
|
|
|
"rag", |
|
|
|
|
"storage_chromadb", |
|
|
|
|
"quant_bnb", |
|
|
|
|
"dbgpts", |
|
|
|
|
], |
|
|
|
|
"config": "configs/dbgpt-local-vllm.toml", |
|
|
|
|
"description": i18n.get('vllm_desc'), |
|
|
|
|
"note": i18n.get('vllm_note') |
|
|
|
|
"description": i18n.get("vllm_desc"), |
|
|
|
|
"note": i18n.get("vllm_note"), |
|
|
|
|
}, |
|
|
|
|
i18n.get('llama_cpp_preset'): { |
|
|
|
|
"extras": ["base", "hf", "cuda121", "llama_cpp", "rag", "storage_chromadb", "quant_bnb", "dbgpts"], |
|
|
|
|
i18n.get("llama_cpp_preset"): { |
|
|
|
|
"extras": [ |
|
|
|
|
"base", |
|
|
|
|
"hf", |
|
|
|
|
"cuda121", |
|
|
|
|
"llama_cpp", |
|
|
|
|
"rag", |
|
|
|
|
"storage_chromadb", |
|
|
|
|
"quant_bnb", |
|
|
|
|
"dbgpts", |
|
|
|
|
], |
|
|
|
|
"config": "configs/dbgpt-local-llama-cpp.toml", |
|
|
|
|
"description": i18n.get('llama_cpp_desc'), |
|
|
|
|
"note": i18n.get('llama_cpp_note') |
|
|
|
|
"description": i18n.get("llama_cpp_desc"), |
|
|
|
|
"note": i18n.get("llama_cpp_note"), |
|
|
|
|
}, |
|
|
|
|
i18n.get('ollama_preset'): { |
|
|
|
|
i18n.get("ollama_preset"): { |
|
|
|
|
"extras": ["base", "proxy_ollama", "rag", "storage_chromadb", "dbgpts"], |
|
|
|
|
"config": "configs/dbgpt-proxy-ollama.toml", |
|
|
|
|
"description": i18n.get('ollama_desc'), |
|
|
|
|
"note": i18n.get('ollama_note') |
|
|
|
|
"description": i18n.get("ollama_desc"), |
|
|
|
|
"note": i18n.get("ollama_note"), |
|
|
|
|
}, |
|
|
|
|
i18n.get('custom_preset'): { |
|
|
|
|
i18n.get("custom_preset"): { |
|
|
|
|
"extras": [], |
|
|
|
|
"config": "", |
|
|
|
|
"description": i18n.get('custom_desc'), |
|
|
|
|
"note": i18n.get('custom_note') |
|
|
|
|
} |
|
|
|
|
"description": i18n.get("custom_desc"), |
|
|
|
|
"note": i18n.get("custom_note"), |
|
|
|
|
}, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@click.group() |
|
|
|
|
@click.option('--language', '-l', type=click.Choice(['en', 'zh']), help=I18N().get('language_option')) |
|
|
|
|
@click.option( |
|
|
|
|
"--language", |
|
|
|
|
"-l", |
|
|
|
|
type=click.Choice(["en", "zh"]), |
|
|
|
|
help=I18N().get("language_option"), |
|
|
|
|
) |
|
|
|
|
def cli(language): |
|
|
|
|
"""UV Workspace Extras Helper - Manage optional dependencies in UV workspace""" |
|
|
|
|
if language: |
|
|
|
|
set_language(language) |
|
|
|
|
# Update command descriptions to the current language |
|
|
|
|
cli.help = i18n.get('cli_description') |
|
|
|
|
list_extras.help = i18n.get('list_cmd_description') |
|
|
|
|
install_command.help = i18n.get('install_cmd_description') |
|
|
|
|
deploy_preset.help = i18n.get('deploy_cmd_description') |
|
|
|
|
cli.help = i18n.get("cli_description") |
|
|
|
|
list_extras.help = i18n.get("list_cmd_description") |
|
|
|
|
install_command.help = i18n.get("install_cmd_description") |
|
|
|
|
deploy_preset.help = i18n.get("deploy_cmd_description") |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cli.command('list') |
|
|
|
|
@click.option('--verbose', '-v', is_flag=True, help=i18n.get('verbose_option')) |
|
|
|
|
@cli.command("list") |
|
|
|
|
@click.option("--verbose", "-v", is_flag=True, help=i18n.get("verbose_option")) |
|
|
|
|
def list_extras(verbose): |
|
|
|
|
"""List all extras in the workspace""" |
|
|
|
|
extras = extract_workspace_extras() |
|
|
|
|
|
|
|
|
|
if not extras: |
|
|
|
|
click.echo(i18n.get('no_extras_found')) |
|
|
|
|
click.echo(i18n.get("no_extras_found")) |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
click.echo(i18n.get('extras_in_workspace')) |
|
|
|
|
click.echo(i18n.get("extras_in_workspace")) |
|
|
|
|
|
|
|
|
|
for package, info in extras.items(): |
|
|
|
|
click.echo(click.style(f"📦 {package}", fg="green") + |
|
|
|
|
click.style(f" ({os.path.relpath(info['path'])})", fg="blue")) |
|
|
|
|
click.echo( |
|
|
|
|
click.style(f"📦 {package}", fg="green") |
|
|
|
|
+ click.style(f" ({os.path.relpath(info['path'])})", fg="blue") |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
if info['extras']: |
|
|
|
|
if info["extras"]: |
|
|
|
|
click.echo(f" {i18n.get('available_extras')}") |
|
|
|
|
for extra in info['extras']: |
|
|
|
|
deps = info['details'][extra] |
|
|
|
|
click.echo(f" - {click.style(extra, fg='yellow')}: {len(deps)} {i18n.get('dependencies')}") |
|
|
|
|
for extra in info["extras"]: |
|
|
|
|
deps = info["details"][extra] |
|
|
|
|
click.echo( |
|
|
|
|
f" - {click.style(extra, fg='yellow')}: {len(deps)} {i18n.get('dependencies')}" |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
if verbose: |
|
|
|
|
for dep in deps: |
|
|
|
|
@ -427,18 +443,18 @@ def list_extras(verbose): |
|
|
|
|
click.echo() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cli.command('install-cmd') |
|
|
|
|
@click.option('--interactive', '-i', is_flag=True, help=i18n.get('interactive_option')) |
|
|
|
|
@click.option('--all', 'install_all', is_flag=True, help=i18n.get('all_option')) |
|
|
|
|
@click.option('--china', is_flag=True, help=i18n.get('china_option')) |
|
|
|
|
@click.argument('package', required=False) |
|
|
|
|
@click.argument('extra', required=False) |
|
|
|
|
@cli.command("install-cmd") |
|
|
|
|
@click.option("--interactive", "-i", is_flag=True, help=i18n.get("interactive_option")) |
|
|
|
|
@click.option("--all", "install_all", is_flag=True, help=i18n.get("all_option")) |
|
|
|
|
@click.option("--china", is_flag=True, help=i18n.get("china_option")) |
|
|
|
|
@click.argument("package", required=False) |
|
|
|
|
@click.argument("extra", required=False) |
|
|
|
|
def install_command(interactive, install_all, china, package, extra): |
|
|
|
|
"""Generate installation commands for extras""" |
|
|
|
|
extras = extract_workspace_extras() |
|
|
|
|
|
|
|
|
|
if not extras: |
|
|
|
|
click.echo(i18n.get('no_extras_found')) |
|
|
|
|
click.echo(i18n.get("no_extras_found")) |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
# Interactive mode |
|
|
|
|
@ -450,202 +466,254 @@ def install_command(interactive, install_all, china, package, extra): |
|
|
|
|
if install_all: |
|
|
|
|
all_extras = [] |
|
|
|
|
for pkg_info in extras.values(): |
|
|
|
|
all_extras.extend(pkg_info['extras']) |
|
|
|
|
all_extras.extend(pkg_info["extras"]) |
|
|
|
|
|
|
|
|
|
if all_extras: |
|
|
|
|
cmd = "uv sync --all-packages " + " ".join([f"--extra \"{e}\"" for e in all_extras]) |
|
|
|
|
cmd = "uv sync --all-packages " + " ".join( |
|
|
|
|
[f'--extra "{e}"' for e in all_extras] |
|
|
|
|
) |
|
|
|
|
if china: |
|
|
|
|
cmd += " --index-url=https://pypi.tuna.tsinghua.edu.cn/simple" |
|
|
|
|
click.echo(i18n.get('install_all_extras')) |
|
|
|
|
click.echo(i18n.get("install_all_extras")) |
|
|
|
|
click.echo(cmd) |
|
|
|
|
else: |
|
|
|
|
click.echo(i18n.get('no_extras_found')) |
|
|
|
|
click.echo(i18n.get("no_extras_found")) |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
# If no package or extra is provided, show all possible installation commands |
|
|
|
|
if not package: |
|
|
|
|
for pkg, info in extras.items(): |
|
|
|
|
if info['extras']: |
|
|
|
|
for e in info['extras']: |
|
|
|
|
cmd = f"uv sync --extra \"{e}\"" |
|
|
|
|
if info["extras"]: |
|
|
|
|
for e in info["extras"]: |
|
|
|
|
cmd = f'uv sync --extra "{e}"' |
|
|
|
|
if china: |
|
|
|
|
cmd += " --index-url=https://pypi.tuna.tsinghua.edu.cn/simple" |
|
|
|
|
click.echo(i18n.get('install_extras_for').format(pkg, e)) |
|
|
|
|
click.echo(i18n.get("install_extras_for").format(pkg, e)) |
|
|
|
|
click.echo(cmd) |
|
|
|
|
click.echo() |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
# Check if the specified package exists |
|
|
|
|
if package not in extras: |
|
|
|
|
click.echo(i18n.get('package_not_in_workspace').format(package)) |
|
|
|
|
click.echo(i18n.get('available_packages').format(', '.join(extras.keys()))) |
|
|
|
|
click.echo(i18n.get("package_not_in_workspace").format(package)) |
|
|
|
|
click.echo(i18n.get("available_packages").format(", ".join(extras.keys()))) |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
# If no extra is provided, show all extras for the package |
|
|
|
|
if not extra: |
|
|
|
|
pkg_extras = extras[package]['extras'] |
|
|
|
|
pkg_extras = extras[package]["extras"] |
|
|
|
|
if not pkg_extras: |
|
|
|
|
click.echo(i18n.get('package_no_extras').format(package)) |
|
|
|
|
click.echo(i18n.get("package_no_extras").format(package)) |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
cmd = "uv sync " + " ".join([f"--extra \"{e}\"" for e in pkg_extras]) |
|
|
|
|
cmd = "uv sync " + " ".join([f'--extra "{e}"' for e in pkg_extras]) |
|
|
|
|
if china: |
|
|
|
|
cmd += " --index-url=https://pypi.tuna.tsinghua.edu.cn/simple" |
|
|
|
|
click.echo(i18n.get('install_extras_for').format(package, ' '.join(pkg_extras))) |
|
|
|
|
click.echo(i18n.get("install_extras_for").format(package, " ".join(pkg_extras))) |
|
|
|
|
click.echo(cmd) |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
# Check if the specified extra exists |
|
|
|
|
if extra not in extras[package]['extras']: |
|
|
|
|
click.echo(i18n.get('extra_not_in_package').format(extra, package)) |
|
|
|
|
click.echo(i18n.get('available_extras_in_package').format(', '.join(extras[package]['extras']))) |
|
|
|
|
if extra not in extras[package]["extras"]: |
|
|
|
|
click.echo(i18n.get("extra_not_in_package").format(extra, package)) |
|
|
|
|
click.echo( |
|
|
|
|
i18n.get("available_extras_in_package").format( |
|
|
|
|
", ".join(extras[package]["extras"]) |
|
|
|
|
) |
|
|
|
|
) |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
# Show the command to install the specified extra |
|
|
|
|
cmd = f"uv sync --extra \"{extra}\"" |
|
|
|
|
cmd = f'uv sync --extra "{extra}"' |
|
|
|
|
if china: |
|
|
|
|
cmd += " --index-url=https://pypi.tuna.tsinghua.edu.cn/simple" |
|
|
|
|
click.echo(i18n.get('install_extras_for').format(package, extra)) |
|
|
|
|
click.echo(i18n.get("install_extras_for").format(package, extra)) |
|
|
|
|
click.echo(cmd) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _interactive_install_guide(extras: Dict[str, Any], use_china_mirror: bool = False): |
|
|
|
|
"""Interactive installation guide""" |
|
|
|
|
click.echo(click.style(i18n.get('welcome'), fg="green", bold=True)) |
|
|
|
|
click.echo(i18n.get('help_message')) |
|
|
|
|
click.echo(click.style(i18n.get("welcome"), fg="green", bold=True)) |
|
|
|
|
click.echo(i18n.get("help_message")) |
|
|
|
|
|
|
|
|
|
# Get deployment presets |
|
|
|
|
deployment_presets = get_deployment_presets() |
|
|
|
|
|
|
|
|
|
# First step: select installation mode |
|
|
|
|
questions = [ |
|
|
|
|
inquirer.List('preset', |
|
|
|
|
message=i18n.get('select_mode'), |
|
|
|
|
choices=[(f"{name} - {info['description']}", name) for name, info in deployment_presets.items()], |
|
|
|
|
carousel=True) |
|
|
|
|
inquirer.List( |
|
|
|
|
"preset", |
|
|
|
|
message=i18n.get("select_mode"), |
|
|
|
|
choices=[ |
|
|
|
|
(f"{name} - {info['description']}", name) |
|
|
|
|
for name, info in deployment_presets.items() |
|
|
|
|
], |
|
|
|
|
carousel=True, |
|
|
|
|
) |
|
|
|
|
] |
|
|
|
|
answers = inquirer.prompt(questions) |
|
|
|
|
|
|
|
|
|
if not answers: |
|
|
|
|
return # Operation canceled |
|
|
|
|
|
|
|
|
|
selected_preset = answers['preset'] |
|
|
|
|
selected_preset = answers["preset"] |
|
|
|
|
preset_info = deployment_presets[selected_preset] |
|
|
|
|
|
|
|
|
|
# Custom mode: let user select extras |
|
|
|
|
if selected_preset == i18n.get('custom_preset'): |
|
|
|
|
if selected_preset == i18n.get("custom_preset"): |
|
|
|
|
# Collect all available extras |
|
|
|
|
all_available_extras = set() |
|
|
|
|
for pkg_info in extras.values(): |
|
|
|
|
all_available_extras.update(pkg_info["extras"]) |
|
|
|
|
|
|
|
|
|
questions = [ |
|
|
|
|
inquirer.Checkbox('selected_extras', |
|
|
|
|
message=i18n.get('select_extras'), |
|
|
|
|
inquirer.Checkbox( |
|
|
|
|
"selected_extras", |
|
|
|
|
message=i18n.get("select_extras"), |
|
|
|
|
choices=sorted(list(all_available_extras)), |
|
|
|
|
carousel=True) |
|
|
|
|
carousel=True, |
|
|
|
|
) |
|
|
|
|
] |
|
|
|
|
answers = inquirer.prompt(questions) |
|
|
|
|
|
|
|
|
|
if not answers or not answers['selected_extras']: |
|
|
|
|
click.echo(i18n.get('operation_canceled')) |
|
|
|
|
if not answers or not answers["selected_extras"]: |
|
|
|
|
click.echo(i18n.get("operation_canceled")) |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
preset_info['extras'] = answers['selected_extras'] |
|
|
|
|
preset_info["extras"] = answers["selected_extras"] |
|
|
|
|
|
|
|
|
|
# Show installation information |
|
|
|
|
click.echo("\n" + click.style(i18n.get('installation_info'), fg="blue", bold=True)) |
|
|
|
|
click.echo(f"{i18n.get('selected_mode')} {click.style(selected_preset, fg='green')}") |
|
|
|
|
click.echo("\n" + click.style(i18n.get("installation_info"), fg="blue", bold=True)) |
|
|
|
|
click.echo( |
|
|
|
|
f"{i18n.get('selected_mode')} {click.style(selected_preset, fg='green')}" |
|
|
|
|
) |
|
|
|
|
click.echo(f"{i18n.get('description')} {preset_info['description']}") |
|
|
|
|
click.echo(f"{i18n.get('note')} {preset_info['note']}") |
|
|
|
|
click.echo(f"{i18n.get('will_install')} {', '.join(preset_info['extras'])}") |
|
|
|
|
|
|
|
|
|
if preset_info['config']: |
|
|
|
|
if preset_info["config"]: |
|
|
|
|
click.echo(f"{i18n.get('config_file')} {preset_info['config']}") |
|
|
|
|
|
|
|
|
|
# Confirm installation |
|
|
|
|
questions = [ |
|
|
|
|
inquirer.Confirm('confirm', |
|
|
|
|
message=i18n.get('generate_command'), |
|
|
|
|
default=True) |
|
|
|
|
inquirer.Confirm("confirm", message=i18n.get("generate_command"), default=True) |
|
|
|
|
] |
|
|
|
|
answers = inquirer.prompt(questions) |
|
|
|
|
|
|
|
|
|
if not answers or not answers['confirm']: |
|
|
|
|
click.echo(i18n.get('operation_canceled')) |
|
|
|
|
if not answers or not answers["confirm"]: |
|
|
|
|
click.echo(i18n.get("operation_canceled")) |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
# Create installation command |
|
|
|
|
cmd = "uv sync --all-packages " + " ".join([f"--extra \"{e}\"" for e in preset_info['extras']]) |
|
|
|
|
cmd = "uv sync --all-packages " + " ".join( |
|
|
|
|
[f'--extra "{e}"' for e in preset_info["extras"]] |
|
|
|
|
) |
|
|
|
|
if use_china_mirror: |
|
|
|
|
cmd += " --index-url=https://pypi.tuna.tsinghua.edu.cn/simple" |
|
|
|
|
|
|
|
|
|
click.echo("\n" + click.style(i18n.get('installation_command'), fg="green", bold=True)) |
|
|
|
|
click.echo( |
|
|
|
|
"\n" + click.style(i18n.get("installation_command"), fg="green", bold=True) |
|
|
|
|
) |
|
|
|
|
click.echo(cmd) |
|
|
|
|
|
|
|
|
|
if preset_info.get('config'): |
|
|
|
|
click.echo("\n" + click.style(i18n.get('startup_command'), fg="green", bold=True)) |
|
|
|
|
if preset_info.get("config"): |
|
|
|
|
click.echo( |
|
|
|
|
"\n" + click.style(i18n.get("startup_command"), fg="green", bold=True) |
|
|
|
|
) |
|
|
|
|
click.echo(f"uv run dbgpt start webserver --config {preset_info['config']}") |
|
|
|
|
|
|
|
|
|
# The step to configure the API key or model path |
|
|
|
|
if i18n.get('openai_note') in preset_info['note'] or i18n.get('deepseek_note') in preset_info['note']: |
|
|
|
|
click.echo("\n" + click.style(i18n.get('further_configuration'), fg="yellow", bold=True)) |
|
|
|
|
if i18n.get('openai_note') in preset_info['note'] or i18n.get('deepseek_note') in preset_info['note']: |
|
|
|
|
click.echo(i18n.get('set_api_key').format(preset_info['config'])) |
|
|
|
|
elif i18n.get('glm4_note') in preset_info['note'] or i18n.get('vllm_note') in preset_info['note'] or i18n.get('llama_cpp_note') in preset_info['note']: |
|
|
|
|
click.echo("\n" + click.style(i18n.get('further_configuration'), fg="yellow", bold=True)) |
|
|
|
|
if i18n.get('glm4_note') in preset_info['note'] or i18n.get('vllm_note') in preset_info['note'] or i18n.get('llama_cpp_note') in preset_info['note']: |
|
|
|
|
click.echo(i18n.get('set_model_path').format(preset_info['config'])) |
|
|
|
|
if ( |
|
|
|
|
i18n.get("openai_note") in preset_info["note"] |
|
|
|
|
or i18n.get("deepseek_note") in preset_info["note"] |
|
|
|
|
): |
|
|
|
|
click.echo( |
|
|
|
|
"\n" |
|
|
|
|
+ click.style(i18n.get("further_configuration"), fg="yellow", bold=True) |
|
|
|
|
) |
|
|
|
|
if ( |
|
|
|
|
i18n.get("openai_note") in preset_info["note"] |
|
|
|
|
or i18n.get("deepseek_note") in preset_info["note"] |
|
|
|
|
): |
|
|
|
|
click.echo(i18n.get("set_api_key").format(preset_info["config"])) |
|
|
|
|
elif ( |
|
|
|
|
i18n.get("glm4_note") in preset_info["note"] |
|
|
|
|
or i18n.get("vllm_note") in preset_info["note"] |
|
|
|
|
or i18n.get("llama_cpp_note") in preset_info["note"] |
|
|
|
|
): |
|
|
|
|
click.echo( |
|
|
|
|
"\n" |
|
|
|
|
+ click.style(i18n.get("further_configuration"), fg="yellow", bold=True) |
|
|
|
|
) |
|
|
|
|
if ( |
|
|
|
|
i18n.get("glm4_note") in preset_info["note"] |
|
|
|
|
or i18n.get("vllm_note") in preset_info["note"] |
|
|
|
|
or i18n.get("llama_cpp_note") in preset_info["note"] |
|
|
|
|
): |
|
|
|
|
click.echo(i18n.get("set_model_path").format(preset_info["config"])) |
|
|
|
|
|
|
|
|
|
click.echo("\n" + click.style(f"🎉 {i18n.get('finished')}", fg="green", bold=True)) |
|
|
|
|
click.echo(i18n.get('copy_command')) |
|
|
|
|
click.echo(i18n.get("copy_command")) |
|
|
|
|
click.echo("uv run install_help.py --help") |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cli.command('deploy') |
|
|
|
|
@click.option('--preset', '-p', help=i18n.get('preset_option')) |
|
|
|
|
@click.option('--china', is_flag=True, help=i18n.get('china_option')) |
|
|
|
|
@click.option('--list', 'list_presets', is_flag=True, help=i18n.get('list_presets_option')) |
|
|
|
|
@cli.command("deploy") |
|
|
|
|
@click.option("--preset", "-p", help=i18n.get("preset_option")) |
|
|
|
|
@click.option("--china", is_flag=True, help=i18n.get("china_option")) |
|
|
|
|
@click.option( |
|
|
|
|
"--list", "list_presets", is_flag=True, help=i18n.get("list_presets_option") |
|
|
|
|
) |
|
|
|
|
def deploy_preset(preset, china, list_presets): |
|
|
|
|
"""Use predefined deployment templates""" |
|
|
|
|
deployment_presets = get_deployment_presets() |
|
|
|
|
|
|
|
|
|
if list_presets: |
|
|
|
|
click.echo(click.style(i18n.get('available_presets'), fg="green", bold=True)) |
|
|
|
|
click.echo(click.style(i18n.get("available_presets"), fg="green", bold=True)) |
|
|
|
|
for name, info in deployment_presets.items(): |
|
|
|
|
click.echo(f"\n{click.style(name, fg='yellow', bold=True)}") |
|
|
|
|
click.echo(f"{i18n.get('description')} {info['description']}") |
|
|
|
|
click.echo(f"{i18n.get('note')} {info['note']}") |
|
|
|
|
click.echo(f"Extras: {', '.join(info['extras'])}") |
|
|
|
|
if info['config']: |
|
|
|
|
if info["config"]: |
|
|
|
|
click.echo(f"{i18n.get('config_file')} {info['config']}") |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
if not preset: |
|
|
|
|
click.echo(i18n.get('specify_preset')) |
|
|
|
|
click.echo(i18n.get("specify_preset")) |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
if preset not in deployment_presets: |
|
|
|
|
click.echo(i18n.get('preset_not_found').format(preset)) |
|
|
|
|
click.echo(i18n.get('available_presets_list').format(', '.join(deployment_presets.keys()))) |
|
|
|
|
click.echo(i18n.get("preset_not_found").format(preset)) |
|
|
|
|
click.echo( |
|
|
|
|
i18n.get("available_presets_list").format( |
|
|
|
|
", ".join(deployment_presets.keys()) |
|
|
|
|
) |
|
|
|
|
) |
|
|
|
|
return |
|
|
|
|
|
|
|
|
|
preset_info = deployment_presets[preset] |
|
|
|
|
|
|
|
|
|
click.echo(i18n.get('using_preset').format(preset)) |
|
|
|
|
click.echo(i18n.get("using_preset").format(preset)) |
|
|
|
|
click.echo(f"{i18n.get('description')} {preset_info['description']}") |
|
|
|
|
click.echo(f"{i18n.get('note')} {preset_info['note']}") |
|
|
|
|
|
|
|
|
|
cmd = "uv sync --all-packages " + " ".join([f"--extra \"{e}\"" for e in preset_info['extras']]) |
|
|
|
|
cmd = "uv sync --all-packages " + " ".join( |
|
|
|
|
[f'--extra "{e}"' for e in preset_info["extras"]] |
|
|
|
|
) |
|
|
|
|
if china: |
|
|
|
|
cmd += " --index-url=https://pypi.tuna.tsinghua.edu.cn/simple" |
|
|
|
|
|
|
|
|
|
click.echo("\n" + click.style(i18n.get('installation_command'), fg="green", bold=True)) |
|
|
|
|
click.echo( |
|
|
|
|
"\n" + click.style(i18n.get("installation_command"), fg="green", bold=True) |
|
|
|
|
) |
|
|
|
|
click.echo(cmd) |
|
|
|
|
|
|
|
|
|
if preset_info.get('config'): |
|
|
|
|
click.echo("\n" + click.style(i18n.get('startup_command'), fg="green", bold=True)) |
|
|
|
|
if preset_info.get("config"): |
|
|
|
|
click.echo( |
|
|
|
|
"\n" + click.style(i18n.get("startup_command"), fg="green", bold=True) |
|
|
|
|
) |
|
|
|
|
click.echo(f"uv run dbgpt start webserver --config {preset_info['config']}") |
|
|
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
|
|
cli() |
|
|
|
|
|
Fangyin Cheng
8 months ago