diff --git a/llama_stack/providers/tests/inference/test_vision_inference.py b/llama_stack/providers/tests/inference/test_vision_inference.py index 964f70901..a2434ac41 100644 --- a/llama_stack/providers/tests/inference/test_vision_inference.py +++ b/llama_stack/providers/tests/inference/test_vision_inference.py @@ -4,12 +4,12 @@ # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. +import base64 from pathlib import Path import pytest -from llama_stack.apis.common.content_types import ImageContentItem, TextContentItem, URL - +from llama_stack.apis.common.content_types import URL, ImageContentItem, TextContentItem from llama_stack.apis.inference import ( ChatCompletionResponse, ChatCompletionResponseEventType, @@ -23,7 +23,7 @@ from .utils import group_chunks THIS_DIR = Path(__file__).parent with open(THIS_DIR / "pasta.jpeg", "rb") as f: - PASTA_IMAGE = f.read() + PASTA_IMAGE = base64.b64encode(f.read()).decode("utf-8") class TestVisionModelInference: