feat: Add support for EditDocumentsByFunction and ContainsFilter in ExperimentalFeatures

Author: A7bari

features.go

@@ -30,6 +30,16 @@ func (ef *ExperimentalFeatures) SetMetrics(metrics bool) *ExperimentalFeatures {
 	return ef
 }
 
+func (ef *ExperimentalFeatures) SetEditDocumentsByFunction(editDocumentsByFunction bool) *ExperimentalFeatures {
+	ef.EditDocumentsByFunction = &editDocumentsByFunction
+	return ef
+}
+
+func (ef *ExperimentalFeatures) SetContainsFilter(containsFilter bool) *ExperimentalFeatures {
+	ef.ContainsFilter = &containsFilter
+	return ef
+}
+
 func (ef *ExperimentalFeatures) Get() (*ExperimentalFeaturesResult, error) {
 	return ef.GetWithContext(context.Background())
 }
@@ -59,9 +69,11 @@ func (ef *ExperimentalFeatures) Update() (*ExperimentalFeaturesResult, error) {
 
 func (ef *ExperimentalFeatures) UpdateWithContext(ctx context.Context) (*ExperimentalFeaturesResult, error) {
 	request := ExperimentalFeaturesBase{
-		VectorStore: ef.VectorStore,
-		LogsRoute:   ef.LogsRoute,
-		Metrics:     ef.Metrics,
+		VectorStore:             ef.VectorStore,
+		LogsRoute:               ef.LogsRoute,
+		Metrics:                 ef.Metrics,
+		EditDocumentsByFunction: ef.EditDocumentsByFunction,
+		ContainsFilter:          ef.ContainsFilter,
 	}
 	resp := new(ExperimentalFeaturesResult)
 	req := &internalRequest{

features_test.go

@@ -63,11 +63,15 @@ func TestUpdate_ExperimentalFeatures(t *testing.T) {
 			ef.SetVectorStore(true)
 			ef.SetLogsRoute(true)
 			ef.SetMetrics(true)
+			ef.SetEditDocumentsByFunction(true)
+			ef.SetContainsFilter(true)
 			gotResp, err := ef.Update()
 			require.NoError(t, err)
 			require.Equal(t, true, gotResp.VectorStore, "ExperimentalFeatures.Update() should return vectorStore as true")
 			require.Equal(t, true, gotResp.LogsRoute, "ExperimentalFeatures.Update() should return logsRoute as true")
 			require.Equal(t, true, gotResp.Metrics, "ExperimentalFeatures.Update() should return metrics as true")
+			require.Equal(t, true, gotResp.EditDocumentsByFunction, "ExperimentalFeatures.Update() should return editDocumentsByFunction as true")
+			require.Equal(t, true, gotResp.ContainsFilter, "ExperimentalFeatures.Update() should return containsFilter as true")
 		})
 	}
 }

types.go

@@ -542,15 +542,19 @@ type DocumentsResult struct {
 
 // ExperimentalFeaturesResult represents the experimental features result from the API.
 type ExperimentalFeaturesBase struct {
-	VectorStore *bool `json:"vectorStore,omitempty"`
-	LogsRoute   *bool `json:"logsRoute,omitempty"`
-	Metrics     *bool `json:"metrics,omitempty"`
+	VectorStore             *bool `json:"vectorStore,omitempty"`
+	LogsRoute               *bool `json:"logsRoute,omitempty"`
+	Metrics                 *bool `json:"metrics,omitempty"`
+	EditDocumentsByFunction *bool `json:"editDocumentsByFunction,omitempty"`
+	ContainsFilter          *bool `json:"containsFilter,omitempty"`
 }
 
 type ExperimentalFeaturesResult struct {
-	VectorStore bool `json:"vectorStore"`
-	LogsRoute   bool `json:"logsRoute"`
-	Metrics     bool `json:"metrics"`
+	VectorStore             bool `json:"vectorStore"`
+	LogsRoute               bool `json:"logsRoute"`
+	Metrics                 bool `json:"metrics"`
+	EditDocumentsByFunction bool `json:"editDocumentsByFunction"`
+	ContainsFilter          bool `json:"containsFilter"`
 }
 
 type SwapIndexesParams struct {