phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-27 18:50:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: expand set of known gemini models (#2471)
Some checks failed
Test Llama Stack Build / build-single-provider (push) Failing after 39s
Details
Test Llama Stack Build / build-ubi9-container-distribution (push) Failing after 37s
Details
Python Package Build Test / build (3.12) (push) Failing after 36s
Details
Test External Providers / test-external-providers (venv) (push) Failing after 45s
Details
Pre-commit / pre-commit (push) Successful in 1m57s
Details
Integration Tests / test-matrix (http, 3.11, agents) (push) Failing after 5s
Details
Integration Auth Tests / test-matrix (oauth2_token) (push) Failing after 8s
Details
Integration Tests / test-matrix (http, 3.11, providers) (push) Failing after 6s
Details
Integration Tests / test-matrix (http, 3.12, datasets) (push) Failing after 4s
Details
Integration Tests / test-matrix (http, 3.11, datasets) (push) Failing after 10s
Details
Integration Tests / test-matrix (http, 3.11, scoring) (push) Failing after 8s
Details
Integration Tests / test-matrix (http, 3.11, inspect) (push) Failing after 9s
Details
Integration Tests / test-matrix (http, 3.11, inference) (push) Failing after 11s
Details
Integration Tests / test-matrix (http, 3.12, inference) (push) Failing after 9s
Details
Integration Tests / test-matrix (http, 3.11, tool_runtime) (push) Failing after 12s
Details
Integration Tests / test-matrix (http, 3.12, post_training) (push) Failing after 7s
Details
Integration Tests / test-matrix (library, 3.11, agents) (push) Failing after 6s
Details
Integration Tests / test-matrix (library, 3.11, datasets) (push) Failing after 13s
Details
Integration Tests / test-matrix (http, 3.11, vector_io) (push) Failing after 8s
Details
Integration Tests / test-matrix (library, 3.11, inspect) (push) Failing after 6s
Details
Integration Tests / test-matrix (http, 3.12, agents) (push) Failing after 13s
Details
Integration Tests / test-matrix (http, 3.12, providers) (push) Failing after 11s
Details
Integration Tests / test-matrix (http, 3.12, tool_runtime) (push) Failing after 6s
Details
Integration Tests / test-matrix (http, 3.12, inspect) (push) Failing after 10s
Details
Integration Tests / test-matrix (library, 3.11, providers) (push) Failing after 11s
Details
Integration Tests / test-matrix (library, 3.12, agents) (push) Failing after 7s
Details
Integration Tests / test-matrix (library, 3.11, vector_io) (push) Failing after 9s
Details
Integration Tests / test-matrix (http, 3.12, vector_io) (push) Failing after 11s
Details
Integration Tests / test-matrix (library, 3.11, post_training) (push) Failing after 9s
Details
Integration Tests / test-matrix (http, 3.11, post_training) (push) Failing after 12s
Details
Integration Tests / test-matrix (library, 3.11, scoring) (push) Failing after 9s
Details
Integration Tests / test-matrix (library, 3.11, tool_runtime) (push) Failing after 6s
Details
Integration Tests / test-matrix (library, 3.11, inference) (push) Failing after 12s
Details
Integration Tests / test-matrix (library, 3.12, inference) (push) Failing after 7s
Details
Integration Tests / test-matrix (http, 3.12, scoring) (push) Failing after 11s
Details
Integration Tests / test-matrix (library, 3.12, datasets) (push) Failing after 8s
Details
Integration Tests / test-matrix (library, 3.12, inspect) (push) Failing after 8s
Details
Integration Tests / test-matrix (library, 3.12, post_training) (push) Failing after 8s
Details
Integration Tests / test-matrix (library, 3.12, tool_runtime) (push) Failing after 6s
Details
Integration Tests / test-matrix (library, 3.12, vector_io) (push) Failing after 6s
Details
Integration Tests / test-matrix (library, 3.12, scoring) (push) Failing after 9s
Details
Integration Tests / test-matrix (library, 3.12, providers) (push) Failing after 11s
Details
Test Llama Stack Build / generate-matrix (push) Successful in 9s
Details
Python Package Build Test / build (3.11) (push) Failing after 7s
Details
Python Package Build Test / build (3.13) (push) Failing after 6s
Details
Test Llama Stack Build / build-custom-container-distribution (push) Failing after 9s
Details
Unit Tests / unit-tests (3.11) (push) Failing after 5s
Details
Unit Tests / unit-tests (3.13) (push) Failing after 4s
Details
Unit Tests / unit-tests (3.12) (push) Failing after 6s
Details
Test Llama Stack Build / build (push) Failing after 3s
Details

Browse source 
feat: Add Gemini 2.0 and 2.5 models

This commit expands the set of known Gemini models by introducing:
- `gemini/gemini-2.0-flash`
- `gemini/gemini-2.5-flash`
- `gemini/gemini-2.5-pro`

These new models are added to `LLM_MODEL_IDS` for broader compatibility
and updated in `run.yaml` to allow for their immediate use in starter
configurations.

Signed-off-by: Eran Cohen <eranco@redhat.com>
This commit is contained in:
Eran Cohen 2025-06-19 19:19:37 +03:00 • committed by GitHub
parent f394c7f2d9
commit 747e594680
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/providers/remote/inference/gemini/models.py
View file

@ -12,6 +12,9 @@ from llama_stack.providers.utils.inference.model_registry import (
LLM_MODEL_IDS = [
"gemini/gemini-1.5-flash",
"gemini/gemini-1.5-pro",
"gemini/gemini-2.0-flash",
"gemini/gemini-2.5-flash",
"gemini/gemini-2.5-pro",
]

15
llama_stack/templates/starter/run.yaml
View file

@ -731,6 +731,21 @@ models:
provider_id: gemini
provider_model_id: gemini/gemini-1.5-pro
model_type: llm
- metadata: {}
model_id: gemini/gemini-2.0-flash
provider_id: gemini
provider_model_id: gemini/gemini-2.0-flash
model_type: llm
- metadata: {}
model_id: gemini/gemini-2.5-flash
provider_id: gemini
provider_model_id: gemini/gemini-2.5-flash
model_type: llm
- metadata: {}
model_id: gemini/gemini-2.5-pro
provider_id: gemini
provider_model_id: gemini/gemini-2.5-pro
model_type: llm
- metadata:
embedding_dimension: 768
context_length: 2048