update distro

This commit is contained in:
Ashwin Bharambe 2025-11-10 14:45:45 -08:00
parent 93b9c65a89
commit a686beb4c6
4 changed files with 15 additions and 4 deletions

View file

@ -153,7 +153,7 @@ providers:
storage_dir: ${env.FILES_STORAGE_DIR:=~/.llama/distributions/ci-tests/files} storage_dir: ${env.FILES_STORAGE_DIR:=~/.llama/distributions/ci-tests/files}
metadata_store: metadata_store:
table_name: files_metadata table_name: files_metadata
backend: sql_default backend: sql_postgres
safety: safety:
- provider_id: llama-guard - provider_id: llama-guard
provider_type: inline::llama-guard provider_type: inline::llama-guard

View file

@ -153,7 +153,7 @@ providers:
storage_dir: ${env.FILES_STORAGE_DIR:=~/.llama/distributions/starter-gpu/files} storage_dir: ${env.FILES_STORAGE_DIR:=~/.llama/distributions/starter-gpu/files}
metadata_store: metadata_store:
table_name: files_metadata table_name: files_metadata
backend: sql_default backend: sql_postgres
safety: safety:
- provider_id: llama-guard - provider_id: llama-guard
provider_type: inline::llama-guard provider_type: inline::llama-guard

View file

@ -153,7 +153,7 @@ providers:
storage_dir: ${env.FILES_STORAGE_DIR:=~/.llama/distributions/starter/files} storage_dir: ${env.FILES_STORAGE_DIR:=~/.llama/distributions/starter/files}
metadata_store: metadata_store:
table_name: files_metadata table_name: files_metadata
backend: sql_default backend: sql_postgres
safety: safety:
- provider_id: llama-guard - provider_id: llama-guard
provider_type: inline::llama-guard provider_type: inline::llama-guard

View file

@ -154,10 +154,20 @@ def get_distribution_template(name: str = "starter") -> DistributionTemplate:
BuildProvider(provider_type="inline::reference"), BuildProvider(provider_type="inline::reference"),
], ],
} }
files_config = LocalfsFilesImplConfig.sample_run_config(f"~/.llama/distributions/{name}")
files_provider = Provider( files_provider = Provider(
provider_id="meta-reference-files", provider_id="meta-reference-files",
provider_type="inline::localfs", provider_type="inline::localfs",
config=LocalfsFilesImplConfig.sample_run_config(f"~/.llama/distributions/{name}"), config=files_config,
)
postgres_files_config = dict(files_config)
postgres_metadata_store = dict(postgres_files_config.get("metadata_store", {}))
postgres_metadata_store["backend"] = "sql_postgres"
postgres_files_config["metadata_store"] = postgres_metadata_store
postgres_files_provider = Provider(
provider_id="meta-reference-files",
provider_type="inline::localfs",
config=postgres_files_config,
) )
embedding_provider = Provider( embedding_provider = Provider(
provider_id="sentence-transformers", provider_id="sentence-transformers",
@ -272,6 +282,7 @@ def get_distribution_template(name: str = "starter") -> DistributionTemplate:
"run-with-postgres-store.yaml": RunConfigSettings( "run-with-postgres-store.yaml": RunConfigSettings(
provider_overrides={ provider_overrides={
**default_overrides, **default_overrides,
"files": [postgres_files_provider],
"agents": [ "agents": [
Provider( Provider(
provider_id="meta-reference", provider_id="meta-reference",