diff --git a/google/generativeai/types/retriever_types.py b/google/generativeai/types/retriever_types.py
index 9931ee58d..5270c1e81 100644
--- a/google/generativeai/types/retriever_types.py
+++ b/google/generativeai/types/retriever_types.py
@@ -96,17 +96,22 @@ def valid_name(name):
     4: Operator.GREATER_EQUAL,
     "operator_greater_equal": Operator.GREATER_EQUAL,
     "greater_equal": Operator.GREATER_EQUAL,
+    Operator.EQUAL: Operator.EQUAL,
+    5: Operator.EQUAL,
+    "operator_equal": Operator.EQUAL,
+    "equal": Operator.EQUAL,
+    "==": Operator.EQUAL,
     Operator.NOT_EQUAL: Operator.NOT_EQUAL,
-    5: Operator.NOT_EQUAL,
+    6: Operator.NOT_EQUAL,
     "operator_not_equal": Operator.NOT_EQUAL,
     "not_equal": Operator.NOT_EQUAL,
     "!=": Operator.NOT_EQUAL,
     Operator.INCLUDES: Operator.INCLUDES,
-    6: Operator.INCLUDES,
+    7: Operator.INCLUDES,
     "operator_includes": Operator.INCLUDES,
     "includes": Operator.INCLUDES,
     Operator.EXCLUDES: Operator.EXCLUDES,
-    6: Operator.EXCLUDES,
+    8: Operator.EXCLUDES,
     "operator_excludes": Operator.EXCLUDES,
     "excludes": Operator.EXCLUDES,
     "not in": Operator.EXCLUDES,