From 2c0acf207c89e7d1f55155c2de55967135808ebe Mon Sep 17 00:00:00 2001 From: Eric Zhu Date: Tue, 1 Oct 2024 14:05:55 -0700 Subject: [PATCH] Fix formatting --- autogen/agentchat/contrib/vectordb/pgvectordb.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/autogen/agentchat/contrib/vectordb/pgvectordb.py b/autogen/agentchat/contrib/vectordb/pgvectordb.py index bfc781964579..64f4ff333a90 100644 --- a/autogen/agentchat/contrib/vectordb/pgvectordb.py +++ b/autogen/agentchat/contrib/vectordb/pgvectordb.py @@ -4,14 +4,15 @@ from typing import Callable, List, Optional, Union import numpy as np + +# try: +import pgvector +from pgvector.psycopg import register_vector from sentence_transformers import SentenceTransformer from .base import Document, ItemID, QueryResults, VectorDB from .utils import get_logger -# try: -import pgvector -from pgvector.psycopg import register_vector # except ImportError: # raise ImportError("Please install pgvector: `pip install pgvector`")