ci dZddlmZddlmZddlmZdZdZdZ dd ejd ee e e ejfd ejfd Zd ej"d ejfdZd ejd dfdZy)z(Utility functions for prompt management.)Optional)typesz1.0.0zKgs://google-cloud-aiplatform/schema/dataset/metadata/text_prompt_1.0.0.yamlmultimodal_freeformNprompt variablesreturnctj}tj}tj|j|_t |_t||_ |rg}|D]~}tj}i|_ |jD]=\}}tjtj|g|j|<?|j|||_|j j"rL|j j"j%dgd} tj&di| |j _||_t*|_|S)zBConvert a types.Prompt into types.SchemaTextPromptDatasetMetadata.)prompt_message)parts) part_listrT)exclude exclude_none)rSchemaTextPromptDatasetMetadataSchemaPromptApiSchema SchemaPromptSpecMultimodalPrompt prompt_datamultimodal_promptDEFAULT_API_SCHEMA_VERSIONapi_schema_versionboolhas_prompt_variable#SchemaPromptInstancePromptExecution argumentsitems!SchemaPromptInstanceVariableValueSchemaPromptSpecPartListappend executionsr model_dumpSchemaPromptSpecPromptMessageprompt_api_schema PROMPT_TYPE prompt_type) rrprompt_metadatar#prompt_execution_list prompt_varprompt_instance_executionkeyvalprompt_message_dicts O/tmp/pip-target-z3e9_cxr/lib/python/vertexai/_genai/_prompt_management_utils.py$_create_dataset_metadata_from_promptr.se ;;=O335*/*P*P))+',F(*.y/O' "# DJ(-(Q(Q(S %24 % /&,,. S;;"'"@"@u"M*33C8  " ( ()B C D(=$**99  / / > > I I$ D J    / / F2E F ++:):O%"-O datasetcNt|dr0|j$t|jtjs t d|jj }tj}||S|jr"|jj}||_ |jr|j}|r|jg|j_ |D]}|js|j}i}|jD]Z\}} | j| jj *| jj } | sC| dj"sS| d||<\|s|jj|jjj%||S)zConstructs a types.Prompt from a types.Dataset resource returned from the API. Args: dataset: The types.Dataset object containing the prompt metadata. Returns: A types.Prompt object reconstructed from the dataset metadata. metadatazGError retrieving prompt: prompt dataset resource is missing 'metadata'.r)hasattrr2 isinstancerr ValueErrorr#Promptrr rr rrrr r textr) r0 api_schemarr r executionargsvar_mapr*r+r s r-$_create_prompt_from_dataset_metadatar<Ps| GZ (    #'**E,Q,QR U  !!33J \\^F ###55DD+  #..Jf00rMs/-%Q$ >B. LL.T#{'7'7"789:. **.b1 ]]1 \\1hL LLL Lr/