added tools support for fireworks.ai

.github/workflows/main.yaml

@@ -208,7 +208,7 @@ jobs:
     runs-on: ubuntu-latest
     needs:
       - build
-    if: github.event.inputs.publish_build == 'true'
+    if: github.event_name != 'workflow_dispatch' || github.event.inputs.publish_build == 'true'
     permissions:
       contents: write
     steps:

.github/workflows/preview.yaml

@@ -204,7 +204,7 @@ jobs:
     runs-on: ubuntu-latest
     needs:
       - build
-    if: github.event.inputs.publish_build == 'true'
+    if: github.event_name != 'workflow_dispatch' || github.event.inputs.publish_build == 'true'
     permissions:
       contents: write
     steps:

core/config/ConfigHandler.ts

@@ -27,7 +27,7 @@ import { localPathToUri } from "../util/pathToUri.js";
 
 import {
   LOCAL_ONBOARDING_CHAT_MODEL,
-  ONBOARDING_LOCAL_MODEL_TITLE,
+  LOCAL_ONBOARDING_PROVIDER_TITLE,
 } from "./onboarding.js";
 import ControlPlaneProfileLoader from "./profile/ControlPlaneProfileLoader.js";
 import LocalProfileLoader from "./profile/LocalProfileLoader.js";
@@ -401,7 +401,7 @@ export class ConfigHandler {
     const { config } = await this.loadConfig();
     const model = config?.models.find((m) => m.title === title);
     if (!model) {
-      if (title === ONBOARDING_LOCAL_MODEL_TITLE) {
+      if (title === LOCAL_ONBOARDING_PROVIDER_TITLE) {
         // Special case, make calls to Ollama before we have it in the config
         const ollama = new Ollama({
           model: LOCAL_ONBOARDING_CHAT_MODEL,

core/config/ProfileLifecycleManager.ts

@@ -27,6 +27,7 @@ export interface OrganizationDescription {
   id: string;
   iconUrl: string;
   name: string;
+  slug: string | undefined; // TODO: This doesn't need to be undefined, just doing while transitioning the backend
 }
 
 export class ProfileLifecycleManager {

core/config/load.ts

@@ -63,15 +63,15 @@ import {
 } from "../util/paths";
 
 import { ConfigResult, ConfigValidationError } from "@continuedev/config-yaml";
+import { useHub } from "../control-plane/env";
+import { localPathToUri } from "../util/pathToUri";
 import {
   defaultContextProvidersJetBrains,
   defaultContextProvidersVsCode,
   defaultSlashCommandsJetBrains,
   defaultSlashCommandsVscode,
 } from "./default";
 import { getSystemPromptDotFile } from "./getSystemPromptDotFile";
-import { useHub } from "../control-plane/env";
-import { localPathToUri } from "../util/pathToUri";
 import { modifyContinueConfigWithSharedConfig } from "./sharedConfig";
 import { validateConfig } from "./validation.js";
 
@@ -517,6 +517,8 @@ async function intermediateToFinalConfig(
               continueConfig,
               mcpId,
               abortController.signal,
+              "MCP Server",
+              server.faviconUrl,
             );
             if (mcpError) {
               errors.push(mcpError);

core/config/migrateSharedConfig.ts

@@ -1,13 +1,9 @@
-import * as fs from "fs";
-
-import * as JSONC from "comment-json";
-
-import { IDE, SerializedContinueConfig } from "..";
-import { SharedConfigSchema } from "./sharedConfig";
+import { IDE } from "..";
+import { deduplicateArray } from "../util";
 import { GlobalContext } from "../util/GlobalContext";
 import { editConfigJson } from "../util/paths";
 import { resolveSerializedConfig } from "./load";
-import { deduplicateArray } from "../util";
+import { SharedConfigSchema } from "./sharedConfig";
 
 /*
   This migration function eliminates deprecated values from the json file
@@ -201,10 +197,10 @@ export function migrateJsonSharedConfig(filepath: string, ide: IDE): void {
     if (effected) {
       new GlobalContext().updateSharedConfig(shareConfigUpdates);
       editConfigJson(() => config);
-      void ide.showToast(
-        "warning",
-        "Migrated deprecated Continue JSON settings. Edit in the Settings Page",
-      );
+      // void ide.showToast(
+      //   "warning",
+      //   "Migrated deprecated Continue JSON settings. Edit in the Settings Page",
+      // );
     }
   } catch (e) {
     throw new Error(`Migration: Failed to parse config.json: ${e}`);

core/config/onboarding.ts

@@ -3,10 +3,12 @@ import { SerializedContinueConfig } from "../";
 import { FREE_TRIAL_MODELS } from "./default";
 
 export const TRIAL_FIM_MODEL = "codestral-latest";
-export const ONBOARDING_LOCAL_MODEL_TITLE = "Ollama";
+export const LOCAL_ONBOARDING_PROVIDER_TITLE = "Ollama";
 export const LOCAL_ONBOARDING_FIM_MODEL = "qwen2.5-coder:1.5b-base";
+export const LOCAL_ONBOARDING_FIM_TITLE = "Qwen2.5-Coder 1.5B";
 export const LOCAL_ONBOARDING_CHAT_MODEL = "llama3.1:8b";
 export const LOCAL_ONBOARDING_CHAT_TITLE = "Llama 3.1 8B";
+export const LOCAL_ONBOARDING_EMBEDDINGS_MODEL = "nomic-embed-text:latest";
 
 /**
  * We set the "best" chat + autocopmlete models by default
@@ -35,13 +37,13 @@ export function setupLocalConfig(
       ...config.models.filter((model) => model.provider !== "free-trial"),
     ],
     tabAutocompleteModel: {
-      title: "Qwen2.5-Coder 1.5B",
+      title: LOCAL_ONBOARDING_FIM_TITLE,
       provider: "ollama",
       model: LOCAL_ONBOARDING_FIM_MODEL,
     },
     embeddingsProvider: {
       provider: "ollama",
-      model: "nomic-embed-text",
+      model: LOCAL_ONBOARDING_EMBEDDINGS_MODEL,
     },
   };
 }
@@ -68,24 +70,3 @@ export function setupQuickstartConfig(
     },
   };
 }
-
-export function setupLocalConfigAfterFreeTrial(
-  config: SerializedContinueConfig,
-): SerializedContinueConfig {
-  return {
-    ...config,
-    models: [
-      {
-        title: LOCAL_ONBOARDING_CHAT_TITLE,
-        provider: "ollama",
-        model: LOCAL_ONBOARDING_CHAT_MODEL,
-      },
-      {
-        title: ONBOARDING_LOCAL_MODEL_TITLE,
-        provider: "ollama",
-        model: "AUTODETECT",
-      },
-      ...config.models.filter((model) => model.provider !== "free-trial"),
-    ],
-  };
-}

core/config/yaml/loadYaml.ts

@@ -3,6 +3,7 @@ import fs from "node:fs";
 import {
   AssistantUnrolled,
   ConfigResult,
+  ConfigValidationError,
   parseAssistantUnrolled,
   validateConfigYaml,
 } from "@continuedev/config-yaml";
@@ -34,6 +35,7 @@ import { getSystemPromptDotFile } from "../getSystemPromptDotFile";
 import { PlatformConfigMetadata } from "../profile/PlatformProfileLoader";
 
 import { slashFromCustomCommand } from "../../commands";
+import DocsContextProvider from "../../context/providers/DocsContextProvider";
 import { allTools } from "../../tools";
 import { GlobalContext } from "../../util/GlobalContext";
 import { modifyContinueConfigWithSharedConfig } from "../sharedConfig";
@@ -100,7 +102,8 @@ async function configYamlToContinueConfig(
   workOsAccessToken: string | undefined,
   platformConfigMetadata: PlatformConfigMetadata | undefined,
   allowFreeTrial: boolean = true,
-): Promise<ContinueConfig> {
+): Promise<{ config: ContinueConfig; errors: ConfigValidationError[] }> {
+  const errors: ConfigValidationError[] = [];
   const continueConfig: ContinueConfig = {
     slashCommands: [
       ...(await slashCommandsFromV1PromptFiles(ide)),
@@ -214,6 +217,15 @@ async function configYamlToContinueConfig(
     .filter((p) => !!p) ?? []) as IContextProvider[];
   continueConfig.contextProviders.push(...DEFAULT_CONTEXT_PROVIDERS);
 
+  if (
+    continueConfig.docs?.length &&
+    !continueConfig.contextProviders?.some(
+      (cp) => cp.description.title === "docs",
+    )
+  ) {
+    continueConfig.contextProviders.push(new DocsContextProvider({}));
+  }
+
   // Embeddings Provider
   const embedConfig = config.models?.find((model) =>
     model.roles?.includes("embed"),
@@ -259,40 +271,51 @@ async function configYamlToContinueConfig(
 
   // Apply MCP if specified
   const mcpManager = MCPManagerSingleton.getInstance();
-  config.mcpServers?.forEach(async (server) => {
-    const mcpId = server.name;
-    const mcpConnection = mcpManager.createConnection(mcpId, {
-      transport: {
-        type: "stdio",
-        args: [],
-        ...server,
-      },
-    });
-    if (!mcpConnection) {
-      return;
-    }
-
-    const abortController = new AbortController();
-    const mcpConnectionTimeout = setTimeout(
-      () => abortController.abort(),
-      2000,
-    );
+  await Promise.all(
+    config.mcpServers?.map(async (server) => {
+      const mcpId = server.name;
+      const mcpConnection = mcpManager.createConnection(mcpId, {
+        transport: {
+          type: "stdio",
+          args: [],
+          ...server,
+        },
+      });
+      if (!mcpConnection) {
+        return;
+      }
 
-    try {
-      await mcpConnection.modifyConfig(
-        continueConfig,
-        mcpId,
-        abortController.signal,
+      const abortController = new AbortController();
+      const mcpConnectionTimeout = setTimeout(
+        () => abortController.abort(),
+        5000,
       );
-    } catch (e: any) {
-      if (e.name !== "AbortError") {
-        throw e;
+
+      try {
+        const mcpError = await mcpConnection.modifyConfig(
+          continueConfig,
+          mcpId,
+          abortController.signal,
+          server.name,
+          server.faviconUrl,
+        );
+        if (mcpError) {
+          errors.push(mcpError);
+        }
+      } catch (e: any) {
+        errors.push({
+          fatal: false,
+          message: `Failed to load MCP server: ${e.message}`,
+        });
+        if (e.name !== "AbortError") {
+          throw e;
+        }
       }
-    }
-    clearTimeout(mcpConnectionTimeout);
-  });
+      clearTimeout(mcpConnectionTimeout);
+    }) ?? [],
+  );
 
-  return continueConfig;
+  return { config: continueConfig, errors };
 }
 
 export async function loadContinueConfigFromYaml(
@@ -328,7 +351,7 @@ export async function loadContinueConfigFromYaml(
     };
   }
 
-  const continueConfig = await configYamlToContinueConfig(
+  const { config: continueConfig, errors } = await configYamlToContinueConfig(
     configYamlResult.config,
     ide,
     ideSettings,
@@ -357,7 +380,7 @@ export async function loadContinueConfigFromYaml(
 
   return {
     config: withShared,
-    errors: configYamlResult.errors,
+    errors: [...(configYamlResult.errors ?? []), ...errors],
     configLoadInterrupted: false,
   };
 }