attempting alternative solution to import config

2 files changed, 48 insertions, 6 deletions

diff --git a/continuedev/src/continuedev/core/sdk.py b/continuedev/src/continuedev/core/sdk.py
index 57e2c099..42cfbcb9 100644
--- a/continuedev/src/continuedev/core/sdk.py
+++ b/continuedev/src/continuedev/core/sdk.py
@@ -1,7 +1,7 @@
-from functools import cached_property
 import traceback
-from typing import Coroutine, Dict, Literal, Union
+from typing import Coroutine, Union
 import os
+import importlib
 
 from ..plugins.steps.core.core import DefaultModelEditCodeStep
 from ..models.main import Range
@@ -164,6 +164,52 @@ class ContinueSDK(AbstractContinueSDK):
     _last_valid_config: ContinueConfig = None
 
     def _load_config_dot_py(self) -> ContinueConfig:
+        # Read the file content
+        with open(os.path.expanduser('~/.continue/config.py')) as file:
+            config_content = file.read()
+
+        def load_module(module_name: str, class_names: List[str]):
+            # from anthropic import AsyncAnthropic
+            module = importlib.import_module(module_name)
+            for class_name in class_names:
+                globals()[class_name] = getattr(module, class_name)
+
+        while True:
+            # Execute the file content
+            locals_var = {}
+            try:
+                import importlib.util
+                spec = importlib.util.spec_from_file_location(
+                    "config", "/Users/natesesti/.continue/config.py")
+                config = importlib.util.module_from_spec(spec)
+                spec.loader.exec_module(config)
+
+                return config.config
+                # exec(config_content, globals(), locals_var)
+                # print("Done executing, ", locals_var)
+                # return locals_var['config']
+            except ModuleNotFoundError as e:
+                print("ModuleNotFoundError")
+                print(e)
+                print(traceback.format_exception(e))
+                formatted = traceback.format_exception(e)
+                line = formatted[-2].split("\n")[-2].strip().split()
+                # Parse the module name and class name from the error message
+                # Example: ModuleNotFoundError: No module named 'continuedev.src.continuedev.plugins.context_providers.google'
+
+                # Get the module name
+                module_name = line[1]
+                # Get the class name
+                class_names = list(map(lambda x: x.replace(",", ""), filter(lambda x: x.strip() != "", line[3:])))
+
+                # Load the module
+                print(
+                    f"Loading module {module_name} with class names {class_names}")
+                load_module(module_name, class_names)
+            except Exception as e:
+                print("Failed to execute config.py: ", e)
+                raise e
+
         # Use importlib to load the config file config.py at the given path
         path = getConfigFilePath()
 
diff --git a/continuedev/src/continuedev/libs/llm/anthropic.py b/continuedev/src/continuedev/libs/llm/anthropic.py
index ec1b7e40..e6b88d03 100644
--- a/continuedev/src/continuedev/libs/llm/anthropic.py
+++ b/continuedev/src/continuedev/libs/llm/anthropic.py
@@ -17,10 +17,6 @@ class AnthropicLLM(LLM):
     class Config:
         arbitrary_types_allowed = True
 
-    def __init__(self, model: str, system_message: str = None):
-        self.model = model
-        self.system_message = system_message
-
     async def start(self, *, api_key: Optional[str] = None, **kwargs):
         self._async_client = AsyncAnthropic(api_key=api_key)