jP_ddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z d d lmZmZej"ej$d ZGd d ZGddZy)N)AsyncClientWrapperSyncClientWrapper)RequestOptions)AgentBlueprintPublic)AgentBlueprintWrite)AgentConfigEnv)BlueprintPageHistory)AsyncRawAgentConfigsClientRawAgentConfigsClient.c eZdZdefdZedefdZeeeddde de je d e je d e je d e je ddf d Zdd de de jed e je ddfdZdddde de de je d e je def dZdd de de de d e je ddf dZdd de de d e je ddfdZdddde de je d e je defdZdddde de de je d e je def dZddddde de jede jed e je def dZdd de d e je defdZdddde de je d e je defd Zy)!AgentConfigsClientclient_wrapperc&t||_yN)r)r _raw_clientselfrs {/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/rest_api/agent_configs/client.py__init__zAgentConfigsClient.__init__s0Oreturnc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawAgentConfigsClient rrs rwith_raw_responsez$AgentConfigsClient.with_raw_responserN project_id project_nameidrequest_options blueprintr r!r"r#cZ|jj|||||}|jS)aJ Creates a new optimizer config with initial blueprint, or adds a new blueprint to existing config Parameters ---------- blueprint : AgentBlueprintWrite project_id : typing.Optional[str] Project ID. Either project_id or project_name must be provided project_name : typing.Optional[str] Project name. Either project_id or project_name must be provided id : typing.Optional[str] Agent config ID. Generated automatically if not provided request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import OpikApi from Opik import AgentBlueprintWrite from Opik import AgentConfigValueWrite client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.agent_configs.create_agent_config(blueprint=AgentBlueprintWrite(type="blueprint", values=[AgentConfigValueWrite(key='key', type="string", )], ), ) r$r r!r"r#rcreate_agent_configdatarr$r r!r"r# _responses rr(z&AgentConfigsClient.create_agent_config s;P$$88!%+ 9 ~~rr#envscV|jj|||}|jS)a Creates or updates environment-to-blueprint mappings Parameters ---------- project_id : str envs : typing.Sequence[AgentConfigEnv] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import OpikApi from Opik import AgentConfigEnv client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.agent_configs.create_or_update_envs(project_id='project_id', envs=[AgentConfigEnv(env_name='env_name', blueprint_id='blueprint_id', )], ) r r-r#rcreate_or_update_envsr)rr r-r#r+s rr1z(AgentConfigsClient.create_or_update_envsQs2<$$::!o; ~~rmask_idr#env_namer4cX|jj||||}|jS)a Retrieves the blueprint associated with a specific environment Parameters ---------- env_name : str project_id : str mask_id : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AgentBlueprintPublic Blueprint retrieved Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.agent_configs.get_blueprint_by_env(env_name='env_name', project_id='project_id', ) r3rget_blueprint_by_envr)rr5r r4r#r+s rr8z'AgentConfigsClient.get_blueprint_by_envts5B$$99 j'?: ~~rblueprint_namecX|jj||||}|jS)a Sets an environment to point to a blueprint identified by name Parameters ---------- env_name : str project_id : str blueprint_name : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.agent_configs.set_env_by_blueprint_name(env_name='env_name', project_id='project_id', blueprint_name='blueprint_name', ) r:r#rset_env_by_blueprint_namer)rr5r r:r#r+s rr>z,AgentConfigsClient.set_env_by_blueprint_names6@$$>> jQ`? ~~rcV|jj|||}|jS)a1 Soft-deletes an environment by setting its ended_at timestamp Parameters ---------- env_name : str project_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.agent_configs.delete_env(env_name='env_name', project_id='project_id', ) r,r delete_envr)rr5r r#r+s rrBzAgentConfigsClient.delete_envs,2$$//*Ve/f ~~r blueprint_idcV|jj|||}|jS)aV Retrieves a specific blueprint by its ID Parameters ---------- blueprint_id : str mask_id : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AgentBlueprintPublic Blueprint retrieved Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.agent_configs.get_blueprint_by_id(blueprint_id='blueprint_id', ) r3rget_blueprint_by_idr)rrDr4r#r+s rrGz&AgentConfigsClient.get_blueprint_by_ids,<$$88whw8x ~~rnamecX|jj||||}|jS)a Retrieves a specific blueprint by its name within a project Parameters ---------- project_id : str name : str mask_id : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AgentBlueprintPublic Blueprint retrieved Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.agent_configs.get_blueprint_by_name(project_id='project_id', name='name', ) r3rget_blueprint_by_namer)rr rIr4r#r+s rrLz(AgentConfigsClient.get_blueprint_by_names5B$$:: g; ~~rpagesizer#rOrPcX|jj||||}|jS)a} Retrieves paginated blueprint history for a project Parameters ---------- project_id : str page : typing.Optional[int] size : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- BlueprintPageHistory History retrieved Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.agent_configs.get_blueprint_history(project_id='project_id', ) rNrget_blueprint_historyr)rr rOrPr#r+s rrSz(AgentConfigsClient.get_blueprint_history"s5B$$:: To; ~~rcT|jj||}|jS)aC Retrieves only the changes (delta) introduced in a specific blueprint Parameters ---------- blueprint_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AgentBlueprintPublic Delta retrieved Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.agent_configs.get_delta_by_id(blueprint_id='blueprint_id', ) r,rget_delta_by_idr)rrDr#r+s rrWz"AgentConfigsClient.get_delta_by_idHs*0$$44\Sb4c ~~rcV|jj|||}|jS)aU Retrieves the latest blueprint for a project Parameters ---------- project_id : str mask_id : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AgentBlueprintPublic Blueprint retrieved Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.agent_configs.get_latest_blueprint(project_id='project_id', ) r3rget_latest_blueprintr)rr r4r#r+s rr[z'AgentConfigsClient.get_latest_blueprintcs,<$$99*ggv9w ~~r)__name__ __module__ __qualname__rrpropertyr rOMITrtypingOptionalstrrr(Sequencer r1rr8r>rBrGrLintr rSrWr[rrrrsP*;P #8  ,0-1#';?/'/OOC( / ooc* / OOC / 8/ /l<@ !!oon- !  8 !  !P)-;? $$$ % $  8 $ $X<@ ###  #  8 # #Lei),BH//R`Ba @)-;? %   8   L)-;? $$$ % $  8 $ $T&*%);? $$ooc" $ ooc" $  8 $ $NX\5;__^5T >)-;? %   8   rrc eZdZdefdZedefdZeeeddde de je d e je d e je d e je ddf d Zdd de de jed e je ddfdZdddde de de je d e je def dZdd de de de d e je ddf dZdd de de d e je ddfdZdddde de je d e je defdZdddde de de je d e je def dZddddde de jede jed e je def dZdd de d e je defdZdddde de je d e je defd Zy)!AsyncAgentConfigsClientrc&t||_yr)r rrs rrz AsyncAgentConfigsClient.__init__s5^Trrc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawAgentConfigsClient rrs rrz)AsyncAgentConfigsClient.with_raw_responserrNrr$r r!r"r#cvK|jj|||||d{}|jS7w)a Creates a new optimizer config with initial blueprint, or adds a new blueprint to existing config Parameters ---------- blueprint : AgentBlueprintWrite project_id : typing.Optional[str] Project ID. Either project_id or project_name must be provided project_name : typing.Optional[str] Project name. Either project_id or project_name must be provided id : typing.Optional[str] Agent config ID. Generated automatically if not provided request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import AsyncOpikApi from Opik import AgentBlueprintWrite from Opik import AgentConfigValueWrite import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.agent_configs.create_agent_config(blueprint=AgentBlueprintWrite(type="blueprint", values=[AgentConfigValueWrite(key='key', type="string", )], ), ) asyncio.run(main()) r&Nr'r*s rr(z+AsyncAgentConfigsClient.create_agent_configsIV**>>!%+ ?  ~~ s $979r,r-crK|jj|||d{}|jS7w)a Creates or updates environment-to-blueprint mappings Parameters ---------- project_id : str envs : typing.Sequence[AgentConfigEnv] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import AsyncOpikApi from Opik import AgentConfigEnv import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.agent_configs.create_or_update_envs(project_id='project_id', envs=[AgentConfigEnv(env_name='env_name', blueprint_id='blueprint_id', )], ) asyncio.run(main()) r/Nr0r2s rr1z-AsyncAgentConfigsClient.create_or_update_envssBB**@@!oA  ~~ "757r3r5r4ctK|jj||||d{}|jS7w)a Retrieves the blueprint associated with a specific environment Parameters ---------- env_name : str project_id : str mask_id : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AgentBlueprintPublic Blueprint retrieved Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.agent_configs.get_blueprint_by_env(env_name='env_name', project_id='project_id', ) asyncio.run(main()) r3Nr7r9s rr8z,AsyncAgentConfigsClient.get_blueprint_by_envsDH**?? j'?@  ~~ #868r:ctK|jj||||d{}|jS7w)a Sets an environment to point to a blueprint identified by name Parameters ---------- env_name : str project_id : str blueprint_name : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.agent_configs.set_env_by_blueprint_name(env_name='env_name', project_id='project_id', blueprint_name='blueprint_name', ) asyncio.run(main()) r<Nr=r?s rr>z1AsyncAgentConfigsClient.set_env_by_blueprint_namesEF**DD jQ`E  ~~ rpcrK|jj|||d{}|jS7w)a Soft-deletes an environment by setting its ended_at timestamp Parameters ---------- env_name : str project_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.agent_configs.delete_env(env_name='env_name', project_id='project_id', ) asyncio.run(main()) r,NrArCs rrBz"AsyncAgentConfigsClient.delete_env?s98**55h \k5ll ~~mrnrDcrK|jj|||d{}|jS7w)a Retrieves a specific blueprint by its ID Parameters ---------- blueprint_id : str mask_id : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AgentBlueprintPublic Blueprint retrieved Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.agent_configs.get_blueprint_by_id(blueprint_id='blueprint_id', ) asyncio.run(main()) r3NrFrHs rrGz+AsyncAgentConfigsClient.get_blueprint_by_id^sAB**>> '??  ~~ rnrIctK|jj||||d{}|jS7w)a Retrieves a specific blueprint by its name within a project Parameters ---------- project_id : str name : str mask_id : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AgentBlueprintPublic Blueprint retrieved Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.agent_configs.get_blueprint_by_name(project_id='project_id', name='name', ) asyncio.run(main()) r3NrKrMs rrLz-AsyncAgentConfigsClient.get_blueprint_by_namesDH**@@ gA  ~~ rprNrOrPctK|jj||||d{}|jS7w)a Retrieves paginated blueprint history for a project Parameters ---------- project_id : str page : typing.Optional[int] size : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- BlueprintPageHistory History retrieved Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.agent_configs.get_blueprint_history(project_id='project_id', ) asyncio.run(main()) rNNrRrTs rrSz-AsyncAgentConfigsClient.get_blueprint_historysDH**@@ ToA  ~~ rpcpK|jj||d{}|jS7w)a Retrieves only the changes (delta) introduced in a specific blueprint Parameters ---------- blueprint_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AgentBlueprintPublic Delta retrieved Examples -------- from Opik import AsyncOpikApi import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.agent_configs.get_delta_by_id(blueprint_id='blueprint_id', ) asyncio.run(main()) r,NrVrXs rrWz'AsyncAgentConfigsClient.get_delta_by_ids76**:: None: await client.agent_configs.get_latest_blueprint(project_id='project_id', ) asyncio.run(main()) r3NrZr\s rr[z,AsyncAgentConfigsClient.get_latest_blueprintsBB**?? @  ~~ rn)r]r^r_rrr`r rrarrbrcrdrr(rer r1rr8r>rBrGrLrfr rSrWr[rgrrririsU*<U #=  ,0-1#';?2'2OOC( 2 ooc* 2 OOC 2 82 2r<@ $$oon- $  8 $  $V)-;? ''' % '  8 ' '^<@ &&&  &  8 & &Rei),BH//R`Ba F)-;? $$% $  8 $  $V)-;? ''' % '  8 ' 'Z&*%);? ''ooc" ' ooc" '  8 ' 'TX\5;__^5T D)-;? $$% $  8 $  $rri)rbcore.client_wrapperrrcore.request_optionsrtypes.agent_blueprint_publicrtypes.agent_blueprint_writertypes.agent_config_envr types.blueprint_page_historyr raw_clientr r castAnyrarrirgrrrsPG1?=3?Iv{{6::s#qqh SSr