fix: :bug: automigrate between short/long imports

1 files changed, 26 insertions, 9 deletions

diff --git a/continuedev/src/continuedev/core/sdk.py b/continuedev/src/continuedev/core/sdk.py
index e8ee8167..c3b50375 100644
--- a/continuedev/src/continuedev/core/sdk.py
+++ b/continuedev/src/continuedev/core/sdk.py
@@ -5,7 +5,11 @@ from typing import Coroutine, List, Optional, Union
 from ..libs.llm import LLM
 from ..libs.util.devdata import dev_data_logger
 from ..libs.util.logging import logger
-from ..libs.util.paths import getConfigFilePath, getDiffsFolderPath
+from ..libs.util.paths import (
+    convertConfigImports,
+    getConfigFilePath,
+    getDiffsFolderPath,
+)
 from ..libs.util.telemetry import posthog_logger
 from ..models.filesystem import RangeInFile
 from ..models.filesystem_edit import (
@@ -236,14 +240,27 @@ class ContinueSDK(AbstractContinueSDK):
 
     _last_valid_config: ContinueConfig = None
 
-    def _load_config_dot_py(self) -> ContinueConfig:
-        path = getConfigFilePath()
-        config = ContinueConfig.from_filepath(path)
-        self._last_valid_config = config
-
-        logger.debug("Loaded Continue config file from %s", path)
-
-        return config
+    def _load_config_dot_py(self, retry: bool = True) -> ContinueConfig:
+        try:
+            path = getConfigFilePath()
+            config = ContinueConfig.from_filepath(path)
+            self._last_valid_config = config
+
+            logger.debug("Loaded Continue config file from %s", path)
+
+            return config
+        except ModuleNotFoundError as e:
+            if not retry:
+                raise e
+            # Check if the module was "continuedev.src"
+            if e.name == "continuedev.src":
+                convertConfigImports(shorten=True)
+                return self._load_config_dot_py(retry=False)
+            elif e.name.startswith("continuedev."):
+                convertConfigImports(shorten=False)
+                return self._load_config_dot_py(retry=False)
+            else:
+                raise e
 
     def get_code_context(
         self, only_editing: bool = False