fix: Doesn't drop entire PG database, just cleans public schema - Cog 1947

cognee/infrastructure/databases/relational/sqlalchemy/SqlAlchemyAdapter.py

@@ -69,7 +69,7 @@ async def create_table(self, schema_name: str, table_name: str, table_config: li
     async def delete_table(self, table_name: str, schema_name: Optional[str] = "public"):
         async with self.engine.begin() as connection:
             if self.engine.dialect.name == "sqlite":
-                # SQLite doesn’t support schema namespaces and the CASCADE keyword.
+                # SQLite doesn't support schema namespaces and the CASCADE keyword.
                 # However, foreign key constraint can be defined with ON DELETE CASCADE during table creation.
                 await connection.execute(text(f'DROP TABLE IF EXISTS "{table_name}";'))
             else:
@@ -327,10 +327,10 @@ async def delete_database(self):
                     file.write("")
             else:
                 async with self.engine.begin() as connection:
-                    schema_list = await self.get_schema_list()
                     # Create a MetaData instance to load table information
                     metadata = MetaData()
-                    # Drop all tables from all schemas
+                    # Drop all tables from the public schema
+                    schema_list = ["public", "public_staging"]
                     for schema_name in schema_list:
                         # Load the schema information into the MetaData object
                         await connection.run_sync(metadata.reflect, schema=schema_name)

cognee/infrastructure/databases/vector/exceptions/exceptions.py

@@ -6,7 +6,7 @@ class CollectionNotFoundError(CriticalError):
     def __init__(
         self,
         message,
-        name: str = "DatabaseNotCreatedError",
+        name: str = "CollectionNotFoundError",
         status_code: int = status.HTTP_422_UNPROCESSABLE_ENTITY,
     ):
         super().__init__(message, name, status_code)