j ddlZddlmZmZddlmZddlmZddlm Z m Z ejejdZ Gd d ZGd d Zy) N)AsyncClientWrapperSyncClientWrapper)RequestOptions)GuardrailWrite)AsyncRawGuardrailsClientRawGuardrailsClient.c|eZdZdefdZedefdZdddeje deje ddfd Z y) GuardrailsClientclient_wrapperc&t||_yN)r )r _raw_clientselfr s x/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/rest_api/guardrails/client.py__init__zGuardrailsClient.__init__s.nMreturnc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawGuardrailsClient rrs rwith_raw_responsez"GuardrailsClient.with_raw_responserNrequest_options guardrailsrcT|jj||}|jS)a Batch guardrails for traces Parameters ---------- guardrails : typing.Sequence[GuardrailWrite] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import OpikApi from Opik import GuardrailWrite client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.guardrails.create_guardrails(guardrails=[GuardrailWrite(entity_id='entity_id', secondary_id='secondary_id', name="TOPIC", result="passed", config={'key': 'value' }, details={'key': 'value' }, )], ) rrrcreate_guardrailsdatarrr _responses rr"z"GuardrailsClient.create_guardrailss*4$$66*^m6n ~~r)__name__ __module__ __qualname__rrpropertyr rtypingSequencerOptionalrr"rrr r sdN*;N #6  rv#__^<OU_mOn rr c|eZdZdefdZedefdZdddeje deje ddfd Z y) AsyncGuardrailsClientr c&t||_yr)r rrs rrzAsyncGuardrailsClient.__init__<s3>Rrrc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawGuardrailsClient rrs rrz'AsyncGuardrailsClient.with_raw_response?rrNrrrcpK|jj||d{}|jS7w)a9 Batch guardrails for traces Parameters ---------- guardrails : typing.Sequence[GuardrailWrite] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import AsyncOpikApi from Opik import GuardrailWrite import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.guardrails.create_guardrails(guardrails=[GuardrailWrite(entity_id='entity_id', secondary_id='secondary_id', name="TOPIC", result="passed", config={'key': 'value' }, details={'key': 'value' }, )], ) asyncio.run(main()) r Nr!r$s rr"z'AsyncGuardrailsClient.create_guardrailsJs7:**<< ds<tt ~~us !646)r&r'r(rrr)r rr*r+rr,rr"r-rrr/r/;sdS*<S #;  rv#__^<OU_mOn rr/)r*core.client_wrapperrrcore.request_optionsrtypes.guardrail_writer raw_clientr r castAnyOMITr r/r-rrr:sCG12Ev{{6::s#**Z--r