From ff269cb70082580fdda869b8271762c38028a084 Mon Sep 17 00:00:00 2001
From: Austin <31991302+austinm911@users.noreply.github.com>
Date: Wed, 24 Jul 2024 23:20:34 -0700
Subject: [PATCH] Return the rest of the properties from the Vercel AI SDK
 generations

---
 src/models.ts | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/models.ts b/src/models.ts
index b8cf316..17735fe 100644
--- a/src/models.ts
+++ b/src/models.ts
@@ -54,10 +54,11 @@ export async function generateAISDKCompletions<T extends z.ZodSchema<any>>(
     topP: options?.topP,
     mode: options?.mode,
   })
-
+	const { object, ...rest } = result
   return {
     data: result.object,
     url: page.url,
+    ...rest
   }
 }
 
@@ -108,11 +109,12 @@ export async function generateAISDKCode<T extends z.ZodSchema<any>>(
     maxTokens: options?.maxTokens,
     topP: options?.topP,
   })
-
-  return {
-    code: result.text,
-    url: page.url,
-  }
+  const { text, ...rest } = result
+	return {
+		code: result.text,
+		url: page.url,
+		...rest,
+	}
 }
 
 export async function generateLlamaCompletions<T extends z.ZodSchema<any>>(