tjldZddlmZddlZddlmZmZmZddlm Z m Z m Z ddl m Z ddlmZddlmZdd lmZmZmZmZmZmZmZmZmZmZmZmZmZm Z Gd d Z!dS) z5Synchronous client for managing threads in LangGraph.) annotationsN)IteratorMappingSequence)AnyLiteraloverload)_quote_path_param)SyncHttpClient)SyncThreadStream) 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~dgZdhddddiddnZdS)SyncThreadsClienta}Synchronous client for managing threads in LangGraph. This class provides methods to create, retrieve, and manage threads, which represent conversations or stateful interactions. ???+ example "Example" ```python client = get_sync_client(url="http://localhost:2024") thread = client.threads.create(metadata={"user_id": "123"}) ``` httpr returnNonec||_dSN)r)selfrs Y/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langgraph_sdk/_sync/threads.py__init__zSyncThreadsClient.__init__,s  N)includeheadersparams thread_idstrr&Sequence[str] | Noner'Mapping[str, str] | Noner(QueryParamTypes | Nonerci}|rd||d<|r|||jdt |||pdS)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_sync_client(url="http://localhost:2024") thread = 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#r4zSyncThreadsClient.get/sR(*  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] | Noneci} |r|| d<|s|ri|pi|rd|ini| d<|r|| d<|rd|D| d<|*t|ttfr |dd | d <n|| d <|jd | || S) aRCreate 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. Returns: The created `Thread`. ???+ example "Example Usage" ```python client = get_sync_client(url="http://localhost:2024") thread = 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# z7SyncThreadsClient.create...sN     '(k'(uuY'7'7'(|   r%)rFss r#rHz,SyncThreadsClient.create..sQ % % %   "#9      % % %r%r8Ndeleter:strategyr:z/threadsjsonr'r() isinstanceintfloatrpost) r"r6r)r7r8r9r:r'r(payloads r#createzSyncThreadsClient.createcsV#%  -#,GK  x #>r#-5=J))2#GJ   -#,GK   % %$ % % %GL ! ?#U|,, %),(!C!C!$y~~jwPV~WWWr%F)r:return_minimalr'r(Mapping[str, Any]rVLiteral[False]cdSr!rIr"r)r6r:rVr'r(s r#r3zSyncThreadsClient.updates r%)r:r'r( Literal[True]cdSr!rIrZs r#r3zSyncThreadsClient.updates sr%bool Thread | NonecdSr!rIrZs r#r3zSyncThreadsClient.updates r%cd|i}|*t|ttfr |dd|d<n||d<t|pi}|rd|d<|jdt |||pd| S) 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_sync_client(url="http://localhost:2024") thread = 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#r3zSyncThreadsClient.updatesH$.x"8 ?#U|,, %),(!C!C!$w}"--  9(8OH %y 6))44 6 6#+t    r%r1c`|jdt|||dS)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.threads.delete( thread_id="my_thread_id" ) ``` r0r1N)rrKr r"r)r'r(s r#rKzSyncThreadsClient.deletesF4  6))44 6 6PV      r% r) r6rCidsstatuslimitoffsetsort_by sort_orderselectextractr'r(rCrhriThreadStatus | NonerjrQrkrlThreadSortBy | NonermSortOrder | Nonernlist[ThreadSelectField] | Nonerodict[str, str] | None list[Thread]c ||d} |r|| d<|r|| d<|r|| d<|r|| d<|r|| d<|r|| d<| r| | d<| r| | d <|jd | | | S) 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_sync_client(url="http://localhost:2024") threads = client.threads.search( metadata={"number":1}, status="interrupted", limit=15, offset=5 ) ``` )rjrkr6rCrhrirlrmrnroz/threads/searchrNrrS)r"r6rCrhrirjrkrlrmrnror'r(rTs r#searchzSyncThreadsClient.search$sl# #   +"*GJ   ' &GH   ! GEN  ' &GH   )!(GI   /$.GL !  ' &GH   )!(GI y~~ GWV   r%)r6rCrir'r(cli}|r||d<|r||d<|r||d<|jd|||S)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/countrNrw)r"r6rCrir'r(rTs r#countzSyncThreadsClient.countqse*#%  +"*GJ   ' &GH   ' &GH y~~ 7GF   r%c`|jdt|dd||S)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_sync_client(url="http://localhost:2024") client.threads.copy( thread_id="my_thread_id" ) ``` r0z/copyNrNrrSr rfs r#copyzSyncThreadsClient.copysA6y~~ ;))44 ; ; ;    r%rK)rMr'r( thread_ids Sequence[str]rMrdict[str, Any]c\d|i}|dkr||d<|jd|||S)aXPrune 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_sync_client(url="http://localhost:2024") result = client.threads.prune( thread_ids=["thread_1", "thread_2"], ) print(result) # {'pruned_count': 2} ``` r~rKrMz/threads/prunerNrw)r"r~rMr'r(rTs r#prunezSyncThreadsClient.prunesNB *#  x  "*GJ y~~ 7GF   r%) subgraphsr'r( checkpointCheckpoint | None checkpoint_idrrc|r2|jdt|d||d||S|rVd|i}|ri|t|}|jdt|dt|||Sd|i}|ri|t|}|jdt|d||S) abGet 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. subgraphs: Include subgraphs states. headers: Optional custom headers to include with the request. Returns: The thread of the state. ???+ example "Example Usage" ```python client = get_sync_client(url="http://localhost:2024") thread_state = 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)rrrNrz/state/)r(r'/state)rrSr rbr4)r"r)rrrr'r( get_paramss r# get_statezSyncThreadsClient.get_states/Z  9>>K-i88KKK$.YGG "   %y1J <; ;d6ll; 9==c-i88ccARS`AaAacc!!  &y1J <; ;d6ll; 9==@-i88@@@!! r%)rErrr'r(&dict[str, Any] | Sequence[dict] | NonerErcd|i}|r||d<|r||d<|r||d<|jdt|d|||S)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. headers: Optional custom headers to include with the request. Returns: Response after updating a thread's state. ???+ example "Example Usage" ```python 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': {} } } ``` rCrrrEr0rrNr|) r"r)rCrErrr'r(rTs r# update_statezSyncThreadsClient.update_statecs` f#   5'4GO $  /$.GL !  )!(GI y~~ <))44 < < <    r%)rjbeforer6rr'r(rstr | Checkpoint | NoneMapping[str, Any] | Nonelist[ThreadState]cd|i}|r||d<|r||d<|r||d<|jdt|d|||S)aCGet 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. Returns: The state history of the `Thread`. ???+ example "Example Usage" ```python thread_state = client.threads.get_history( thread_id="my_thread_id", limit=5, before="my_timestamp", metadata={"name":"my_name"} ) ``` rjrr6rr0z/historyrNr|) r"r)rjrr6rr'r(rTs r# get_historyzSyncThreadsClient.get_historysL 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)aOpen a v3 thread-centric streaming session. 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 SSE request for this stream session. transport: event transport to use, `"sse"` (default) or `"websocket"`. Returns: A `SyncThreadStream` to use as a 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#streamzSyncThreadsClient.streamsi. 0 0 0FGG G#,#8iic$*,,>O>O%/(4$    r% run_modes) stream_mode last_event_idr'r(r-ThreadStreamMode | Sequence[ThreadStreamMode]rIterator[StreamPart]cd|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", stream_mode="run_modes", ): print(chunk) ``` rr0z/streamGETz Last-Event-IDr1)r3rrr )r"r)rrr'r(r5s r# join_streamzSyncThreadsClient.join_streamsD ;   (    ' ' '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*rrrr;r'r,r(r-rr)__name__ __module__ __qualname____doc__r$r4rUr r3rKrxrzr}rrrrrrrIr%r#rrsf  )-,0)- 2 2 2 2 2 2 n $/3KO#.2,0)-IXIXIXIXIXIXV /3).,0)-     X  /3,0)-     X  /3,0)-     X  /3$,0)-2 2 2 2 2 2 p-1)-       B$(&*'+'+15)-,0)-K K K K K K `&*,0)-      H-1)-       L#+,0)- ' ' ' ' ' ' X)-$( E  ,0)-EEEEEEX#(,$(,0)-= = = = = = F*.-1(,,0)-3 3 3 3 3 3 n!%! -1*.16! ! ! ! ! ! NFQ$(,0)-. . . . . . . . r%r)"r __future__rrcollections.abcrrrtypingrrr langgraph_sdk._shared.utilitiesr langgraph_sdk._sync.httpr langgraph_sdk._sync.streamr langgraph_sdk.schemar rrrrrrrrrrrrrrrIr%r#rst;;"""""" 7777777777))))))))))======333333777777$J  J  J  J  J  J  J  J  J  J  r%