in tablestore-python-mcp-server/src/tablestore_mcp_server/embedding/provider.py [0:0]
def create_embedding(settings: EmbeddingProviderSettings) -> BaseEmbedding:
logger.info(f"Using embedding provider {settings.provider_type} with model {settings.model_name}")
if settings.model_name is None or len(settings.model_name) == 0:
raise ValueError("`model_name` is empty")
if settings.provider_type == EmbeddingProviderType.HUGGING_FACE:
embed_model = HuggingFaceEmbedding(model_name=settings.model_name)
return embed_model
elif settings.provider_type == EmbeddingProviderType.DASH_SCOPE:
logger.info(f"api_key: {settings.dash_scope_api_key}")
embed_model = DashScopeEmbedding(model_name=settings.model_name, api_key=settings.dash_scope_api_key)
return embed_model
else:
raise ValueError(f"unsupported embedding type: {settings.provider_type}")