From 0df0c6158ca4c38eacea7cb92bebcc5a0ea8e46a Mon Sep 17 00:00:00 2001
From: Quin Lynch <quin@edgedb.com>
Date: Fri, 26 Apr 2024 15:26:01 -0300
Subject: [PATCH] Add overloads for adding/removing sets for update

---
 .../Builders/UpdateShapeBuilder.cs                  | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/EdgeDB.Net.QueryBuilder/Builders/UpdateShapeBuilder.cs b/src/EdgeDB.Net.QueryBuilder/Builders/UpdateShapeBuilder.cs
index fd63076c..e0273a6e 100644
--- a/src/EdgeDB.Net.QueryBuilder/Builders/UpdateShapeBuilder.cs
+++ b/src/EdgeDB.Net.QueryBuilder/Builders/UpdateShapeBuilder.cs
@@ -96,11 +96,24 @@ public UpdateShapeBuilder<T, U> Add<V>(Expression<Func<T, IEnumerable<V>?>> sele
     public UpdateShapeBuilder<T, U> Add<V>(Expression<Func<T, IEnumerable<V>?>> selector, Expression<Func<T, V>> value)
         => AddElement(selector, value, ExpressionType.AddAssign);
 
+    public UpdateShapeBuilder<T, U> Add<V>(Expression<Func<T, IEnumerable<V>?>> selector,
+        Expression<Func<T, IEnumerable<V>>> value)
+        => AddElement(selector, value, ExpressionType.AddAssign);
+
+    public UpdateShapeBuilder<T, U> Add<V>(Expression<Func<T, IEnumerable<V>?>> selector,
+        Expression<Func<T, U, IEnumerable<V>>> value)
+        => AddElement(selector, value, ExpressionType.AddAssign);
+
     public UpdateShapeBuilder<T, U> Remove<V>(Expression<Func<T, IEnumerable<V>?>> selector, Expression<Func<T, U, V>> value)
         => AddElement(selector, value, ExpressionType.SubtractAssign);
     public UpdateShapeBuilder<T, U> Remove<V>(Expression<Func<T, IEnumerable<V>?>> selector, Expression<Func<T, V>> value)
         => AddElement(selector, value, ExpressionType.SubtractAssign);
 
+    public UpdateShapeBuilder<T, U> Remove<V>(Expression<Func<T, IEnumerable<V>?>> selector, Expression<Func<T, U, IEnumerable<V>>> value)
+        => AddElement(selector, value, ExpressionType.SubtractAssign);
+    public UpdateShapeBuilder<T, U> Remove<V>(Expression<Func<T, IEnumerable<V>?>> selector, Expression<Func<T, IEnumerable<V>>> value)
+        => AddElement(selector, value, ExpressionType.SubtractAssign);
+
     private UpdateShapeBuilder<T, U> AddElement(LambdaExpression selector, LambdaExpression value, ExpressionType type)
     {
         var key = ExpressionUtils.GetMemberSelection(selector);