bugfix: exclude null types in anyOf list

dottxt-ai · May 20, 2024 · dbf193e · dbf193e
1 parent edf0320
commit dbf193e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/outlines/fsm/json_schema.py b/outlines/fsm/json_schema.py
@@ -137,7 +137,7 @@ def optimize_schema(instance):
             elif "anyOf" in subinstance and _has_null_type(subinstance["anyOf"]):
                 any_of_list = subinstance.pop("anyOf")
                 filtered_any_of_list = list(
-                    filter(lambda d: _is_null_type(d), any_of_list)
+                    filter(lambda d: not _is_null_type(d), any_of_list)
                 )
                 if len(filtered_any_of_list) == 0:
                     keys_to_remove.add(key)