openclaw/extensions/deepinfra/embedding-adapter.ts

// Deepinfra plugin module adapts its text embedding runtime to the generic provider contract.
import type { EmbeddingProviderAdapter } from "openclaw/plugin-sdk/embedding-providers";
import {
  embeddingProviderOwnsDestination,
  sanitizeEmbeddingCacheHeaders,
} from "openclaw/plugin-sdk/memory-core-host-engine-embeddings";
import {
  createDeepInfraEmbeddingProvider,
  DEFAULT_DEEPINFRA_EMBEDDING_MODEL,
} from "./embedding-provider.js";
import { DEEPINFRA_BASE_URL, type DeepInfraSurfaceModel } from "./provider-models.js";

const EXCLUDED_EMBEDDING_HEADERS = ["authorization", "content-type", "x-api-key", "api-key"];

// First entry of embedModels becomes the default embedding model.
export function buildDeepInfraEmbeddingAdapter(options?: {
  embedModels?: readonly DeepInfraSurfaceModel[];
}): EmbeddingProviderAdapter {
  const defaultModel = options?.embedModels?.[0]?.id ?? DEFAULT_DEEPINFRA_EMBEDDING_MODEL;
  return {
    id: "deepinfra",
    defaultModel,
    transport: "remote",
    authProviderId: "deepinfra",
    create: async (createOptions) => {
      const { provider, client } = await createDeepInfraEmbeddingProvider({
        ...createOptions,
        provider: "deepinfra",
        defaultModel,
      });
      const headers = sanitizeEmbeddingCacheHeaders(client.headers, EXCLUDED_EMBEDDING_HEADERS);
      const usesDefaultIdentity =
        headers.length === 0 &&
        embeddingProviderOwnsDestination({
          baseUrl: client.baseUrl,
          providerBaseUrl: DEEPINFRA_BASE_URL,
        });
      return {
        provider,
        runtime: {
          id: "deepinfra",
          cacheKeyData: {
            provider: "deepinfra",
            model: client.model,
            ...(usesDefaultIdentity ? {} : { baseUrl: client.baseUrl, headers }),
          },
        },
      };
    },
  };
}

export const deepinfraEmbeddingProviderAdapter: EmbeddingProviderAdapter =
  buildDeepInfraEmbeddingAdapter();