Update the size parameter to be dimension

datastax · Oct 30, 2023 · aabd563 · aabd563
1 parent b5f7747
commit aabd563
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -74,9 +74,9 @@ api_endpoint = f"https://{database_id}-{database_region}.{database_base_url}"
 astra_db = AstraDB(token=token, api_endpoint=api_endpoint)
 
 # Possible Operations
-astra_db.create_collection(collection_name="collection_test_delete", size=5)
+astra_db.create_collection(collection_name="collection_test_delete", dimension=5)
 astra_db.delete_collection(collection_name="collection_test_delete")
-astra_db.create_collection(collection_name="collection_test", size=5)
+astra_db.create_collection(collection_name="collection_test", dimension=5)
 
 # Collections
 astra_db_collection = AstraDBCollection(

diff --git a/astrapy/db.py b/astrapy/db.py
@@ -347,9 +347,11 @@ def get_collections(self):
 
         return response
 
-    def create_collection(self, size=None, options={}, function="", collection_name=""):
-        if size and not options:
-            options = {"vector": {"size": size}}
+    def create_collection(
+        self, dimension=None, options={}, function="", collection_name=""
+    ):
+        if dimension and not options:
+            options = {"vector": {"dimension": dimension}}
             if function:
                 options["vector"]["function"] = function
         if options:

diff --git a/scripts/astrapy_latest_interface.py b/scripts/astrapy_latest_interface.py
@@ -43,9 +43,9 @@
 astra_db = AstraDB(token=token, api_endpoint=api_endpoint)
 
 # Possible Operations
-astra_db.create_collection(collection_name="collection_test_delete", size=5)
+astra_db.create_collection(collection_name="collection_test_delete", dimension=5)
 astra_db.delete_collection(collection_name="collection_test_delete")
-astra_db.create_collection(collection_name="collection_test", size=5)
+astra_db.create_collection(collection_name="collection_test", dimension=5)
 
 # Collections
 astra_db_collection = AstraDBCollection(