From 7331ec971247c75ba45c6426c51d04932aa8551f Mon Sep 17 00:00:00 2001
From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com>
Date: Tue, 3 Dec 2024 13:49:50 -0500
Subject: [PATCH] Fix FileUrl

---
 tests/prompts/test_base.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/tests/prompts/test_base.py b/tests/prompts/test_base.py
index 78418e81d..4dca37d6d 100644
--- a/tests/prompts/test_base.py
+++ b/tests/prompts/test_base.py
@@ -1,3 +1,4 @@
+from pydantic import FileUrl
 import pytest
 from fastmcp.prompts.base import (
     Prompt,
@@ -102,7 +103,7 @@ class TestRenderPrompt:
                 content=EmbeddedResource(
                     type="resource",
                     resource=TextResourceContents(
-                        uri="file://file.txt",
+                        uri=FileUrl("file://file.txt"),
                         text="File contents",
                         mimeType="text/plain",
                     ),
@@ -115,7 +116,7 @@ class TestRenderPrompt:
                 content=EmbeddedResource(
                     type="resource",
                     resource=TextResourceContents(
-                        uri="file://file.txt",
+                        uri=FileUrl("file://file.txt"),
                         text="File contents",
                         mimeType="text/plain",
                     ),
@@ -133,7 +134,7 @@ class TestRenderPrompt:
                     content=EmbeddedResource(
                         type="resource",
                         resource=TextResourceContents(
-                            uri="file://file.txt",
+                            uri=FileUrl("file://file.txt"),
                             text="File contents",
                             mimeType="text/plain",
                         ),
@@ -151,7 +152,7 @@ class TestRenderPrompt:
                 content=EmbeddedResource(
                     type="resource",
                     resource=TextResourceContents(
-                        uri="file://file.txt",
+                        uri=FileUrl("file://file.txt"),
                         text="File contents",
                         mimeType="text/plain",
                     ),
@@ -171,7 +172,7 @@ class TestRenderPrompt:
                 "content": {
                     "type": "resource",
                     "resource": {
-                        "uri": "file://file.txt",
+                        "uri": FileUrl("file://file.txt"),
                         "text": "File contents",
                         "mimeType": "text/plain",
                     },
@@ -184,7 +185,7 @@ class TestRenderPrompt:
                 content=EmbeddedResource(
                     type="resource",
                     resource=TextResourceContents(
-                        uri="file://file.txt",
+                        uri=FileUrl("file://file.txt"),
                         text="File contents",
                         mimeType="text/plain",
                     ),