feat: add pip_index_url configuration and update installation commands

Author: ai-chen2050

src/parallax/config/config_manager.cpp

@@ -18,6 +18,7 @@ namespace parallax
         const std::string KEY_WSL_KERNEL_URL = "wsl_kernel_url";
         const std::string KEY_PRAKASA_GIT_REPO_URL = "prakasa_git_repo_url";
         const std::string KEY_PRAKASA_GIT_BRANCH = "prakasa_git_branch";
+        const std::string KEY_PIP_INDEX_URL = "pip_index_url";
 
         // Default configuration file name
         const std::string ConfigManager::DEFAULT_CONFIG_PATH = "parallax_config.txt";
@@ -62,7 +63,7 @@ namespace parallax
             config_values_[KEY_PRAKASA_GIT_REPO_URL] =
                 "https://github.com/hetu-project/prakasa.git";
             config_values_[KEY_PRAKASA_GIT_BRANCH] = "main";
-            // proxy_url has no default value
+            // proxy_url and pip_index_url have no default value (use official PyPI by default)
         }
 
         // Load configuration file
@@ -109,7 +110,8 @@ namespace parallax
                 {KEY_WSL_INSTALLER_URL, config_values_[KEY_WSL_INSTALLER_URL]},
                 {KEY_WSL_KERNEL_URL, config_values_[KEY_WSL_KERNEL_URL]},
                 {KEY_PRAKASA_GIT_REPO_URL, config_values_[KEY_PRAKASA_GIT_REPO_URL]},
-                {KEY_PRAKASA_GIT_BRANCH, config_values_[KEY_PRAKASA_GIT_BRANCH]}};
+                {KEY_PRAKASA_GIT_BRANCH, config_values_[KEY_PRAKASA_GIT_BRANCH]},
+                {KEY_PIP_INDEX_URL, config_values_[KEY_PIP_INDEX_URL]}};
 
             std::string line;
             while (std::getline(file, line))
@@ -325,7 +327,8 @@ namespace parallax
         {
             static const std::set<std::string> valid_keys = {
                 KEY_PROXY_URL, KEY_WSL_LINUX_DISTRO, KEY_WSL_INSTALLER_URL,
-                KEY_WSL_KERNEL_URL, KEY_PRAKASA_GIT_REPO_URL, KEY_PRAKASA_GIT_BRANCH};
+                KEY_WSL_KERNEL_URL, KEY_PRAKASA_GIT_REPO_URL, KEY_PRAKASA_GIT_BRANCH,
+                KEY_PIP_INDEX_URL};
 
             return valid_keys.find(key) != valid_keys.end();
         }

src/parallax/config/config_manager.h

@@ -15,6 +15,7 @@ namespace parallax
       extern const std::string KEY_WSL_KERNEL_URL;
       extern const std::string KEY_PRAKASA_GIT_REPO_URL;
       extern const std::string KEY_PRAKASA_GIT_BRANCH;
+      extern const std::string KEY_PIP_INDEX_URL;
 
       // Configuration file manager class
       class ConfigManager

src/parallax/environment/software_installer2.cpp

@@ -175,6 +175,9 @@ namespace parallax
             std::string git_branch =
                 parallax::config::ConfigManager::GetInstance().GetConfigValue(
                     parallax::config::KEY_PRAKASA_GIT_BRANCH);
+            std::string pip_index_url =
+                parallax::config::ConfigManager::GetInstance().GetConfigValue(
+                    parallax::config::KEY_PIP_INDEX_URL);
 
             const std::string &proxy_url = context_->GetProxyUrl();
 
@@ -283,6 +286,14 @@ namespace parallax
                 "cd ~/prakasa && ([ -d ./venv ] || python3 -m venv ./venv) && source "
                 "./venv/bin/activate "
                 "&& pip install -e '.[gpu]'";
+            if (!pip_index_url.empty())
+            {
+                install_base_cmd =
+                    "cd ~/prakasa && ([ -d ./venv ] || python3 -m venv ./venv) && source "
+                    "./venv/bin/activate "
+                    "&& pip install -i " +
+                    pip_index_url + " -e '.[gpu]'";
+            }
             if (!proxy_url.empty())
             {
                 install_base_cmd =
@@ -291,6 +302,15 @@ namespace parallax
                     "./venv/bin/activate && HTTP_PROXY=\"" +
                     proxy_url + "\" HTTPS_PROXY=\"" + proxy_url +
                     "\" pip install -e '.[gpu]'";
+                if (!pip_index_url.empty())
+                {
+                    install_base_cmd =
+                        "cd ~/prakasa && ([ -d ./venv ] || python3 -m venv ./venv) && "
+                        "source "
+                        "./venv/bin/activate && HTTP_PROXY=\"" +
+                        proxy_url + "\" HTTPS_PROXY=\"" + proxy_url +
+                        "\" pip install -i " + pip_index_url + " -e '.[gpu]'";
+                }
             }
             commands.emplace_back("install_prakasa_base", install_base_cmd, 1800,
                                   true);