From a686beb4c6a3e37ff821b4083dd7535656e923e8 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Mon, 10 Nov 2025 14:45:45 -0800 Subject: [PATCH] update distro --- .../ci-tests/run-with-postgres-store.yaml | 2 +- .../starter-gpu/run-with-postgres-store.yaml | 2 +- .../starter/run-with-postgres-store.yaml | 2 +- src/llama_stack/distributions/starter/starter.py | 13 ++++++++++++- 4 files changed, 15 insertions(+), 4 deletions(-) diff --git a/src/llama_stack/distributions/ci-tests/run-with-postgres-store.yaml b/src/llama_stack/distributions/ci-tests/run-with-postgres-store.yaml index 839612257..c149fd696 100644 --- a/src/llama_stack/distributions/ci-tests/run-with-postgres-store.yaml +++ b/src/llama_stack/distributions/ci-tests/run-with-postgres-store.yaml @@ -153,7 +153,7 @@ providers: storage_dir: ${env.FILES_STORAGE_DIR:=~/.llama/distributions/ci-tests/files} metadata_store: table_name: files_metadata - backend: sql_default + backend: sql_postgres safety: - provider_id: llama-guard provider_type: inline::llama-guard diff --git a/src/llama_stack/distributions/starter-gpu/run-with-postgres-store.yaml b/src/llama_stack/distributions/starter-gpu/run-with-postgres-store.yaml index 1920ebd9d..07a60b69c 100644 --- a/src/llama_stack/distributions/starter-gpu/run-with-postgres-store.yaml +++ b/src/llama_stack/distributions/starter-gpu/run-with-postgres-store.yaml @@ -153,7 +153,7 @@ providers: storage_dir: ${env.FILES_STORAGE_DIR:=~/.llama/distributions/starter-gpu/files} metadata_store: table_name: files_metadata - backend: sql_default + backend: sql_postgres safety: - provider_id: llama-guard provider_type: inline::llama-guard diff --git a/src/llama_stack/distributions/starter/run-with-postgres-store.yaml b/src/llama_stack/distributions/starter/run-with-postgres-store.yaml index 702f95381..fdbe3ef8e 100644 --- a/src/llama_stack/distributions/starter/run-with-postgres-store.yaml +++ b/src/llama_stack/distributions/starter/run-with-postgres-store.yaml @@ -153,7 +153,7 @@ providers: storage_dir: ${env.FILES_STORAGE_DIR:=~/.llama/distributions/starter/files} metadata_store: table_name: files_metadata - backend: sql_default + backend: sql_postgres safety: - provider_id: llama-guard provider_type: inline::llama-guard diff --git a/src/llama_stack/distributions/starter/starter.py b/src/llama_stack/distributions/starter/starter.py index 88cd3a4fe..0e8993f76 100644 --- a/src/llama_stack/distributions/starter/starter.py +++ b/src/llama_stack/distributions/starter/starter.py @@ -154,10 +154,20 @@ def get_distribution_template(name: str = "starter") -> DistributionTemplate: BuildProvider(provider_type="inline::reference"), ], } + files_config = LocalfsFilesImplConfig.sample_run_config(f"~/.llama/distributions/{name}") files_provider = Provider( provider_id="meta-reference-files", 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( provider_id="sentence-transformers", @@ -272,6 +282,7 @@ def get_distribution_template(name: str = "starter") -> DistributionTemplate: "run-with-postgres-store.yaml": RunConfigSettings( provider_overrides={ **default_overrides, + "files": [postgres_files_provider], "agents": [ Provider( provider_id="meta-reference",