phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-05 18:22:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix

Browse source
This commit is contained in:
Sixian Yi 2025-01-15 15:27:23 -08:00
parent 27e07b44b5
commit 685baa7f2d
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/providers/tests/inference/test_embeddings.py
View file

@ -6,7 +6,8 @@
import pytest import pytest
from llama_stack.apis.inference import EmbeddingsResponse, ModelType from llama_stack.apis.inference import EmbeddingsResponse
from llama_stack.apis.models import ModelType
# How to run this test: # How to run this test:
# pytest -v -s llama_stack/providers/tests/inference/test_embeddings.py # pytest -v -s llama_stack/providers/tests/inference/test_embeddings.py

2
llama_stack/providers/tests/inference/test_vision_inference.py
View file

@ -143,7 +143,7 @@ class TestVisionModelInference:
assert len(grouped[ChatCompletionResponseEventType.complete]) == 1 assert len(grouped[ChatCompletionResponseEventType.complete]) == 1
content = "".join( content = "".join(
chunk.event.delta chunk.event.delta.text
for chunk in grouped[ChatCompletionResponseEventType.progress] for chunk in grouped[ChatCompletionResponseEventType.progress]
) )
for expected_string in expected_strings: for expected_string in expected_strings: