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

chore: Updating Vector IO integration tests to use llama stack build

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-10-17 08:38:50 -04:00
parent a701f68bd7
commit da7b39a3e3
13 changed files with 298 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

28
tests/integration/suites.py
View file

@ -153,6 +153,29 @@ SETUP_DEFINITIONS: dict[str, Setup] = {
"text_model": "groq/llama-3.3-70b-versatile",
},
),
"milvus": Setup(
name="milvus",
description="Milvus vector database provider for vector_io tests",
env={
"MILVUS_URL": "dummy",
},
),
"chromadb": Setup(
name="chromadb",
description="ChromaDB vector database provider for vector_io tests",
env={
"CHROMADB_URL": "http://localhost:8000",
},
),
"pgvector": Setup(
name="pgvector",
description="PGVector database provider for vector_io tests",
env={
"PGVECTOR_DB": "llama_stack_test",
"PGVECTOR_USER": "postgres",
"PGVECTOR_PASSWORD": "password",
},
),
}
@ -179,4 +202,9 @@ SUITE_DEFINITIONS: dict[str, Suite] = {
roots=["tests/integration/inference/test_vision_inference.py"],
default_setup="ollama-vision",
),
"vector_io": Suite(
name="vector_io",
roots=["tests/integration/vector_io"],
default_setup="milvus",
),
}