better-qdrant-mcp-server
Version:
MCP server for enhanced Qdrant vector database functionality
34 lines • 1.21 kB
JavaScript
import { BaseEmbeddingService } from './base.js';
export class FastEmbedService extends BaseEmbeddingService {
constructor(model) {
super(undefined, undefined, model || 'BAAI/bge-small-en');
// FastEmbed models typically produce 384-dimensional embeddings
this.vectorSize = 384;
this.defaultModel = 'BAAI/bge-small-en';
this.embedder = null;
}
async initializeEmbedder() {
if (!this.embedder) {
// Dynamic import to handle CommonJS module
const fastembed = await import('fastembed');
this.embedder = new fastembed.FastEmbed({
model: this.model || this.defaultModel
});
}
}
async generateEmbeddings(texts) {
await this.initializeEmbedder();
if (!this.embedder) {
throw new Error('FastEmbed embedder not initialized');
}
const embeddings = await this.embedder.embed(texts);
return embeddings.map((embedding) => Array.from(embedding));
}
requiresApiKey() {
return false;
}
validateConfig() {
// No validation needed as FastEmbed runs locally
}
}
//# sourceMappingURL=fastembed.js.map