Bläddra i källkod

[Improvement] fix slack loader max_count (#1127)

Co-authored-by: Deven Patel <deven298@yahoo.com>
Deven Patel 1 år sedan
förälder
incheckning
1b661bb2fd
2 ändrade filer med 6 tillägg och 6 borttagningar
  1. 5 5
      embedchain/loaders/slack.py
  2. 1 1
      pyproject.toml

+ 5 - 5
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")

+ 1 - 1
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 <taranjeet@embedchain.ai>",