Skip to content

Commit 63a15fe

Browse files
authored
Fix pperator conversion. (#419)
1 parent 90d0cdb commit 63a15fe

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

google/generativeai/types/retriever_types.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -96,17 +96,22 @@ def valid_name(name):
9696
4: Operator.GREATER_EQUAL,
9797
"operator_greater_equal": Operator.GREATER_EQUAL,
9898
"greater_equal": Operator.GREATER_EQUAL,
99+
Operator.EQUAL: Operator.EQUAL,
100+
5: Operator.EQUAL,
101+
"operator_equal": Operator.EQUAL,
102+
"equal": Operator.EQUAL,
103+
"==": Operator.EQUAL,
99104
Operator.NOT_EQUAL: Operator.NOT_EQUAL,
100-
5: Operator.NOT_EQUAL,
105+
6: Operator.NOT_EQUAL,
101106
"operator_not_equal": Operator.NOT_EQUAL,
102107
"not_equal": Operator.NOT_EQUAL,
103108
"!=": Operator.NOT_EQUAL,
104109
Operator.INCLUDES: Operator.INCLUDES,
105-
6: Operator.INCLUDES,
110+
7: Operator.INCLUDES,
106111
"operator_includes": Operator.INCLUDES,
107112
"includes": Operator.INCLUDES,
108113
Operator.EXCLUDES: Operator.EXCLUDES,
109-
6: Operator.EXCLUDES,
114+
8: Operator.EXCLUDES,
110115
"operator_excludes": Operator.EXCLUDES,
111116
"excludes": Operator.EXCLUDES,
112117
"not in": Operator.EXCLUDES,

0 commit comments

Comments
 (0)