ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-25 23:14:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

Add path expansion to image/audio/file (#1038)

Browse source 
* Add path expansion to image/audio/file

* Make path tests platform agnostic
This commit is contained in:
Jeremiah Lowin 2025-07-04 09:51:30 -04:00 • committed by GitHub
commit e182231fc8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 58 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/fastmcp/utilities/types.py
View file

@ -3,6 +3,7 @@
import base64
import inspect
import mimetypes
import os
from collections.abc import Callable
from functools import lru_cache
from pathlib import Path
@ -101,7 +102,7 @@ class Image:
if path is not None and data is not None:
raise ValueError("Only one of path or data can be provided")
self.path = Path(path) if path else None
self.path = Path(os.path.expandvars(str(path))).expanduser() if path else None
self.data = data
self._format = format
self._mime_type = self._get_mime_type()
@ -160,7 +161,7 @@ class Audio:
if path is not None and data is not None:
raise ValueError("Only one of path or data can be provided")
self.path = Path(path) if path else None
self.path = Path(os.path.expandvars(str(path))).expanduser() if path else None
self.data = data
self._format = format
self._mime_type = self._get_mime_type()
@ -219,7 +220,7 @@ class File:
if path is not None and data is not None:
raise ValueError("Only one of path or data can be provided")
self.path = Path(path) if path else None
self.path = Path(os.path.expandvars(str(path))).expanduser() if path else None
self.data = data
self._format = format
self._mime_type = self._get_mime_type()