Skip to content

Commit 2802493

Browse files
committed
fix: resolve rebase conflicts and add missing nebius provider types
1 parent 87be2d8 commit 2802493

File tree

4 files changed

+30
-3
lines changed

4 files changed

+30
-3
lines changed

packages/types/src/provider-settings.ts

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ export const providerNames = [
3030
"groq",
3131
"chutes",
3232
"litellm",
33+
"nebius",
3334
] as const
3435

3536
export const providerNamesSchema = z.enum(providerNames)
@@ -200,6 +201,12 @@ const litellmSchema = baseProviderSettingsSchema.extend({
200201
litellmModelId: z.string().optional(),
201202
})
202203

204+
const nebiusSchema = baseProviderSettingsSchema.extend({
205+
nebiusBaseUrl: z.string().optional(),
206+
nebiusApiKey: z.string().optional(),
207+
nebiusModelId: z.string().optional(),
208+
})
209+
203210
const defaultSchema = z.object({
204211
apiProvider: z.undefined(),
205212
})
@@ -226,6 +233,7 @@ export const providerSettingsSchemaDiscriminated = z.discriminatedUnion("apiProv
226233
groqSchema.merge(z.object({ apiProvider: z.literal("groq") })),
227234
chutesSchema.merge(z.object({ apiProvider: z.literal("chutes") })),
228235
litellmSchema.merge(z.object({ apiProvider: z.literal("litellm") })),
236+
nebiusSchema.merge(z.object({ apiProvider: z.literal("nebius") })),
229237
defaultSchema,
230238
])
231239

@@ -252,6 +260,7 @@ export const providerSettingsSchema = z.object({
252260
...groqSchema.shape,
253261
...chutesSchema.shape,
254262
...litellmSchema.shape,
263+
...nebiusSchema.shape,
255264
...codebaseIndexProviderSchema.shape,
256265
})
257266

@@ -353,4 +362,8 @@ export const PROVIDER_SETTINGS_KEYS = keysOf<ProviderSettings>()([
353362
"litellmBaseUrl",
354363
"litellmApiKey",
355364
"litellmModelId",
365+
// Nebius
366+
"nebiusBaseUrl",
367+
"nebiusApiKey",
368+
"nebiusModelId",
356369
])

src/api/providers/index.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,3 +20,4 @@ export { XAIHandler } from "./xai"
2020
export { GroqHandler } from "./groq"
2121
export { ChutesHandler } from "./chutes"
2222
export { LiteLLMHandler } from "./litellm"
23+
export { NebiusHandler } from "./nebius"

webview-ui/src/components/settings/ApiOptions.tsx

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -418,7 +418,11 @@ const ApiOptions = ({
418418
)}
419419

420420
{selectedProvider === "nebius" && (
421-
<Nebius apiConfiguration={apiConfiguration} setApiConfigurationField={setApiConfigurationField} />
421+
<Nebius
422+
apiConfiguration={apiConfiguration}
423+
setApiConfigurationField={setApiConfigurationField}
424+
organizationAllowList={organizationAllowList}
425+
/>
422426
)}
423427

424428
{selectedProvider === "human-relay" && (

webview-ui/src/components/settings/providers/Nebius.tsx

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,9 @@
11
import { useCallback } from "react"
22
import { VSCodeTextField } from "@vscode/webview-ui-toolkit/react"
33

4-
import { ProviderSettings, RouterModels, nebiusDefaultModelId } from "@roo/shared/api"
4+
import type { ProviderSettings, OrganizationAllowList } from "@roo-code/types"
5+
6+
import { RouterModels, nebiusDefaultModelId } from "@roo/api"
57

68
import { useAppTranslation } from "@src/i18n/TranslationContext"
79

@@ -12,9 +14,15 @@ type NebiusProps = {
1214
apiConfiguration: ProviderSettings
1315
setApiConfigurationField: (field: keyof ProviderSettings, value: ProviderSettings[keyof ProviderSettings]) => void
1416
routerModels?: RouterModels
17+
organizationAllowList: OrganizationAllowList
1518
}
1619

17-
export const Nebius = ({ apiConfiguration, setApiConfigurationField, routerModels }: NebiusProps) => {
20+
export const Nebius = ({
21+
apiConfiguration,
22+
setApiConfigurationField,
23+
routerModels,
24+
organizationAllowList,
25+
}: NebiusProps) => {
1826
const { t } = useAppTranslation()
1927

2028
const handleInputChange = useCallback(
@@ -59,6 +67,7 @@ export const Nebius = ({ apiConfiguration, setApiConfigurationField, routerModel
5967
serviceName="Nebius"
6068
serviceUrl="https://docs.nebius.ai/"
6169
setApiConfigurationField={setApiConfigurationField}
70+
organizationAllowList={organizationAllowList}
6271
/>
6372
</>
6473
)

0 commit comments

Comments
 (0)