c iSdZddlmZddlZddlmZmZddlmZmZddl m Z ddl m Z ddl mZdd lmZmZmZmZmZmZmZmZmZmZmZmZmZGd d ZdS) z*Synchronous cron client for LangGraph SDK.) annotationsN)MappingSequence)datetimetzinfo)Any)_resolve_timezone)SyncHttpClient) AllConfigContextCronCronSelectField CronSortBy DurabilityInputOnCompletionBehaviorQueryParamTypesRun SortOrder StreamModeceZdZdZdHdZddddddddddddddddddddId-Zdddddddddddddddddddd.dJd1Zddd2dKd4Zddddddddddddddddddd5dLd7Zdddd8d9dddddd: dMdEZ dddddFdNdGZ dS)OSyncCronClientaSynchronous client for managing cron jobs in LangGraph. This class provides methods to create and manage scheduled tasks (cron jobs) for automated graph executions. ???+ example "Example" ```python client = get_sync_client(url="http://localhost:8123") cron_job = client.crons.create_for_thread(thread_id="thread_123", assistant_id="asst_456", schedule="0 * * * *") ``` !!! note "Feature Availability" The crons client functionality is not supported on all licenses. Please check the relevant license documentation for the most up-to-date details on feature availability. http_clientr returnNonec||_dSN)http)selfrs C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langgraph_sdk/_sync/cron.py__init__zSyncCronClient.__init__0s  N)inputmetadataconfigcontextcheckpoint_duringinterrupt_beforeinterrupt_afterwebhookmultitask_strategyend_timeenabledtimezone stream_modestream_subgraphsstream_resumable durabilityheadersparams thread_idstr assistant_idscheduler$ Input | Noner%Mapping[str, Any] | Noner& Config | Noner'Context | Noner( bool | Noner)All | list[str] | Noner*r+ str | Noner,r-datetime | Noner.r/str | tzinfo | Noner0(StreamMode | Sequence[StreamMode] | Noner1r2r3Durability | Noner4Mapping[str, str] | Noner5QueryParamTypes | Nonercv|tjdtdid|d|d|d|d |d |d | d | d |d| d| d| r| ndd|dt |d|d|d|d|i}d|D}|jd|d|||S)a Create a cron job for a thread. Args: thread_id: the thread ID to run the cron job on. assistant_id: The assistant ID or graph name to use for the cron job. If using graph name, will default to first assistant created from that graph. schedule: The cron schedule to execute this job on. Schedules are interpreted in UTC unless a timezone is specified. input: The input to the graph. metadata: Metadata to assign to the cron job runs. config: The configuration for the assistant. context: Static context to add to the assistant. !!! version-added "Added in version 0.6.0" checkpoint_during: (deprecated) Whether to checkpoint during the run (or only at the end/interruption). interrupt_before: Nodes to interrupt immediately before they get executed. interrupt_after: Nodes to Nodes to interrupt immediately after they get executed. webhook: Webhook to call after LangGraph API call is done. multitask_strategy: Multitask strategy to use. Must be one of 'reject', 'interrupt', 'rollback', or 'enqueue'. end_time: The time to stop running the cron job. If not provided, the cron job will run indefinitely. enabled: Whether the cron job is enabled. By default, it is considered enabled. timezone: IANA timezone for the cron schedule. Accepts a string (e.g. 'America/New_York') or a ``datetime.tzinfo`` instance (e.g. ``ZoneInfo("America/New_York")``). stream_mode: The stream mode(s) to use. stream_subgraphs: Whether to stream output from subgraphs. stream_resumable: Whether to persist the stream chunks in order to resume the stream later. durability: Durability level for the run. Must be one of 'sync', 'async', or 'exit'. "async" means checkpoints are persisted async while next graph step executes, replaces checkpoint_during=True "sync" means checkpoints are persisted sync after graph step executes, replaces checkpoint_during=False "exit" means checkpoints are only persisted when the run exits, does not save intermediate steps headers: Optional custom headers to include with the request. Returns: The cron `Run`. ???+ example "Example Usage" ```python client = get_sync_client(url="http://localhost:8123") cron_run = client.crons.create_for_thread( thread_id="my-thread-id", assistant_id="agent", schedule="27 15 * * *", input={"messages": [{"role": "user", "content": "hello!"}]}, metadata={"name":"my_run"}, context={"model_name": "openai"}, interrupt_before=["node_to_stop_before_1","node_to_stop_before_2"], interrupt_after=["node_to_stop_after_1","node_to_stop_after_2"], webhook="https://my.fake.webhook.com", multitask_strategy="interrupt", enabled=True ) ``` Nd`checkpoint_during` is deprecated and will be removed in a future version. Use `durability` instead. stacklevelr9r$r&r%r'r8r)r*r(r+r,r-r.r/r0r1r2r3ci|] \}}||| Sr.0kvs r! z4SyncCronClient.create_for_thread..EEEDAqq}1a}}}r#z /threads/ /runs/cronsjsonr4r5warningswarnDeprecationWarning isoformatr itemsrpost)r r6r8r9r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5payloads r!create_for_threadz SyncCronClient.create_for_thread3s\  ( Mv"       U  f    w  L    0    !2  w  !"4  B**,,,d  w  )(33  ;  0! "  0# $ *%  (FEGMMOOEEEy~~ . . . .    r#)r$r%r&r'r(r)r*r+on_run_completedr,r-r.r/r0r1r2r3r4r5r`OnCompletionBehavior | Nonecp|tjdtdid|d|d|d|d |d |d |d | d | d|d| d| d| r| ndd|dt |d|d|||d}d|D}|jd|||S)a Create a cron run. Args: assistant_id: The assistant ID or graph name to use for the cron job. If using graph name, will default to first assistant created from that graph. schedule: The cron schedule to execute this job on. Schedules are interpreted in UTC unless a timezone is specified. input: The input to the graph. metadata: Metadata to assign to the cron job runs. config: The configuration for the assistant. context: Static context to add to the assistant. !!! version-added "Added in version 0.6.0" checkpoint_during: (deprecated) Whether to checkpoint during the run (or only at the end/interruption). interrupt_before: Nodes to interrupt immediately before they get executed. interrupt_after: Nodes to Nodes to interrupt immediately after they get executed. webhook: Webhook to call after LangGraph API call is done. on_run_completed: What to do with the thread after the run completes. Must be one of 'delete' (default) or 'keep'. 'delete' removes the thread after execution. 'keep' creates a new thread for each execution but does not clean them up. Clients are responsible for cleaning up kept threads. multitask_strategy: Multitask strategy to use. Must be one of 'reject', 'interrupt', 'rollback', or 'enqueue'. end_time: The time to stop running the cron job. If not provided, the cron job will run indefinitely. enabled: Whether the cron job is enabled. By default, it is considered enabled. timezone: IANA timezone for the cron schedule. Accepts a string (e.g. 'America/New_York') or a ``datetime.tzinfo`` instance (e.g. ``ZoneInfo("America/New_York")``). stream_mode: The stream mode(s) to use. stream_subgraphs: Whether to stream output from subgraphs. stream_resumable: Whether to persist the stream chunks in order to resume the stream later. durability: Durability level for the run. Must be one of 'sync', 'async', or 'exit'. "async" means checkpoints are persisted async while next graph step executes, replaces checkpoint_during=True "sync" means checkpoints are persisted sync after graph step executes, replaces checkpoint_during=False "exit" means checkpoints are only persisted when the run exits, does not save intermediate steps headers: Optional custom headers to include with the request. Returns: The cron `Run`. ???+ example "Example Usage" ```python client = get_sync_client(url="http://localhost:8123") cron_run = client.crons.create( assistant_id="agent", schedule="27 15 * * *", input={"messages": [{"role": "user", "content": "hello!"}]}, metadata={"name":"my_run"}, context={"model_name": "openai"}, checkpoint_during=True, interrupt_before=["node_to_stop_before_1","node_to_stop_before_2"], interrupt_after=["node_to_stop_after_1","node_to_stop_after_2"], webhook="https://my.fake.webhook.com", multitask_strategy="interrupt", enabled=True ) ``` NrHrIrJr9r$r&r%r'r8r)r*r+r(r`r,r-r.r/r0r1)r2r3ci|] \}}||| SrrMrNs r!rRz)SyncCronClient.create..rSr#rTrUrW)r r8r9r$r%r&r'r(r)r*r+r`r,r-r.r/r0r1r2r3r4r5r^s r!createzSyncCronClient.creates{d  ( Mv"       U  f    w  L    0    w  !2   0  !"4  B**,,,d  w  )(33 ;! "  0# $!1$'   *FEGMMOOEEEy~~    r#r4r5cron_idcF|jd|||dS)aDelete a cron. Args: cron_id: The cron ID 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_sync_client(url="http://localhost:8123") client.crons.delete( cron_id="cron_to_delete" ) ``` /runs/crons/reN)rdelete)r rfr4r5s r!rizSyncCronClient.deletes06 11176RRRRRr#)r9r-r$r%r&r'r+r)r*r`r.r/r0r1r2r3r4r5rc*id|d|r|ndd|d|d|d|d|d | d | d | d | d t| d|d|d|d|}d|D}|jd||||S)a`Update a cron job by ID. Args: cron_id: The cron ID to update. schedule: The cron schedule to execute this job on. Schedules are interpreted in UTC unless a timezone is specified. end_time: The end date to stop running the cron. input: The input to the graph. metadata: Metadata to assign to the cron job runs. config: The configuration for the assistant. context: Static context added to the assistant. webhook: Webhook to call after LangGraph API call is done. interrupt_before: Nodes to interrupt immediately before they get executed. interrupt_after: Nodes to interrupt immediately after they get executed. on_run_completed: What to do with the thread after the run completes. Must be one of 'delete' or 'keep'. 'delete' removes the thread after execution. 'keep' creates a new thread for each execution but does not clean them up. enabled: Enable or disable the cron job. timezone: IANA timezone for the cron schedule. Accepts a string (e.g. 'America/New_York') or a ``datetime.tzinfo`` instance (e.g. ``ZoneInfo("America/New_York")``). stream_mode: The stream mode(s) to use. stream_subgraphs: Whether to stream output from subgraphs. stream_resumable: Whether to persist the stream chunks in order to resume the stream later. durability: Durability level for the run. Must be one of 'sync', 'async', or 'exit'. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: The updated cron job. ???+ example "Example Usage" ```python client = get_sync_client(url="http://localhost:8123") updated_cron = client.crons.update( cron_id="1ef3cefa-4c09-6926-96d0-3dc97fd5e39b", schedule="0 10 * * *", enabled=False, ) ``` r9r-Nr$r%r&r'r+r)r*r`r.r/r0r1r2r3ci|] \}}||| SrrMrNs r!rRz)SyncCronClient.update..rSr#rhrU)r[r r\rpatch)r rfr9r-r$r%r&r'r+r)r*r`r.r/r0r1r2r3r4r5r^s r!updatezSyncCronClient.update4s8B  B**,,,d  U    f  w   w   0     0  w  )(33  ;   0   0 *! $FEGMMOOEEEy $7 $ $    r# r) r8r6r.limitoffsetsort_by sort_orderselectr4r5rointrprqCronSortBy | NonerrSortOrder | Nonerslist[CronSelectField] | None list[Cron]c |||||d} |r|| d<|r|| d<|r|| d<d| D} |jd| | | S)aGet a list of cron jobs. Args: assistant_id: The assistant ID or graph name to search for. thread_id: the thread ID to search for. enabled: Whether the cron job is enabled. limit: The maximum number of results to return. offset: The number of results to skip. headers: Optional custom headers to include with the request. Returns: The list of cron jobs returned by the search, ???+ example "Example Usage" ```python client = get_sync_client(url="http://localhost:8123") cron_jobs = client.crons.search( assistant_id="my_assistant_id", thread_id="my_thread_id", enabled=True, limit=5, offset=5, ) print(cron_jobs) ``` ```shell ---------------------------------------------------------- [ { 'cron_id': '1ef3cefa-4c09-6926-96d0-3dc97fd5e39b', 'assistant_id': 'my_assistant_id', 'thread_id': 'my_thread_id', 'user_id': None, 'payload': { 'input': {'start_time': ''}, 'schedule': '4 * * * *', 'assistant_id': 'my_assistant_id' }, 'schedule': '4 * * * *', 'next_run_date': '2024-07-25T17:04:00+00:00', 'end_time': None, 'created_at': '2024-07-08T06:02:23.073257+00:00', 'updated_at': '2024-07-08T06:02:23.073257+00:00' } ] ``` )r8r6r.rorprqrrrsci|] \}}||| SrrMrNs r!rRz)SyncCronClient.search..rSr#z/runs/crons/searchrU)r\rr]) r r8r6r.rorprqrrrsr4r5r^s r!searchzSyncCronClient.searchsD)" # #   )!(GI   /$.GL !  ' &GH EEGMMOOEEEy~~ w   r#)r8r6r4r5c^i}|r||d<|r||d<|jd|||S)a|Count cron jobs matching filters. Args: assistant_id: Assistant ID to filter by. thread_id: Thread ID to filter by. headers: Optional custom headers to include with the request. params: Optional query parameters to include with the request. Returns: int: Number of crons matching the criteria. r8r6z/runs/crons/countrU)rr])r r8r6r4r5r^s r!countzSyncCronClient.countsS&#%  3&2GN #  -#,GK y~~ gwv   r#)rr rr),r6r7r8r7r9r7r$r:r%r;r&r<r'r=r(r>r)r?r*r?r+r@r,r@r-rAr.r>r/rBr0rCr1r>r2r>r3rDr4rEr5rFrr),r8r7r9r7r$r:r%r;r&r<r'r=r(r>r)r?r*r?r+r@r`rar,r@r-rAr.r>r/rBr0rCr1r>r2r>r3rDr4rEr5rFrr)rfr7r4rEr5rFrr)(rfr7r9r@r-rAr$r:r%r;r&r<r'r=r+r@r)r?r*r?r`rar.r>r/rBr0rCr1r>r2r>r3rDr4rEr5rFrr)r8r@r6r@r.r>rortrprtrqrurrrvrsrwr4rEr5rFrrx) r8r@r6r@r4rEr5rFrrt) __name__ __module__ __qualname____doc__r"r_rdrirmr{r}rMr#r!rrs$    #-1 $"&)-3726")-$(#(,@D(,(,(,,0)-/o o o o o o l#-1 $"&)-3726"8<)-$(#(,@D(,(,(,,0)-/q q q q q q n-1)- SSSSSSB $$("-1 $"&"37268<#(,@D(,(,(,,0)-+Y Y Y Y Y Y |$( $#%)'+/3,0)-Q Q Q Q Q Q l$( $,0)-         r#r)r __future__rrXcollections.abcrrrrtypingrlanggraph_sdk._shared.utilitiesr langgraph_sdk._sync.httpr langgraph_sdk.schemar r r rrrrrrrrrrrrMr#r!rs^00""""""--------%%%%%%%%======333333"_ _ _ _ _ _ _ _ _ _ r#