Skip to content

Commit

Permalink
Fix formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
ekzhu committed Oct 1, 2024
1 parent 1aeee61 commit 2c0acf2
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions autogen/agentchat/contrib/vectordb/pgvectordb.py
Original file line number Diff line number Diff line change
Expand Up @@ -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`")

Expand Down

0 comments on commit 2c0acf2

Please sign in to comment.