52 lines
1.5 KiB
TypeScript
52 lines
1.5 KiB
TypeScript
import type { SsrFPolicy } from "../infra/net/ssrf.js";
|
|
import { normalizeEmbeddingModelWithPrefixes } from "./embeddings-model-normalize.js";
|
|
import {
|
|
createRemoteEmbeddingProvider,
|
|
resolveRemoteEmbeddingClient,
|
|
} from "./embeddings-remote-provider.js";
|
|
import type { EmbeddingProvider, EmbeddingProviderOptions } from "./embeddings.js";
|
|
|
|
export type MistralEmbeddingClient = {
|
|
baseUrl: string;
|
|
headers: Record<string, string>;
|
|
ssrfPolicy?: SsrFPolicy;
|
|
model: string;
|
|
};
|
|
|
|
export const DEFAULT_MISTRAL_EMBEDDING_MODEL = "mistral-embed";
|
|
const DEFAULT_MISTRAL_BASE_URL = "https://api.mistral.ai/v1";
|
|
|
|
export function normalizeMistralModel(model: string): string {
|
|
return normalizeEmbeddingModelWithPrefixes({
|
|
model,
|
|
defaultModel: DEFAULT_MISTRAL_EMBEDDING_MODEL,
|
|
prefixes: ["mistral/"],
|
|
});
|
|
}
|
|
|
|
export async function createMistralEmbeddingProvider(
|
|
options: EmbeddingProviderOptions,
|
|
): Promise<{ provider: EmbeddingProvider; client: MistralEmbeddingClient }> {
|
|
const client = await resolveMistralEmbeddingClient(options);
|
|
|
|
return {
|
|
provider: createRemoteEmbeddingProvider({
|
|
id: "mistral",
|
|
client,
|
|
errorPrefix: "mistral embeddings failed",
|
|
}),
|
|
client,
|
|
};
|
|
}
|
|
|
|
export async function resolveMistralEmbeddingClient(
|
|
options: EmbeddingProviderOptions,
|
|
): Promise<MistralEmbeddingClient> {
|
|
return await resolveRemoteEmbeddingClient({
|
|
provider: "mistral",
|
|
options,
|
|
defaultBaseUrl: DEFAULT_MISTRAL_BASE_URL,
|
|
normalizeModel: normalizeMistralModel,
|
|
});
|
|
}
|