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:
44
44
45
45
def get_next_request (self ) -> GenerationRequest :
46
46
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 )
48
48
base_request .content = "" .join (
49
49
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 + [ "" ] )
51
51
)
52
52
)
53
53
base_request .stats ["prompt_tokens" ] = sum (
54
54
x .stats ["prompt_tokens" ] for x in self .prompts [: completed_responses + 1 ]
55
55
)
56
56
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 ]
58
59
)
59
60
60
61
return base_request
You can’t perform that action at this time.
0 commit comments