Parcourir la source

Update history_dialogue_service: add limit

StrayWarrior il y a 2 semaines
Parent
commit
a17992c343
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. 2 2
      history_dialogue_service.py

+ 2 - 2
history_dialogue_service.py

@@ -13,7 +13,7 @@ class HistoryDialogueService:
         self.base_url = base_url
 
     def get_dialogue_history(self, staff_id: str, user_id: str, max_count: int = 10):
-        url = f"{self.base_url}?sender={staff_id}&receiver={user_id}"
+        url = f"{self.base_url}?sender={staff_id}&receiver={user_id}&limit={max_count}"
         response = requests.post(url, headers={
             'Content-Type': 'application/json'
         })
@@ -39,7 +39,7 @@ class HistoryDialogueService:
                 'timestamp': record.get('sendtime', 0)
             })
         ret = sorted(ret, key=lambda x: x['timestamp'])
-        return ret[-max_count:] if len(ret) > max_count else ret
+        return ret
 
 
 if __name__ == '__main__':