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

Merge pull request #1904 from langchain-ai/nc/29sep/thread-history-subgraph

Browse Source 
sdk-py: Add arg to get history for subgraphs
This commit is contained in:
Nuno Campos
2024-09-29 19:32:23 -07:00
committed by GitHub
parent 620cffd62d 7fba675e8a
commit 904904c419
1 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/sdk-py/langgraph_sdk/client.py
+16 -8
View File
@@ -1031,17 +1031,20 @@ class ThreadsClient:
async def get_history(
self,
thread_id: str,
*,
limit: int = 10,
before: Optional[str] = None,
before: Optional[str | Checkpoint] = None,
metadata: Optional[dict] = None,
checkpoint: Optional[Checkpoint] = None,
) -> list[ThreadState]:
"""Get the state history of a thread.
Args:
thread_id: The ID of the thread to get the state of.
checkpoint: Get history for this subgraph. If empty defaults to root.
limit: The maximum number of results to return.
before: Thread timestamp to get history before.
metadata: The metadata of the thread history to get.
before: Get history before this checkpoint.
metadata: Filter checkpoints by metadata.
Returns:
list[ThreadState]: the state history of the thread.
@@ -1051,8 +1054,6 @@ class ThreadsClient:
thread_state = await client.threads.get_history(
thread_id="my_thread_id",
limit=5,
before="my_timestamp",
metadata={"name":"my_name"}
)
""" # noqa: E501
@@ -1063,6 +1064,8 @@ class ThreadsClient:
payload["before"] = before
if metadata:
payload["metadata"] = metadata
if checkpoint:
payload["checkpoint"] = checkpoint
return await self.http.post(f"/threads/{thread_id}/history", json=payload)
@@ -2994,17 +2997,20 @@ class SyncThreadsClient:
def get_history(
self,
thread_id: str,
*,
limit: int = 10,
before: Optional[str] = None,
before: Optional[str | Checkpoint] = None,
metadata: Optional[dict] = None,
checkpoint: Optional[Checkpoint] = None,
) -> list[ThreadState]:
"""Get the state history of a thread.
Args:
thread_id: The ID of the thread to get the state of.
checkpoint: Get history for this subgraph. If empty defaults to root.
limit: The maximum number of results to return.
before: Thread timestamp to get history before.
metadata: The metadata of the thread history to get.
before: Get history before this checkpoint.
metadata: Filter checkpoints by metadata.
Returns:
list[ThreadState]: the state history of the thread.
@@ -3026,6 +3032,8 @@ class SyncThreadsClient:
payload["before"] = before
if metadata:
payload["metadata"] = metadata
if checkpoint:
payload["checkpoint"] = checkpoint
return self.http.post(f"/threads/{thread_id}/history", json=payload)