wehub-resource-sync
a7d6d88f6f
CI / changes (push) Has been cancelled
CI / cd libs/checkpoint (push) Has been cancelled
CI / cd libs/checkpoint-conformance (push) Has been cancelled
CI / cd libs/checkpoint-postgres (push) Has been cancelled
CI / cd libs/checkpoint-sqlite (push) Has been cancelled
CI / cd libs/cli (push) Has been cancelled
CI / cd libs/prebuilt (push) Has been cancelled
CI / cd libs/sdk-py (push) Has been cancelled
CI / cd libs/langgraph (push) Has been cancelled
CI / Check SDK methods matching (push) Has been cancelled
CI / Check CLI schema hasn't changed #3.13 (push) Has been cancelled
CI / CLI integration test (push) Has been cancelled
CI / sdk-py integration test (push) Has been cancelled
CI / CI Success (push) Has been cancelled
baseline / benchmark (push) Has been cancelled
Deploy Redirects to GitHub Pages / deploy (push) Has been cancelled
28 lines
683 B
Python
28 lines
683 B
Python
from typing_extensions import NotRequired, Required, TypedDict
|
|
|
|
from langgraph.graph import StateGraph
|
|
from langgraph.managed import RemainingSteps
|
|
|
|
|
|
class StatePlain(TypedDict):
|
|
remaining_steps: RemainingSteps
|
|
|
|
|
|
class StateNotRequired(TypedDict):
|
|
remaining_steps: NotRequired[RemainingSteps]
|
|
|
|
|
|
class StateRequired(TypedDict):
|
|
remaining_steps: Required[RemainingSteps]
|
|
|
|
|
|
def test_managed_values_recognized() -> None:
|
|
graph = StateGraph(StatePlain)
|
|
assert "remaining_steps" in graph.managed
|
|
|
|
graph = StateGraph(StateNotRequired)
|
|
assert "remaining_steps" in graph.managed
|
|
|
|
graph = StateGraph(StateRequired)
|
|
assert "remaining_steps" in graph.managed
|