Fix validator not returning parsed value

packages/lms-client/src/Chat.ts

@@ -406,11 +406,11 @@ export class Chat extends MaybeMutable<ChatHistoryData> {
  */
 export type ChatLike = ChatInput | string | Chat | ChatMessageInput | ChatHistoryData;
 export const chatHistoryLikeSchema = z.union([
+  z.instanceof(Chat as any),
   chatHistoryDataSchema,
   z.string(),
   chatHistoryInputSchema,
   chatMessageInputSchema,
-  z.instanceof(Chat as any),
 ]) as ZodSchema<ChatLike>;
 
 /**
@@ -772,8 +772,8 @@ export class ChatMessage extends MaybeMutable<ChatMessageData> {
  */
 export type ChatMessageLike = ChatMessageInput | string | ChatMessage | ChatMessageData;
 export const chatMessageLikeSchema = z.union([
+  z.instanceof(ChatMessage as any),
   chatMessageInputSchema,
   z.string(),
-  z.instanceof(ChatMessage as any),
   chatMessageDataSchema,
 ]) as ZodSchema<ChatMessageLike>;

packages/lms-client/src/llm/LLMDynamicHandle.ts

@@ -134,7 +134,9 @@ export interface LLMRespondOpts<TStructuredOutputType = unknown>
    */
   onMessage?: (message: ChatMessage) => void;
 }
-const llmRespondOptsSchema = llmPredictionOptsSchema;
+const llmRespondOptsSchema = llmPredictionOptsSchema.extend({
+  onMessage: z.function().optional(),
+});
 
 type LLMRespondExtraOpts<TStructuredOutputType = unknown> = Omit<
   LLMRespondOpts<TStructuredOutputType>,

packages/lms-common/src/Validator.ts

@@ -91,7 +91,7 @@ export class Validator {
         index,
       }));
     if (errors.length === 0) {
-      return values as T;
+      return results.map(result => (result as any).data) as T;
     } else {
       const erroredValues = new Set(errors.map(({ index }) => index));
       const lead = leadProducer(erroredValues);