fix codestral templating

continuedev · May 29, 2024 · 58691db · 58691db
1 parent 9702e6d
commit 58691db
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/core/llm/autodetect.ts b/core/llm/autodetect.ts
@@ -176,6 +176,10 @@ function autodetectTemplateType(model: string): TemplateType | undefined {
     return "none";
   }
 
+  if (lower.includes("codestral")) {
+    return "none";
+  }
+
   if (lower.includes("alpaca") || lower.includes("wizard")) {
     return "alpaca";
   }
@@ -303,6 +307,8 @@ function autodetectPromptTemplates(
     editTemplate = null;
   } else if (templateType) {
     editTemplate = gptEditPrompt;
+  } else if (model.includes("codestral")) {
+    editTemplate = osModelsEditPrompt;
   }
 
   if (editTemplate !== null) {

diff --git a/extensions/vscode/src/webviewProtocol.ts b/extensions/vscode/src/webviewProtocol.ts
@@ -138,10 +138,6 @@ export class VsCodeWebviewProtocol
             if (message.includes("exceeded")) {
               message +=
                 " To keep using Continue, you can set up a local model or use your own API key.";
-            } else if (message.includes("upgrade Continue")) {
-            } else {
-              message +=
-                " To avoid rate limiting, you can set up a local model or use your own API key.";
             }
 
             vscode.window