File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -42,6 +42,7 @@ def set_message(
4242 def get_messages (
4343 self ,
4444 last_updated : datetime ,
45+ sort_fields : list [str ] | None = None ,
4546 num_chunks : int | None = None ,
4647 chunk_size : int = 1000 ,
4748 all_fields : bool = True ,
@@ -51,6 +52,7 @@ def get_messages(
5152
5253 Args:
5354 last_updated (datetime): Datetime to use to query for newer messages
55+ sort_fields (List[str]): Fields used to sort results. Prefix with '-' to sort in descending order.
5456 num_chunks (int): Maximum number of chunks of data to yield. None will yield all possible.
5557 chunk_size (int): Number of data entries per chunk.
5658 all_fields (bool): Whether to return all fields in the document. Defaults to True.
@@ -65,6 +67,11 @@ def get_messages(
6567 """
6668 query_params = {}
6769
70+ if sort_fields :
71+ query_params .update (
72+ {"_sort_fields" : "," .join ([s .strip () for s in sort_fields ])}
73+ )
74+
6875 return self ._search (
6976 last_updated = last_updated ,
7077 num_chunks = num_chunks ,
You can’t perform that action at this time.
0 commit comments