tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning¶

Example Airflow DAG for Google Vertex AI Generative Model Tuning Tasks.

Attributes¶

`PROJECT_ID`
`DAG_ID`
`REGION`
`GEMINI_API_KEY`
`SOURCE_MODEL`
`TRAIN_DATASET`
`TUNED_MODEL_DISPLAY_NAME`
`TUNING_JOB_CONFIG`
`TUNED_VIDEO_MODEL_DISPLAY_NAME`
`TUNING_JOB_VIDEO_MODEL_CONFIG`
`BUCKET_NAME`
`FILE_NAME`
`UPLOAD_FILE_PATH`
`TRAIN_VIDEO_DATASET`
`test_run`

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.PROJECT_ID[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.DAG_ID = 'gen_ai_generative_model_tuning_dag'[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.REGION = 'us-central1'[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.GEMINI_API_KEY = 'api_key'[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.SOURCE_MODEL = "{{ task_instance.xcom_pull('get_actual_model') }}"[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.TRAIN_DATASET[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.TUNED_MODEL_DISPLAY_NAME = 'my_tuned_gemini_model'[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.TUNING_JOB_CONFIG[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.TUNED_VIDEO_MODEL_DISPLAY_NAME = 'my_tuned_gemini_video_model'[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.TUNING_JOB_VIDEO_MODEL_CONFIG[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.BUCKET_NAME[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.FILE_NAME = 'video_tuning_dataset.jsonl'[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.UPLOAD_FILE_PATH = ''[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.TRAIN_VIDEO_DATASET[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.get_gemini_api_key()[source]¶

tests.system.google.cloud.gen_ai.example_gen_ai_generative_model_tuning.test_run[source]¶