From cef574818a156a75b5b1426911e9da88a46362fa Mon Sep 17 00:00:00 2001 From: paituo <330435863@qq.com> Date: Mon, 19 Aug 2024 10:24:07 +0800 Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E8=BF=9Brerank=E6=95=88=E6=9E=9C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/app/engine/loaders/db.py | 16 ++++++++-------- backend/app/xinference/base.py | 4 ++-- 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/backend/app/engine/loaders/db.py b/backend/app/engine/loaders/db.py index 69d3279..63a7c02 100644 --- a/backend/app/engine/loaders/db.py +++ b/backend/app/engine/loaders/db.py @@ -168,14 +168,14 @@ def get_db_documents(configs: list[DBLoaderConfig]): engine = create_engine(entry.uri) sql_database = SQLDatabase(engine) - table_schema_objs = makeDescriptionByEngine(sql_database) - table_node_mapping = SQLTableNodeMapping(sql_database) - - nodes = table_node_mapping.to_nodes(table_schema_objs) - for node in nodes: - node.metadata.update(metadata) - - docs.extend(nodes) + # table_schema_objs = makeDescriptionByEngine(sql_database) + # table_node_mapping = SQLTableNodeMapping(sql_database) + # + # nodes = table_node_mapping.to_nodes(table_schema_objs) + # for node in nodes: + # node.metadata.update(metadata) + # + # docs.extend(nodes) queries = entry.queries or [] loader = CustomDatabaseReader(sql_database) diff --git a/backend/app/xinference/base.py b/backend/app/xinference/base.py index 92f87e1..54cb16d 100644 --- a/backend/app/xinference/base.py +++ b/backend/app/xinference/base.py @@ -155,8 +155,8 @@ class XinferenceRerank(BaseNodePostprocessor): description="The model description from Xinference." ) _generator: Any = PrivateAttr() - _model_uid: str - _endpoint: str + _model_uid: str = Field(description="The Xinference model to use.") + _endpoint: str = Field(description="The Xinference endpoint URL to use.") model: str = Field(description="Dashscope rerank model name.") top_n: int = Field(description="Top N nodes to return.") threshold: float = Field(description="threshold nodes to return.")