From 246111bba7fe76ebd0e0e5b43129261058acc126 Mon Sep 17 00:00:00 2001
From: William Easton <bill.easton@elastic.co>
Date: Mon, 2 Jun 2025 10:08:48 -0500
Subject: [PATCH] Fix missing import

---
 tests/tools/test_tool.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/tools/test_tool.py b/tests/tools/test_tool.py
index 7669c75f7..97f6fdf81 100644
--- a/tests/tools/test_tool.py
+++ b/tests/tools/test_tool.py
@@ -1,6 +1,6 @@
 import pytest
 from mcp.types import EmbeddedResource, ImageContent, TextContent, TextResourceContents
-from pydantic import AnyUrl
+from pydantic import AnyUrl, BaseModel
 
 from fastmcp import FastMCP, Image
 from fastmcp.client import Client