diff --git a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
index c558ea1ba..50b625fea 100644
--- a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
+++ b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
@@ -1,5 +1,5 @@
 import asyncio
-from typing import List, Optional, Union, get_type_hints
+from typing import List, Optional, get_type_hints
 from sqlalchemy.inspection import inspect
 from sqlalchemy.orm import Mapped, mapped_column
 from sqlalchemy.dialects.postgresql import insert
diff --git a/cognee/infrastructure/databases/vector/pgvector/__init__.py b/cognee/infrastructure/databases/vector/pgvector/__init__.py
index 2f8f08620..8cde08139 100644
--- a/cognee/infrastructure/databases/vector/pgvector/__init__.py
+++ b/cognee/infrastructure/databases/vector/pgvector/__init__.py
@@ -1,2 +1 @@
-from .PGVectorAdapter import PGVectorAdapter
 from .create_db_and_tables import create_db_and_tables