Skip to content

feat(veo): add Veo 3.1 model support #733

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.

Sign up for GitHub

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

Open
CharlesWong wants to merge 1 commit into
base: master
Choose a base branch
from
Open

feat(veo): add Veo 3.1 model support #733

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 5 additions & 3 deletions packages/app/server/src/providers/VertexAIProvider.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,11 @@ import { env } from '../env';

// Constants
export const PROXY_PASSTHROUGH_ONLY_MODEL = 'PROXY_PLACEHOLDER_VERTEX_AI';
const VEO3_MODELS = [
const VEO_MODELS = [
'veo-3.0-fast-generate-preview',
'veo-3.0-generate-preview',
'veo-3.1-generate-preview',
'veo-3.1-fast-generate-preview',
];
const GCS_BUCKET_NAME = 'echo-veo3-videos';
const GCS_URI_PREFIX = 'gs://';
Expand Down Expand Up @@ -496,7 +498,7 @@ export class VertexAIProvider extends BaseProvider {
}
}

private isVeo3Model(): boolean {
return VEO3_MODELS.includes(this.getModel());
private isVeoModel(): boolean {
return VEO_MODELS.includes(this.getModel());
}
}
28 changes: 22 additions & 6 deletions packages/sdk/ts/src/supported-models/video/vertex-ai.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,25 +2,41 @@ import type { SupportedVideoModel } from '../types';

export type VertexAIVideoModel =
| 'veo-3.0-fast-generate-preview'
| 'veo-3.0-generate-preview';
| 'veo-3.0-generate-preview'
| 'veo-3.1-generate-preview'
| 'veo-3.1-fast-generate-preview';
/**
* Vertex AI video models with official pricing information
* Based on: https://cloud.google.com/vertex-ai/generative-ai/pricing
*
* Veo 3: $0.40/second with audio, $0.20/second video only
* Veo 3 Fast: $0.15/second with audio, $0.10/second video only
* Veo 3.0: $0.40/second with audio, $0.20/second video only
* Veo 3.0 Fast: $0.15/second with audio, $0.10/second video only
* Veo 3.1: $0.40/second with audio, $0.20/second video only
* Veo 3.1 Fast: $0.15/second with audio, $0.10/second video only
*/
export const VertexAIVideoModels: SupportedVideoModel[] = [
{
model_id: 'veo-3.0-fast-generate-preview',
cost_per_second_with_audio: 0.15,
cost_per_second_without_audio: 0.1, // Fixed: was 0.1, now 0.10 for clarity
cost_per_second_without_audio: 0.1,
provider: 'VertexAI',
},
{
model_id: 'veo-3.0-generate-preview',
cost_per_second_with_audio: 0.4, // Fixed: was 0.4, now 0.40 for clarity
cost_per_second_without_audio: 0.2, // Fixed: was 0.2, now 0.20 for clarity
cost_per_second_with_audio: 0.4,
cost_per_second_without_audio: 0.2,
provider: 'VertexAI',
},
{
model_id: 'veo-3.1-generate-preview',
cost_per_second_with_audio: 0.4,
cost_per_second_without_audio: 0.2,
provider: 'VertexAI',
},
{
model_id: 'veo-3.1-fast-generate-preview',
cost_per_second_with_audio: 0.15,
cost_per_second_without_audio: 0.1,
provider: 'VertexAI',
},
];
2 changes: 2 additions & 0 deletions templates/next-video-template/src/app/api/generate-video/validation.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,8 @@ export function validateGenerateVideoRequest(body: unknown): ValidationResult {
const validModels: VideoModelOption[] = [
'veo-3.0-fast-generate-preview',
'veo-3.0-generate-preview',
'veo-3.1-generate-preview',
'veo-3.1-fast-generate-preview',
];
if (!model || !validModels.includes(model as VideoModelOption)) {
return {
Expand Down
2 changes: 1 addition & 1 deletion templates/next-video-template/src/app/api/generate-video/vertex.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ import {
*/
export async function handleGeminiGenerate(
prompt: string,
model: 'veo-3.0-fast-generate-preview' | 'veo-3.0-generate-preview',
model: 'veo-3.0-fast-generate-preview' | 'veo-3.0-generate-preview' | 'veo-3.1-generate-preview' | 'veo-3.1-fast-generate-preview',
durationSeconds: number = 4,
generateAudio: boolean = false,
image?: string, // Base64 encoded image or data URL (first frame)
Expand Down
4 changes: 3 additions & 1 deletion templates/next-video-template/src/components/video-generator.tsx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,8 @@ import { VideoHistory } from './video-history';
const models: VideoModelConfig[] = [
{ id: 'veo-3.0-fast-generate-preview', name: 'Veo 3 Fast' },
{ id: 'veo-3.0-generate-preview', name: 'Veo 3' },
{ id: 'veo-3.1-generate-preview', name: 'Veo 3.1' },
{ id: 'veo-3.1-fast-generate-preview', name: 'Veo 3.1 Fast' },
];

/**
Expand All @@ -57,7 +59,7 @@ const models: VideoModelConfig[] = [
*/
export default function VideoGenerator() {
const [model, setModel] = useState<VideoModelOption>(
'veo-3.0-fast-generate-preview'
'veo-3.1-fast-generate-preview'
);
const [durationSeconds, setDurationSeconds] = useState<4 | 6 | 8>(4);
const [generateAudio, setGenerateAudio] = useState<boolean>(false);
Expand Down
4 changes: 3 additions & 1 deletion templates/next-video-template/src/lib/types.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,9 @@ export type ModelOption = 'openai' | 'gemini';
*/
export type VideoModelOption =
| 'veo-3.0-fast-generate-preview'
| 'veo-3.0-generate-preview';
| 'veo-3.0-generate-preview'
| 'veo-3.1-generate-preview'
| 'veo-3.1-fast-generate-preview';

/**
* Model configuration with display names
Expand Down