From 16cfe1014e8a632dd8805ccbde2e0ba92c30a46f Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 26 Dec 2024 17:31:42 -0800 Subject: [PATCH] vision inference --- .../tests/inference/test_vision_inference.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/llama_stack/providers/tests/inference/test_vision_inference.py b/llama_stack/providers/tests/inference/test_vision_inference.py index d58164676..1bdee051f 100644 --- a/llama_stack/providers/tests/inference/test_vision_inference.py +++ b/llama_stack/providers/tests/inference/test_vision_inference.py @@ -8,11 +8,16 @@ from pathlib import Path import pytest - -from llama_models.llama3.api.datatypes import * # noqa: F403 -from llama_stack.apis.inference import * # noqa: F403 from llama_stack.apis.common.content_types import ImageContentItem, TextContentItem, URL +from llama_stack.apis.inference import ( + ChatCompletionResponse, + ChatCompletionResponseEventType, + ChatCompletionResponseStreamChunk, + SamplingParams, + UserMessage, +) + from .utils import group_chunks THIS_DIR = Path(__file__).parent