tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api

Example Airflow DAG for Google Gen AI Gemini Batch API and File API.

Attributes

ENV_ID

PROJECT_ID

REGION

DAG_ID

GEMINI_API_KEY

INLINED_REQUESTS_FOR_BATCH_JOB

INLINED_REQUESTS_FOR_EMBEDDINGS_BATCH_JOB

GEMINI_XCOM_API_KEY

LOCAL_FILE_NAME

LOCAL_EMBEDDINGS_FILE_NAME

UPLOAD_FILE_PATH

PATH_TO_SAVE_RESULTS

UPLOAD_EMBEDDINGS_FILE_PATH

UPLOADED_FILE_NAME

UPLOADED_EMBEDDINGS_FILE_NAME

BATCH_JOB_WITH_INLINED_REQUESTS_NAME

BATCH_JOB_WITH_FILE_NAME

EMBEDDINGS_BATCH_JOB_WITH_INLINED_REQUESTS_NAME

EMBEDDINGS_BATCH_JOB_WITH_FILE_NAME

test_run

Functions

get_gemini_api_key()

Module Contents

tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.ENV_ID[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.PROJECT_ID[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.REGION = 'us-central1'[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.DAG_ID = 'gen_ai_gemini_batch_api'[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.GEMINI_API_KEY = 'api_key'[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.INLINED_REQUESTS_FOR_BATCH_JOB[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.INLINED_REQUESTS_FOR_EMBEDDINGS_BATCH_JOB[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.GEMINI_XCOM_API_KEY = "{{ task_instance.xcom_pull('get_gemini_api_key') }}"[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.LOCAL_FILE_NAME = 'gemini_batch_requests.jsonl'[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.LOCAL_EMBEDDINGS_FILE_NAME = 'gemini_batch_embeddings_requests.jsonl'[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.UPLOAD_FILE_PATH = ''[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.PATH_TO_SAVE_RESULTS = ''[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.UPLOAD_EMBEDDINGS_FILE_PATH = ''[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.UPLOADED_FILE_NAME = "{{ task_instance.xcom_pull(task_ids='upload_file_for_batch_job_task', key='file_name') }}"[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.UPLOADED_EMBEDDINGS_FILE_NAME = "{{ task_instance.xcom_pull(task_ids='upload_file_for_embeddings_batch_job_task', key='file_name') }}"[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.BATCH_JOB_WITH_INLINED_REQUESTS_NAME = "{{ task_instance.xcom_pull(task_ids='create_batch_job_using_inlined_requests_task', key='job_name') }}"[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.BATCH_JOB_WITH_FILE_NAME = "{{ task_instance.xcom_pull(task_ids='create_batch_job_using_file_task', key='job_name') }}"[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.EMBEDDINGS_BATCH_JOB_WITH_INLINED_REQUESTS_NAME = "{{ task_instance.xcom_pull(task_ids='create_embeddings_job_using_inlined_requests_task',...[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.EMBEDDINGS_BATCH_JOB_WITH_FILE_NAME = "{{ task_instance.xcom_pull(task_ids='create_embeddings_job_using_file_task', key='job_name') }}"[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.get_gemini_api_key()[source]
tests.system.google.cloud.gen_ai.example_gen_ai_gemini_batch_api.test_run[source]

Was this entry helpful?