singleton pattern

norton120 · Dec 2, 2024 · 2b8d2ad · 2b8d2ad
1 parent 62b0147
commit 2b8d2ad
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/letta/server/rest_api/app.py b/letta/server/rest_api/app.py
@@ -182,6 +182,7 @@ def on_startup():
         # Tool.load_default_tools(get_db_session())
 
         generate_openapi_schema(app)
+        # always migrate now that we have a default in-memory postgres
 
     @app.on_event("shutdown")
     def on_shutdown():

diff --git a/letta/settings.py b/letta/settings.py
@@ -88,8 +88,8 @@ def letta_pg_uri(self) -> str:
             return f"postgresql+pg8000://{self.pg_user}:{self.pg_password}@{self.pg_host}:{self.pg_port}/{self.pg_db}"
         else:
             # start the pg binary. This is the default in-memory postgres that replaces SQLite/Chroma
-            db_uri = self.launch_pg_binary()
-            return db_uri
+            self.pg_uri = self.launch_pg_binary()
+            return self.pg_uri
 
     @property
     def letta_pg_uri_no_default(self) -> str: