From 1d4cc10826c9b64887118ac2483289d144db5f5f Mon Sep 17 00:00:00 2001 From: Mark Daoust Date: Mon, 1 Jul 2024 10:19:15 -0700 Subject: [PATCH] Fix pperator conversion. --- google/generativeai/types/retriever_types.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) 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,