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

update tests given new type signature

Browse source
This commit is contained in:
Ashwin Bharambe 2025-02-20 16:57:17 -08:00
parent 32aa380e69
commit 587c429e08
1 changed files with 0 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -67,22 +67,12 @@ DUMMY_IMAGE_BASE64 = ImageContentItem(image=ImageContentItemImage(data="base64st
@pytest.mark.parametrize(
"contents",
[
[DUMMY_STRING],
[DUMMY_TEXT],
[DUMMY_STRING, DUMMY_STRING2],
[DUMMY_TEXT, DUMMY_TEXT2],
[[DUMMY_TEXT]],
[[DUMMY_TEXT], [DUMMY_TEXT, DUMMY_TEXT2]],
[DUMMY_STRING, [DUMMY_TEXT, DUMMY_TEXT2]],
],
ids=[
"string",
"text",
"list[string]",
"list[text]",
"list[list[text]]",
"list[list[text],list[text,text]]",
"string,list[text,text]",
],
)
def test_embedding_text(llama_stack_client, embedding_model_id, contents):
@ -96,26 +86,12 @@ def test_embedding_text(llama_stack_client, embedding_model_id, contents):
@pytest.mark.parametrize(
"contents",
[
[DUMMY_IMAGE_URL],
[DUMMY_IMAGE_BASE64],
[DUMMY_IMAGE_URL, DUMMY_IMAGE_BASE64],
[DUMMY_IMAGE_URL, DUMMY_STRING, DUMMY_IMAGE_BASE64, DUMMY_TEXT],
[[DUMMY_IMAGE_URL]],
[[DUMMY_IMAGE_BASE64]],
[[DUMMY_IMAGE_URL, DUMMY_TEXT, DUMMY_IMAGE_BASE64]],
[[DUMMY_IMAGE_URL], [DUMMY_IMAGE_BASE64]],
[[DUMMY_IMAGE_URL], [DUMMY_TEXT, DUMMY_IMAGE_BASE64]],
],
ids=[
"url",
"base64",
"list[url,base64]",
"list[url,string,base64,text]",
"list[list[url]]",
"list[list[base64]]",
"list[list[url,text,base64]]",
"list[list[url],list[base64]]",
"list[list[url],list[text,base64]]",
],
)
@pytest.mark.skip(reason="Media is not supported")