From 9d6dc8f883f340e703aea256c435f324609e4a49 Mon Sep 17 00:00:00 2001
From: cpacker <packercharles@gmail.com>
Date: Wed, 25 Sep 2024 11:46:46 -0700
Subject: [PATCH] chore: fix typing errors (also black)

---
 letta/agent_store/chroma.py |  4 ++--
 letta/agent_store/db.py     |  6 +++---
 letta/server/server.py      | 18 +++++++++---------
 3 files changed, 14 insertions(+), 14 deletions(-)

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 1096ff6900..00861edacf 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:
@@ -1963,18 +1963,18 @@ def get_current_user(self) -> User:
 
         return self.get_default_user()
         ## NOTE: same code as local client to get the default user
-        #config = LettaConfig.load()
-        #user_id = config.anon_clientid
-        #user = self.get_user(user_id)
+        # config = LettaConfig.load()
+        # user_id = config.anon_clientid
+        # user = self.get_user(user_id)
 
-        #if not user:
+        # if not user:
         #    user = self.create_user(UserCreate())
 
         #    # # update config
         #    config.anon_clientid = str(user.id)
         #    config.save()
 
-        #return user
+        # return user
 
     def list_models(self) -> List[LLMConfig]:
         """List available models"""