diff --git a/changelog.d/fixes/12784-arcee-ai-provider-registry.md b/changelog.d/fixes/12784-arcee-ai-provider-registry.md new file mode 100644 index 0000000000..f211b561b8 --- /dev/null +++ b/changelog.d/fixes/12784-arcee-ai-provider-registry.md @@ -0,0 +1 @@ +- fix(sse): register Arcee AI in the executor provider registry so requests reach api.arcee.ai instead of silently falling back to OpenAI (#12784) diff --git a/open-sse/config/providers/index.ts b/open-sse/config/providers/index.ts index 8c670411b1..fa9ba3b069 100644 --- a/open-sse/config/providers/index.ts +++ b/open-sse/config/providers/index.ts @@ -136,6 +136,7 @@ import { freemodel_devProvider } from "./registry/freemodel-dev/index.ts"; import { gitlawb_gmiProvider } from "./registry/gitlawb/gmi/index.ts"; import { gitlawbProvider } from "./registry/gitlawb/index.ts"; import { liquidProvider } from "./registry/liquid/index.ts"; +import { arceeAiProvider } from "./registry/arcee-ai/index.ts"; import { deepinfraProvider } from "./registry/deepinfra/index.ts"; import { agyProvider } from "./registry/agy/index.ts"; import { agnesProvider } from "./registry/agnes/index.ts"; @@ -409,6 +410,7 @@ export const REGISTRY: Record = { "gitlawb-gmi": gitlawb_gmiProvider, gitlawb: gitlawbProvider, liquid: liquidProvider, + "arcee-ai": arceeAiProvider, deepinfra: deepinfraProvider, agy: agyProvider, agnes: agnesProvider, diff --git a/open-sse/config/providers/registry/arcee-ai/index.ts b/open-sse/config/providers/registry/arcee-ai/index.ts new file mode 100644 index 0000000000..e75ebe0957 --- /dev/null +++ b/open-sse/config/providers/registry/arcee-ai/index.ts @@ -0,0 +1,10 @@ +import type { RegistryEntry } from "../../shared.ts"; +import { buildOpenAiCompatibleRegistryEntry } from "../../shared.ts"; + +export const arceeAiProvider: RegistryEntry = buildOpenAiCompatibleRegistryEntry({ + id: "arcee-ai", + alias: "arcee", + baseUrl: "https://api.arcee.ai/api/v1/chat/completions", + models: [], + passthroughModels: true, +}); diff --git a/tests/unit/arcee-ai-provider.test.ts b/tests/unit/arcee-ai-provider.test.ts new file mode 100644 index 0000000000..579c7d4fc8 --- /dev/null +++ b/tests/unit/arcee-ai-provider.test.ts @@ -0,0 +1,44 @@ +import test from "node:test"; +import assert from "node:assert/strict"; +import fs from "node:fs"; +import os from "node:os"; +import path from "node:path"; + +const TEST_DATA_DIR = fs.mkdtempSync(path.join(os.tmpdir(), "omniroute-arcee-provider-")); +process.env.DATA_DIR = TEST_DATA_DIR; + +const { PROVIDERS } = await import("../../open-sse/config/constants.ts"); +const { REGISTRY: providerRegistry } = await import("../../open-sse/config/providerRegistry.ts"); +const { APIKEY_PROVIDERS } = await import("../../src/shared/constants/providers.ts"); +const { DefaultExecutor } = await import("../../open-sse/executors/default.ts"); +const dbCore = await import("../../src/lib/db/core.ts"); + +test.after(() => { + dbCore.closeDbInstance(); + fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true, maxRetries: 5, retryDelay: 100 }); +}); + +const ARCEE_CHAT_URL = "https://api.arcee.ai/api/v1/chat/completions"; + +test("arcee-ai is offered in the onboarding catalog", () => { + assert.ok(APIKEY_PROVIDERS["arcee-ai"]); +}); + +test("arcee-ai has a routing entry in the executor REGISTRY", () => { + const entry = providerRegistry["arcee-ai"]; + assert.ok(entry, "providerRegistry['arcee-ai'] must be defined"); + assert.equal(entry.id, "arcee-ai"); + assert.equal(entry.alias, "arcee"); + assert.equal(entry.format, "openai"); + assert.equal(entry.executor, "default"); + assert.equal(entry.baseUrl, ARCEE_CHAT_URL); + assert.equal(entry.authType, "apikey"); + assert.equal(entry.authHeader, "bearer"); + assert.equal(entry.passthroughModels, true); +}); + +test("DefaultExecutor routes arcee-ai to Arcee's own base URL, not OpenAI's", () => { + const executor = new DefaultExecutor("arcee-ai"); + assert.equal(executor.config.baseUrl, ARCEE_CHAT_URL); + assert.notEqual(executor.config.baseUrl, PROVIDERS.openai.baseUrl); +});