mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-30 16:53:12 +00:00
657bb12e85
This gets the fireworks provider passing 100% of our OpenAI API
verification tests when run against a Llama Stack server using the
fireworks provider. Testing against Fireworks directly, without Llama
Stack in the middle, has a lower pass rate.
The main changes are are in how we divert Llama model OpenAI chat
completion requests to the Llama Stack chat completion API (vs
OpenAI), which applies all the client-side formatting necessary to get
tool calls working properly on Fireworks.
A side-effect of this work is any provider using the
OpenAIChatCompletionToLlamaStackMixin (renamed from
OpenAIChatCompletioonUnsupportedMixin) will also get a better
conversion from OpenAI to Llama Stack, including streaming and
non-stream responses.
A small change was required to
`llama_stack/models/llama/llama3/tool_utils.py` to get tests to 100%
because code there was incorrectly assuming any JSON response with a
`name` key was a tool call response. One of our verification tests
produces JSON keys with a `name` key that is not a tool call response,
so I tightened up the logic there to require both a `name` and
`parameters` key in the JSON response before it gets considered a
potential tool call. The `parameters` key was required by the code
anyway, but it wasn't explicitly checking for its existence.
Lastly, this adds some new verification test configs so we can see the
results of using OpenAI APIs against SaaS services directly compared
to hitting Llama Stack with a remote provider pointing at that SaaS
service.
You can run these verification tests like:
```
llama stack run \
--image-type venv \
tests/verifications/openai-api-verification-run.yaml
python tests/verifications/generate_report.py \
--run-tests \
--provider together fireworks openai \
together-llama-stack \
fireworks-llama-stack \
openai-llama-stack
```
Signed-off-by: Ben Browning <bbrownin@redhat.com>
126 lines
3.7 KiB
YAML
126 lines
3.7 KiB
YAML
version: '2'
|
|
image_name: openai-api-verification
|
|
apis:
|
|
- inference
|
|
- telemetry
|
|
- tool_runtime
|
|
- vector_io
|
|
providers:
|
|
inference:
|
|
- provider_id: together
|
|
provider_type: remote::together
|
|
config:
|
|
url: https://api.together.xyz/v1
|
|
api_key: ${env.TOGETHER_API_KEY:}
|
|
- provider_id: fireworks
|
|
provider_type: remote::fireworks
|
|
config:
|
|
url: https://api.fireworks.ai/inference/v1
|
|
api_key: ${env.FIREWORKS_API_KEY}
|
|
- provider_id: openai
|
|
provider_type: remote::openai
|
|
config:
|
|
url: https://api.openai.com/v1
|
|
api_key: ${env.OPENAI_API_KEY:}
|
|
- provider_id: sentence-transformers
|
|
provider_type: inline::sentence-transformers
|
|
config: {}
|
|
vector_io:
|
|
- provider_id: faiss
|
|
provider_type: inline::faiss
|
|
config:
|
|
kvstore:
|
|
type: sqlite
|
|
namespace: null
|
|
db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/openai}/faiss_store.db
|
|
telemetry:
|
|
- provider_id: meta-reference
|
|
provider_type: inline::meta-reference
|
|
config:
|
|
service_name: "${env.OTEL_SERVICE_NAME:\u200B}"
|
|
sinks: ${env.TELEMETRY_SINKS:console,sqlite}
|
|
sqlite_db_path: ${env.SQLITE_DB_PATH:~/.llama/distributions/openai/trace_store.db}
|
|
tool_runtime:
|
|
- provider_id: brave-search
|
|
provider_type: remote::brave-search
|
|
config:
|
|
api_key: ${env.BRAVE_SEARCH_API_KEY:}
|
|
max_results: 3
|
|
- provider_id: tavily-search
|
|
provider_type: remote::tavily-search
|
|
config:
|
|
api_key: ${env.TAVILY_SEARCH_API_KEY:}
|
|
max_results: 3
|
|
- provider_id: code-interpreter
|
|
provider_type: inline::code-interpreter
|
|
config: {}
|
|
- provider_id: rag-runtime
|
|
provider_type: inline::rag-runtime
|
|
config: {}
|
|
- provider_id: model-context-protocol
|
|
provider_type: remote::model-context-protocol
|
|
config: {}
|
|
- provider_id: wolfram-alpha
|
|
provider_type: remote::wolfram-alpha
|
|
config:
|
|
api_key: ${env.WOLFRAM_ALPHA_API_KEY:}
|
|
metadata_store:
|
|
type: sqlite
|
|
db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/openai}/registry.db
|
|
models:
|
|
- metadata: {}
|
|
model_id: together/meta-llama/Llama-3.3-70B-Instruct-Turbo
|
|
provider_id: together
|
|
provider_model_id: meta-llama/Llama-3.3-70B-Instruct-Turbo
|
|
model_type: llm
|
|
- metadata: {}
|
|
model_id: together/meta-llama/Llama-4-Scout-17B-16E-Instruct
|
|
provider_id: together
|
|
provider_model_id: meta-llama/Llama-4-Scout-17B-16E-Instruct
|
|
model_type: llm
|
|
- metadata: {}
|
|
model_id: together/meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8
|
|
provider_id: together
|
|
provider_model_id: meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8
|
|
model_type: llm
|
|
- metadata: {}
|
|
model_id: fireworks/llama-v3p3-70b-instruct
|
|
provider_id: fireworks
|
|
provider_model_id: accounts/fireworks/models/llama-v3p3-70b-instruct
|
|
model_type: llm
|
|
- metadata: {}
|
|
model_id: fireworks/llama4-scout-instruct-basic
|
|
provider_id: fireworks
|
|
provider_model_id: accounts/fireworks/models/llama4-scout-instruct-basic
|
|
model_type: llm
|
|
- metadata: {}
|
|
model_id: fireworks/llama4-maverick-instruct-basic
|
|
provider_id: fireworks
|
|
provider_model_id: accounts/fireworks/models/llama4-maverick-instruct-basic
|
|
model_type: llm
|
|
- metadata: {}
|
|
model_id: openai/gpt-4o
|
|
provider_id: openai
|
|
provider_model_id: openai/gpt-4o
|
|
model_type: llm
|
|
- metadata: {}
|
|
model_id: openai/gpt-4o-mini
|
|
provider_id: openai
|
|
provider_model_id: openai/gpt-4o-mini
|
|
model_type: llm
|
|
shields: []
|
|
vector_dbs: []
|
|
datasets: []
|
|
scoring_fns: []
|
|
benchmarks: []
|
|
tool_groups:
|
|
- toolgroup_id: builtin::websearch
|
|
provider_id: tavily-search
|
|
- toolgroup_id: builtin::rag
|
|
provider_id: rag-runtime
|
|
- toolgroup_id: builtin::code_interpreter
|
|
provider_id: code-interpreter
|
|
- toolgroup_id: builtin::wolfram_alpha
|
|
provider_id: wolfram-alpha
|
|
server:
|
|
port: 8321
|