diff options
| -rw-r--r-- | continuedev/src/continuedev/core/models.py | 3 | ||||
| -rw-r--r-- | continuedev/src/continuedev/libs/util/edit_config.py | 16 |
2 files changed, 12 insertions, 7 deletions
diff --git a/continuedev/src/continuedev/core/models.py b/continuedev/src/continuedev/core/models.py index f03a8f29..f3bf8125 100644 --- a/continuedev/src/continuedev/core/models.py +++ b/continuedev/src/continuedev/core/models.py @@ -10,7 +10,7 @@ from ..libs.llm.ollama import Ollama from ..libs.llm.openai import OpenAI from ..libs.llm.replicate import ReplicateLLM from ..libs.llm.together import TogetherLLM - +from ..libs.llm.llamacpp import LlamaCpp class ContinueSDK(BaseModel): pass @@ -35,6 +35,7 @@ MODEL_CLASSES = { AnthropicLLM, ReplicateLLM, Ollama, + LlamaCpp ] } diff --git a/continuedev/src/continuedev/libs/util/edit_config.py b/continuedev/src/continuedev/libs/util/edit_config.py index 7bdffc8e..3dd4646c 100644 --- a/continuedev/src/continuedev/libs/util/edit_config.py +++ b/continuedev/src/continuedev/libs/util/edit_config.py @@ -5,11 +5,14 @@ import redbaron from .paths import getConfigFilePath - -def load_red(): +def get_config_source(): config_file_path = getConfigFilePath() with open(config_file_path, "r") as file: source_code = file.read() + return source_code + +def load_red(): + source_code = get_config_source() red = redbaron.RedBaron(source_code) return red @@ -54,12 +57,13 @@ def edit_config_property(key_path: List[str], value: redbaron.RedBaron): def add_config_import(line: str): + # check if the import already exists + source = get_config_source() + if line in source: + return + with edit_lock: red = load_red() - # check if the import already exists - for node in red: - if node.type == "import" and node.dumps() == line: - return # if it doesn't exist, add it red.insert(1, line) |