From 41c879cd867ecc894f7200d68204dabe69d85a8d Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <95534749+ItzCrazyKns@users.noreply.github.com>
Date: Mon, 1 Dec 2025 18:34:36 +0530
Subject: [PATCH] feat(llm): use zod inferences at return type

---
 src/lib/models/base/llm.ts | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/lib/models/base/llm.ts b/src/lib/models/base/llm.ts
index fb24af4..0e175e7 100644
--- a/src/lib/models/base/llm.ts
+++ b/src/lib/models/base/llm.ts
@@ -1,3 +1,4 @@
+import z from 'zod';
 import {
   GenerateObjectInput,
   GenerateOptions,
@@ -12,10 +13,10 @@ abstract class BaseLLM<CONFIG> {
   abstract streamText(
     input: GenerateTextInput,
   ): AsyncGenerator<StreamTextOutput>;
-  abstract generateObject<T>(input: GenerateObjectInput): Promise<T>;
+  abstract generateObject<T>(input: GenerateObjectInput): Promise<z.infer<T>>;
   abstract streamObject<T>(
     input: GenerateObjectInput,
-  ): AsyncGenerator<Partial<T>>;
+  ): AsyncGenerator<Partial<z.infer<T>>>;
 }
 
 export default BaseLLM;