Merge d211f5188b into 1d3f27fe5b

llama_stack/templates/starter/build.yaml

@@ -39,8 +39,11 @@ distribution_spec:
     - remote::tavily-search
     - inline::rag-runtime
     - remote::model-context-protocol
+  internal_providers:
+    sql_store:
+      - inline::sqlite-store
+      - inline::postgres-store
+    kv_store:
+      - inline::sqlite-kv
+      - inline::postgres-kv
 image_type: conda
-additional_pip_packages:
-- aiosqlite
-- asyncpg
-- sqlalchemy[asyncio]

llama_stack/templates/starter/run.yaml

@@ -97,6 +97,24 @@ providers:
     provider_type: inline::llama-guard
     config:
       excluded_categories: []
+  sql_store:
+  - provider_id: sqlite-store
+    provider_type: inline::sqlite-store
+    config:
+      db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/starter}/sqlite_store.db
+  - provider_id: postgres-store
+    provider_type: inline::postgres-store
+    config:
+      db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/starter}/postgres_store.db
+  kv_store:
+  - provider_id: sqlite-kv
+    provider_type: inline::sqlite-kv
+    config:
+      db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/starter}/sqlite_kv.db
+  - provider_id: postgres-kv
+    provider_type: inline::postgres-kv
+    config:
+      db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/starter}/postgres_kv.db
   agents:
   - provider_id: meta-reference
     provider_type: inline::meta-reference
@@ -106,8 +124,7 @@ providers:
         namespace: null
         db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/starter}/agents_store.db
       responses_store:
-        type: sqlite
+        store_id: default_store
-        db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/starter}/responses_store.db
   telemetry:
   - provider_id: meta-reference
     provider_type: inline::meta-reference
@@ -167,11 +184,23 @@ providers:
     provider_type: remote::model-context-protocol
     config: {}
 metadata_store:
-  type: sqlite
+  store_id: default_kv_store
-  db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/starter}/registry.db
 inference_store:
-  type: sqlite
+  store_id: default_store
-  db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/starter}/inference_store.db
+sql_stores:
+  - store_id: default_store
+    provider_id: postgres-store
+    config:
+      host: ${env.POSTGRES_HOST:localhost}
+      port: ${env.POSTGRES_PORT:5432}
+      db: ${env.POSTGRES_DB:llamastack}
+      user: ${env.POSTGRES_USER:llamastack}
+      password: ${env.POSTGRES_PASSWORD:llamastack}
+kv_stores:
+  - store_id: default_kv_store
+    provider_id: sqlite-kv
+    config:
+      db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/starter}/sqlite_kv.db
 models:
 - metadata: {}
   model_id: openai/gpt-4o