add ranking rules

#97
tgbot-collection · May 27, 2023 · 1a594ca · 1a594ca
1 parent d2cd6bd
commit 1a594ca
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/searchgram/engine.py b/searchgram/engine.py
@@ -29,6 +29,9 @@ def __init__(self):
             self.client = meilisearch.Client(MEILI_HOST, MEILI_PASS)
             self.client.create_index("telegram", {"primaryKey": "ID"})
             self.client.index("telegram").update_filterable_attributes(["chat.id", "chat.username", "chat.type"])
+            self.client.index("telegram").update_ranking_rules(
+                ["timestamp:desc", "words", "typo", "proximity", "attribute", "sort", "exactness"]
+            )
             self.client.index("telegram").update_sortable_attributes(["timestamp"])
         except:
             logging.critical("Failed to connect to MeiliSearch")
@@ -61,7 +64,7 @@ def upsert(self, message):
         if self.check_ignore(message):
             return
         data = self.set_uid(message)
-        self.client.index("telegram").add_documents([data])
+        self.client.index("telegram").add_documents([data], primary_key="ID")
 
     @staticmethod
     def __clean_user(user: "str"):