Add custom server implementation to ease the development of the tools

This commit is contained in:
Kacper Łukawski
2025-03-07 18:11:15 +01:00
parent b2c96ba7de
commit f8d3cc474b
8 changed files with 304 additions and 90 deletions

View File

@@ -7,14 +7,12 @@ from .base import EmbeddingProvider
class FastEmbedProvider(EmbeddingProvider):
"""FastEmbed implementation of the embedding provider."""
"""
FastEmbed implementation of the embedding provider.
:param model_name: The name of the FastEmbed model to use.
"""
def __init__(self, model_name: str):
"""
Initialize the FastEmbed provider.
:param model_name: The name of the FastEmbed model to use.
"""
self.model_name = model_name
self.embedding_model = TextEmbedding(model_name)