WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-20 22:52:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

sdk-py: Update return type annotation for Thread.update_state() methods. (#2050)

Browse Source 
### Summary
The response body of the endpoint `POST /threads/{thread_id}/state`
looks like this:
```
{
    "checkpoint": {
        "thread_id": "e2496803-ecd5-4e0c-a779-3226296181c2",
        "checkpoint_ns": "",
        "checkpoint_id": "1ef4a9b8-e6fb-67b1-8001-abd5184439d1",
        "checkpoint_map": {}
    }
}
```
This commit is contained in:
Andrew Nguonly
2024-10-09 14:33:14 +02:00
committed by Tat Dat Duong
parent 0628c6402f
commit 598bb5a641
2 changed files with 38 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/sdk-py/langgraph_sdk/client.py
+31 -6
View File
@@ -53,6 +53,7 @@ from langgraph_sdk.schema import (
Thread,
ThreadState,
ThreadStatus,
ThreadUpdateStateResponse,
)
logger = logging.getLogger(__name__)
@@ -1060,7 +1061,7 @@ class ThreadsClient:
as_node: Optional[str] = None,
checkpoint: Optional[Checkpoint] = None,
checkpoint_id: Optional[str] = None, # deprecated
) -> None:
) -> ThreadUpdateStateResponse:
"""Update the state of a thread.
Args:
@@ -1070,15 +1071,27 @@ class ThreadsClient:
checkpoint: The checkpoint to update the state of.
Returns:
None
ThreadUpdateStateResponse: Response after updating a thread's state.
Example Usage:
await client.threads.update_state(
response = await client.threads.update_state(
thread_id="my_thread_id",
values={"messages":[{"role": "user", "content": "hello!"}]},
as_node="my_node",
)
print(response)
----------------------------------------------------------------------------------------------------------------------------------------------------------------------
{
'checkpoint': {
'thread_id': 'e2496803-ecd5-4e0c-a779-3226296181c2',
'checkpoint_ns': '',
'checkpoint_id': '1ef4a9b8-e6fb-67b1-8001-abd5184439d1',
'checkpoint_map': {}
}
}
""" # noqa: E501
payload: Dict[str, Any] = {
@@ -3109,7 +3122,7 @@ class SyncThreadsClient:
as_node: Optional[str] = None,
checkpoint: Optional[Checkpoint] = None,
checkpoint_id: Optional[str] = None, # deprecated
) -> None:
) -> ThreadUpdateStateResponse:
"""Update the state of a thread.
Args:
@@ -3119,15 +3132,27 @@ class SyncThreadsClient:
checkpoint: The checkpoint to update the state of.
Returns:
None
ThreadUpdateStateResponse: Response after updating a thread's state.
Example Usage:
await client.threads.update_state(
response = client.threads.update_state(
thread_id="my_thread_id",
values={"messages":[{"role": "user", "content": "hello!"}]},
as_node="my_node",
)
print(response)
----------------------------------------------------------------------------------------------------------------------------------------------------------------------
{
'checkpoint': {
'thread_id': 'e2496803-ecd5-4e0c-a779-3226296181c2',
'checkpoint_ns': '',
'checkpoint_id': '1ef4a9b8-e6fb-67b1-8001-abd5184439d1',
'checkpoint_map': {}
}
}
""" # noqa: E501
payload: Dict[str, Any] = {
libs/sdk-py/langgraph_sdk/schema.py
+7
View File
@@ -207,6 +207,13 @@ class ThreadState(TypedDict):
"""Tasks to execute in this step. If already attempted, may contain an error."""
class ThreadUpdateStateResponse(TypedDict):
"""Represents the response from updating a thread's state."""
checkpoint: Checkpoint
"""Checkpoint of the latest state."""
class Run(TypedDict):
"""Represents a single execution run."""