phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

fix: update URL import, URL -> ImageContentItemImageURL (#1204)

Browse source 
# What does this PR do?

fixes test to use new name for URL import

## Test Plan

`LLAMA_STACK_BASE_URL=http://localhost:8321 pytest -v
tests/client-sdk/inference/test_embedding.py --embedding-model
baai/bge-m3`
This commit is contained in:
Matthew Farrellee 2025-02-21 10:02:21 -06:00 • committed by GitHub
parent 34226d6c93
commit 3099c5243f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/client-sdk/inference/test_embedding.py
View file

@ -47,9 +47,9 @@
import pytest import pytest
from llama_stack_client.types import EmbeddingsResponse from llama_stack_client.types import EmbeddingsResponse
from llama_stack_client.types.shared.interleaved_content import ( from llama_stack_client.types.shared.interleaved_content import (
URL,
ImageContentItem, ImageContentItem,
ImageContentItemImage, ImageContentItemImage,
ImageContentItemImageURL,
TextContentItem, TextContentItem,
) )
@ -59,7 +59,7 @@ DUMMY_TEXT = TextContentItem(text=DUMMY_STRING, type="text")
DUMMY_TEXT2 = TextContentItem(text=DUMMY_STRING2, type="text") DUMMY_TEXT2 = TextContentItem(text=DUMMY_STRING2, type="text")
# TODO(mf): add a real image URL and base64 string # TODO(mf): add a real image URL and base64 string
DUMMY_IMAGE_URL = ImageContentItem( DUMMY_IMAGE_URL = ImageContentItem(
image=ImageContentItemImage(url=URL(uri="https://example.com/image.jpg")), type="image" image=ImageContentItemImage(url=ImageContentItemImageURL(uri="https://example.com/image.jpg")), type="image"
) )
DUMMY_IMAGE_BASE64 = ImageContentItem(image=ImageContentItemImage(data="base64string"), type="image") DUMMY_IMAGE_BASE64 = ImageContentItem(image=ImageContentItemImage(data="base64string"), type="image")