mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-03 09:53:45 +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:
|
||||
kvstore:
|
||||
namespace: batches
|
||||
backend: kv_default
|
||||
backend: kv_postgres
|
||||
storage:
|
||||
backends:
|
||||
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={
|
||||
"kv_postgres": PostgresKVStoreConfig.sample_run_config(),
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue