googleapis / googleapis/python-aiplatform

Question about Vertex AI EvalTask Judge Model

Open
#6,042 0 comments 0 reactions 0 assignees View on GitHub
api: vertex-ai
Dominant language
Python
Stars
905
Forks
465
Avg merge
1d 13h
Merged PRs (30d)
44

Description

Hi, I am using the Vertex AI Gen AI Evaluation SDK, specifically the EvalTask for evaluating model responses. I see that the judge model for evaluation is typically a Vertex AI hosted Gemini model, and you can customize it by tuning a Gemini model using the SDK. (https://cloud.google.com/vertex-ai/generative-ai/docs/models/configure-judge-model )

My question is:
Is it possible to replace the judge model with an external model served outside of Vertex AI, such as a custom REST API endpoint (for example, a Databricks serving endpoint or any other external model API)? In other words, can the judge model in EvalTask be configured to call an external model API instead of a Vertex AI managed model?

If not supported, are there any recommended ways or workarounds to integrate external judge models into the Vertex AI evaluation workflow?

Thank you very much for any guidance!

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.