chore: Bring back driver close call

getzep · Sep 22, 2024 · f651132 · f651132
1 parent cd3906d
commit f651132
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/graphiti_core/graphiti.py b/graphiti_core/graphiti.py
@@ -158,8 +158,8 @@ def close(self):
                 # Use graphiti...
             finally:
                 graphiti.close()
-        self.driver.close()
         """
+        self.driver.close()
 
     async def build_indices_and_constraints(self):
         """

diff --git a/server/graph_service/main.py b/server/graph_service/main.py
@@ -13,6 +13,8 @@ async def lifespan(_: FastAPI):
     settings = get_settings()
     await initialize_graphiti(settings)
     yield
+    # Shutdown
+    # No need to close Graphiti here, as it's handled per-request
 
 
 app = FastAPI(lifespan=lifespan)