From f1c5acf16c105f47277263cbfb45c6bb5d6843b7 Mon Sep 17 00:00:00 2001 From: Boris Arzentar Date: Tue, 8 Jul 2025 11:54:49 +0200 Subject: [PATCH] fix: pgvector dependency import --- .../infrastructure/databases/vector/pgvector/PGVectorAdapter.py | 2 +- cognee/infrastructure/databases/vector/pgvector/__init__.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) 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