i6ddlmZmZmZmZmZmZddlZddlm Z m Z ddl m Z ddl mZddddddddeeed ed eed eed eed eedeeddeeee fdeeeffdZdeeefdeeeffdZdddddddde deeed ed eed eed eed eedeeddeeee fde fdZde dede fdZdddde dededeede f dZdddddddddde deeed ed eed eededeed eed eedeeddeeee fde fdZde dedefd ZdS)!)AnyDictListLiteralOptionalUnionN) AgentResponseAgentWebhookConfig)AsyncHttpClient)_normalize_schema)schema integration max_creditsstrict_constrain_to_urlsmodelwebhookurlspromptrrrrr)z spark-1-proz spark-1-minirreturnci}|||d<||d<|)r7getattrrM) r0rrrrr?r@rrrrstartedr8s r$agentrRls  !9         GWdD ) )F FF-QXYYY Y Y Y Y Y YYr&cK|d|d{V}|ddS)a Cancel a running agent job. Args: client: Async HTTP client instance job_id: ID of the agent job to cancel Returns: bool: True if the agent was cancelled, False otherwise Raises: Exception: If the cancellation fails r:NsuccessF)deleter5r;)r0r8r6s r$ cancel_agentrVsP4F4455 5 5 5 5 5 5D 99;;??9e , ,,r&)typingrrrrrrrFtypesr r utils.http_client_asyncr utils.validationr rintboolr%r/r7r<rMrRrVr3r&r$r]s<<<<<<<<<<<<<<<<66666666666666111111!!%!%/3>B8<%%% 49 % % SM % # % # %'tn% G9: ;%eC!334 5% #s(^%%%%PtCH~$sCx.!!%!%/3>B8<$$$ $ 49 $  $ SM $ # $#$'tn$ G9: ;$eC!334 5$$$$$6$?$C$M$$$$! 333 3 3 3 c] 3  3333,!!%!!%/3>B8<ZZZ Z 49 Z  Z SM Z # ZZc]Z#Z'tnZ G9: ;ZeC!334 5ZZZZZ>---------r&