-
Notifications
You must be signed in to change notification settings - Fork 623
Add GCPVertexGeminiSFTConfig to python OptimizationConfig type #2962
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pull Request Overview
This PR adds support for GCP Vertex Gemini SFT (Supervised Fine-Tuning) configuration to the Python client's optimization configuration options. This expands the available optimization providers for the experimental_launch_optimization
function.
- Adds GCPVertexGeminiSFTConfig to the OptimizationConfig union type
- Enables users to specify GCP Vertex Gemini as an optimization provider in Python
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Tests complained, I believe needs change in tensorzero-core/src/inference/types/pyo3_helpers.rs
…dd-gcp-sft-config-to-python
This PR adds the GCPVertexGeminiSFTConfig to OptimizationConfig in
clients/python/tensorzero/__init__.py
to make it visible to the python client when callingexperimental_launch_optimization
Important
Add
GCPVertexGeminiSFTConfig
toOptimizationConfig
and update tests and deserialization to support it.GCPVertexGeminiSFTConfig
toOptimizationConfig
in__init__.py
for Python client visibility.deserialize_optimization_config
inpyo3_helpers.rs
to handleGCPVertexGeminiSFTConfig
.test_sync_gcp_vertex_gemini_sft
andtest_async_gcp_vertex_gemini_sft
intest_optimization.py
for new config.deserialize_optimization_config
to includeGCPVertexGeminiSFTConfig
.This description was created by
for 2825d39. You can customize this summary. It will automatically update as commits are pushed.