Vertex AI中Gemini批量预测返回UNIMPLEMENTED错误求助

核心问题分析 #

你遇到的StatusCode.UNIMPLEMENTED和HTTP 404头错误，主要由以下配置问题导致：

1. 模型引用路径错误 #

Gemini预训练模型属于Google公共发布者项目，不能直接引用自己项目下的路径。必须使用正确的公共模型地址：

model=f"projects/cloud-large-language-models/locations/us-central1/models/gemini-1.0-pro-001"

2. BatchPredictionJob的name参数误用 #

name字段是批量预测任务的自定义ID（比如my-batch-job-001），不能直接填写项目ID。建议直接省略该参数，系统会自动生成唯一ID，避免冲突。

3. BigQuery URI格式不完整 #

正确的BigQuery URI必须包含项目ID、数据集ID、表ID三层结构：

• 输入URI格式：bq://<你的项目ID>.<数据集ID>.<输入表ID>
• 输出URI格式：bq://<你的项目ID>.<数据集ID>.<输出表前缀或完整表名>
  你当前的bq://sample.text_input缺少必要层级，系统无法定位到目标表。

4. 输入表数据格式要求 #

Gemini批量预测的输入表必须包含名为content的字符串列，用于存储待处理的文本内容；多模态场景需对应其他列名，纯文本场景仅需content列。

修正后的完整代码 #

import os
import aiplatform_v1

async def sample_create_batch_prediction_job():
    project_id = os.getenv("PROJECT_ID")
    location = "us-central1"
    
    input_config = aiplatform_v1.types.BatchPredictionJob.InputConfig(
        bigquery_source=aiplatform_v1.types.BigQuerySource(
            input_uri=f"bq://{project_id}.your_dataset.your_input_table"
        )
    )
    output_config = aiplatform_v1.types.BatchPredictionJob.OutputConfig(
        bigquery_destination=aiplatform_v1.types.BigQueryDestination(
            output_uri=f"bq://{project_id}.your_dataset.output_prefix_"
        )
    )

    batch_prediction_job_1 = aiplatform_v1.types.BatchPredictionJob(
        display_name='my-batch-prediction-job',
        model=f"projects/cloud-large-language-models/locations/{location}/models/gemini-1.0-pro-001",
        input_config=input_config,
        output_config=output_config,
        model_parameters={"temperature": 0.0}
    )

    client = aiplatform_v1.JobServiceAsyncClient(
        client_options={"api_endpoint": f"{location}-aiplatform.googleapis.com"}
    )

    request = aiplatform_v1.CreateBatchPredictionJobRequest(
        parent=f"projects/{project_id}/locations/{location}",
        batch_prediction_job=batch_prediction_job_1,
    )

    response = await client.create_batch_prediction_job(request=request)
    print("Batch prediction job created:", response.name)

额外注意事项 #

• 确保服务账号拥有BigQuery数据读写权限和Vertex AI Job创建权限
• 输出URI可仅指定数据集+前缀，系统会自动生成唯一输出表名，避免重复
• 更换Gemini模型时，替换路径中的模型ID即可（如gemini-1.5-flash-001）

内容的提问来源于stack exchange，提问作者Linda Lawton - DaImTo