(id, finished: bool = False)
| 278 | |
| 279 | |
| 280 | def create_rsp(id, finished: bool = False): |
| 281 | result = tllm.Result() |
| 282 | result.output_token_ids = [[id]] |
| 283 | result.context_logits = None |
| 284 | result.generation_logits = None |
| 285 | result.log_probs = None |
| 286 | result.cum_log_probs = None |
| 287 | if finished: |
| 288 | result.finish_reasons = [tllm.FinishReason.END_ID] |
| 289 | result.is_final = finished |
| 290 | result.sequence_index = 0 |
| 291 | return tllm.Response(request_id=0, result=result, client_id=0) |
| 292 | |
| 293 | |
| 294 | def test_GenerationResultBase(): |
no outgoing calls
no test coverage detected