From 5f7d2853b8e8b2231a9f332ecd176fca54900d96 Mon Sep 17 00:00:00 2001 From: Weves Date: Tue, 16 May 2023 22:41:12 -0700 Subject: [PATCH] Fix connectors --- backend/danswer/connectors/factory.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/backend/danswer/connectors/factory.py b/backend/danswer/connectors/factory.py index 4486b1e7a2ea..2f6ebcb9a562 100644 --- a/backend/danswer/connectors/factory.py +++ b/backend/danswer/connectors/factory.py @@ -48,8 +48,11 @@ def build_connector( def build_pull_connector( source: DocumentSource, connector_specific_config: dict[str, Any] ) -> PullLoader: - return _range_pull_to_pull( - build_connector(source, InputType.PULL, connector_specific_config) + connector = build_connector(source, InputType.PULL, connector_specific_config) + return ( + _range_pull_to_pull(connector) + if isinstance(connector, RangePullLoader) + else connector )