TensorRT-LLMs/tests/unittest/api_stability/references_committed/request_output.yaml
Enwei Zhu 224469b096
test: [TRTLLM-4334] Create 1.0 criteria scope from API stability references (#3069)
* committed APIs validation

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* fix

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* clean name

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* separate

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* add TODOs

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* fix naming

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* fix

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

---------

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>
2025-03-26 18:14:35 +08:00

34 lines
861 B
YAML

methods:
aresult:
parameters: {}
return_annotation: tensorrt_llm.executor.result.GenerationResult
result:
parameters:
timeout:
annotation: Optional[float]
default: None
return_annotation: tensorrt_llm.executor.result.GenerationResult
properties:
request_id:
annotation: int
default: inspect._empty
prompt:
annotation: Optional[str]
default: inspect._empty
prompt_token_ids:
annotation: List[int]
default: inspect._empty
outputs:
annotation: List[tensorrt_llm.executor.result.CompletionOutput]
default: inspect._empty
context_logits:
annotation: Optional[torch.Tensor]
default: inspect._empty
finished:
annotation: bool
default: inspect._empty
# TODO [TRTLLM-876]
# prompt_logprobs:
# annotation: Optional[PromptLogprobs]
# default: inspect._empty