packages/sdk/src/embedding-service.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

import type { EmbeddingProvider } from "./embedding-provider.js";
import {
	type OpenAIEmbeddingModel,
	OpenAIEmbeddingProvider,
} from "./openai-embedding-provider.js";

export type EmbeddingModel = OpenAIEmbeddingModel;

export type EmbeddingServiceConfiguration = {
	apiKey: string;
	model?: EmbeddingModel;
};

export class EmbeddingService implements EmbeddingProvider {
	private provider: EmbeddingProvider;

	constructor(configuration: EmbeddingServiceConfiguration) {
		this.provider = new OpenAIEmbeddingProvider({
			apiKey: configuration.apiKey,
			model: configuration.model,
		});
	}

	get dimensions(): number {
		return this.provider.dimensions;
	}

	async generate(text: string): Promise<number[]> {
		return this.provider.generate(text);
	}

	async generateBatch(texts: string[]): Promise<number[][]> {
		return this.provider.generateBatch(texts);
	}
}

export function createEmbeddingService(
	provider: EmbeddingProvider,
): EmbeddingProvider {
	return provider;
}