From 3099c5243fb4d93cc9df4282395371ec97660812 Mon Sep 17 00:00:00 2001 From: Matthew Farrellee Date: Fri, 21 Feb 2025 10:02:21 -0600 Subject: [PATCH] fix: update URL import, URL -> ImageContentItemImageURL (#1204) # 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` --- tests/client-sdk/inference/test_embedding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/client-sdk/inference/test_embedding.py b/tests/client-sdk/inference/test_embedding.py index a25382866..602f9c062 100644 --- a/tests/client-sdk/inference/test_embedding.py +++ b/tests/client-sdk/inference/test_embedding.py @@ -47,9 +47,9 @@ import pytest from llama_stack_client.types import EmbeddingsResponse from llama_stack_client.types.shared.interleaved_content import ( - URL, ImageContentItem, ImageContentItemImage, + ImageContentItemImageURL, TextContentItem, ) @@ -59,7 +59,7 @@ DUMMY_TEXT = TextContentItem(text=DUMMY_STRING, type="text") DUMMY_TEXT2 = TextContentItem(text=DUMMY_STRING2, type="text") # TODO(mf): add a real image URL and base64 string 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")