forked from phoenix-oss/llama-stack-mirror
add distro report (#847)
# What does this PR do? Generate distro reports to cover inference, agents, and vector_io. ## Test Plan Report generated through `/opt/miniconda3/envs/stack/bin/pytest -s -v tests/client-sdk/ --report` ## Sources Please link relevant resources if necessary. ## Before submitting - [ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case). - [ ] Ran pre-commit to handle lint / formatting issues. - [ ] Read the [contributor guideline](https://github.com/meta-llama/llama-stack/blob/main/CONTRIBUTING.md), Pull Request section? - [ ] Updated relevant documentation. - [ ] Wrote necessary unit or integration tests.
This commit is contained in:
parent
23f1980f9c
commit
597869a2aa
8 changed files with 328 additions and 60 deletions
|
@ -4,6 +4,7 @@
|
|||
# This source code is licensed under the terms described in the LICENSE file in
|
||||
# the root directory of this source tree.
|
||||
|
||||
from llama_stack.providers.datatypes import Api
|
||||
|
||||
INFERENCE_API_CAPA_TEST_MAP = {
|
||||
"chat_completion": {
|
||||
|
@ -27,10 +28,9 @@ INFERENCE_API_CAPA_TEST_MAP = {
|
|||
},
|
||||
}
|
||||
|
||||
MEMORY_API_TEST_MAP = {
|
||||
"/insert, /query": {
|
||||
"inline": ["test_memory_bank_insert_inline_and_query"],
|
||||
"url": ["test_memory_bank_insert_from_url_and_query"],
|
||||
VECTORIO_API_TEST_MAP = {
|
||||
"retrieve": {
|
||||
"": ["test_vector_db_retrieve"],
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -44,7 +44,7 @@ AGENTS_API_TEST_MAP = {
|
|||
|
||||
|
||||
API_MAPS = {
|
||||
"inference": INFERENCE_API_CAPA_TEST_MAP,
|
||||
"memory": MEMORY_API_TEST_MAP,
|
||||
"agents": AGENTS_API_TEST_MAP,
|
||||
Api.inference: INFERENCE_API_CAPA_TEST_MAP,
|
||||
Api.vector_io: VECTORIO_API_TEST_MAP,
|
||||
Api.agents: AGENTS_API_TEST_MAP,
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue