fix version gate

alexander-alderman-webb · alexander-alderman-webb · commit 19755806403d · 2026-02-02T14:24:14.000+01:00
diff --git a/tests/integrations/fastmcp/test_fastmcp.py b/tests/integrations/fastmcp/test_fastmcp.py
@@ -625,7 +625,7 @@ def code_help_prompt(language: str):
                 },
             }
 
-            if Message is not None:
+            if FASTMCP_VERSION is not None and FASTMCP_VERSION.startswith("3"):
                 message = Message(message)
 
             return [message]
@@ -714,7 +714,7 @@ async def async_prompt(topic: str):
                 },
             }
 
-            if Message is not None:
+            if FASTMCP_VERSION is not None and FASTMCP_VERSION.startswith("3"):
                 message1 = Message(message1)
                 message2 = Message(message2)
 

Original file line number	Diff line number	Diff line change
`@@ -625,7 +625,7 @@ def code_help_prompt(language: str):`
`625`	`625`	`},`
`626`	`626`	`}`
`627`	`627`
`628`		`- if Message is not None:`
	`628`	`+ if FASTMCP_VERSION is not None and FASTMCP_VERSION.startswith("3"):`
`629`	`629`	`message = Message(message)`
`630`	`630`
`631`	`631`	`return [message]`
`@@ -714,7 +714,7 @@ async def async_prompt(topic: str):`
`714`	`714`	`},`
`715`	`715`	`}`
`716`	`716`
`717`		`- if Message is not None:`
	`717`	`+ if FASTMCP_VERSION is not None and FASTMCP_VERSION.startswith("3"):`
`718`	`718`	`message1 = Message(message1)`
`719`	`719`	`message2 = Message(message2)`
`720`	`720`