diff --git a/memgpt/agent_store/db.py b/memgpt/agent_store/db.py index 52146db4..d4220d45 100644 --- a/memgpt/agent_store/db.py +++ b/memgpt/agent_store/db.py @@ -532,15 +532,15 @@ class PostgresStorageConnector(SQLStorageConnector): # Commit the changes to the database session.commit() - def str_to_datetime(self, str_date): + def str_to_datetime(self, str_date: str) -> datetime: val = str_date.split("-") _datetime = datetime(int(val[0]), int(val[1]), int(val[2])) return _datetime def query_date(self, start_date, end_date, limit=None, offset=0): filters = self.get_filters({}) - _start_date = self.str_to_datetime(start_date) - _end_date = self.str_to_datetime(end_date) + _start_date = self.str_to_datetime(start_date) if isinstance(start_date, str) else start_date + _end_date = self.str_to_datetime(end_date) if isinstance(end_date, str) else end_date with self.session_maker() as session: query = ( session.query(self.db_model) diff --git a/memgpt/settings.py b/memgpt/settings.py index 6fceff37..1c982433 100644 --- a/memgpt/settings.py +++ b/memgpt/settings.py @@ -27,7 +27,7 @@ class Settings(BaseSettings): # add this property to avoid being returned the default # reference: https://github.com/cpacker/MemGPT/issues/1362 @property - def memgpt_pt_uri_no_default(self) -> str: + def memgpt_pg_uri_no_default(self) -> str: if self.pg_uri: return self.pg_uri elif self.pg_db and self.pg_user and self.pg_password and self.pg_host and self.pg_port: