File tree 1 file changed +3
-1
lines changed
1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -92,6 +92,7 @@ class InputRequest:
92
92
class RequestFuncOutput :
93
93
input_request : InputRequest = None
94
94
generated_token_list : list [str ] = None
95
+ generated_text : str = None
95
96
success : bool = False
96
97
latency : float = 0
97
98
ttft : float = 0
@@ -102,7 +103,7 @@ def to_dict(self):
102
103
return {
103
104
"prompt" : self .input_request .prompt ,
104
105
"original_output" : self .input_request .output ,
105
- "generated_token_list " : self .generated_token_list ,
106
+ "generated_text " : self .generated_text ,
106
107
"success" : self .success ,
107
108
"latency" : self .latency ,
108
109
"prompt_len" : self .prompt_len
@@ -280,6 +281,7 @@ async def send_request(
280
281
output .ttft = ttft
281
282
output .latency = latency
282
283
output .generated_token_list = generated_token_list
284
+ output .generated_text = "" .join (generated_token_list )
283
285
output .success = True
284
286
if pbar :
285
287
pbar .update (1 )
You can’t perform that action at this time.
0 commit comments