Complete the airflow survey & get a free airflow 3 certification!

tests.system.google.cloud.vertex_ai.example_vertex_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`
`test_run`

Functions¶

get_gemini_api_key()

Module Contents¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.PROJECT_ID[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.DAG_ID = 'vertex_ai_generative_model_tuning_dag'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.REGION = 'us-central1'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.GEMINI_API_KEY = 'api_key'[source]¶

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

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.TRAIN_DATASET = 'gs://cloud-samples-data/ai-platform/generative_ai/gemini-2_0/text/sft_train_data.jsonl'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.TUNED_MODEL_DISPLAY_NAME = 'my_tuned_gemini_model'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.get_gemini_api_key()[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_generative_model_tuning.test_run[source]¶

Was this entry helpful?