@@ -56,7 +56,8 @@ describe('ref tags', async () => {
5656 child : 'child message' ,
5757 }
5858
59- const action = ( ) => render ( { prompt : prompts [ 'parent' ] , adapter : Adapters . default } )
59+ const action = ( ) =>
60+ render ( { prompt : prompts [ 'parent' ] , adapter : Adapters . default } )
6061 const error = await getExpectedError ( action , CompileError )
6162 expect ( error . code ) . toBe ( 'missing-reference-function' )
6263 } )
@@ -83,16 +84,19 @@ describe('ref tags', async () => {
8384 parent : '<prompt path="child" />' ,
8485 }
8586
86- const action = ( ) =>
87- render ( {
88- prompt : prompts [ 'parent' ] ,
89- parameters : { foo : 'bar' } ,
90- referenceFn : buildReferenceFn ( prompts ) ,
91- adapter : Adapters . default ,
92- } )
93-
94- const error = await getExpectedError ( action , CompileError )
95- expect ( error . code ) . toBe ( 'variable-not-declared' )
87+ const result = await render ( {
88+ prompt : prompts [ 'parent' ] ,
89+ referenceFn : buildReferenceFn ( prompts ) ,
90+ adapter : Adapters . default ,
91+ } )
92+ expect ( result . messages . length ) . toBe ( 1 )
93+ const message = result . messages [ 0 ] ! as SystemMessage
94+ expect ( message . content ) . toEqual ( [
95+ {
96+ type : 'text' ,
97+ text : 'Child message:' ,
98+ } ,
99+ ] )
96100 } )
97101
98102 it ( 'referenced prompts can receive parameters as tag attributes' , async ( ) => {
@@ -336,12 +340,12 @@ describe('ref tags', async () => {
336340 {{ bar }}
337341 </content-text>
338342 {{ endfor }}
339- ` )
343+ ` ) ,
340344 }
341345
342346 const metadata = await scan ( {
343347 prompt : prompts [ 'parent' ] ,
344- referenceFn : buildReferenceFn ( prompts )
348+ referenceFn : buildReferenceFn ( prompts ) ,
345349 } )
346350
347351 const result = await render ( {
@@ -351,18 +355,19 @@ describe('ref tags', async () => {
351355 } )
352356
353357 expect ( result . messages . length ) . toBe ( 2 )
354- const [ firstMessage , secondMessage ] = result . messages as [ UserMessage , SystemMessage ]
358+ const [ firstMessage , secondMessage ] = result . messages as [
359+ UserMessage ,
360+ SystemMessage ,
361+ ]
355362 expect ( firstMessage . role ) . toBe ( MessageRole . user )
356363 expect ( firstMessage . content . length ) . toBe ( 3 )
357364 expect ( firstMessage . content ) . toEqual ( [
358- { type : 'text' , text : '11' } ,
359- { type : 'text' , text : '12' } ,
360- { type : 'text' , text : '13' } ,
365+ { type : 'text' , text : '11' } ,
366+ { type : 'text' , text : '12' } ,
367+ { type : 'text' , text : '13' } ,
361368 ] )
362369 expect ( secondMessage . role ) . toBe ( MessageRole . system )
363370 expect ( secondMessage . content . length ) . toBe ( 1 )
364- expect ( secondMessage . content ) . toEqual ( [
365- { type : 'text' , text : '10' }
366- ] )
371+ expect ( secondMessage . content ) . toEqual ( [ { type : 'text' , text : '10' } ] )
367372 } )
368373} )
0 commit comments