From 213bfcf79b5bce58987459f755e00958d01464d2 Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Fri, 1 May 2026 11:33:05 +0100 Subject: [PATCH] refactor: trim embedding provider internal exports --- extensions/google/embedding-provider.ts | 12 ++++++------ extensions/mistral/embedding-provider.ts | 6 +++--- extensions/openai/embedding-provider.ts | 4 ++-- extensions/voyage/embedding-provider.ts | 4 ++-- 4 files changed, 13 insertions(+), 13 deletions(-) diff --git a/extensions/google/embedding-provider.ts b/extensions/google/embedding-provider.ts index d01eb569238..118f585266a 100644 --- a/extensions/google/embedding-provider.ts +++ b/extensions/google/embedding-provider.ts @@ -37,7 +37,7 @@ const GEMINI_MAX_INPUT_TOKENS: Record = { "gemini-embedding-2-preview": 8192, }; -export type GeminiTaskType = NonNullable; +type GeminiTaskType = NonNullable; // --- gemini-embedding-2-preview support --- @@ -49,12 +49,12 @@ export const GEMINI_EMBEDDING_2_MODELS = new Set([ const GEMINI_EMBEDDING_2_DEFAULT_DIMENSIONS = 3072; const GEMINI_EMBEDDING_2_VALID_DIMENSIONS = [768, 1536, 3072] as const; -export type GeminiTextPart = { text: string }; -export type GeminiInlinePart = { +type GeminiTextPart = { text: string }; +type GeminiInlinePart = { inlineData: { mimeType: string; data: string }; }; -export type GeminiPart = GeminiTextPart | GeminiInlinePart; -export type GeminiEmbeddingRequest = { +type GeminiPart = GeminiTextPart | GeminiInlinePart; +type GeminiEmbeddingRequest = { content: { parts: GeminiPart[] }; taskType: GeminiTaskType; outputDimensionality?: number; @@ -305,7 +305,7 @@ export async function createGeminiEmbeddingProvider( }; } -export async function resolveGeminiEmbeddingClient( +async function resolveGeminiEmbeddingClient( options: MemoryEmbeddingProviderCreateOptions, ): Promise { const remote = options.remote; diff --git a/extensions/mistral/embedding-provider.ts b/extensions/mistral/embedding-provider.ts index 4ef3c25d5a2..7635e9a64b0 100644 --- a/extensions/mistral/embedding-provider.ts +++ b/extensions/mistral/embedding-provider.ts @@ -7,7 +7,7 @@ import { } from "openclaw/plugin-sdk/memory-core-host-engine-embeddings"; import type { SsrFPolicy } from "openclaw/plugin-sdk/ssrf-runtime"; -export type MistralEmbeddingClient = { +type MistralEmbeddingClient = { baseUrl: string; headers: Record; ssrfPolicy?: SsrFPolicy; @@ -17,7 +17,7 @@ export type MistralEmbeddingClient = { export const DEFAULT_MISTRAL_EMBEDDING_MODEL = "mistral-embed"; const DEFAULT_MISTRAL_BASE_URL = "https://api.mistral.ai/v1"; -export function normalizeMistralModel(model: string): string { +function normalizeMistralModel(model: string): string { return normalizeEmbeddingModelWithPrefixes({ model, defaultModel: DEFAULT_MISTRAL_EMBEDDING_MODEL, @@ -40,7 +40,7 @@ export async function createMistralEmbeddingProvider( }; } -export async function resolveMistralEmbeddingClient( +async function resolveMistralEmbeddingClient( options: MemoryEmbeddingProviderCreateOptions, ): Promise { return await resolveRemoteEmbeddingClient({ diff --git a/extensions/openai/embedding-provider.ts b/extensions/openai/embedding-provider.ts index 0df74f9da3e..94a30383541 100644 --- a/extensions/openai/embedding-provider.ts +++ b/extensions/openai/embedding-provider.ts @@ -26,7 +26,7 @@ const OPENAI_MAX_INPUT_TOKENS: Record = { "text-embedding-ada-002": 8191, }; -export function normalizeOpenAiModel(model: string): string { +function normalizeOpenAiModel(model: string): string { const trimmed = model.trim(); if (!trimmed) { return DEFAULT_OPENAI_EMBEDDING_MODEL; @@ -82,7 +82,7 @@ export async function createOpenAiEmbeddingProvider( }; } -export async function resolveOpenAiEmbeddingClient( +async function resolveOpenAiEmbeddingClient( options: MemoryEmbeddingProviderCreateOptions, ): Promise { const client = await resolveRemoteEmbeddingClient({ diff --git a/extensions/voyage/embedding-provider.ts b/extensions/voyage/embedding-provider.ts index f4d218c80cc..67cc4b184e1 100644 --- a/extensions/voyage/embedding-provider.ts +++ b/extensions/voyage/embedding-provider.ts @@ -22,7 +22,7 @@ const VOYAGE_MAX_INPUT_TOKENS: Record = { "voyage-code-3": 32000, }; -export function normalizeVoyageModel(model: string): string { +function normalizeVoyageModel(model: string): string { return normalizeEmbeddingModelWithPrefixes({ model, defaultModel: DEFAULT_VOYAGE_EMBEDDING_MODEL, @@ -72,7 +72,7 @@ export async function createVoyageEmbeddingProvider( }; } -export async function resolveVoyageEmbeddingClient( +async function resolveVoyageEmbeddingClient( options: MemoryEmbeddingProviderCreateOptions, ): Promise { const { baseUrl, headers, ssrfPolicy } = await resolveRemoteEmbeddingBearerClient({