diff --git a/src/lib/models/base/llm.ts b/src/lib/models/base/llm.ts
index 5d6f52d..e701fa9 100644
--- a/src/lib/models/base/llm.ts
+++ b/src/lib/models/base/llm.ts
@@ -1,10 +1,8 @@
 import {
   GenerateObjectInput,
-  GenerateObjectOutput,
   GenerateOptions,
   GenerateTextInput,
   GenerateTextOutput,
-  StreamObjectOutput,
   StreamTextOutput,
 } from '../types';
 
@@ -15,12 +13,10 @@ abstract class BaseLLM<CONFIG> {
   abstract streamText(
     input: GenerateTextInput,
   ): AsyncGenerator<StreamTextOutput>;
-  abstract generateObject<T>(
-    input: GenerateObjectInput,
-  ): Promise<GenerateObjectOutput<T>>;
+  abstract generateObject<T>(input: GenerateObjectInput): Promise<T>;
   abstract streamObject<T>(
     input: GenerateObjectInput,
-  ): AsyncGenerator<StreamObjectOutput<T>>;
+  ): AsyncGenerator<Partial<T>>;
 }
 
 export default BaseLLM;