@codai/cbd
Version:
Codai Better Database - High-Performance Vector Memory System with HPKV-inspired architecture and MCP server
40 lines • 1.51 kB
TypeScript
/**
* CBD Embedding Service
* Supports multiple embedding models (OpenAI, local, custom)
*/
import { EmbeddingModel } from '../types/index.js';
export declare class OpenAIEmbeddingModel implements EmbeddingModel {
readonly name = "openai";
readonly dimensions = 1536;
private apiKey;
private modelName;
constructor(apiKey: string, modelName?: string);
generateEmbedding(text: string): Promise<Float32Array>;
generateBatchEmbeddings(texts: string[]): Promise<Float32Array[]>;
}
export declare class LocalEmbeddingModel implements EmbeddingModel {
readonly name = "local";
readonly dimensions = 384;
private pipeline;
private modelName;
private initialized;
constructor(modelName?: string);
initialize(): Promise<void>;
generateEmbedding(text: string): Promise<Float32Array>;
generateBatchEmbeddings(texts: string[]): Promise<Float32Array[]>;
}
export declare class EmbeddingService {
private model;
constructor(model: EmbeddingModel);
generateEmbedding(text: string): Promise<Float32Array>;
generateBatchEmbeddings(texts: string[]): Promise<Float32Array[]>;
generateConversationEmbedding(userRequest: string, assistantResponse: string): Promise<Float32Array>;
getDimensions(): number;
getModelName(): string;
static createFromConfig(config: {
type: 'openai' | 'local' | 'custom';
apiKey?: string;
modelName?: string;
}): EmbeddingService;
}
//# sourceMappingURL=EmbeddingService.d.ts.map