init

main.py

@@ -62,18 +62,20 @@ def create_a_collection(milvus_client, collection_name):
         schema=schema
     )
 
+    collection = Collection(name=collection_name, milvus_client=milvus_client)
+
     # Create an index for the collection
     # IVF_FLAT index is used here, with metric_type COSINE
     index_params = {
-    "index_type": "IVF_FLAT",
-    "metric_type": "COSINE",
-    "params": {
-        "nlist": 128
+        "index_type": "IVF_FLAT",
+        "metric_type": "COSINE",
+        "params": {
+            "nlist": 128
+        }
     }
-}
 
     # Create the index on the vector field
-    milvus_client.create_index(
+    collection.create_index(
         collection_name=collection_name,
         field_name="vector",
         index_params=index_params  # Pass the dictionary, not a string