j^RddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z d d lmZmZej"ej$d ZGd d ZGddZy)N)AsyncClientWrapperSyncClientWrapper)RequestOptions)AnnotationQueuePagePublic)AnnotationQueuePublic)AnnotationQueueWrite)AnnotationQueueWriteScope)AsyncRawAnnotationQueuesClientRawAnnotationQueuesClient.cVeZdZdefdZedefdZdddede jed e je ddfd Z ddddddd d e jed e jede jede jede jed e je defdZeeeeedddededede jede jede jede jede je jed e je ddfdZddde jed e je ddfdZddde jed e je ddfdZddded e je defdZeeeeedddede jede jede jede jede je jed e je ddfdZdddede jed e je ddfd Zy)!AnnotationQueuesClientclient_wrapperc&t||_yN)r)r _raw_clientselfrs /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/rest_api/annotation_queues/client.py__init__zAnnotationQueuesClient.__init__s4NSreturnc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawAnnotationQueuesClient rrs rwith_raw_responsez(AnnotationQueuesClient.with_raw_responserNrequest_optionsididsr cV|jj|||}|jS)a  Add traces or threads to annotation queue Parameters ---------- id : str ids : typing.Sequence[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.annotation_queues.add_items_to_annotation_queue(id='id', ids=['ids'], ) r"r radd_items_to_annotation_queuedatarr!r"r _responses rr&z4AnnotationQueuesClient.add_items_to_annotation_queue s,2$$BB23`oBp ~~rpagesizenamefilterssortingr r+r,r-r.r/c\|jj||||||}|jS)a Find annotation queues with filtering and sorting Parameters ---------- page : typing.Optional[int] size : typing.Optional[int] name : typing.Optional[str] filters : typing.Optional[str] sorting : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AnnotationQueuePagePublic Annotation queues page Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.annotation_queues.find_annotation_queues() r*rfind_annotation_queuesr'rr+r,r-r.r/r r)s rr2z-AnnotationQueuesClient.find_annotation_queues<s:N$$;;DtWg_n< ~~rr! description instructionscomments_enabledfeedback_definition_namesr project_idscoper5r6r7r8c b|jj|||||||||  } | jS)ab Create annotation queue for human annotation workflows Parameters ---------- project_id : str name : str scope : AnnotationQueueWriteScope id : typing.Optional[str] description : typing.Optional[str] instructions : typing.Optional[str] comments_enabled : typing.Optional[bool] feedback_definition_names : typing.Optional[typing.Sequence[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.annotation_queues.create_annotation_queue(project_id='project_id', name='name', scope="trace", ) r9r-r:r!r5r6r7r8r rcreate_annotation_queuer' rr9r-r:r!r5r6r7r8r r)s rr>z.AnnotationQueuesClient.create_annotation_queuehsG^$$<<!#%-&?+=  ~~rannotation_queuescT|jj||}|jS)a Create multiple annotation queues for human annotation workflows Parameters ---------- annotation_queues : typing.Sequence[AnnotationQueueWrite] List of annotation queues to create request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import OpikApi from Opik import AnnotationQueueWrite client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.annotation_queues.create_annotation_queue_batch(annotation_queues=[AnnotationQueueWrite(project_id='project_id', name='name', scope="trace", )], ) r@r rcreate_annotation_queue_batchr'rr@r r)s rrDz4AnnotationQueuesClient.create_annotation_queue_batchs18$$BB/C ~~rcT|jj||}|jS)a  Delete multiple annotation queues by their IDs Parameters ---------- ids : typing.Sequence[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.annotation_queues.delete_annotation_queue_batch(ids=['ids'], ) r$rdelete_annotation_queue_batchr'rr"r r)s rrHz4AnnotationQueuesClient.delete_annotation_queue_batchs*.$$BBs\kBl ~~rcT|jj||}|jS)a Get annotation queue by id Parameters ---------- id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AnnotationQueuePublic Annotation queue resource Examples -------- from Opik import OpikApi client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) client.annotation_queues.get_annotation_queue_by_id(id='id', ) rrget_annotation_queue_by_idr'rr!r r)s rrLz1AnnotationQueuesClient.get_annotation_queue_by_ids*0$$??Tc?d ~~rr-r5r6r7r8r c ^|jj|||||||}|jS)a Update annotation queue by id Parameters ---------- id : str name : typing.Optional[str] description : typing.Optional[str] instructions : typing.Optional[str] comments_enabled : typing.Optional[bool] feedback_definition_names : typing.Optional[typing.Sequence[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.annotation_queues.update_annotation_queue(id='id', ) rNrupdate_annotation_queuer' rr!r-r5r6r7r8r r)s rrQz.AnnotationQueuesClient.update_annotation_queuesAR$$<< #%-&?+= ~~rcV|jj|||}|jS)a Remove items from annotation queue Parameters ---------- id : str ids : typing.Sequence[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.annotation_queues.remove_items_from_annotation_queue(id='id', ids=['ids'], ) r$r"remove_items_from_annotation_queuer'r(s rrUz9AnnotationQueuesClient.remove_items_from_annotation_queue.s-2$$GGPSetGu ~~r)__name__ __module__ __qualname__rrpropertyr rstrtypingSequenceOptionalrr&intrr2OMITr boolr>r rDrHrrLrQrUrrrrsET*;T #<  im%s3FLooVdFe >&*%)%)(,(,;?*ooc"*ooc" * ooc" * % *%* 8* #*d$(,0-126KO;?:: : ) : OOC :__S):ooc*:!//$/:$*??6??33G#H: 8: :@<@ "??+?@ 8   D`dooc*=C__^=\ 6NR+1??>+J >&*,0-126KO;?2 2ooc" 2 __S) 2 ooc* 2!//$/2$*??6??33G#H2 82 2jim%s3FLooVdFe rrcVeZdZdefdZedefdZdddede jed e je ddfd Z ddddddd d e jed e jede jede jede jed e je defdZeeeeedddededede jede jede jede jede je jed e je ddfdZddde jed e je ddfdZddde jed e je ddfdZddded e je defdZeeeeedddede jede jede jede jede je jed e je ddfdZdddede jed e je ddfd Zy)!AsyncAnnotationQueuesClientrc&t||_yr)r rrs rrz$AsyncAnnotationQueuesClient.__init__Ls9Xrrc|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawAnnotationQueuesClient rrs rrz-AsyncAnnotationQueuesClient.with_raw_responseOrrNrr!r"r crK|jj|||d{}|jS7w)a Add traces or threads to annotation queue Parameters ---------- id : str ids : typing.Sequence[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.annotation_queues.add_items_to_annotation_queue(id='id', ids=['ids'], ) asyncio.run(main()) r$Nr%r(s rr&z9AsyncAnnotationQueuesClient.add_items_to_annotation_queueZs:8**HHQTfuHvv ~~w "757r*r+r,r-r.r/cxK|jj||||||d{}|jS7w)a5 Find annotation queues with filtering and sorting Parameters ---------- page : typing.Optional[int] size : typing.Optional[int] name : typing.Optional[str] filters : typing.Optional[str] sorting : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AnnotationQueuePagePublic Annotation queues page 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.annotation_queues.find_annotation_queues() asyncio.run(main()) r*Nr1r3s rr2z2AsyncAnnotationQueuesClient.find_annotation_queuesysIT**AADtWg_nB  ~~ s %:8:r4r9r:r5r6r7r8c ~K|jj|||||||||  d{} | jS7w)a Create annotation queue for human annotation workflows Parameters ---------- project_id : str name : str scope : AnnotationQueueWriteScope id : typing.Optional[str] description : typing.Optional[str] instructions : typing.Optional[str] comments_enabled : typing.Optional[bool] feedback_definition_names : typing.Optional[typing.Sequence[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.annotation_queues.create_annotation_queue(project_id='project_id', name='name', scope="trace", ) asyncio.run(main()) r<Nr=r?s rr>z3AsyncAnnotationQueuesClient.create_annotation_queuesVd**BB!#%-&?+C   ~~  s (=;=r@cpK|jj||d{}|jS7w)aV Create multiple annotation queues for human annotation workflows Parameters ---------- annotation_queues : typing.Sequence[AnnotationQueueWrite] List of annotation queues to create request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from Opik import AsyncOpikApi from Opik import AnnotationQueueWrite import asyncio client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", ) async def main() -> None: await client.annotation_queues.create_annotation_queue_batch(annotation_queues=[AnnotationQueueWrite(project_id='project_id', name='name', scope="trace", )], ) asyncio.run(main()) rBNrCrEs rrDz9AsyncAnnotationQueuesClient.create_annotation_queue_batchs?>**HH/I  ~~ !646cpK|jj||d{}|jS7w)as Delete multiple annotation queues by their IDs Parameters ---------- ids : typing.Sequence[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.annotation_queues.delete_annotation_queue_batch(ids=['ids'], ) asyncio.run(main()) r$NrGrIs rrHz9AsyncAnnotationQueuesClient.delete_annotation_queue_batch s74**HHSbqHrr ~~srkcpK|jj||d{}|jS7w)a} Get annotation queue by id Parameters ---------- id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AnnotationQueuePublic Annotation queue resource 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.annotation_queues.get_annotation_queue_by_id(id='id', ) asyncio.run(main()) rNrKrMs rrLz6AsyncAnnotationQueuesClient.get_annotation_queue_by_id(s76**EEbZiEjj ~~krkrNc zK|jj|||||||d{}|jS7w)aA Update annotation queue by id Parameters ---------- id : str name : typing.Optional[str] description : typing.Optional[str] instructions : typing.Optional[str] comments_enabled : typing.Optional[bool] feedback_definition_names : typing.Optional[typing.Sequence[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.annotation_queues.update_annotation_queue(id='id', ) asyncio.run(main()) rNNrPrRs rrQz3AsyncAnnotationQueuesClient.update_annotation_queueFsPX**BB #%-&?+C  ~~ s &;9;crK|jj|||d{}|jS7w)a Remove items from annotation queue Parameters ---------- id : str ids : typing.Sequence[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.annotation_queues.remove_items_from_annotation_queue(id='id', ids=['ids'], ) asyncio.run(main()) r$NrTr(s rrUz>AsyncAnnotationQueuesClient.remove_items_from_annotation_queue}sA8**MM CN  ~~ rg)rVrWrXrrrYr rrZr[r\r]rr&r^rr2r_r r`r>r rDrHrrLrQrUrarrrcrcKsGY*<Y #A  im%s3FLooVdFe D&*%)%)(,(,;?-ooc"-ooc" - ooc" - % -%- 8- #-j$(,0-126KO;?== = ) = OOC =__S)=ooc*=!//$/=$*??6??33G#H= 8= =F<@ ""??+?@" 8 "  "J`dooc*=C__^=\ <NR+1??>+J D&*,0-126KO;?5 5ooc" 5 __S) 5 ooc* 5!//$/5$*??6??33G#H5 85 5pim%s3FLooVdFe rrc)r[core.client_wrapperrrcore.request_optionsr"types.annotation_queue_page_publicrtypes.annotation_queue_publicrtypes.annotation_queue_writer "types.annotation_queue_write_scoper raw_clientr r castAnyr_rrcrarrrysPG1JA?JQv{{6::s#wwt QQr