*j4ddlmZddlmZmZmZddlmZmZm Z ddl m Z ddl m Z mZddlmZmZddlmZddlmZmZdd lmZd ZddZGdde eedZedieGddeeZeddedZdddZdS) ) annotations) dataclassfieldreplace)AnyGenericcast) BaseStore) TypedDictUnpack)CONFCONFIG_KEY_RUNTIME) get_config) _DC_KWARGS StreamWriter)ContextT)Runtime get_runtime_rreturnNonecdSN)rs [/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/langgraph/runtime.py_no_op_stream_writerrsc8eZdZUded<ded<ded<ded<d S) _RuntimeOverridesrcontextBaseStore | Nonestorer stream_writerrpreviousN)__name__ __module__ __qualname____annotations__rrrrrs=MMMMMrrF)totalceZdZUdZedZded< edZded< eeZ ded < edZ d ed < ddZ ddZ dS)rat Convenience class that bundles run-scoped context and other runtime utilities. This class is injected into graph nodes and middleware. It provides access to `context`, `store`, `stream_writer`, and `previous`. !!! note "Accessing `config`" `Runtime` does not include `config`. To access `RunnableConfig`, you can inject it directly by adding a `config: RunnableConfig` parameter to your node function (recommended), or use `get_config()` from `langgraph.config`. !!! note `ToolRuntime` (from `langgraph.prebuilt`) is a subclass that provides similar functionality but is designed specifically for tools. It shares `context`, `store`, and `stream_writer` with `Runtime`, and adds tool-specific attributes like `config`, `state`, and `tool_call_id`. !!! version-added "Added in version v0.6.0" Example: ```python from typing import TypedDict from langgraph.graph import StateGraph from dataclasses import dataclass from langgraph.runtime import Runtime from langgraph.store.memory import InMemoryStore @dataclass class Context: # (1)! user_id: str class State(TypedDict, total=False): response: str store = InMemoryStore() # (2)! store.put(("users",), "user_123", {"name": "Alice"}) def personalized_greeting(state: State, runtime: Runtime[Context]) -> State: '''Generate personalized greeting using runtime context and store.''' user_id = runtime.context.user_id # (3)! name = "unknown_user" if runtime.store: if memory := runtime.store.get(("users",), user_id): name = memory.value["name"] response = f"Hello {name}! Nice to see you again." return {"response": response} graph = ( StateGraph(state_schema=State, context_schema=Context) .add_node("personalized_greeting", personalized_greeting) .set_entry_point("personalized_greeting") .set_finish_point("personalized_greeting") .compile(store=store) ) result = graph.invoke({}, context=Context(user_id="user_123")) print(result) # > {'response': 'Hello Alice! Nice to see you again.'} ``` 1. Define a schema for the runtime context. 2. Create a store to persist memories and other information. 3. Use the runtime context to access the `user_id`. N)defaultrr r!r"rr#rr$otherRuntime[ContextT]rct|jp|j|jp|j|jtur|jn|j|j|jn|jS)zMerge two runtimes together. If a value is not provided in the other runtime, the value from the current runtime is used. Nr r"r#r$)rr r"r#rr$)selfr,s rmergez Runtime.mergevse M1T\++"*>>> --#&+n&rGs""""""1111111111%%%%%%%%%%******////////CCCCCCCC''''''44444444%%%%%% $.--- 78#4E Zl*l*l*l*l*ghl*l*l*^'  &  r