This commit is contained in:
Charles Packer
2024-05-12 20:28:02 -07:00
committed by GitHub
parent e9c9513f84
commit c5ac6850c7
2 changed files with 4 additions and 4 deletions

View File

@@ -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)

View File

@@ -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: