kKjdZddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z mZmZddl mZdd lmZdd lmZdd lmZeeZGd d eZGddeZGddeZGddeZy)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) get_logger)FastMCPBaseModelc0eZdZUdZded<ded<ded<y) EventEntryzStored event entry.strevent_id stream_idz dict | NonemessageN__name__ __module__ __qualname____doc____annotations__k/Users/ahmed/devFolder/Ultron/claude-voice/.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 cPeZdZdZddZd dZd dZ d dZ d dZy) SessionScopedEventStorezEEventStore adapter that isolates stream IDs to one transport session.c>||_t|d|d|_y)N:) _event_storelen_stream_prefix)self event_store session_ids r__init__z SessionScopedEventStore.__init__+s$'!$Z 1:,a@rc"|j|SN)r(r)rs r_scope_stream_idz(SessionScopedEventStore._scope_stream_id/s%%&yk22rcj|j|jsy|t|jdSr.) startswithr(r'r/s r_unscope_stream_idz*SessionScopedEventStore._unscope_stream_id2s2##D$7$78T001344rctK|jj|j||d{S7wr.)r& store_eventr0)r)rrs rr5z#SessionScopedEventStore.store_event7s;&&22  ! !) ,g    s /868cKgdfd }|jj||d{}|y|j|}|tj d|yD]}||d{|S7J7 w)Nc0Kj|ywr.)append)eventreplayed_eventss r buffer_eventzASessionScopedEventStore.replay_events_after..buffer_eventEs  " "5 )sz>Event ID %s does not belong to this session-scoped event store)r9r returnNone)r&replay_events_afterr3loggerwarning)r) last_event_id send_callbackr;scoped_stream_idrr9r:s @rr>z+SessionScopedEventStore.replay_events_after>s /1 *"&!2!2!F!F <"    #++,<=   NNP $E& & &%#  's"(A:A6AA:-A8. A:8A:N)r* SDKEventStorer+r)rr r<r )rr r<StreamId | Nonerr rzJSONRPCMessage | Noner<r rAr rBrr<rE) rrrrr,r0r3r5r>rrrr#r#(sSOA35  ! ,A  %  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_ttlrrr&r _stream_store)r)storagemax_events_per_streamttls rr,zEventStore.__init__zsm (/'?+- &;# :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)keyvaluerUrY)r!)rrr model_dumpr&putrQrRgetr!r8r'rPdeleter )r)rrrentry stream_datar!old_ids rr5zEventStore.store_eventsTuw<7>G&&F&3D  ##499#MMM!..22y2AA -8K))b " y>D77 7#$Bt'B'B&BC''..6.:::D!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 r[Nz Event ID z not found in storezStream z not found in stream )r&r^r?r@rrRr!index ValueErrorrr model_validater r) r)rArBr`rrar! start_idxrr9msgs rr>zEventStore.replay_events_aftersW''++ +>> NNY}o5HI JOO  ..22y2AA  NNWYK/BC D))  ! 6:I ")*-H++//H/==E$33EMMB#Lenn$EFFF . 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)rSzAsyncKeyValue | NonerTintrUz int | NonerFrG)rrrrr,r5r>rrrr 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 rD mcp.typesr fastmcp.utilities.loggingrfastmcp.utilities.typesrrr?rr r#rrrrusl#;13UUB$04 H !& 1m1hIIr