diff --git a/app/(playground)/p/[agentId]/beta-proto/graph/server-actions.ts b/app/(playground)/p/[agentId]/beta-proto/graph/server-actions.ts
index 81a03aca..7a403764 100644
--- a/app/(playground)/p/[agentId]/beta-proto/graph/server-actions.ts
+++ b/app/(playground)/p/[agentId]/beta-proto/graph/server-actions.ts
@@ -33,7 +33,7 @@ export async function generateArtifactStream(
 	const stream = createStreamableValue();
 
 	(async () => {
-		const model = "gpt-4o-mini";
+		const model = "gpt-4o";
 		const generation = trace.generation({
 			input: params.userPrompt,
 			model,
diff --git a/app/(playground)/p/[agentId]/beta-proto/web-search/server-action.ts b/app/(playground)/p/[agentId]/beta-proto/web-search/server-action.ts
index efba5275..672548e0 100644
--- a/app/(playground)/p/[agentId]/beta-proto/web-search/server-action.ts
+++ b/app/(playground)/p/[agentId]/beta-proto/web-search/server-action.ts
@@ -35,7 +35,7 @@ export async function generateWebSearchStream(
 	const stream = createStreamableValue();
 
 	(async () => {
-		const model = "gpt-4o-mini";
+		const model = "gpt-4o";
 		const generation = trace.generation({
 			input: inputs.userPrompt,
 			model,