diff --git a/src/lib/agents/media/image.ts b/src/lib/agents/media/image.ts
index f146824..b04d532 100644
--- a/src/lib/agents/media/image.ts
+++ b/src/lib/agents/media/image.ts
@@ -29,7 +29,7 @@ const searchImages = async (
     query: z.string().describe('The image search query.'),
   });
 
-  const res = await llm.generateObject<z.infer<typeof schema>>({
+  const res = await llm.generateObject<typeof schema>({
     messages: [
       {
         role: 'system',
diff --git a/src/lib/agents/media/video.ts b/src/lib/agents/media/video.ts
index feac720..c8f19b6 100644
--- a/src/lib/agents/media/video.ts
+++ b/src/lib/agents/media/video.ts
@@ -28,7 +28,7 @@ const searchVideos = async (
     query: z.string().describe('The video search query.'),
   });
 
-  const res = await llm.generateObject<z.infer<typeof schema>>({
+  const res = await llm.generateObject<typeof schema>({
     messages: [
       {
         role: 'system',