KidZddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z mZmZddl mZdd lmZdd lmZdd lmZeeZGd d eZGddeZGddeZy)ayEventStore implementation backed by AsyncKeyValue. This module provides an EventStore implementation that enables SSE polling/resumability for Streamable HTTP transports. Events are stored using the key_value package's AsyncKeyValue protocol, allowing users to configure any compatible backend (in-memory, Redis, etc.) following the same pattern as ResponseCachingMiddleware. ) annotations)uuid4)PydanticAdapter) AsyncKeyValue) MemoryStore) EventCallbackEventId EventMessageStreamId) EventStore)JSONRPCMessage) BaseModel) get_loggerc0eZdZUdZded<ded<ded<y) EventEntryzStored event entry.strevent_id stream_idz dict | NonemessageN__name__ __module__ __qualname____doc____annotations__f/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/server/event_store.pyrrsMN rrceZdZUdZded<y)StreamEventListzList of event IDs for a stream.z list[str] event_idsNrrrrr r "s )rr cReZdZdZ d ddZ ddZ d dZy) r a-EventStore implementation backed by AsyncKeyValue. Enables SSE polling/resumability by storing events that can be replayed when clients reconnect. Works with any AsyncKeyValue backend (memory, Redis, etc.) following the same pattern as ResponseCachingMiddleware and OAuthProxy. Example: ```python from fastmcp import FastMCP from fastmcp.server.event_store import EventStore # Default in-memory storage event_store = EventStore() # Or with a custom backend from key_value.aio.stores.redis import RedisStore redis_backend = RedisStore(url="redis://localhost") event_store = EventStore(storage=redis_backend) mcp = FastMCP("MyServer") app = mcp.http_app(event_store=event_store, retry_interval=2000) ``` Args: storage: AsyncKeyValue backend. Defaults to MemoryStore. max_events_per_stream: Maximum events to retain per stream. Default 100. ttl: Event TTL in seconds. Default 3600 (1 hour). Set to None for no expiration. Nc|xs t|_||_||_t t |jt d|_t t|jtd|_y)Nfastmcp_events) key_valuepydantic_modeldefault_collectionfastmcp_streams) r_storage_max_events_per_stream_ttlrr _event_storer _stream_store)selfstoragemax_events_per_streamttls r__init__zEventStore.__init__Fsm (/'?+- &;# :I9Tmm%/:  @O @ mm*0 @ rcKtt}t|||r|jdnd}|jj |||j d{|jj|d{}|r |jng}|j|t||jkDrI|d|j D]&}|jj|d{(||j d}|jj |t||j d{|S777V7 w)aStore an event and return its ID. Args: stream_id: ID of the stream the event belongs to message: The JSON-RPC message to store, or None for priming events Returns: The generated event ID for the stored event json)modeN)rrr)keyvaluer1r6)r!)rrr model_dumpr,putr+r-getr!appendlenr*deleter )r.rrrentry stream_datar!old_ids r store_eventzEventStore.store_event^sXuw<7>G&&F&3D  ##499#MMM!..22y2AA -8K))b " y>D77 7#$Bt'B'B&BC ;''..6.::: ;!4#>#>">"@AI  $$!I6 %   ' NB; sJA!E#E $#EE A/E7E8A EEE EEEcK|jj|d{}|stjd|dy|j}|j j|d{}|stjd|dy|j } |j|dz}||dD]y}|jj|d{} | s*| js7tj| j} |t| | jd{{|S77#t$rtjd|d|YywxYw779w)aOReplay events that occurred after the specified event ID. Args: last_event_id: The ID of the last event the client received send_callback: A callback function to send events to the client Returns: The stream ID of the replayed events, or None if the event ID was not found r8Nz Event ID z not found in storezStream z not found in stream )r,r;loggerwarningrr-r!index ValueErrorrr model_validater r) r. last_event_id send_callbackr?rr@r! start_idxreventmsgs rreplay_events_afterzEventStore.replay_events_afters^''++ +>> NNY}o5HI JOO  ..22y2AA  NNWYK/BC D))  ! 6:I ")*- GH++//H/==E$33EMMB#Lenn$EFFF  G 7? B  NNY}o5J9+V W  >Gsp E&D3A E&.D6/,E&D80'E&E"E& E&,>E&*E$+ E&6E&8$EE&EE&$E&)Ndi)r/zAsyncKeyValue | Noner0intr1z int | None)rr rzJSONRPCMessage | Nonereturnr )rJr rKrrRzStreamId | None)rrrrr2rBrOrrrr r (sr>)-%(  %  #   0'!',A' 'R**%*  *rr N)r __future__ruuidrkey_value.aio.adapters.pydanticrkey_value.aio.protocolsrkey_value.aio.stores.memoryrmcp.server.streamable_httprr r r r SDKEventStore mcp.typesr pydanticrfastmcp.utilities.loggingrrrErr rrrr]s]#;13UUB$0 H i IIr