Skip to content

Commit 1f58d87

Browse files
committed
fix: update nebius test to use LITELLM_COMPUTER_USE_MODELS
1 parent 2802493 commit 1f58d87

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/api/providers/fetchers/__tests__/nebius.test.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import axios from "axios"
44
import { getNebiusModels } from "../nebius"
5-
import { COMPUTER_USE_MODELS } from "../../../../shared/api"
5+
import { LITELLM_COMPUTER_USE_MODELS } from "../../../../shared/api"
66

77
jest.mock("axios")
88

@@ -129,9 +129,9 @@ describe("Nebius API", () => {
129129
expect(computerUseModels).toContain("anthropic/claude-3.5-sonnet")
130130
expect(computerUseModels).toContain("anthropic/claude-3.7-sonnet")
131131

132-
// Verify these models are in the COMPUTER_USE_MODELS set
132+
// Verify these models are in the LITELLM_COMPUTER_USE_MODELS set
133133
computerUseModels.forEach((modelId) => {
134-
expect(COMPUTER_USE_MODELS.has(modelId)).toBe(true)
134+
expect(LITELLM_COMPUTER_USE_MODELS.has(modelId)).toBe(true)
135135
})
136136
})
137137

0 commit comments

Comments
 (0)