Small PR cleanup

Author: strawgate

pyproject.toml

@@ -118,7 +118,7 @@ testpaths = ["tests"]
 python_files = ["test_*.py", "*_test.py"]
 python_classes = ["Test*"]
 python_functions = ["test_*"]
-addopts = ["--inline-snapshot=fix,create"]
+addopts = ["--inline-snapshot=disable"]
 
 [tool.ty.src]
 include = ["src", "tests"]

src/fastmcp/client/auth/oauth.py

@@ -5,23 +5,20 @@
 import webbrowser
 from asyncio import Future
 from collections.abc import AsyncGenerator
-from datetime import datetime
-from pathlib import Path
 from typing import Any
 from urllib.parse import urlparse
 
 import anyio
 import httpx
 from kv_store_adapter.adapters.pydantic import PydanticAdapter
-from kv_store_adapter.stores.disk import DiskStore
 from kv_store_adapter.types import KVStoreProtocol
 from mcp.client.auth import OAuthClientProvider, TokenStorage
 from mcp.shared.auth import (
     OAuthClientInformationFull,
     OAuthClientMetadata,
     OAuthToken,
 )
-from pydantic import AnyHttpUrl, BaseModel
+from pydantic import AnyHttpUrl
 from uvicorn.server import Server
 
 from fastmcp import settings

src/fastmcp/server/auth/oauth_proxy.py

@@ -29,7 +29,6 @@
 from authlib.common.security import generate_token
 from authlib.integrations.httpx_client import AsyncOAuth2Client
 from kv_store_adapter.adapters.pydantic import PydanticAdapter
-from kv_store_adapter.stores.disk import DiskStore
 from kv_store_adapter.types import KVStoreProtocol
 from mcp.server.auth.provider import (
     AccessToken,

src/fastmcp/settings.py

@@ -6,7 +6,7 @@
 from pathlib import Path
 from typing import TYPE_CHECKING, Annotated, Any, Literal
 
-from kv_store_adapter.stores.disk import DiskStore
+from kv_store_adapter.types import KVStoreProtocol
 from pydantic import Field, ImportString, field_validator
 from pydantic.fields import FieldInfo
 from pydantic_settings import (
@@ -149,7 +149,7 @@ def settings(self) -> Self:
 
     home: Path = Path.home() / ".fastmcp"
 
-    data_path: Path = home / "data.db"
+    data_path: Path | None = home / "data.db"
 
     test_mode: bool = False
 
@@ -384,8 +384,15 @@ def server_auth_class(self) -> AuthProvider | None:
         return auth_class
 
     @cached_property
-    def data_store(self) -> DiskStore:
-        return DiskStore(path=str(self.data_path), size_limit=1024 * 1024 * 10) # 10MB
+    def data_store(self) -> KVStoreProtocol:
+        if not self.data_path:
+            from kv_store_adapter.stores.memory import MemoryStore
+
+            return MemoryStore()
+
+        from kv_store_adapter.stores.disk import DiskStore
+
+        return DiskStore(path=str(self.data_path), size_limit=1024 * 1024 * 10)  # 10MB
 
 
 def __getattr__(name: str):