feat: Support Milvus partition names (#39)

Signed-off-by: Anush008 <anushshetty90@gmail.com>

Author: Anush008

README.md

@@ -172,14 +172,15 @@ docker run --net=host --rm -it registry.cloud.qdrant.io/library/qdrant-migration
 
 | Flag                       | Description                                             |
 | -------------------------- | ------------------------------------------------------- |
-| `--milvus.url`             | Source Milvus URL (e.g. `https://your-milvus-hostname`) |
-| `--milvus.collection`      | Source collection name                                  |
-| `--milvus.api-key`         | Source API key (`$SOURCE_API_KEY`)                      |
-| `--milvus.enable-tls-auth` | Enable TLS Auth                                         |
+| `--milvus.url`             | Milvus URL (e.g. `https://your-milvus-hostname`)        |
+| `--milvus.collection`      | Milvus collection name                                  |
+| `--milvus.api-key`         | Milvus API key for authentication                       |
+| `--milvus.enable-tls-auth` | Whether to enable TLS Auth                              |
 | `--milvus.username`        | Username for Milvus                                     |
 | `--milvus.password`        | Password for Milvus                                     |
-| `--milvus.db-name`         | Milvus database name                                    |
-| `--milvus.server-version`  | Server version                                          |
+| `--milvus.db-name`         | Optional database name                                  |
+| `--milvus.server-version`  | Milvus server version                                   |
+| `--milvus.partitions`      | List of partition names                                 |
 
 #### Qdrant Options
 

cmd/migrate_from_milvus.go

@@ -222,6 +222,7 @@ func (r *MigrateFromMilvusCmd) migrateData(ctx context.Context, sourceClient *mi
 		}
 
 		result, err := sourceClient.Query(ctx, milvusclient.NewQueryOption(r.Milvus.Collection).
+			WithPartitions(r.Milvus.Partitions...).
 			WithFilter(filter).
 			WithOutputFields("*").
 			WithLimit(batchSize))

pkg/commons/config.go

@@ -15,14 +15,15 @@ type MigrationConfig struct {
 }
 
 type MilvusConfig struct {
-	Url           string `help:"Source Milvus URL, e.g. https://your-milvus-hostname" required:"true"`
-	Collection    string `help:"Source collection" required:"true"`
-	APIKey        string `help:"Source API key"`
-	EnableTLSAuth bool   `help:"Enable TLS Auth for Milvus" default:"false"`
-	Username      string `help:"Milvus username"`
-	Password      string `help:"Milvus password"`
-	DBName        string `help:"Milvus database name"`
-	ServerVersion string `help:"Milvus server version"`
+	Url           string   `help:"Source Milvus URL, e.g. https://your-milvus-hostname" required:"true"`
+	Collection    string   `help:"Source collection" required:"true"`
+	APIKey        string   `help:"Source API key"`
+	EnableTLSAuth bool     `help:"Enable TLS Auth for Milvus" default:"false"`
+	Username      string   `help:"Milvus username"`
+	Password      string   `help:"Milvus password"`
+	DBName        string   `help:"Milvus database name"`
+	ServerVersion string   `help:"Milvus server version"`
+	Partitions    []string `help:"Milvus partition names"`
 }
 
 type PineconeConfig struct {