simstudioai--sim/apps/sim/app/workspace/[workspaceId]/providers/provider-models-loader.tsx

'use client'

import { useEffect } from 'react'
import { createLogger } from '@sim/logger'
import { useParams } from 'next/navigation'
import { useProviderModels } from '@/hooks/queries/providers'
import {
  updateBasetenProviderModels,
  updateFireworksProviderModels,
  updateLiteLLMProviderModels,
  updateOllamaCloudProviderModels,
  updateOllamaProviderModels,
  updateOpenRouterProviderModels,
  updateTogetherProviderModels,
  updateVLLMProviderModels,
} from '@/providers/utils'
import { type ProviderName, useProvidersStore } from '@/stores/providers'

const logger = createLogger('ProviderModelsLoader')

function useSyncProvider(provider: ProviderName, workspaceId?: string) {
  const setProviderModels = useProvidersStore((state) => state.setProviderModels)
  const setProviderLoading = useProvidersStore((state) => state.setProviderLoading)
  const setOpenRouterModelInfo = useProvidersStore((state) => state.setOpenRouterModelInfo)
  const { data, isLoading, isFetching, error } = useProviderModels(provider, workspaceId)

  useEffect(() => {
    setProviderLoading(provider, isLoading || isFetching)
  }, [provider, isLoading, isFetching, setProviderLoading])

  useEffect(() => {
    if (!data) return

    try {
      if (provider === 'ollama') {
        updateOllamaProviderModels(data.models)
      } else if (provider === 'ollama-cloud') {
        void updateOllamaCloudProviderModels(data.models)
      } else if (provider === 'vllm') {
        updateVLLMProviderModels(data.models)
      } else if (provider === 'litellm') {
        updateLiteLLMProviderModels(data.models)
      } else if (provider === 'openrouter') {
        void updateOpenRouterProviderModels(data.models)
        if (data.modelInfo) {
          setOpenRouterModelInfo(data.modelInfo)
        }
      } else if (provider === 'fireworks') {
        void updateFireworksProviderModels(data.models)
      } else if (provider === 'together') {
        void updateTogetherProviderModels(data.models)
      } else if (provider === 'baseten') {
        void updateBasetenProviderModels(data.models)
      }
    } catch (syncError) {
      logger.warn(`Failed to sync provider definitions for ${provider}`, syncError as Error)
    }

    setProviderModels(provider, data.models)
  }, [provider, data, setProviderModels, setOpenRouterModelInfo])

  useEffect(() => {
    if (error) {
      logger.error(`Failed to load ${provider} models`, error)
    }
  }, [provider, error])
}

export function ProviderModelsLoader() {
  const params = useParams()
  const workspaceId = params?.workspaceId as string | undefined

  useSyncProvider('base')
  useSyncProvider('ollama')
  useSyncProvider('ollama-cloud', workspaceId)
  useSyncProvider('vllm')
  useSyncProvider('litellm')
  useSyncProvider('openrouter')
  useSyncProvider('fireworks', workspaceId)
  useSyncProvider('together', workspaceId)
  useSyncProvider('baseten', workspaceId)
  return null
}