-
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
to Python clientOptimizationConfig
and update tests and Rust deserialization to support it.GCPVertexGeminiSFTConfig
toOptimizationConfig
in__init__.py
for visibility in Python client.deserialize_optimization_config()
inpyo3_helpers.rs
to handleUninitializedGCPVertexGeminiSFTConfig
.test_sync_gcp_vertex_gemini_sft()
andtest_async_gcp_vertex_gemini_sft()
intest_optimization.py
for synchronous and asynchronous testing ofGCPVertexGeminiSFTConfig
.mixed_rendered_samples
inconftest.py
to adjust tool parameters structure.This description was created by
for 193a3a8. You can customize this summary. It will automatically update as commits are pushed.