mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-03 18:00:36 +00:00
fix: configure batches provider to use kv_postgres backend
The batches provider was still referencing kv_default backend which doesn't exist in the postgres configuration. This would cause runtime failures when using the batches API with postgres store enabled. Now properly configured to use kv_postgres backend with batches namespace.
This commit is contained in:
parent
3b34efb81a
commit
2efaecd887
2 changed files with 13 additions and 1 deletions
|
|
@ -231,7 +231,7 @@ providers:
|
||||||
config:
|
config:
|
||||||
kvstore:
|
kvstore:
|
||||||
namespace: batches
|
namespace: batches
|
||||||
backend: kv_default
|
backend: kv_postgres
|
||||||
storage:
|
storage:
|
||||||
backends:
|
backends:
|
||||||
kv_postgres:
|
kv_postgres:
|
||||||
|
|
|
||||||
|
|
@ -282,6 +282,18 @@ def get_distribution_template(name: str = "starter") -> DistributionTemplate:
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
],
|
],
|
||||||
|
"batches": [
|
||||||
|
Provider(
|
||||||
|
provider_id="reference",
|
||||||
|
provider_type="inline::reference",
|
||||||
|
config=dict(
|
||||||
|
kvstore=KVStoreReference(
|
||||||
|
backend="kv_postgres",
|
||||||
|
namespace="batches",
|
||||||
|
).model_dump(exclude_none=True),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
],
|
||||||
},
|
},
|
||||||
storage_backends={
|
storage_backends={
|
||||||
"kv_postgres": PostgresKVStoreConfig.sample_run_config(),
|
"kv_postgres": PostgresKVStoreConfig.sample_run_config(),
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue