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

vision inference

Browse source
This commit is contained in:
Xi Yan 2024-12-26 17:31:42 -08:00
parent 3b1f20ac00
commit 16cfe1014e
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -8,11 +8,16 @@ from pathlib import Path
import pytest 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.common.content_types import ImageContentItem, TextContentItem, URL
from llama_stack.apis.inference import (
ChatCompletionResponse,
ChatCompletionResponseEventType,
ChatCompletionResponseStreamChunk,
SamplingParams,
UserMessage,
)
from .utils import group_chunks from .utils import group_chunks
THIS_DIR = Path(__file__).parent THIS_DIR = Path(__file__).parent