【Dify】 vanna 的依赖包安装，完善

api/extensions/utils/vanna_text2sql.py

@@ -24,7 +24,7 @@ from pymilvus.model.base import BaseEmbeddingFunction
 class CustomEmbeddingFunction(BaseEmbeddingFunction):
     def __init__(self, config=None):
         model_host = config['host'] if "host" in config else 'http://wsd.wisdomidata.com:19042'
-        self.embed_model = config['embed_model'] if "embed_model" in config else 'bge-m3'
+        self.embed_model = config['embed_model'] if "embed_model" in config else 'BAAI/bge-m3'
         self.embedding_model = ollama.Client(model_host)
         self.keep_alive = config.get('keep_alive', None)
         self.ollama_options = config.get('options', {})
@@ -70,7 +70,7 @@ class VannaServer:
         milvus_client = MilvusClient(uri=milvus_uri,db_name=milvus_database)
 
         embedding_host = config["embedding_host"] if "embedding_host" in config else 'http://wsd.wisdomidata.com:19042'
-        embedding_model = config["embedding_model"] if "embedding_model" in config else "bge-m3" # BAAI/bge-m3
+        embedding_model = config["embedding_model"] if "embedding_model" in config else "BAAI/bge-m3" # BAAI/bge-m3
         embedding_function = model.dense.SentenceTransformerEmbeddingFunction(
             model_name=embedding_model,
             device='cpu'  # 'cpu' or 'cuda:0'

api/pyproject.toml

@@ -6,6 +6,7 @@ dependencies = [
     "vanna[milvus,mysql,ollama,postgres]>=0.7.9,<0.8.0",
     "scikit-learn~=1.6.1",
     "sentence-transformers~=2.2.2",
+    "huggingface_hub==0.19.3",
     "pymilvus[model]>=2.5.8,<2.6.0",
     "pillow>=11.2.1,<12.0.0",
     "pymupdf>=1.25.5,<1.26.0",
@@ -83,7 +84,7 @@ dependencies = [
     "sqlalchemy~=2.0.29",
     "starlette==0.41.0",
     "tiktoken~=0.9.0",
-    "transformers~=4.51.0",
+    "transformers~=4.36.2",
     "unstructured[docx,epub,md,ppt,pptx]~=0.16.1",
     "weave~=0.51.0",
     "yarl~=1.18.3",