i.SSKrSSKrSSKJr SSKJrJrJrJrJ r SSK J r J r SSK Jr SSKJr SSKJr S r"S S \5r\R,"S \5 "S S\5r\"S\ \S9r"SS\\5rg)N)asynccontextmanager)AnyAsyncGeneratorGenericOptionalType) BaseModelValidationError)TypeVar) DictLikeModel)BaseSerializeric\rSrSrSrg)UnserializableKeyWarningN)__name__ __module__ __qualname____firstlineno____static_attributes__rW/home/james-whalen/.local/lib/python3.13/site-packages/workflows/context/state_store.pyrrsrroncec4^\rSrSrSrS\4U4SjjrSrU=r$) DictStatea Dynamic, dict-like Pydantic model for workflow state. Used as the default state model when no typed state is provided. Behaves like a mapping while retaining Pydantic validation and serialization. Examples: ```python from workflows.context.state_store import DictState state = DictState() state["foo"] = 1 state.bar = 2 # attribute-style access works for nested structures ``` See Also: - [InMemoryStateStore][workflows.context.state_store.InMemoryStateStore] paramsc &>[TU]"S0UD6 g)Nr)super__init__)selfr __class__s rr!DictState.__init__,s "6"rr) rrrr__doc__rr!r __classcell__)r#s@rrrs&###rrMODEL_T)bounddefaultc6\rSrSr%SrSr\\\S'S\4Sjr S\4Sjr S \SS 4S jr S S S\ \ \44Sjr\S\ \ \4S S SS4Sj5r\S\\S 44Sj5r\4S\ S\\S\4SjjrS\ S\SS 4SjrSSjrS\S\ S\4SjrS\S\ S\SS 4SjrSrg )InMemoryStateStore4a Async, in-memory, type-safe state manager for workflows. This store holds a single Pydantic model instance representing global workflow state. When the generic parameter is omitted, it defaults to [DictState][workflows.context.state_store.DictState] for flexible, dictionary-like usage. Thread-safety is ensured with an internal `asyncio.Lock`. Consumers can either perform atomic reads/writes via `get_state` and `set_state`, or make in-place, transactional edits via the `edit_state` context manager. Examples: Typed state model: ```python from pydantic import BaseModel from workflows.context.state_store import InMemoryStateStore class MyState(BaseModel): count: int = 0 store = InMemoryStateStore(MyState()) async with store.edit_state() as state: state.count += 1 ``` Dynamic state with `DictState`: ```python from workflows.context.state_store import InMemoryStateStore, DictState store = InMemoryStateStore(DictState()) await store.set("user.profile.name", "Ada") name = await store.get("user.profile.name") ``` See Also: - [Context.store][workflows.context.context.Context.store] )memory state_type initial_statecdXl[R"5Ul[ U5Ulg)N)_stateasyncioLock_locktyper.)r"r/s rr!InMemoryStateStore.__init__ds # \\^ }-rreturnc># URR5$7f)z{Return a shallow copy of the current state model. Returns: MODEL_T: A `.model_copy()` of the internal Pydantic model. )r1 model_copyr"s r get_stateInMemoryStateStore.get_stateis {{%%''sstateNc(# [U[UR55(d![S[UR535eURIShvN XlSSS5IShvN gNN!,IShvN (df  g=f7f)zReplace the current state model. Args: state (MODEL_T): New state of the same type as the existing model. Raises: ValueError: If the type differs from the existing state type. zState must be of type N) isinstancer5r1 ValueErrorr4r"r=s r set_stateInMemoryStateStore.set_stateqsW%dkk!2335d4;;6G5HIJ J:::K:::::sHABA4BA8# B.A6/B6B8B>B ? B B serializerrc[UR[5(a{0nURR5Hup4UR U5X#'M SU0[UR5R[UR5RS.$UR UR5nU[UR5R[UR5RS.$![ aHnX0R ;a#[R"SUS3[S9 SnAM[SUSU35eSnAff=f) aSerialize the state and model metadata for persistence. For `DictState`, each individual item is serialized using the provided serializer since values can be arbitrary Python objects. For other Pydantic models, defers to the serializer (e.g. JSON) which can leverage model-aware encoding. Args: serializer (BaseSerializer): Strategy used to encode values. Returns: dict[str, Any]: A payload suitable for [from_dict][workflows.context.state_store.InMemoryStateStore.from_dict]. z4Skipping serialization of known unserializable key: zY -- This is expected but will require this item to be set manually after deserialization.)categoryNz(Failed to serialize state value for key : _data) state_datar. state_module)r?r1ritems serialize Exceptionknown_unserializable_keyswarningswarnrr@r5rr)r"rDserialized_datakeyvalueeserialized_states rto_dictInMemoryStateStore.to_dicts2 dkk9 - - O"kk//1  +5+?+?+FO(2 '8"4;;/88 $T[[ 1 < <  *33DKK@ /"4;;/88 $T[[ 1 < < +! <<< RSVRWXtt%= !$B3%r!M sC66 E,E2EErUzInMemoryStateStore[MODEL_T]cU(dU"[55$URS05nURSS5nUS:XaKURS05n0nUR5HupxURU5Xg'M [US9n OURU5n U"U 5$![an [ SUSU 35eSn A ff=f) aXRestore a state store from a serialized payload. Args: serialized_state (dict[str, Any]): The payload produced by [to_dict][workflows.context.state_store.InMemoryStateStore.to_dict]. serializer (BaseSerializer): Strategy to decode stored values. Returns: InMemoryStateStore[MODEL_T]: A store with the reconstructed model. rIr.rrHz*Failed to deserialize state value for key rGN)rH)rgetrK deserializerMr@) clsrUrDrIr._data_serializeddeserialized_datarRrSrTstate_instances r from_dictInMemoryStateStore.from_dicts y{# #%)),; %)), D   $)~~gr:  " .446 -7-C-CE-J%*7'->?N'33J?N>""!$DSEA3Os,B&& C0CCc# URIShvN URnU7v XlSSS5IShvN gN,N!,IShvN (df  g=f7f)zEdit state transactionally under a lock. Yields the mutable model and writes it back on exit. This pattern avoids read-modify-write races and keeps updates atomic. Yields: MODEL_T: The current state model for in-place mutation. N)r4r1rAs r edit_stateInMemoryStateStore.edit_states2:::KKEKK :::::sAAAAA AAAAA A AApathr)c# U(aURS5O/n[U5[:a[S[S35eURIShvN UR nUHnUR XE5nM SSS5IShvN U$N=![a1 U[LaUssSSS5IShvN $SUS3n[U5ef=fNF!,IShvN (df  W$=f7f)aGet a nested value using dot-separated paths. Supports dict keys, list indices, and attribute access transparently at each segment. Args: path (str): Dot-separated path, e.g. "user.profile.name". default (Any): If provided, return this when the path does not exist; otherwise, raise `ValueError`. Returns: Any: The resolved value. Raises: ValueError: If the path is invalid and no default is provided or if the path depth exceeds limits. .Path length exceeds segmentsNzPath 'z' not found in state) splitlen MAX_DEPTHr@r4r1_traverse_steprMEllipsis)r"rdr)segmentsrSsegmentmsgs rrYInMemoryStateStore.gets$'+4::c? x=9 $3I;iHI I::: &![['G //?E (:   &(*"N::tf$89 o%  & :: sxAC.BC.C&B? C. C C.C(C) C.5B86C.<CCC.C+C  C+&C.rSc# U(d [S5eURS5n[U5[:a[S[S35eURIShvN UR nUSSHnUR XE5nM URXCSU5 SSS5IShvN gNT![[[[4a 0nURXEU5 UnMtf=fN>!,IShvN (df  g=f7f)aKSet a nested value using dot-separated paths. Intermediate containers are created as needed. Dicts, lists, tuples, and Pydantic models are supported where appropriate. Args: path (str): Dot-separated path to write. value (Any): Value to assign. Raises: ValueError: If the path is empty or exceeds the maximum depth. zPath cannot be emptyrfrgrhN) r@rirjrkr4r1rlKeyErrorAttributeError IndexError TypeError _assign_step)r"rdrSrncurrentro intermediates rsetInMemoryStateStore.set s34 4::c? x=9 $3I;iHI I:::kkG$CR=+"11'CG)   g|U ;::!.*iH+(*L%%g E*G +:::slADB1D C,6B3C, D+C*,D30C'#C,&C''C,*D,D2C5 3 D?Dc# URURR55IShvN gN![a [ S5ef=f7f)zReset the state to its type defaults. Raises: ValueError: If the model type cannot be instantiated from defaults (i.e., fields missing default values). Nz'State must have defaults for all fields)rBr1r#r r@r:s rclearInMemoryStateStore.clear1sF H..!6!6!89 9 9 HFG G Hs%A,868A8AAobjroc[U[5(aX$[U5nX$![[[ 4a Of=f[ X5$)zCFollow one segment into *obj* (dict key, list index, or attribute).)r?dictintr@rwrvgetattr)r"rroidxs rrl!InMemoryStateStore._traverse_step=sR c4 <  g,C8OIz2   s$$s*AAc[U[5(aX1U'g[U5nX1U'g![[[ 4a Of=f[ XU5 g)zJAssign *value* to *segment* of *obj* (dict key, list index, or attribute).N)r?rrr@rwrvsetattr)r"rrorSrs rrxInMemoryStateStore._assign_stepLsV c4  L  g,CH Iz2    e$s,AA)r4r1r.)r7N)rrrrr%rNrr'__annotations__r!r;rBrstrrrV classmethodr_rrrbrmrrYr{r~rlrxrrrrr+r+4s<'V!,W .g. ((  W    ."2.tCH~.`%##CH~%#;K%# &%#%#N .$"?  =E"c"HSM""H"rss*??/%*'   w  f67# #2 )9i @g%)g%r