feat: support vertex imagen3

This commit is contained in:
Laisky.Cai
2025-01-12 04:18:57 +00:00
parent bfe28fc1f8
commit feacea0321
30 changed files with 227 additions and 49 deletions

View File

@@ -0,0 +1,23 @@
package imagen
type CreateImageRequest struct {
Instances []createImageInstance `json:"instances" binding:"required,min=1"`
Parameters createImageParameters `json:"parameters" binding:"required"`
}
type createImageInstance struct {
Prompt string `json:"prompt"`
}
type createImageParameters struct {
SampleCount int `json:"sample_count" binding:"required,min=1"`
}
type CreateImageResponse struct {
Predictions []createImageResponsePrediction `json:"predictions"`
}
type createImageResponsePrediction struct {
MimeType string `json:"mimeType"`
BytesBase64Encoded string `json:"bytesBase64Encoded"`
}