add sentence-transformers provider

This commit is contained in:
Vladislav 2025-02-20 14:45:50 +01:00
parent ecdba4b030
commit 7bb7597c00
4 changed files with 10 additions and 12 deletions

View file

@ -143,7 +143,6 @@
"autoevals", "autoevals",
"blobfile", "blobfile",
"chardet", "chardet",
"chromadb-client",
"datasets", "datasets",
"faiss-cpu", "faiss-cpu",
"fastapi", "fastapi",

View file

@ -23,7 +23,7 @@ The `llamastack/distribution-groq` distribution consists of the following provid
| scoring | `inline::basic`, `inline::llm-as-judge`, `inline::braintrust` | | scoring | `inline::basic`, `inline::llm-as-judge`, `inline::braintrust` |
| telemetry | `inline::meta-reference` | | telemetry | `inline::meta-reference` |
| tool_runtime | `remote::brave-search`, `remote::tavily-search`, `inline::code-interpreter`, `inline::rag-runtime` | | tool_runtime | `remote::brave-search`, `remote::tavily-search`, `inline::code-interpreter`, `inline::rag-runtime` |
| vector_io | `inline::faiss`, `remote::chromadb`, `remote::pgvector` | | vector_io | `inline::faiss` |
### Environment Variables ### Environment Variables

View file

@ -6,8 +6,6 @@ distribution_spec:
- remote::groq - remote::groq
vector_io: vector_io:
- inline::faiss - inline::faiss
- remote::chromadb
- remote::pgvector
safety: safety:
- inline::llama-guard - inline::llama-guard
agents: agents:

View file

@ -17,6 +17,9 @@ providers:
config: config:
url: https://api.groq.com url: https://api.groq.com
api_key: ${env.GROQ_API_KEY} api_key: ${env.GROQ_API_KEY}
- provider_id: sentence-transformers
provider_type: inline::sentence-transformers
config: {}
vector_io: vector_io:
- provider_id: faiss - provider_id: faiss
provider_type: inline::faiss provider_type: inline::faiss
@ -25,12 +28,6 @@ providers:
type: sqlite type: sqlite
namespace: null namespace: null
db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/groq}/faiss_store.db db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/groq}/faiss_store.db
- provider_id: chromadb
provider_type: remote::chromadb
config: {}
- provider_id: pgvector
provider_type: remote::pgvector
config: {}
safety: safety:
- provider_id: llama-guard - provider_id: llama-guard
provider_type: inline::llama-guard provider_type: inline::llama-guard
@ -118,8 +115,12 @@ models:
provider_id: groq provider_id: groq
provider_model_id: llama-3.2-3b-preview provider_model_id: llama-3.2-3b-preview
model_type: llm model_type: llm
shields: - metadata:
- shield_id: meta-llama/Llama-Guard-3-8B embedding_dimension: 384
model_id: all-MiniLM-L6-v2
provider_id: sentence-transformers
model_type: embedding
shields: []
vector_dbs: [] vector_dbs: []
datasets: [] datasets: []
scoring_fns: [] scoring_fns: []