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:
@@ -1,5 +0,0 @@
|
||||
from .base import EmbeddingProvider
|
||||
from .factory import create_embedding_provider
|
||||
from .fastembed import FastEmbedProvider
|
||||
|
||||
__all__ = ["EmbeddingProvider", "FastEmbedProvider", "create_embedding_provider"]
|
||||
|
||||
@@ -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)
|
||||
|
||||
5
src/mcp_server_qdrant/embeddings/types.py
Normal file
5
src/mcp_server_qdrant/embeddings/types.py
Normal file
@@ -0,0 +1,5 @@
|
||||
from enum import Enum
|
||||
|
||||
|
||||
class EmbeddingProviderType(Enum):
|
||||
FASTEMBED = "fastembed"
|
||||
Reference in New Issue
Block a user