fix yaml schema registration logic

Author: RomneyDa

extensions/vscode/src/activation/activate.ts

@@ -34,34 +34,33 @@ export async function activateExtension(context: vscode.ExtensionContext) {
   }
 
   // Register config.yaml schema by removing old entries and adding new one (uri.fsPath changes with each version)
+  const yamlMatcher = ".continue/**/*.yaml";
   const yamlConfig = vscode.workspace.getConfiguration("yaml");
+
   const existingSchemas = yamlConfig.get("schemas") || {};
+  const newSchemas = Object.entries(existingSchemas).filter(
+    ([_, value]) => Array.isArray(value) && value.includes(yamlMatcher), // remove old ones
+  );
 
-  const oldContinueSchemaKeys = Object.entries(existingSchemas)
-    .filter(
-      ([_, value]) =>
-        Array.isArray(value) && value.includes(".continue/**/*.yaml"),
-    )
-    .map(([key]) => key);
+  const newPath = path.join(
+    context.extension.extensionUri.fsPath,
+    "config-yaml-schema.json",
+  );
+  newSchemas.push([newPath, [yamlMatcher]]);
 
-  for (const oldKey of oldContinueSchemaKeys) {
+  try {
     await yamlConfig.update(
-      `schemas.${oldKey}`,
-      undefined,
+      "schemas",
+      Object.fromEntries(newSchemas),
       vscode.ConfigurationTarget.Global,
     );
+  } catch (error) {
+    console.error(
+      "Failed to register Continue config.yaml schema, most likely, YAML extension is not installed",
+      error,
+    );
   }
 
-  const newPath = path.join(
-    context.extension.extensionUri.fsPath,
-    "config-yaml-schema.json",
-  );
-  await yamlConfig.update(
-    `schemas.${newPath}`,
-    [".continue/**/*.yaml"],
-    vscode.ConfigurationTarget.Global,
-  );
-
   const api = new VsCodeContinueApi(vscodeExtension);
   const continuePublicApi = {
     registerCustomContextProvider: api.registerCustomContextProvider.bind(api),