Support custom conda registory

devchat-ai · Sep 20, 2024 · b89aae5 · b89aae5
1 parent 3ecbcd4
commit b89aae5
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/main/kotlin/ai/devchat/installer/PythonEnvManager.kt b/src/main/kotlin/ai/devchat/installer/PythonEnvManager.kt
@@ -3,6 +3,7 @@ package ai.devchat.installer
 import ai.devchat.common.Log
 import ai.devchat.common.OSInfo
 import ai.devchat.common.PathUtils
+import ai.devchat.storage.CONFIG
 import java.io.File
 import java.io.IOException
 import java.nio.file.Paths
@@ -56,7 +57,7 @@ class PythonEnvManager {
             PathUtils.mambaBinPath,
             "create",
             "-n", name,
-            "-c", "conda-forge",
+            "-c", getCondaForgeUrl(),
             "-r", PathUtils.mambaWorkPath,
             "python=$version",
             "--yes"
@@ -91,6 +92,16 @@ class PythonEnvManager {
         }
         throw RuntimeException("$errPrefix Maximum retries exceed")
     }
+
+    private fun getCondaForgeUrl(): String {
+        val defaultUrl = "https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/"
+        return try {
+            (CONFIG["conda-forge-url"] as? String ) ?: defaultUrl
+        } catch (error: Exception) {
+            Log.warn("Error reading conda-forge URL from config: $error")
+            defaultUrl
+        }
+    }
 }