llama-stack-mirror/llama_stack/templates
Xi Yan 9d005154d7
fix vllm template (#813)
# What does this PR do?

- Fix vLLM template to resolve
https://github.com/meta-llama/llama-stack/issues/805
- Fix agents test with shields

## Test Plan

```
vllm serve meta-llama/Llama-3.1-8B-Instruct
VLLM_URL="http://localhost:8000/v1" INFERENCE_MODEL="meta-llama/Llama-3.1-8B-Instruct" llama stack run ./llama_stack/templates/remote-vllm/run.yaml
```

```
LLAMA_STACK_BASE_URL=http://localhost:8321 pytest -v ./tests/client-sdk/
```

<img width="1245" alt="image"
src="https://github.com/user-attachments/assets/9af27684-5a9c-4187-b338-cbfc5211bd99"
/>


- custom tool flaky due to model outputs
- /completions API not implemented

**Vision Model**
- 11B-Vision-Instruct
<img width="1240" alt="image"
src="https://github.com/user-attachments/assets/1d3b3b17-fa09-43a7-b56c-3f77263825c5"
/>


## 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.
2025-01-17 15:34:29 -08:00
..
bedrock cerebras template update for memory (#792) 2025-01-16 16:07:53 -08:00
cerebras Remove llama-guard in Cerebras template & improve agent test (#798) 2025-01-16 18:11:35 -08:00
experimental-post-training add braintrust to experimental-post-training template (#763) 2025-01-14 13:42:59 -08:00
fireworks cerebras template update for memory (#792) 2025-01-16 16:07:53 -08:00
hf-endpoint cerebras template update for memory (#792) 2025-01-16 16:07:53 -08:00
hf-serverless add default toolgroups to all providers (#795) 2025-01-16 16:54:59 -08:00
meta-reference-gpu add default toolgroups to all providers (#795) 2025-01-16 16:54:59 -08:00
meta-reference-quantized-gpu cerebras template update for memory (#792) 2025-01-16 16:07:53 -08:00
nvidia add default toolgroups to all providers (#795) 2025-01-16 16:54:59 -08:00
ollama add default toolgroups to all providers (#795) 2025-01-16 16:54:59 -08:00
remote-vllm fix vllm template (#813) 2025-01-17 15:34:29 -08:00
tgi add default toolgroups to all providers (#795) 2025-01-16 16:54:59 -08:00
together cerebras template update for memory (#792) 2025-01-16 16:07:53 -08:00
vllm-gpu cerebras template update for memory (#792) 2025-01-16 16:07:53 -08:00
__init__.py Auto-generate distro yamls + docs (#468) 2024-11-18 14:57:06 -08:00
template.py agents to use tools api (#673) 2025-01-08 19:01:00 -08:00