[hotfix] Fix argument name mismatch in PythonVectorStore. (#635)

(cherry picked from commit 13d543e)

Author: wenjin272

api/src/main/java/org/apache/flink/agents/api/vectorstores/python/PythonVectorStore.java

@@ -80,7 +80,7 @@ public List<String> add(
         kwargs.put("documents", pythonDocuments);
 
         if (collection != null) {
-            kwargs.put("collection", collection);
+            kwargs.put("collection_name", collection);
         }
 
         return (List<String>) adapter.callMethod(vectorStore, "add", kwargs);
@@ -109,7 +109,7 @@ public List<Document> get(
             kwargs.put("ids", ids);
         }
         if (collection != null) {
-            kwargs.put("collection", collection);
+            kwargs.put("collection_name", collection);
         }
 
         Object pythonDocuments = adapter.callMethod(vectorStore, "get", kwargs);
@@ -126,7 +126,7 @@ public void delete(
             kwargs.put("ids", ids);
         }
         if (collection != null) {
-            kwargs.put("collection", collection);
+            kwargs.put("collection_name", collection);
         }
         adapter.callMethod(vectorStore, "delete", kwargs);
     }

api/src/test/java/org/apache/flink/agents/api/vectorstores/python/PythonCollectionManageableVectorStoreTest.java

@@ -261,7 +261,7 @@ void testAddDocuments() throws Exception {
                         argThat(
                                 kwargs -> {
                                     assertThat(kwargs).containsKey("documents");
-                                    assertThat(kwargs).containsKey("collection");
+                                    assertThat(kwargs).containsKey("collection_name");
                                     assertThat(kwargs).containsKey("batch_size");
                                     return true;
                                 }));
@@ -294,7 +294,7 @@ void testGetDocuments() throws Exception {
                         argThat(
                                 kwargs -> {
                                     assertThat(kwargs).containsKey("ids");
-                                    assertThat(kwargs).containsKey("collection");
+                                    assertThat(kwargs).containsKey("collection_name");
                                     return true;
                                 }));
     }
@@ -317,7 +317,7 @@ void testDeleteDocuments() throws Exception {
                         argThat(
                                 kwargs -> {
                                     assertThat(kwargs).containsKey("ids");
-                                    assertThat(kwargs).containsKey("collection");
+                                    assertThat(kwargs).containsKey("collection_name");
                                     return true;
                                 }));
     }