j] XddlZddlmZmZddlmZmZddlmZddlm Z Gd d Z y) N)ListOptional)rest_operationsexperiment_item)opik_query_language)clientc~eZdZdZdej fdZ d dedeede de d e ed ee jf d Zy) ExperimentsClientz5Client for managing and querying experiments in bulk. rest_clientc||_y)N) _rest_client)selfrs /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/api_objects/experiment/experiments_client.py__init__zExperimentsClient.__init__ s 'N dataset_nameexperiment_idstruncate max_results filter_stringreturnc6|Atjtjj |j }nd}|j jj|j}tj|||j |||S)a Find experiment items associated with a specific dataset among a list of experiments. This method queries the dataset for experiment items matching the criteria provided by the input parameters. It leverages the ExperimentsClient to perform the underlying operation. Args: dataset_name: Name of the dataset to query for experiment items. experiment_ids: List of experiment IDs to filter the results. filter_string: Optional filter string to refine the query based on additional criteria (dataset fields, feedback scores, etc.). truncate: Whether to truncate image data stored in input, output, or metadata. Defaults to True. max_results: Maximum number of results to return. Defaults to 1000. Returns: A list of experiment item content objects that match the criteria. N)r) dataset_idrrrrfilter_expression) jsondumpsr OpikQueryLanguage for_tracesget_filter_expressionsrdatasetsget_dataset_by_identifieridr!find_experiment_items_for_dataset)rrrrrrrrs rr&z3ExperimentsClient.find_experiment_items_for_datasets8  $ $ #55@@!((*!  !% &&//II%J " @@!)))#/   r)TiN)__name__ __module__ __qualname____doc__rest_api_clientOpikApirstrrboolintrrExperimentItemContentr&rrr r st?(O$;$;('+ 1 1 S 1  1  1  } 1  o33 41 rr ) rtypingrrrrr rest_apir r+r r1rrr5s !."17 7 r