tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_generative_model¶
Example Airflow DAG for Google Vertex AI Generative Model prompting.
Module Contents¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_generative_model.PROJECT_ID[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_generative_model.DAG_ID = 'example_vertex_ai_generative_model_dag'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_generative_model.REGION = 'us-central1'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_generative_model.PROMPT = 'In 10 words or less, why is Apache Airflow amazing?'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_generative_model.LANGUAGE_MODEL = 'text-bison'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_generative_model.TEXT_EMBEDDING_MODEL = 'textembedding-gecko'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_generative_model.MULTIMODAL_MODEL = 'gemini-pro'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_generative_model.MULTIMODAL_VISION_MODEL = 'gemini-pro-vision'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_generative_model.VISION_PROMPT = 'In 10 words or less, describe this content.'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_generative_model.MEDIA_GCS_PATH = 'gs://download.tensorflow.org/example_images/320px-Felis_catus-cat_on_snow.jpg'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_generative_model.MIME_TYPE = 'image/jpeg'[source]¶