fix: hotfix for server test (#1753)

letta-ai · Sep 13, 2024 · bba1a73 · bba1a73
1 parent 220658f
commit bba1a73
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/memgpt/client/admin.py b/memgpt/client/admin.py
@@ -61,7 +61,6 @@ def delete_key(self, api_key: str) -> APIKey:
 
     def create_user(self, name: Optional[str] = None) -> User:
         request = UserCreate(name=name)
-        print("YYYYY Pinging", f"{self.base_url}/{self.api_prefix}/admin/users")
         response = requests.post(f"{self.base_url}/{self.api_prefix}/admin/users", headers=self.headers, json=request.model_dump())
         if response.status_code != 200:
             raise HTTPError(response.json())

diff --git a/memgpt/server/rest_api/app.py b/memgpt/server/rest_api/app.py
@@ -38,7 +38,9 @@
 # TODO(ethan)
 # NOTE(charles): @ethan I had to add this to get the global as the bottom to work
 interface: StreamingServerInterface = StreamingServerInterface
-server: SyncServer = None
+# global server
+# server: SyncServer = None
+server = SyncServer(default_interface_factory=lambda: interface())
 
 # TODO(ethan): eventuall remove
 if password := settings.server_pass:
@@ -57,8 +59,8 @@
 
 def create_application() -> "FastAPI":
     """the application start routine"""
-    global server
-    server = SyncServer(default_interface_factory=lambda: interface())
+    # global server
+    # server = SyncServer(default_interface_factory=lambda: interface())
 
     app = FastAPI(
         swagger_ui_parameters={"docExpansion": "none"},
@@ -136,7 +138,7 @@ def on_startup():
     def on_shutdown():
         global server
         server.save_agents()
-        server = None
+        # server = None
 
     return app
 

diff --git a/memgpt/server/rest_api/utils.py b/memgpt/server/rest_api/utils.py
@@ -61,7 +61,7 @@ def get_memgpt_server() -> SyncServer:
     # Check if a global server is already instantiated
     from memgpt.server.rest_api.app import server
 
-    assert isinstance(server, SyncServer)
+    # assert isinstance(server, SyncServer)
     return server