chore: import upstream snapshot with attribution

apps/sim/tools/falai-hosting.test.ts

@@ -0,0 +1,108 @@
+/**
+ * @vitest-environment node
+ */
+import { afterEach, describe, expect, it, vi } from 'vitest'
+import {
+  FALAI_HOSTED_KEY_MARKUP_MULTIPLIER,
+  FALAI_IMAGE_FALLBACK_PROVIDER_COST_DOLLARS,
+  getFalAICostMetadata,
+} from '@/lib/tools/falai-pricing'
+import { imageGenerateTool } from '@/tools/image/generate'
+import { falaiVideoTool } from '@/tools/video/falai'
+
+afterEach(() => {
+  vi.useRealTimers()
+  vi.unstubAllGlobals()
+})
+
+describe('Fal.ai hosted key pricing', () => {
+  it('applies hosted markup to image generation provider cost', () => {
+    const pricing = imageGenerateTool.hosting?.pricing
+    expect(pricing?.type).toBe('custom')
+    if (!pricing || pricing.type !== 'custom') throw new Error('Expected custom pricing')
+
+    const result = pricing.getCost(
+      {},
+      {
+        __falaiCostDollars: 0.1,
+        __falaiBilling: {
+          source: 'billing_events',
+          endpointId: 'fal-ai/nano-banana-2',
+        },
+      }
+    )
+
+    expect(typeof result).toBe('object')
+    if (typeof result === 'number') throw new Error('Expected structured pricing result')
+    expect(result.cost).toBeCloseTo(0.1 * FALAI_HOSTED_KEY_MARKUP_MULTIPLIER)
+    expect(result.metadata).toMatchObject({
+      providerCostDollars: 0.1,
+      markupMultiplier: FALAI_HOSTED_KEY_MARKUP_MULTIPLIER,
+      source: 'billing_events',
+    })
+  })
+
+  it('applies hosted markup to video generation provider cost', () => {
+    const pricing = falaiVideoTool.hosting?.pricing
+    expect(pricing?.type).toBe('custom')
+    if (!pricing || pricing.type !== 'custom') throw new Error('Expected custom pricing')
+
+    const result = pricing.getCost(
+      {},
+      {
+        __falaiCostDollars: 0.4,
+        __falaiBilling: {
+          source: 'billing_events',
+          endpointId: 'fal-ai/veo3.1',
+        },
+      }
+    )
+
+    expect(typeof result).toBe('object')
+    if (typeof result === 'number') throw new Error('Expected structured pricing result')
+    expect(result.cost).toBeCloseTo(0.4 * FALAI_HOSTED_KEY_MARKUP_MULTIPLIER)
+    expect(result.metadata).toMatchObject({
+      providerCostDollars: 0.4,
+      markupMultiplier: FALAI_HOSTED_KEY_MARKUP_MULTIPLIER,
+      source: 'billing_events',
+    })
+  })
+
+  it('returns fallback floor cost metadata instead of throwing when billing and estimate fail', async () => {
+    vi.useFakeTimers()
+
+    const mockFetch = vi
+      .fn()
+      .mockResolvedValueOnce(
+        new Response(JSON.stringify({ billing_events: [] }), {
+          status: 200,
+          headers: { 'Content-Type': 'application/json' },
+        })
+      )
+      .mockResolvedValueOnce(
+        new Response(JSON.stringify({ billing_events: [] }), {
+          status: 200,
+          headers: { 'Content-Type': 'application/json' },
+        })
+      )
+      .mockResolvedValueOnce(new Response('pricing unavailable', { status: 500 }))
+    vi.stubGlobal('fetch', mockFetch)
+
+    const resultPromise = getFalAICostMetadata({
+      apiKey: 'fal-key',
+      endpointId: 'fal-ai/nano-banana-2',
+      requestId: 'request-1',
+    })
+
+    await vi.advanceTimersByTimeAsync(500)
+    const result = await resultPromise
+
+    expect(result).toMatchObject({
+      endpointId: 'fal-ai/nano-banana-2',
+      requestId: 'request-1',
+      costDollars: FALAI_IMAGE_FALLBACK_PROVIDER_COST_DOLLARS,
+      source: 'fallback_floor',
+      currency: 'USD',
+    })
+  })
+})