From fe75f43ee6a6dfcb25f4f8b3c35c2745b491f4ac Mon Sep 17 00:00:00 2001
From: David Soria Parra <davidsp@anthropic.com>
Date: Wed, 18 Dec 2024 14:02:51 +0000
Subject: [PATCH] fix: Update test to match Pydantic's JSON schema output for
 model references with defaults

Pydantic now uses allOf to combine model references with default values in JSON schema output.
---
 tests/server/fastmcp/test_func_metadata.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/server/fastmcp/test_func_metadata.py b/tests/server/fastmcp/test_func_metadata.py
index b329c90..044798d 100644
--- a/tests/server/fastmcp/test_func_metadata.py
+++ b/tests/server/fastmcp/test_func_metadata.py
@@ -316,7 +316,11 @@ def test_complex_function_json_schema():
                 "type": "string",
             },
             "my_model_a_with_default": {
-                "$ref": "#/$defs/SomeInputModelA",
+                "allOf": [
+                    {
+                        "$ref": "#/$defs/SomeInputModelA"
+                    }
+                ],
                 "default": {},
             },
             "an_int_with_default": {