feat(llm): use zod inferences at return type

This commit is contained in:
ItzCrazyKns
2025-12-01 18:34:36 +05:30
parent 9b3833f933
commit 41c879cd86

View File

@@ -1,3 +1,4 @@
import z from 'zod';
import { import {
GenerateObjectInput, GenerateObjectInput,
GenerateOptions, GenerateOptions,
@@ -12,10 +13,10 @@ abstract class BaseLLM<CONFIG> {
abstract streamText( abstract streamText(
input: GenerateTextInput, input: GenerateTextInput,
): AsyncGenerator<StreamTextOutput>; ): AsyncGenerator<StreamTextOutput>;
abstract generateObject<T>(input: GenerateObjectInput): Promise<T>; abstract generateObject<T>(input: GenerateObjectInput): Promise<z.infer<T>>;
abstract streamObject<T>( abstract streamObject<T>(
input: GenerateObjectInput, input: GenerateObjectInput,
): AsyncGenerator<Partial<T>>; ): AsyncGenerator<Partial<z.infer<T>>>;
} }
export default BaseLLM; export default BaseLLM;