Skip to content

Commit ab0d7c2

Browse files
committed
test(api): add cognima provider expectations to requestRouterModels test
1 parent ad187f7 commit ab0d7c2

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/core/webview/__tests__/webviewMessageHandler.spec.ts

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -221,7 +221,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
221221
expect(mockGetModels).toHaveBeenCalledWith({ provider: "openrouter" })
222222
expect(mockGetModels).toHaveBeenCalledWith({ provider: "requesty", apiKey: "requesty-key" })
223223
expect(mockGetModels).toHaveBeenCalledWith({ provider: "glama" })
224-
expect(mockGetModels).toHaveBeenCalledWith({ provider: "unbound", apiKey: "unbound-key" })
224+
expect(mockGetModels).toHaveBeenCalledWith({ provider: "unbound",apiKey: "unbound-key" })
225225
expect(mockGetModels).toHaveBeenCalledWith({ provider: "vercel-ai-gateway" })
226226
expect(mockGetModels).toHaveBeenCalledWith({ provider: "deepinfra" })
227227
expect(mockGetModels).toHaveBeenCalledWith(
@@ -230,6 +230,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
230230
baseUrl: expect.any(String),
231231
}),
232232
)
233+
expect(mockGetModels).toHaveBeenCalledWith({ provider: "cognima", apiKey: undefined })
233234
expect(mockGetModels).toHaveBeenCalledWith({
234235
provider: "litellm",
235236
apiKey: "litellm-key",
@@ -249,6 +250,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
249250
unbound: mockModels,
250251
litellm: mockModels,
251252
roo: mockModels,
253+
cognima: {},
252254
ollama: {},
253255
lmstudio: {},
254256
"vercel-ai-gateway": mockModels,
@@ -340,6 +342,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
340342
glama: mockModels,
341343
unbound: mockModels,
342344
roo: mockModels,
345+
cognima: {},
343346
litellm: {},
344347
ollama: {},
345348
lmstudio: {},
@@ -385,6 +388,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
385388
glama: mockModels,
386389
unbound: {},
387390
roo: mockModels,
391+
cognima: {},
388392
litellm: {},
389393
ollama: {},
390394
lmstudio: {},

0 commit comments

Comments
 (0)