From 870e37b4ad308315453c2374f6af3221cc5936ab Mon Sep 17 00:00:00 2001 From: "lengjs@wsidomidata.com" Date: Wed, 25 Jun 2025 17:25:10 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E9=85=8D=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- api/core/workflow/nodes/vanna/vanna_node.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/api/core/workflow/nodes/vanna/vanna_node.py b/api/core/workflow/nodes/vanna/vanna_node.py index 64d2bdc2aa..145bbc2c90 100644 --- a/api/core/workflow/nodes/vanna/vanna_node.py +++ b/api/core/workflow/nodes/vanna/vanna_node.py @@ -16,16 +16,19 @@ from .entities import VannaNodeData class Config: def __init__(self, supplier): self.embedding_supplier = "SiliconFlow" - self.milvus_uri = dify_config.MILVUS_URI - self.milvus_database = 'vanna_demo' + self.milvus_uri = dify_config.VANNA_MILVUS_URI + self.milvus_database = dify_config.VANNA_MILVUS_DATABASE + self.embedding_host = dify_config.VANNA_EMBEDDING_HOST + self.embedding_model = dify_config.VANNA_EMBEDDING_MODEL + self.embedding_type = dify_config.VANNA_EMBEDDING_TYPE self.supplier = supplier self.sql_type = 'postgres' self.sql_config = { - "host": dify_config.DB_HOST, - "dbname": 'vanna_demo', - "user": dify_config.DB_USERNAME, - "password": dify_config.DB_PASSWORD, - "port": dify_config.DB_PORT + "host": dify_config.VANNA_DB_HOST, + "dbname": dify_config.VANNA_DB_DATABASE, + "user": dify_config.VANNA_DB_USERNAME, + "password": dify_config.VANNA_DB_PASSWORD, + "port": dify_config.VANNA_DB_PORT } vn_instances = {}