fix: rename huggingface TEI class

Author: liwenshipro

modelcache/embedding/__init__.py

@@ -6,7 +6,7 @@
 fasttext = LazyImport("fasttext", globals(), "modelcache.embedding.fasttext")
 paddlenlp = LazyImport("paddlenlp", globals(), "modelcache.embedding.paddlenlp")
 timm = LazyImport("timm", globals(), "modelcache.embedding.timm")
-text_embeddings_inference = LazyImport("text_embeddings_inference", globals(), "modelcache.embedding.text_embeddings_inference")
+huggingface_tei = LazyImport("huggingface_tei", globals(), "modelcache.embedding.huggingface_tei")
 
 
 def Huggingface(model="sentence-transformers/all-mpnet-base-v2"):
@@ -33,4 +33,4 @@ def Timm(model="resnet50", device="default"):
     return timm.Timm(model, device)
 
 def TextEmbeddingsInference(base_url, model):
-    return text_embeddings_inference.TextEmbeddingsInference(base_url, model)
+    return huggingface_tei.HuggingfaceTEI(base_url, model)

modelcache/embedding/text_embeddings_inference.py renamed to modelcache/embedding/huggingface_tei.py

@@ -3,7 +3,7 @@
 import numpy as np
 from modelcache.embedding.base import BaseEmbedding
 
-class TextEmbeddingsInference(BaseEmbedding):
+class HuggingfaceTEI(BaseEmbedding):
     def __init__(self, base_url: str, model: str):
         self.base_url = base_url
         self.model = model
@@ -12,6 +12,7 @@ def __init__(self, base_url: str, model: str):
             'Content-Type': 'application/json',
         }
         self.__dimension = self.to_embeddings('test').shape[0]
+
     def to_embeddings(self, data, **_):
         json_data = {
             'input': data,