Handle parameters with environmental variables only (#24)

* Switch to Typer to read the CLI parameters and options

* Rely on environmental variables only

* Fix tests

* Update README
This commit is contained in:
Kacper Łukawski
2025-03-10 16:36:31 +01:00
committed by GitHub
parent 349abbb3ec
commit b9f773e99c
11 changed files with 84 additions and 161 deletions

View File

@@ -1,5 +0,0 @@
from .base import EmbeddingProvider
from .factory import create_embedding_provider
from .fastembed import FastEmbedProvider
__all__ = ["EmbeddingProvider", "FastEmbedProvider", "create_embedding_provider"]

View File

@@ -1,4 +1,5 @@
from mcp_server_qdrant.embeddings import EmbeddingProvider
from mcp_server_qdrant.embeddings.types import EmbeddingProviderType
from mcp_server_qdrant.settings import EmbeddingProviderSettings
@@ -8,7 +9,7 @@ def create_embedding_provider(settings: EmbeddingProviderSettings) -> EmbeddingP
:param settings: The settings for the embedding provider.
:return: An instance of the specified embedding provider.
"""
if settings.provider_type.lower() == "fastembed":
if settings.provider_type == EmbeddingProviderType.FASTEMBED:
from mcp_server_qdrant.embeddings.fastembed import FastEmbedProvider
return FastEmbedProvider(settings.model_name)

View File

@@ -0,0 +1,5 @@
from enum import Enum
class EmbeddingProviderType(Enum):
FASTEMBED = "fastembed"