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

Add warnings regarding memory store

Browse source
This commit is contained in:
William Easton 2025-10-10 12:30:27 -04:00
commit d1f7c74cde
No known key found for this signature in database
2 changed files with 16 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/fastmcp/client/auth/oauth.py
View file

@ -191,6 +191,14 @@ class OAuth(OAuthClientProvider):
# Create server-specific token storage
token_storage = token_storage or MemoryStore()
if isinstance(token_storage, MemoryStore):
from warnings import warn
warn(
message="Using in-memory token storage is not recommended for production use -- "
+ "tokens will be lost on server restart."
)
self.token_storage_adapter: TokenStorageAdapter = TokenStorageAdapter(
async_key_value=token_storage, server_url=server_base_url
)

8
src/fastmcp/server/auth/oauth_proxy.py
View file

@ -388,6 +388,14 @@ class OAuthProxy(OAuthProvider):
self._client_storage: AsyncKeyValue = client_storage or MemoryStore()
if isinstance(self._client_storage, MemoryStore):
from warnings import warn
warn(
message="Using in-memory client storage is not recommended for production use -- "
+ "clients will be lost on server restart which may require manual clean-up of oauth information on the client."
)
self._client_store = PydanticAdapter[ProxyDCRClient](
key_value=self._client_storage,
pydantic_model=ProxyDCRClient,