postgresml
diff --git a/‎pgml-sdks/python/pgml/README.md
Lines changed: 16 additions & 255 deletions b/‎pgml-sdks/python/pgml/README.md
Lines changed: 16 additions & 255 deletions
diff --git a/‎pgml-sdks/python/pgml/docs/pgml/collection.md
Lines changed: 0 additions & 128 deletions b/‎pgml-sdks/python/pgml/docs/pgml/collection.md
Lines changed: 0 additions & 128 deletions
diff --git a/‎pgml-sdks/python/pgml/docs/pgml/database.md
Lines changed: 0 additions & 34 deletions b/‎pgml-sdks/python/pgml/docs/pgml/database.md
Lines changed: 0 additions & 34 deletions
diff --git a/‎pgml-sdks/python/pgml/docs/pgml/dbutils.md
Lines changed: 0 additions & 48 deletions b/‎pgml-sdks/python/pgml/docs/pgml/dbutils.md
Lines changed: 0 additions & 48 deletions
diff --git a/‎pgml-sdks/python/pgml/docs/pgml/index.md
Lines changed: 0 additions & 8 deletions b/‎pgml-sdks/python/pgml/docs/pgml/index.md
Lines changed: 0 additions & 8 deletions
diff --git a/‎pgml-sdks/python/pgml/examples/vector_search.py
Lines changed: 1 addition & 1 deletion b/‎pgml-sdks/python/pgml/examples/vector_search.py
Lines changed: 1 addition & 1 deletion
diff --git a/‎pgml-sdks/python/pgml/pgml/collection.py
Lines changed: 4 additions & 2 deletions b/‎pgml-sdks/python/pgml/pgml/collection.py
Lines changed: 4 additions & 2 deletions
diff --git a/‎pgml-sdks/python/pgml/pyproject.toml
Lines changed: 3 additions & 7 deletions b/‎pgml-sdks/python/pgml/pyproject.toml
Lines changed: 3 additions & 7 deletions
@@ -29,6 +29,6 @@
 
 start = time()
 results = collection.vector_search("Who won 20 grammy awards?", top_k=2)
-rprint("Query time %0.3f"%(time()-start))
 rprint(json.dumps(results, indent=2))
+rprint("Query time %0.3f"%(time()-start))
 db.archive_collection(collection_name)
@@ -264,9 +264,10 @@ def _create_chunks_table(self):
 
     def upsert_documents(
         self,
-        documents: List[Dict[str, Any]],
+        documents: List[Dict[str, str]],
         text_key: Optional[str] = "text",
         id_key: Optional[str] = "id",
+        verbose: bool = False,
     ) -> None:
         """
         The function `upsert_documents` inserts or updates documents in a database table based on their ID,
@@ -275,7 +276,7 @@ def upsert_documents(
         :param documents: A list of dictionaries, where each dictionary represents a document to be upserted
         into a database table. Each dictionary should contain metadata about the document, as well as the
         actual text of the document
-        :type documents: List[Dict[str, Any]]
+        :type documents: List[Dict[str, str]]
         :param text_key: The key in the dictionary that corresponds to the text of the document, defaults to
         text
         :type text_key: Optional[str] (optional)
@@ -287,6 +288,7 @@ def upsert_documents(
         :param verbose: A boolean parameter that determines whether or not to print verbose output during
         the upsert process. If set to True, additional information will be printed to the console during the
         upsert process. If set to False, only essential information will be printed, defaults to False
+        :type verbose: bool (optional)
         """
         conn = self.pool.getconn()
         for document in track(documents, description="Upserting documents"):
 
@@ -1,14 +1,10 @@
 [tool.poetry]
 name = "pgml"
-version = "0.6.0"
-description = "Python SDK is designed to facilitate the development of scalable vector search applications on PostgreSQL databases."
-authors = ["PostgresML <team@postgresml.org>"]
-homepage = "https://postgresml.org"
-repository = "https://github.com/postgresml/postgresml"
-documentation = "https://github.com/postgresml/postgresml/tree/master/pgml-sdks/python/pgml"
+version = "0.1.0"
+description = ""
+authors = ["Santi Adavani <santi@hyperparam.ai>"]
 readme = "README.md"
 packages = [{include = "pgml"}]
-keywords = ["postgres","machine learning","vector databases","embeddings"]
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"