fix: streaming parser overflow

Author: dosco

src/ax/dsp/generate.ts

@@ -351,6 +351,10 @@ export class AxGen<
           streamingValidation
         )
 
+        if (skip) {
+          continue
+        }
+
         if (this.streamingAsserts.length !== 0) {
           await assertStreamingAssertions(
             this.streamingAsserts,
@@ -372,10 +376,6 @@ export class AxGen<
 
         yield* streamValues<OUT>(this.signature, content, values, xstate)
 
-        if (skip) {
-          continue
-        }
-
         await assertAssertions(this.asserts, values)
       }
 

src/examples/extract.ts

@@ -33,7 +33,7 @@ const ai = new AxAI({
 // ai.setOptions({ debug: true })
 
 const gen = new AxGen<{ chatMessage: string; currentDate: Date }>(
-  `chatMessage, currentDate:datetime -> subject, foundMeeting:boolean, ticketNumber?:number, customerNumber?:number, datesMentioned:datetime[], shortSummary, messageType:class "reminder, follow-up, meeting, other"`
+  `chatMessage, currentDate:datetime -> subject, thinking, reasoning, foundMeeting:boolean, ticketNumber?:number, customerNumber?:number, datesMentioned:datetime[], shortSummary, messageType:class "reminder, follow-up, meeting, other"`
 )
 
 const stream = await gen.streamingForward(ai, { chatMessage, currentDate })
@@ -44,7 +44,7 @@ for await (const chunk of stream) {
   console.log('>', chunk)
 }
 
-console.log('\n\n# Not Streaming')
+// console.log('\n\n# Not Streaming')
 
-const res = await gen.forward(ai, { chatMessage, currentDate })
-console.log('>', res)
+// const res = await gen.forward(ai, { chatMessage, currentDate })
+// console.log('>', res)