File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -44,17 +44,18 @@ def __len__(self) -> int:
4444
4545 def get_next_request (self ) -> GenerationRequest :
4646 completed_responses = len (self .responses )
47- base_request = self .prompts [completed_responses ].model_copy ()
47+ base_request = self .prompts [completed_responses ].model_copy (deep = True )
4848 base_request .content = "" .join (
4949 itertools .chain .from_iterable (
50- zip ((x .content for x in self .prompts ), self .responses )
50+ zip ((x .content for x in self .prompts ), self .responses + [ "" ] )
5151 )
5252 )
5353 base_request .stats ["prompt_tokens" ] = sum (
5454 x .stats ["prompt_tokens" ] for x in self .prompts [: completed_responses + 1 ]
5555 )
5656 base_request .constraints ["output_tokens" ] = sum (
57- x .constraints ["output_tokens" ] for x in self .prompts [:completed_responses ]
57+ x .constraints ["output_tokens" ]
58+ for x in self .prompts [: completed_responses + 1 ]
5859 )
5960
6061 return base_request
You can’t perform that action at this time.
0 commit comments