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.")