feat: add mongodb provider

This commit is contained in:
Young Han 2025-10-10 10:42:42 -07:00
parent d8c82c10c2
commit 885631d071
8 changed files with 61 additions and 5 deletions

View file

@ -26,6 +26,7 @@ distribution_spec:
- provider_type: inline::milvus
- provider_type: remote::chromadb
- provider_type: remote::pgvector
- provider_type: remote::mongodb
files:
- provider_type: inline::localfs
safety:

View file

@ -129,6 +129,19 @@ providers:
kvstore:
type: sqlite
db_path: ${env.SQLITE_STORE_DIR:=~/.llama/distributions/starter-gpu}/pgvector_registry.db
- provider_id: ${env.MONGODB_CONNECTION_STRING:+mongodb_atlas}
provider_type: remote::mongodb
config:
connection_string: ${env.MONGODB_CONNECTION_STRING}
database_name: llama_stack
index_name: vector_index
path_field: embedding
similarity_metric: cosine
max_pool_size: 100
timeout_ms: 30000
kvstore:
type: sqlite
db_path: ${env.SQLITE_STORE_DIR:=~/.llama/distributions/starter-gpu}/mongodb_registry.db
files:
- provider_id: meta-reference-files
provider_type: inline::localfs