diff --git a/letta/agent_store/chroma.py b/letta/agent_store/chroma.py index e192a1543b..8907b5f96e 100644 --- a/letta/agent_store/chroma.py +++ b/letta/agent_store/chroma.py @@ -256,8 +256,8 @@ def query_text(self, query, count=None, start=None, filters: Optional[Dict] = {} def get_all_cursor( self, filters: Optional[Dict] = {}, - after: str = None, - before: str = None, + after: Optional[str] = None, + before: Optional[str] = None, limit: Optional[int] = 1000, order_by: str = "created_at", reverse: bool = False, diff --git a/letta/agent_store/db.py b/letta/agent_store/db.py index b84dc56be9..c746591595 100644 --- a/letta/agent_store/db.py +++ b/letta/agent_store/db.py @@ -213,9 +213,9 @@ def get_all_paginated(self, filters: Optional[Dict] = {}, page_size: Optional[in def get_all_cursor( self, filters: Optional[Dict] = {}, - after: str = None, - before: str = None, - limit: Optional[int] = 1000, + after: Optional[str] = None, + before: Optional[str] = None, + limit: int = 1000, order_by: str = "created_at", reverse: bool = False, ): diff --git a/letta/server/server.py b/letta/server/server.py index 2ab79144da..6d498eea7e 100644 --- a/letta/server/server.py +++ b/letta/server/server.py @@ -1257,10 +1257,10 @@ def get_agent_recall_cursor( agent_id: str, after: Optional[str] = None, before: Optional[str] = None, - limit: Optional[int] = 100, - order_by: Optional[str] = "created_at", - order: Optional[str] = "asc", - reverse: Optional[bool] = False, + limit: int = 100, + order_by: str = "created_at", + order: str = "asc", + reverse: bool = False, return_message_object: bool = True, ) -> Union[List[Message], List[LettaMessage]]: if self.ms.get_user(user_id=user_id) is None: