jY"ddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZd d lmZd d lmZGd dZy)N)client)core)RequestOptions)AgentBlueprintWrite)AgentConfigValueWrite)AgentConfigEnv) rest_helpers) id_helpers) Blueprint) type_helperscJeZdZdZdedej ddfdZedefdZ e de je jee jfd e je jee je je je jeffde jee je je je jefffd Z dd e jee je je je jeffd e jed e jed edef dZddddddde jede jede jede je jee jfde jede jef dZ dde je jee jfd e je jee je je je jeffd e jede je jee jfdef dZdededdfdZ dde je jee jfd e je jee je je je jeffd e jedefdZy)AgentConfigManageru?Project-level agent config entity — internal REST operations. project_name rest_client_returnNc ||_||_yN) _project_name _rest_client)selfrrs }/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/api_objects/agent_config/config.py__init__zAgentConfigManager.__init__s *(c|jSr)r)rs rrzAgentConfigManager.project_names!!!r parametersfields_with_valuesc||S|xsijDcic]\}}||t||dfc}}Scc}}wr)itemstype)rrkvs r_resolve_fields_with_valuesz.AgentConfigManager._resolve_fields_with_valuess^  )% % $)r002 21} QD! !2   s; descriptionid config_typec g}|jD]Q\}\}}} | |jt|tj|tj ||| St ||||S)N)keyr valuer$)r%r valuesr$)rappendrr python_type_to_backend_typepython_value_to_backend_valuer) rrr$r%r&backend_values field_namepy_typer) field_descs r_build_blueprint_payloadz+AgentConfigManager._build_blueprint_payload/s8J8P8P8R 4J4%}  ! !%"%AA'J&DDUGT *  9S#!#   r)nameenvmask_id field_typestimeout_in_secondsr3r4r5r6r7c| t|nd} tj|j|j}|*|jj j ||||}nT|*|jj j||||}n(|jj j|||}t|||jS#tj$r} | jdk(rYd} ~ yd} ~ wwxYw)a0Fetch a blueprint by name, environment name, or latest. Priority: ``name`` > ``env`` > latest. Returns ``None`` if not found. Args: name: Fetch the blueprint with this version name. env: Fetch the blueprint tagged with this environment name. mask_id: ID of a mask blueprint to overlay on the result. field_types: Mapping of prefixed field key to Python type used for deserialising backend values. timeout_in_seconds: HTTP request timeout in seconds. N)r7) project_idr3r5request_options)env_namer9r5r:)r9r5r:i raw_blueprintr6r) rr resolve_project_id_by_namerr agent_configsget_blueprint_by_nameget_blueprint_by_envget_latest_blueprint rest_api_coreApiError status_coder ) rr3r4r5r6r7r:r9rawes r get_blueprintz AgentConfigManager.get_blueprintKs#0"- .@ A   %@@!!4#5#5J''55KK)#$3 L ''55JJ )#$3 K''55JJ)#$3K#**  %% }}#  sB*CD'C<;C<<DcR|j||}tj}|j|||}|jj j ||j|jj j|}t|||jS)aCreate a new blueprint and return it. Pass either ``parameters`` (plain key-value pairs whose types are inferred) or ``fields_with_values`` (explicit ``{key: (type, value)}`` mapping). If both are given ``fields_with_values`` takes precedence. Args: parameters: Plain ``{field_name: value}`` dict; types are inferred via ``type(value)``. fields_with_values: Explicit ``{field_name: (python_type, value)}`` mapping, bypassing type inference. description: Human-readable description stored with the blueprint. field_types: Mapping of prefixed field key to Python type used when fetching back the created blueprint. )r% blueprintrr<) r#r generate_idr2rr?create_agent_configrget_blueprint_by_idr )rrrr$r6 blueprint_idpayloadrFs rcreate_blueprintz#AgentConfigManager.create_blueprints0"== * "--/ //  0  '';;++ < --AA,O#**  rrOctj|j|j}|jjj |t ||gy)zAssociate a blueprint with an environment name. Args: env: Environment name (e.g. ``"production"``). blueprint_id: ID of the blueprint to tag. )r;rO)r9envsN)r r>rrr?create_or_update_envsr)rr4rOr9s rtag_blueprint_with_envz)AgentConfigManager.tag_blueprint_with_envsV"<<   t11  ''==! #LIJ > rc|j||}tj}|j|||d}|jj j ||j|S)aCreate a mask blueprint and return its ID. A mask overlays a subset of fields on top of an existing blueprint. Apply it by passing the returned ID to ``get_blueprint(mask_id=...)``. Args: parameters: Plain ``{field_name: value}`` dict; types are inferred via ``type(value)``. fields_with_values: Explicit ``{field_name: (python_type, value)}`` mapping, bypassing type inference. description: Human-readable description stored with the mask. mask)r%r&rJ)r#r rLr2rr?rMr)rrrr$r5rPs r create_maskzAgentConfigManager.create_masks}("== * ((*//  V0  '';;++ < r)NrK)NNNN)NNN)__name__ __module__ __qualname____doc__str rest_clientOpikApirpropertyr staticmethodtypingOptionalDictAnyTupler#rr2intr rHrQrUrXrrrrsI))")))  )"c"" OOFKKVZZ$@A "OO KKV\\&**fjj&//RUBV*VWW X  S&,,vzz6::vs?S'STT U   ,$(& "KK fjj&**fooc6JJK K  __S)  OOC     >&*$((,EI37; ooc"; __S ! ; % ; __V[[fjj%AB ; #OOC0;   #; ~EI ,0EI( OOFKKVZZ$@A( #OO KKV\\&**fjj&//RUBV*VWW X ( __S) ( __V[[fjj%AB(  ( T  #  S  T  "EI ,0 OOFKKVZZ$@A#OO KKV\\&**fjj&//RUBV*VWW X  __S)  rr)rb opik.rest_apirr^rrC"opik.rest_api.core.request_optionsr)opik.rest_api.types.agent_blueprint_writer,opik.rest_api.types.agent_config_value_writer$opik.rest_api.types.agent_config_envropik.api_objectsr opikr rKr r rrhrrrqs2 //=IN?) QQr