Explorar o código

[BugFixes] slack loader (#1135)

Co-authored-by: Deven Patel <deven298@yahoo.com>
Deven Patel hai 1 ano
pai
achega
4dfce44c1a
Modificáronse 1 ficheiros con 7 adicións e 3 borrados
  1. 7 3
      embedchain/loaders/slack.py

+ 7 - 3
embedchain/loaders/slack.py

@@ -77,7 +77,7 @@ class SlackLoader(BaseLoader):
             )
 
             messages = results.get("messages")
-            num_message = results.get("total")
+            num_message = len(messages)
             logging.info(f"Found {num_message} messages for query: {query}")
 
             matches = messages.get("matches", [])
@@ -86,9 +86,13 @@ class SlackLoader(BaseLoader):
                 text = message.get("text")
                 content = clean_string(text)
 
-                message_meta_data_keys = ["channel", "iid", "team", "ts", "type", "user", "username"]
-                meta_data = message.fromkeys(message_meta_data_keys, "")
+                message_meta_data_keys = ["iid", "team", "ts", "type", "user", "username"]
+                meta_data = {}
+                for key in message.keys():
+                    if key in message_meta_data_keys:
+                        meta_data[key] = message.get(key)
                 meta_data.update({"url": url})
+
                 data.append(
                     {
                         "content": content,