vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-06 05:10:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: trim embedding provider internal exports

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-01 11:33:05 +01:00
parent d4645373e7
commit 213bfcf79b
4 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
extensions/google/embedding-provider.ts
View File

@@ -37,7 +37,7 @@ const GEMINI_MAX_INPUT_TOKENS: Record<string, number> = {
"gemini-embedding-2-preview": 8192,
};
export type GeminiTaskType = NonNullable<MemoryEmbeddingProviderCreateOptions["taskType"]>;
type GeminiTaskType = NonNullable<MemoryEmbeddingProviderCreateOptions["taskType"]>;
// --- 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<GeminiEmbeddingClient> {
const remote = options.remote;

6
extensions/mistral/embedding-provider.ts
View File

@@ -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<string, string>;
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<MistralEmbeddingClient> {
return await resolveRemoteEmbeddingClient({

4
extensions/openai/embedding-provider.ts
View File

@@ -26,7 +26,7 @@ const OPENAI_MAX_INPUT_TOKENS: Record<string, number> = {
"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<OpenAiEmbeddingClient> {
const client = await resolveRemoteEmbeddingClient({

4
extensions/voyage/embedding-provider.ts
View File

@@ -22,7 +22,7 @@ const VOYAGE_MAX_INPUT_TOKENS: Record<string, number> = {
"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<VoyageEmbeddingClient> {
const { baseUrl, headers, ssrfPolicy } = await resolveRemoteEmbeddingBearerClient({