run-with-safety memory

llama_stack/templates/hf-serverless/hf_serverless.py

@@ -70,7 +70,8 @@ def get_distribution_template() -> DistributionTemplate:
                                 repo="${env.SAFETY_MODEL}",
                             ),
                         ),
-                    ]
+                    ],
+                    "memory": [memory_provider],
                 },
                 default_models=[
                     inference_model,

llama_stack/templates/hf-serverless/run-with-safety.yaml

@@ -31,12 +31,6 @@ providers:
         type: sqlite
         namespace: null
         db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/hf-serverless}/faiss_store.db
-  - provider_id: chromadb
-    provider_type: remote::chromadb
-    config: {}
-  - provider_id: pgvector
-    provider_type: remote::pgvector
-    config: {}
   safety:
   - provider_id: llama-guard
     provider_type: inline::llama-guard