update output formatting in test

rgbkrk · Sep 25, 2024 · 2c5ab63 · 2c5ab63
1 parent cd3317e
commit 2c5ab63
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tests/test_registry.py b/tests/test_registry.py
@@ -176,7 +176,7 @@ def test_generate_function_schema_with_model_argument():
                 "x": {"type": "integer"},
                 "y": {"type": "string"},
                 "z": {"default": False, "type": "boolean"},
-                "model": {"allOf": [{"$ref": "#/$defs/SimpleModel"}], "default": None},
+                "model": {"$ref": "#/$defs/SimpleModel", "default": None},
             },
             "required": ["x", "y"],
             "$defs": {
@@ -197,6 +197,7 @@ def test_generate_function_schema_with_model_argument():
                 }
             },
         },
+        "strict": None,
     }
     assert schema.model_dump() == expected_schema
 
@@ -212,8 +213,8 @@ def test_generate_function_schema_with_model_and_nested_model_arguments():
                 "x": {"type": "integer"},
                 "y": {"type": "string"},
                 "z": {"default": False, "type": "boolean"},
-                "model": {"allOf": [{"$ref": "#/$defs/SimpleModel"}], "default": None},
-                "nested_model": {"allOf": [{"$ref": "#/$defs/NestedModel"}], "default": None},
+                "model": {"$ref": "#/$defs/SimpleModel", "default": None},
+                "nested_model": {"$ref": "#/$defs/NestedModel", "default": None},
             },
             "required": ["x", "y"],
             "$defs": {
@@ -249,6 +250,7 @@ def test_generate_function_schema_with_model_and_nested_model_arguments():
                 },
             },
         },
+        "strict": None,
     }
     assert schema.model_dump() == expected_schema
 
@@ -268,6 +270,7 @@ def test_generate_function_schema_with_uuid_argument():
             },
             "required": ["x", "y"],
         },
+        "strict": None,
     }
     assert schema.model_dump() == expected_schema
 
@@ -383,7 +386,6 @@ async def test_function_registry_call_edge_cases():
 async def test_function_registry_call_tool():
     registry = FunctionRegistry()
     registry.register(simple_func, SimpleModel)
-
     registry.register(simple_func_with_model_arg)
 
     tools = registry.tools
@@ -416,7 +418,7 @@ async def test_function_registry_call_tool():
                         "x": {"type": "integer"},
                         "y": {"type": "string"},
                         "z": {"default": False, "type": "boolean"},
-                        "model": {"allOf": [{"$ref": "#/$defs/SimpleModel"}], "default": None},
+                        "model": {"$ref": "#/$defs/SimpleModel", "default": None},
                     },
                     "required": ["x", "y"],
                     "$defs": {