Upgrade to tastytrade 11.0.0 and replace OAuthSession with Session

Breaking changes in tastytrade 11.0.0:
- OAuthSession class has been removed
- OAuth is now the only authentication method
- The Session class now handles OAuth authentication

Changes:
- Replace all OAuthSession imports with Session
- Update tastytrade dependency from >=10.3.0 to >=11.0.0
- Update type hints in ServerContext and function signatures

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <[email protected]>

Author: Ferdous Bhai

background.py

@@ -5,7 +5,7 @@
 from zoneinfo import ZoneInfo
 
 import typer
-from tastytrade.session import OAuthSession
+from tastytrade.session import Session
 from tastytrade.market_sessions import a_get_market_sessions, ExchangeType, MarketStatus
 
 from agent import create_tastytrader_agent # loads .env internally
@@ -21,13 +21,13 @@
 logger = logging.getLogger(__name__)
 
 
-def create_tastytrade_session() -> OAuthSession:
+def create_tastytrade_session() -> Session:
     """Create and return a configured TastyTrade session."""
     client_secret = os.getenv("TASTYTRADE_CLIENT_SECRET")
     refresh_token = os.getenv("TASTYTRADE_REFRESH_TOKEN")
     if not client_secret or not refresh_token:
         raise ValueError("Missing required TastyTrade environment variables")
-    return OAuthSession(client_secret, refresh_token)
+    return Session(client_secret, refresh_token)
 
 
 def check_market_open() -> bool:

pyproject.toml

@@ -6,7 +6,7 @@ readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
     "mcp[cli]>=v1.14.0",
-    "tastytrade>=10.3.0",
+    "tastytrade>=11.0.0",
     "humanize>=4.12.3",
     "aiolimiter==1.2.1",
     "tabulate>=0.9.0",

tasty_agent/server.py

@@ -17,7 +17,7 @@
 from mcp.server.session import ServerSession
 from mcp.types import SamplingMessage, TextContent
 from tabulate import tabulate
-from tastytrade import OAuthSession, Account
+from tastytrade import Session, Account
 from tastytrade.dxfeed import Quote, Greeks
 from tastytrade.instruments import Equity, Option, a_get_option_chain
 from tastytrade.market_sessions import a_get_market_sessions, a_get_market_holidays, ExchangeType, MarketStatus
@@ -40,7 +40,7 @@ def to_table(data: Sequence[BaseModel]) -> str:
 
 @dataclass
 class ServerContext:
-    session: OAuthSession
+    session: Session
     account: Account
 
 
@@ -64,7 +64,7 @@ async def lifespan(_) -> AsyncIterator[ServerContext]:
         )
 
     try:
-        session = OAuthSession(client_secret, refresh_token)
+        session = Session(client_secret, refresh_token)
         accounts = Account.get(session)
         logger.info(f"Successfully authenticated with Tastytrade. Found {len(accounts)} account(s).")
     except Exception as e:
@@ -170,12 +170,12 @@ def validate_strike_price(strike_price: Any) -> float:
 
 
 @cached(ttl=86400, cache=Cache.MEMORY, serializer=PickleSerializer())  # 24 hours
-async def get_cached_option_chain(session: OAuthSession, symbol: str):
+async def get_cached_option_chain(session: Session, symbol: str):
     """Cache option chains for 24 hours as they rarely change during that timeframe."""
     return await a_get_option_chain(session, symbol)
 
 
-async def get_instrument_details(session: OAuthSession, instrument_specs: list[InstrumentSpec]) -> list[InstrumentDetail]:
+async def get_instrument_details(session: Session, instrument_specs: list[InstrumentSpec]) -> list[InstrumentDetail]:
     """Get instrument details with validation and caching."""
     async def lookup_single_instrument(spec: InstrumentSpec) -> InstrumentDetail:
         symbol = spec.symbol.upper()