mirror of
https://github.com/meta-llama/llama-stack.git
synced 2026-01-03 06:32:16 +00:00
restructure config
This commit is contained in:
parent
702cf2d563
commit
26d9804efd
8 changed files with 218 additions and 62 deletions
|
|
@ -1,24 +1,59 @@
|
|||
tests:
|
||||
- path: inference/test_vision_inference.py
|
||||
functions:
|
||||
- test_vision_chat_completion_streaming
|
||||
- test_vision_chat_completion_non_streaming
|
||||
inference:
|
||||
tests:
|
||||
- inference/test_vision_inference.py::test_vision_chat_completion_streaming
|
||||
- inference/test_vision_inference.py::test_vision_chat_completion_non_streaming
|
||||
- inference/test_text_inference.py::test_structured_output
|
||||
- inference/test_text_inference.py::test_chat_completion_streaming
|
||||
- inference/test_text_inference.py::test_chat_completion_non_streaming
|
||||
- inference/test_text_inference.py::test_chat_completion_with_tool_calling
|
||||
- inference/test_text_inference.py::test_chat_completion_with_tool_calling_streaming
|
||||
|
||||
- path: inference/test_text_inference.py
|
||||
functions:
|
||||
- test_structured_output
|
||||
- test_chat_completion_streaming
|
||||
- test_chat_completion_non_streaming
|
||||
- test_chat_completion_with_tool_calling
|
||||
- test_chat_completion_with_tool_calling_streaming
|
||||
fixtures:
|
||||
provider_fixtures:
|
||||
- inference: ollama
|
||||
- default_fixture_param_id: fireworks
|
||||
- inference: together
|
||||
# - inference: tgi
|
||||
# - inference: vllm_remote
|
||||
inference_models:
|
||||
- meta-llama/Llama-3.1-8B-Instruct
|
||||
- meta-llama/Llama-3.2-11B-Vision-Instruct
|
||||
|
||||
inference_fixtures:
|
||||
- ollama
|
||||
- fireworks
|
||||
- together
|
||||
- tgi
|
||||
- vllm_remote
|
||||
safety_shield: ~
|
||||
embedding_model: ~
|
||||
|
||||
test_models:
|
||||
text: meta-llama/Llama-3.1-8B-Instruct
|
||||
vision: meta-llama/Llama-3.2-11B-Vision-Instruct
|
||||
|
||||
agent:
|
||||
tests:
|
||||
- agents/test_agents.py::test_agent_turns_with_safety
|
||||
- agents/test_agents.py::test_rag_agent
|
||||
|
||||
fixtures:
|
||||
provider_fixtures:
|
||||
- default_fixture_param_id: ollama
|
||||
- default_fixture_param_id: together
|
||||
- default_fixture_param_id: fireworks
|
||||
|
||||
safety_shield: ~
|
||||
embedding_model: ~
|
||||
|
||||
inference_models:
|
||||
- meta-llama/Llama-3.2-1B-Instruct
|
||||
|
||||
|
||||
memory:
|
||||
tests:
|
||||
- memory/test_memory.py::test_query_documents
|
||||
|
||||
fixtures:
|
||||
provider_fixtures:
|
||||
- default_fixture_param_id: ollama
|
||||
- inference: sentence_transformers
|
||||
memory: faiss
|
||||
- default_fixture_param_id: chroma
|
||||
|
||||
inference_models:
|
||||
- meta-llama/Llama-3.2-1B-Instruct
|
||||
|
||||
safety_shield: ~
|
||||
embedding_model: ~
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue