mirror of
https://github.com/meta-llama/llama-stack.git
synced 2026-01-02 02:30:00 +00:00
fix tests
This commit is contained in:
parent
d9d827ff25
commit
0edd3ce78b
3 changed files with 7 additions and 8 deletions
|
|
@ -32,6 +32,7 @@ from llama_stack.apis.inference import (
|
|||
UserMessage,
|
||||
)
|
||||
from llama_stack.apis.models import Model
|
||||
|
||||
from .utils import group_chunks
|
||||
|
||||
|
||||
|
|
@ -476,7 +477,7 @@ class TestInference:
|
|||
last = grouped[ChatCompletionResponseEventType.progress][-1]
|
||||
# assert last.event.stop_reason == expected_stop_reason
|
||||
assert last.event.delta.parse_status == ToolCallParseStatus.succeeded
|
||||
assert last.event.delta.content.type == "tool_call"
|
||||
assert isinstance(last.event.delta.content, ToolCall)
|
||||
|
||||
call = last.event.delta.content
|
||||
assert call.tool_name == "get_weather"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue