Fix format

Author: ekzhu

autogen/agentchat/contrib/vectordb/pgvectordb.py

@@ -9,11 +9,11 @@
 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`")
+# try:
+import pgvector
+from pgvector.psycopg import register_vector
+# except ImportError:
+#     raise ImportError("Please install pgvector: `pip install pgvector`")
 
 try:
     import psycopg
@@ -416,6 +416,7 @@ def query(
         results = []
         for query_text in query_texts:
             vector = self.embedding_function(query_text)
+            vector_string = "[" + ",".join([f"{x:.8f}" for x in vector]) + "]"
 
             if distance_type.lower() == "cosine":
                 index_function = "<=>"
@@ -428,7 +429,7 @@ def query(
             query = (
                 f"SELECT id, documents, embedding, metadatas "
                 f"FROM {self.name} "
-                f"{clause} embedding {index_function} '{str(vector)}' {distance_threshold} "
+                f"{clause} embedding {index_function} '{vector_string}' {distance_threshold} "
                 f"LIMIT {n_results}"
             )
             cursor.execute(query)