From 1b661bb2fd134d5063a45806f5a726d2fb0e49e2 Mon Sep 17 00:00:00 2001 From: Deven Patel Date: Sun, 7 Jan 2024 13:03:42 +0530 Subject: [PATCH] [Improvement] fix slack loader max_count (#1127) Co-authored-by: Deven Patel --- embedchain/loaders/slack.py | 10 +++++----- pyproject.toml | 2 +- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/embedchain/loaders/slack.py b/embedchain/loaders/slack.py index 44de53b5..a426d980 100644 --- a/embedchain/loaders/slack.py +++ b/embedchain/loaders/slack.py @@ -16,10 +16,10 @@ class SlackLoader(BaseLoader): def __init__(self, config: Optional[Dict[str, Any]] = None): super().__init__() - if config is not None: - self.config = config - else: - self.config = {"base_url": SLACK_API_BASE_URL} + self.config = config if config else {} + + if "base_url" not in self.config: + self.config["base_url"] = SLACK_API_BASE_URL self.client = None self._setup_loader(self.config) @@ -73,7 +73,7 @@ class SlackLoader(BaseLoader): query=query, sort="timestamp", sort_dir="desc", - count=1000, + count=self.config.get("count", 100), ) messages = results.get("messages") diff --git a/pyproject.toml b/pyproject.toml index 289264ad..6d8e7dd2 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "embedchain" -version = "0.1.54" +version = "0.1.55" description = "Data platform for LLMs - Load, index, retrieve and sync any unstructured data" authors = [ "Taranjeet Singh ",