Function resolveProvider

apps/sim/lib/knowledge/embeddings.ts:200–257

(
  embeddingModel: string,
  workspaceId?: string | null
)

}

async function resolveProvider(
  embeddingModel: string,
  workspaceId?: string | null
): Promise<ResolvedProvider> {
  const azureApiKey = env.AZURE_OPENAI_API_KEY
  const azureEndpoint = env.AZURE_OPENAI_ENDPOINT
  const azureApiVersion = env.AZURE_OPENAI_API_VERSION
  const isOpenAIModel = SUPPORTED_EMBEDDING_MODELS[embeddingModel]?.provider === 'openai'
  /**
   * Azure deployment names default to the embedding model name when
   * `KB_OPENAI_MODEL_NAME` is unset — this matches the pre-existing
   * convention where deployments are named after the model they host.
   */
  const azureDeploymentName = env.KB_OPENAI_MODEL_NAME || embeddingModel
  const useAzure = Boolean(isOpenAIModel && azureApiKey && azureEndpoint && azureApiVersion)

  const info = getEmbeddingModelInfo(embeddingModel)

  if (useAzure) {
    return {
      modelName: azureDeploymentName,
      pricingId: info.pricingId,
      isBYOK: false,
      tokenizerProvider: info.tokenizerProvider,
      buildRequest: buildAzureOpenAIProvider(
        azureDeploymentName,
        azureApiKey!,
        azureEndpoint!,
        azureApiVersion!
      ),
    }
  }

  if (info.provider === 'openai') {
    const { apiKey, isBYOK } = await resolveOpenAIKey(workspaceId)
    return {
      modelName: embeddingModel,
      pricingId: info.pricingId,
      isBYOK,
      tokenizerProvider: info.tokenizerProvider,
      buildRequest: buildOpenAIProvider(embeddingModel, apiKey),
    }
  }

  if (info.provider === 'gemini') {
    const { apiKey, isBYOK } = await resolveGeminiKey(workspaceId)
    return {
      modelName: embeddingModel,
      pricingId: info.pricingId,
      isBYOK,
      tokenizerProvider: info.tokenizerProvider,
      maxItemsPerRequest: GEMINI_MAX_ITEMS_PER_REQUEST,
      buildRequest: buildGeminiProvider(embeddingModel, apiKey),
    }
  }

  throw new Error(`Unknown embedding provider for model ${embeddingModel}`)
}