tjrdZddlmZddlZddlmZmZmZddlm Z m Z m Z ddl m Z ddlmZddlmZdd lmZmZmZmZmZmZmZmZmZmZmZmZmZm Z Gd d Z!dS) z/Async client for managing threads in LangGraph.) annotationsN) AsyncIteratorMappingSequence)AnyLiteraloverload) HttpClient)AsyncThreadStream)_quote_path_param) CheckpointJsonOnConflictBehavior PruneStrategyQueryParamTypes SortOrder StreamPartThreadThreadSelectField ThreadSortBy ThreadState ThreadStatusThreadStreamModeThreadUpdateStateResponsec eZdZdZdodZdddddpdZddddddddddqdZeddddd drd$Zedddd%dsd'Zedddd%dtd*Zddddd dtd+Zddd,dud-Z ddddd.d/ddddddd0 dvdAZ ddddddBdwdCZ ddd,dudDZ dEdddFdxdLZ dyddddMdzdSZddddddTd{dXZd.ddddddYd|d^Z d}ddd_d`d~dgZddhdddiddnZdS) ThreadsClientaClient for managing threads in LangGraph. A thread maintains the state of a graph across multiple interactions/invocations (aka runs). It accumulates and persists the graph's state, allowing for continuity between separate invocations of the graph. ???+ example "Example" ```python client = get_client(url="http://localhost:2024")) new_thread = await client.threads.create(metadata={"user_id": "123"}) ``` httpr returnNonec||_dSN)r)selfrs Z/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langgraph_sdk/_async/threads.py__init__zThreadsClient.__init__-s  N)includeheadersparams thread_idstrr&Sequence[str] | Noner'Mapping[str, str] | Noner(QueryParamTypes | NonercKi}|rd||d<|r|||jdt |||pdd{VS)aGet a thread by ID. Args: thread_id: The ID of the thread to get. include: Additional fields to include in the response. Supported values: `"ttl"`. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: Thread object. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") thread = await client.threads.get( thread_id="my_thread_id" ) print(thread) ``` ```shell ----------------------------------------------------- { 'thread_id': 'my_thread_id', 'created_at': '2024-07-18T18:35:15.540834+00:00', 'updated_at': '2024-07-18T18:35:15.540834+00:00', 'metadata': {'graph_id': 'agent'} } ``` ,r& /threads/Nr'r()joinupdatergetr )r"r)r&r'r( query_paramss r#r4zThreadsClient.get0sT(*  8&)hhw&7&7L #  (    ' ' 'Y]] 6))44 6 6'4#         r%)metadatar) if_exists superstepsgraph_idttlr'r(r6r str | Noner7OnConflictBehavior | Noner84Sequence[dict[str, Sequence[dict[str, Any]]]] | Noner9r:int | Mapping[str, Any] | NonecKi} |r|| d<|s|ri|pi|rd|ini| d<|r|| d<|rd|D| d<|*t|ttfr |dd | d <n|| d <|jd | || d{VS) aCreate a new thread. Args: metadata: Metadata to add to thread. thread_id: ID of thread. If `None`, ID will be a randomly generated UUID. if_exists: How to handle duplicate creation. Defaults to 'raise' under the hood. Must be either 'raise' (raise error if duplicate), or 'do_nothing' (return existing thread). supersteps: Apply a list of supersteps when creating a thread, each containing a sequence of updates. Each update has `values` or `command` and `as_node`. Used for copying a thread between deployments. graph_id: Optional graph ID to associate with the thread. ttl: Optional time-to-live in minutes for the thread. You can pass an integer (minutes) or a mapping with keys `ttl` and optional `strategy` (defaults to "delete"). headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: The created thread. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") thread = await client.threads.create( metadata={"number":1}, thread_id="my-thread-id", if_exists="raise" ) ``` r)r9r6r7c6g|]}dd|dDiS)updatescXg|]'}|d|d|dd(S)valuescommandas_node)rCrDrE)r4).0us r# z3ThreadsClient.create...sN     '(k'(uuY'7'7'(|   r%)rFss r#rHz(ThreadsClient.create..sQ % % %   "#9      % % %r%r8Ndeleter:strategyr:z/threadsjsonr'r() isinstanceintfloatrpost) r"r6r)r7r8r9r:r'r(payloads r#createzThreadsClient.createesV#%  -#,GK  x #>r#-5=J))2#GJ   -#,GK   % %$ % % %GL ! ?#U|,, %),(!C!C!$Y^^ Wgf$         r%F)r:return_minimalr'r(Mapping[str, Any]rVLiteral[False]c KdSr!rIr"r)r6r:rVr'r(s r#r3zThreadsClient.updates r%)r:r'r( Literal[True]c KdSr!rIrZs r#r3zThreadsClient.updates sr%bool Thread | Nonec KdSr!rIrZs r#r3zThreadsClient.updates r%cKd|i}|*t|ttfr |dd|d<n||d<t|pi}|rd|d<|jdt |||pd| d{VS) aUpdate a thread. Args: thread_id: ID of thread to update. metadata: Metadata to merge with existing thread metadata. ttl: Optional time-to-live in minutes for the thread. You can pass an integer (minutes) or a mapping with keys `ttl` and optional `strategy` (defaults to "delete"). return_minimal: If `True`, request a 204 response with no body. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: The updated thread, or `None` when `return_minimal=True`. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") thread = await client.threads.update( thread_id="my-thread-id", metadata={"number":1}, ttl=43_200, ) ``` r6NrKrLr:zreturn=minimalPreferr0rN)rPrQrRdictrpatchr ) r"r)r6r:rVr'r(rTrequest_headerss r#r3zThreadsClient.updatesH$.x"8 ?#U|,, %),(!C!C!$w}"--  9(8OH %Y__ 6))44 6 6#+t %         r%r1cpK|jdt|||d{VdS)aDelete a thread. Args: thread_id: The ID of the thread to delete. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: `None` ???+ example "Example Usage" ```python client = get_client(url="http://localhost2024) await client.threads.delete( thread_id="my_thread_id" ) ``` r0r1N)rrKr r"r)r'r(s r#rKzThreadsClient.delete sh6i 6))44 6 6PV           r% r) r6rCidsstatuslimitoffsetsort_by sort_orderselectextractr'r(rCrhriThreadStatus | NonerjrQrkrlThreadSortBy | NonermSortOrder | Nonernlist[ThreadSelectField] | Nonerodict[str, str] | None list[Thread]c K||d} |r|| d<|r|| d<|r|| d<|r|| d<|r|| d<|r|| d<| r| | d<| r| | d <|jd | | | d {VS) aSearch for threads. Args: metadata: Thread metadata to filter on. values: State values to filter on. ids: List of thread IDs to filter by. status: Thread status to filter on. Must be one of 'idle', 'busy', 'interrupted' or 'error'. limit: Limit on number of threads to return. offset: Offset in threads table to start search from. sort_by: Sort by field. sort_order: Sort order. select: List of fields to include in the response. extract: Dictionary mapping aliases to JSONB paths to extract from thread data. Paths use dot notation for nested keys and bracket notation for array indices (e.g., `{"last_msg": "values.messages[-1]"}`). Extracted values are returned in an `extracted` field on each thread. Maximum 10 paths per request. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: List of the threads matching the search parameters. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") threads = await client.threads.search( metadata={"number":1}, status="interrupted", limit=15, offset=5 ) ``` )rjrkr6rCrhrirlrmrnroz/threads/searchrNNrrS)r"r6rCrhrirjrkrlrmrnror'r(rTs r#searchzThreadsClient.search)sn# #   +"*GJ   ' &GH   ! GEN  ' &GH   )!(GI   /$.GL !  ' &GH   )!(GI Y^^  $         r%)r6rCrir'r(c|Ki}|r||d<|r||d<|r||d<|jd|||d{VS)aCount threads matching filters. Args: metadata: Thread metadata to filter on. values: State values to filter on. status: Thread status to filter on. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: int: Number of threads matching the criteria. r6rCriz/threads/countrNNrw)r"r6rCrir'r(rTs r#countzThreadsClient.countzs*#%  +"*GJ   ' &GH   ' &GH Y^^ 7GF$         r%cpK|jdt|dd||d{VS)aCopy a thread. Args: thread_id: The ID of the thread to copy. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: `None` ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024) await client.threads.copy( thread_id="my_thread_id" ) ``` r0z/copyNrNrrSr rfs r#copyzThreadsClient.copysc6Y^^ ;))44 ; ; ; $         r%rK)rMr'r( thread_ids Sequence[str]rMrdict[str, Any]clKd|i}|dkr||d<|jd|||d{VS)aYPrune threads by ID. Args: thread_ids: List of thread IDs to prune. strategy: The prune strategy. `"delete"` removes threads entirely. `"keep_latest"` prunes old checkpoints but keeps threads and their latest state. Defaults to `"delete"`. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: A dict containing `pruned_count` (number of threads pruned). ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024") result = await client.threads.prune( thread_ids=["thread_1", "thread_2"], ) print(result) # {'pruned_count': 2} ``` r~rKrMz/threads/prunerNNrw)r"r~rMr'r(rTs r#prunezThreadsClient.prunespB *#  x  "*GJ Y^^ 7GF$         r%) subgraphsr'r( checkpointCheckpoint | None checkpoint_idrrcK|r8|jdt|d||d||d{VS|r\d|i}|ri|t|}|jdt|dt|||d{VSd|i}|ri|t|}|jdt|d ||d{VS) aGet the state of a thread. Args: thread_id: The ID of the thread to get the state of. checkpoint: The checkpoint to get the state of. checkpoint_id: (deprecated) The checkpoint ID to get the state of. subgraphs: Include subgraphs states. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: The thread of the state. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024) thread_state = await client.threads.get_state( thread_id="my_thread_id", checkpoint_id="my_checkpoint_id" ) print(thread_state) ``` ```shell ---------------------------------------------------------------------------------------------------------------------------------------------------------------------- { 'values': { 'messages': [ { 'content': 'how are you?', 'additional_kwargs': {}, 'response_metadata': {}, 'type': 'human', 'name': None, 'id': 'fe0a5778-cfe9-42ee-b807-0adaa1873c10', 'example': False }, { 'content': "I'm doing well, thanks for asking! I'm an AI assistant created by Anthropic to be helpful, honest, and harmless.", 'additional_kwargs': {}, 'response_metadata': {}, 'type': 'ai', 'name': None, 'id': 'run-159b782c-b679-4830-83c6-cef87798fe8b', 'example': False, 'tool_calls': [], 'invalid_tool_calls': [], 'usage_metadata': None } ] }, 'next': [], 'checkpoint': { 'thread_id': 'e2496803-ecd5-4e0c-a779-3226296181c2', 'checkpoint_ns': '', 'checkpoint_id': '1ef4a9b8-e6fb-67b1-8001-abd5184439d1' } 'metadata': { 'step': 1, 'run_id': '1ef4a9b8-d7da-679a-a45a-872054341df2', 'source': 'loop', 'writes': { 'agent': { 'messages': [ { 'id': 'run-159b782c-b679-4830-83c6-cef87798fe8b', 'name': None, 'type': 'ai', 'content': "I'm doing well, thanks for asking! I'm an AI assistant created by Anthropic to be helpful, honest, and harmless.", 'example': False, 'tool_calls': [], 'usage_metadata': None, 'additional_kwargs': {}, 'response_metadata': {}, 'invalid_tool_calls': [] } ] } }, 'user_id': None, 'graph_id': 'agent', 'thread_id': 'e2496803-ecd5-4e0c-a779-3226296181c2', 'created_by': 'system', 'assistant_id': 'fe096781-5601-53d2-b2f6-0d3403f7e9ca'}, 'created_at': '2024-07-25T15:35:44.184703+00:00', 'parent_config': { 'thread_id': 'e2496803-ecd5-4e0c-a779-3226296181c2', 'checkpoint_ns': '', 'checkpoint_id': '1ef4a9b8-d80d-6fa7-8000-9300467fad0f' } } ``` r0z/state/checkpoint)rrrNNrz/state/)r(r'/state)rrSr rbr4)r"r)rrrr'r( get_paramss r# get_statezThreadsClient.get_states\  K-i88KKK$.YGG (   %y1J <; ;d6ll; c-i88ccARS`AaAacc!'  &y1J <; ;d6ll; @-i88@@@!' r%)rErrr'r(&dict[str, Any] | Sequence[dict] | NonerErcKd|i}|r||d<|r||d<|r||d<|jdt|d|||d{VS) aUpdate the state of a thread. Args: thread_id: The ID of the thread to update. values: The values to update the state with. as_node: Update the state as if this node had just executed. checkpoint: The checkpoint to update the state of. checkpoint_id: (deprecated) The checkpoint ID to update the state of. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: Response after updating a thread's state. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024) response = await client.threads.update_state( thread_id="my_thread_id", values={"messages":[{"role": "user", "content": "hello!"}]}, as_node="my_node", ) print(response) ``` ```shell ---------------------------------------------------------------------------------------------------------------------------------------------------------------------- { 'checkpoint': { 'thread_id': 'e2496803-ecd5-4e0c-a779-3226296181c2', 'checkpoint_ns': '', 'checkpoint_id': '1ef4a9b8-e6fb-67b1-8001-abd5184439d1', 'checkpoint_map': {} } } ``` rCrrrEr0rrNNr|) r"r)rCrErrr'r(rTs r# update_statezThreadsClient.update_statemsf f#   5'4GO $  /$.GL !  )!(GI Y^^ <))44 < < < $         r%)rjbeforer6rr'r(rstr | Checkpoint | NoneMapping[str, Any] | Nonelist[ThreadState]cKd|i}|r||d<|r||d<|r||d<|jdt|d|||d{VS) azGet the state history of a thread. Args: thread_id: The ID of the thread to get the state history for. checkpoint: Return states for this subgraph. If empty defaults to root. limit: The maximum number of states to return. before: Return states before this checkpoint. metadata: Filter states by metadata key-value pairs. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: The state history of the thread. ???+ example "Example Usage" ```python client = get_client(url="http://localhost:2024) thread_state = await client.threads.get_history( thread_id="my_thread_id", limit=5, ) ``` rjrr6rr0z/historyrNNr|) r"r)rjrr6rr'r(rTs r# get_historyzThreadsClient.get_historysJ U#   ' &GH   +"*GJ   /$.GL !Y^^ >))44 > > > $         r%sse)r'run_start_timeout transport assistant_idr float | NonerLiteral['sse', 'websocket']r c |dvrtdt|j||ntt j||||du|S)uOpen a v3 thread-centric streaming session. When `thread_id` is None, a fresh UUIDv4 is minted client-side and included in the URL of subsequent `POST /threads/{thread_id}/...` calls. The server creates the thread row lazily on the first `run.start` (internal server detail — the SDK does not send any `if_not_exists` flag). The v3 protocol response carries only `run_id`, never `thread_id` — that's why the SDK mints the id client-side. Args: thread_id: optional explicit thread identifier. Defaults to a fresh UUIDv4. assistant_id: assistant the run will use. Required. headers: optional headers forwarded on every command and event request for this stream session. run_start_timeout: optional seconds to wait for an in-flight `run.start` before subscribing operations raise `asyncio.TimeoutError`. Defaults to `None` (wait forever). transport: event transport to use — `"sse"` (default) or `"websocket"`. Returns: An `AsyncThreadStream` to use as an async context manager. )r websocketz'transport must be 'sse' or 'websocket'.N)rr)rr'rexplicit_thread_idtransport_kind) ValueErrorr rr*uuiduuid4)r"r)rr'rrs r#streamzThreadsClient.streamsjD 0 0 0FGG G #,#8iic$*,,>O>O%/(4$    r% run_modes) last_event_id stream_moder'r(rr-ThreadStreamMode | Sequence[ThreadStreamMode]AsyncIterator[StreamPart]cKd|i}|r|||jdt|ddi|rd|ini|pi|S)aGet a stream of events for a thread. Args: thread_id: The ID of the thread to get the stream for. last_event_id: The ID of the last event to get. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: An iterator of stream parts. ???+ example "Example Usage" ```python for chunk in client.threads.join_stream( thread_id="my_thread_id", last_event_id="my_event_id", ): print(chunk) ``` rr0z/streamGETz Last-Event-IDr1)r3rrr )r"r)rrr'r(r5s r# join_streamzThreadsClient.join_streamsB ;   (    ' ' 'y =))44 = = = 7DLO]33"=b     r%)rr rr) r)r*r&r+r'r,r(r-rr)r6rr)r;r7r<r8r=r9r;r:r>r'r,r(r-rr)r)r*r6rWr:r>rVrXr'r,r(r-rr)r)r*r6rWr:r>rVr[r'r,r(r-rr)r)r*r6rWr:r>rVr]r'r,r(r-rr^)r)r*r'r,r(r-rr)r6rrCrrhr+rirprjrQrkrQrlrqrmrrrnrsrortr'r,r(r-rru) r6rrCrrirpr'r,r(r-rrQ) r~rrMrr'r,r(r-rr)NN)r)r*rrrr;rr]r'r,r(r-rr)r)r*rCrrEr;rrrr;r'r,r(r-rr)r)r*rjrQrrr6rrrr'r,r(r-rrr!) r)r;rr*r'r,rrrrrr ) r)r*rr;rrr'r,r(r-rr)__name__ __module__ __qualname____doc__r$r4rUr r3rKrxrzr}rrrrrrrIr%r#rrsd  )-,0)- 3 3 3 3 3 3 p $/3KO#.2,0)-K K K K K K Z /3).,0)-     X  /3,0)-     X  /3,0)-     X  /3$,0)-2 2 2 2 2 2 p-1)-       D$(&*'+'+15)-,0)-O O O O O O h&*,0)-      H-1)-       L#+,0)- ' ' ' ' ' ' X)-$( F  ,0)-FFFFFFZ#(,$(,0)-@ @ @ @ @ @ L*.-1(,,0)-2 2 2 2 2 2 l!%, -1*.16, , , , , , d%)EP,0)-- - - - - - - - r%r)"r __future__rrcollections.abcrrrtypingrrr langgraph_sdk._async.httpr langgraph_sdk._async.streamr langgraph_sdk._shared.utilitiesr langgraph_sdk.schemar rrrrrrrrrrrrrrrIr%r#rst55"""""" <<<<<<<<<<))))))))))000000999999======$`  `  `  `  `  `  `  `  `  `  r%