From 4dfce44c1a4abc8d42191d1d57e286ba31f87008 Mon Sep 17 00:00:00 2001 From: Deven Patel Date: Sun, 7 Jan 2024 23:13:25 +0530 Subject: [PATCH] [BugFixes] slack loader (#1135) Co-authored-by: Deven Patel --- embedchain/loaders/slack.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/embedchain/loaders/slack.py b/embedchain/loaders/slack.py index a426d980..56f895ba 100644 --- a/embedchain/loaders/slack.py +++ b/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,