fix(nebius): normalize unsupported forced tools

Author: waleedlatif1

apps/sim/components/icons/nebius.tsx

@@ -1,6 +1,8 @@
 import type { SVGProps } from 'react'
 
-export function NebiusIcon(props: SVGProps<SVGSVGElement>) {
+interface NebiusIconProps extends SVGProps<SVGSVGElement> {}
+
+export function NebiusIcon(props: NebiusIconProps) {
   return (
     <svg {...props} viewBox='0 0 32 32' fill='none' xmlns='http://www.w3.org/2000/svg'>
       <rect width='32' height='32' fill='#052B42' rx='8' />

…sim/providers/nebius/nebius.live.test.ts …/sim/providers/nebius/index.live.test.tsapps/sim/providers/nebius/nebius.live.test.ts renamed to apps/sim/providers/nebius/index.live.test.ts apps/sim/providers/nebius/nebius.live.test.ts renamed to apps/sim/providers/nebius/index.live.test.ts

@@ -5,7 +5,7 @@ import { providersTraceEnrichmentMock } from '@sim/testing/mocks/providers-trace
 import { beforeEach, describe, expect, it, vi } from 'vitest'
 import type { StreamingExecution } from '@/executor/types'
 import { nebiusProvider } from '@/providers/nebius'
-import type { ProviderResponse } from '@/providers/types'
+import type { ProviderResponse, ProviderToolConfig } from '@/providers/types'
 
 vi.mock('@/providers', () => providersMock)
 vi.mock('@/providers/trace-enrichment', () => providersTraceEnrichmentMock)
@@ -90,6 +90,50 @@ describe('Nebius Chat Completions wire contract', () => {
     })
   })
 
+  it.each([
+    { model: 'nebius/deepseek-ai/DeepSeek-V4.1-Flash', forced: false },
+    { model: 'nebius/openbmb/MiniCPM-V-4_5', forced: false },
+    { model: 'nebius/openbmb/minicpm-v-4_5', forced: false },
+    { model: MODEL, forced: true },
+  ])('honors the forced-tool wire contract for $model', async (scenario) => {
+    const tools: ProviderToolConfig[] = [
+      {
+        id: 'http_request',
+        description: 'Fetch a URL',
+        params: {},
+        parameters: { type: 'object', properties: {}, required: [] },
+        usageControl: 'force',
+      },
+      {
+        id: 'disabled_tool',
+        description: 'Disabled tool',
+        params: {},
+        parameters: { type: 'object', properties: {}, required: [] },
+        usageControl: 'none',
+      },
+    ]
+    const payloads: Array<{
+      tool_choice: unknown
+      tools: Array<{ function: { name: string } }>
+    }> = []
+    vi.mocked(fetch).mockImplementationOnce(async (_url, init) => {
+      payloads.push(JSON.parse(String(init?.body)))
+      return jsonResponse(completion('42'))
+    })
+    const result = (await nebiusProvider.executeRequest({
+      ...REQUEST,
+      model: scenario.model,
+      tools,
+    })) as ProviderResponse
+    expect(payloads[0]?.tool_choice).toEqual(
+      scenario.forced ? { type: 'function', function: { name: 'http_request' } } : 'auto'
+    )
+    expect(payloads[0]?.tools.map((tool) => tool.function.name)).toEqual(['http_request'])
+    expect(result.content).toBe('42')
+    expect(tools[0].usageControl).toBe('force')
+    expect(tools[1].usageControl).toBe('none')
+  })
+
   it('decodes SSE reasoning and usage while retaining Nebius pricing', async () => {
     vi.mocked(fetch).mockImplementationOnce(async (_url, init) => {
       const payload = JSON.parse(String(init?.body))

apps/sim/providers/nebius/index.test.ts

@@ -6,6 +6,7 @@ import {
   getModelCapabilities,
   getProviderDefaultModel,
   getProviderModels,
+  supportsForcedToolUse,
 } from '@/providers/models'
 import {
   type ChatCompletionPayload,
@@ -58,16 +59,24 @@ export const nebiusProvider: ProviderConfig = {
       (model) => model.toLowerCase() === request.model.toLowerCase()
     )
     const requestedModel = (catalogModel ?? request.model).replace(/^nebius\//i, '')
-    return executeChatCompletionRequest(request, {
-      providerId: 'nebius',
-      providerName: 'Nebius',
-      client,
-      requestedModel,
-      reportedModel: request.model,
-      logger,
-      applyResponseFormat,
-      reasoningFields: ['reasoning_content', 'reasoning'],
-      recordPendingUsage: true,
-    })
+    const tools = supportsForcedToolUse(`nebius/${requestedModel}`)
+      ? request.tools
+      : request.tools?.map((tool) =>
+          tool.usageControl === 'force' ? { ...tool, usageControl: 'auto' as const } : tool
+        )
+    return executeChatCompletionRequest(
+      { ...request, tools },
+      {
+        providerId: 'nebius',
+        providerName: 'Nebius',
+        client,
+        requestedModel,
+        reportedModel: request.model,
+        logger,
+        applyResponseFormat,
+        reasoningFields: ['reasoning_content', 'reasoning'],
+        recordPendingUsage: true,
+      }
+    )
   },
 }