fix: quotes in milvus filter (#122)

ansys · Feb 13, 2025 · 07d500a · 07d500a
1 parent f7fee62
commit 07d500a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/pkg/privatefunctions/milvus/milvus.go b/pkg/privatefunctions/milvus/milvus.go
@@ -21,7 +21,7 @@ import (
 
 // Global variables
 var (
-	CallsBatchSize          = 10000
+	CallsBatchSize          = 500
 	DumpBatchSize           = 10000
 	MilvusConnectionTimeout = 5 * time.Second
 	MilvusConnectionRetries = 40
@@ -934,6 +934,13 @@ func createStringFilterExpression(filterType string, filter []string) (filterExp
 		}
 	}()
 
+	for i := range filter {
+		// Escape double backslashes in the filter value.
+		filter[i] = strings.ReplaceAll(filter[i], "\\", "\\\\")
+		// Escape single quotes in the filter value
+		filter[i] = strings.ReplaceAll(filter[i], "'", "\\'")
+	}
+
 	filterExpression = fmt.Sprintf("%s in ['%s'", filterType, filter[0])
 	if len(filter) > 1 {
 		filterExpression += fmt.Sprintf(", '%s'", strings.Join(filter[1:], "', '"))