c iddZddlmZddlmZmZmZddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZGddZd S) z/Async client for managing threads in LangGraph.) annotations) AsyncIteratorMappingSequence)Any) HttpClient) CheckpointJsonOnConflictBehavior PruneStrategyQueryParamTypes SortOrder StreamPartThreadThreadSelectField ThreadSortBy ThreadState ThreadStatusThreadStreamModeThreadUpdateStateResponsec  eZdZdZd^dZddddd_dZdddddddddd`dZdddddad!Zddd"dbd#Zddddd$d%ddddddd& dcd7Z dddddd8ddd9Z ddd"dbd:Z d;ddd<dedBZ dfdCdddDdgdKZ ddddddLdhdPZd$ddddddQdidVZddWdddXdjd]ZdS)k 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"}) ``` httprreturnNonec||_dS)N)r)selfrs C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\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|||jd|||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#)joinupdaterget)rr$r!r"r# query_paramss rr/zThreadsClient.get-sT(*  8&)hhw&7&7L #  (    ' ' 'Y]] # # #'4#         r )metadatar$ if_exists superstepsgraph_idttlr"r#r1r str | Noner2OnConflictBehavior | Noner34Sequence[dict[str, Sequence[dict[str, Any]]]] | Noner4r5int | 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$r4r1r2c6g|]}dd|dDiS)updatescXg|]'}|d|d|dd(S)valuescommandas_node)r>r?r@)r/).0us r z3ThreadsClient.create...sN     '(k'(uuY'7'7'(|   r )rAss rrCz(ThreadsClient.create..sQ % % %   "#9      % % %r r3Ndeleter5strategyr5z/threadsjsonr"r#) isinstanceintfloatrpost) rr1r$r2r3r4r5r"r#payloads rcreatezThreadsClient.createbsV#%  -#,GK  x #>r#-5=J))2#GJ   -#,GK   % %$ % % %GL ! ?#U|,, %),(!C!C!$Y^^ Wgf$         r )r5r"r#Mapping[str, Any]cKd|i}|*t|ttfr |dd|d<n||d<|jd||||d{VS)amUpdate 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"). 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.update( thread_id="my-thread-id", metadata={"number":1}, ttl=43_200, ) ``` r1NrFrGr5r+rI)rKrLrMrpatch)rr$r1r5r"r#rOs rr.zThreadsClient.updatesD$.x"8 ?#U|,, %),(!C!C!$Y__ # # # %         r r,cVK|jd|||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" ) ``` r+r,N)rrFrr$r"r#s rrFzThreadsClient.deletesG6i6966PVWWWWWWWWWWWr r) r1r>idsstatuslimitoffsetsort_by sort_orderselectextractr"r#r>rWrXThreadStatus | NonerYrLrZr[ThreadSortBy | Noner\SortOrder | Noner]list[ThreadSelectField] | Noner^dict[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 ) ``` )rYrZr1r>rWrXr[r\r]r^z/threads/searchrINrrN)rr1r>rWrXrYrZr[r\r]r^r"r#rOs rsearchzThreadsClient.searchsn# #   +"*GJ   ' &GH   ! GEN  ' &GH   )!(GI   /$.GL !  ' &GH   )!(GI Y^^  $         r )r1r>rXr"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. r1r>rXz/threads/countrINrf)rr1r>rXr"r#rOs rcountzThreadsClient.countLs*#%  +"*GJ   ' &GH   ' &GH Y^^ 7GF$         r cVK|jd|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" ) ``` r+z/copyNrIrfrUs rcopyzThreadsClient.copylsW6Y^^ ( ( ( (tWV$         r rF)rHr"r# thread_ids Sequence[str]rHr dict[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} ``` rlrFrHz/threads/prunerINrf)rrlrHr"r#rOs rprunezThreadsClient.prunespB *#  x  "*GJ Y^^ 7GF$         r F) subgraphsr"r# checkpointCheckpoint | None checkpoint_idrqboolrchK|r+|jd|d||d||d{VS|rBd|i}|ri|t|}|jd|d|||d{VSd|i}|ri|t|}|jd|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' } } ``` r+z/state/checkpoint)rrrqrINrqz/state/)r#r"/state)rrNdictr/)rr$rrrtrqr"r# get_paramss r get_statezThreadsClient.get_statesh\  8I888$.YGG (   %y1J <; ;d6ll; =I==m==!'  &y1J <; ;d6ll; -I---!' r )r@rrrtr"r#&dict[str, Any] | Sequence[dict] | Noner@rcKd|i}|r||d<|r||d<|r||d<|jd|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': {} } } ``` r>rtrrr@r+rwrINrf) rr$r>r@rrrtr"r#rOs r update_statezThreadsClient.update_state<sf f#   5'4GO $  /$.GL !  )!(GI Y^^ ) ) ) )QW$         r )rYbeforer1rrr"r#r~str | Checkpoint | NoneMapping[str, Any] | Nonelist[ThreadState]cKd|i}|r||d<|r||d<|r||d<|jd|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, ) ``` rYr~r1rrr+z/historyrINrf) rr$rYr~r1rrr"r#rOs r get_historyzThreadsClient.get_history{sJ U#   ' &GH   +"*GJ   /$.GL !Y^^ + + + + $         r run_modes) last_event_id stream_moder"r#rr-ThreadStreamMode | Sequence[ThreadStreamMode]AsyncIterator[StreamPart]cKd|i}|r|||jd|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) ``` rr+z/streamGETz Last-Event-IDr,)r.rstream)rr$rrr"r#r0s r join_streamzThreadsClient.join_streamsB ;   (    ' ' 'y * * * * 7DLO]33"=b     r )rrrr) r$r%r!r&r"r'r#r(rr)r1r r$r6r2r7r3r8r4r6r5r9r"r'r#r(rr) r$r%r1rQr5r9r"r'r#r(rr)r$r%r"r'r#r(rr)r1r r>r rWr&rXr_rYrLrZrLr[r`r\rar]rbr^rcr"r'r#r(rrd) r1r r>r rXr_r"r'r#r(rrL) rlrmrHr r"r'r#r(rrn)NN)r$r%rrrsrtr6rqrur"r'r#r(rr)r$r%r>r{r@r6rrrsrtr6r"r'r#r(rr)r$r%rYrLr~rr1rrrrsr"r'r#r(rr) r$r%rr6rrr"r'r#r(rr)__name__ __module__ __qualname____doc__rr/rPr.rFrgrirkrprzr}rrrDr rrrs  )-,0)- 3 3 3 3 3 3 p $/3KO#.2,0)-K K K K K K d/3,0)-- - - - - - f-1)- XXXXXX@$(&*'+'+15)-,0)-O O O O O O h&*,0)-      H-1)-       F#+,0)- ' ' ' ' ' ' X)-$( F  ,0)-FFFFFFZ#(,$(,0)-= = = = = = F*.-1(,,0)-2 2 2 2 2 2 p%)EP,0)-- - - - - - - - r rN)r __future__rcollections.abcrrrtypingrlanggraph_sdk._async.httprlanggraph_sdk.schemar r r r r rrrrrrrrrrrDr rrs;55""""""<<<<<<<<<<000000$A  A  A  A  A  A  A  A  A  A  r