Add remote::postgresql dataset provider to the registry

Signed-off-by: Josh Salomon <jsalomon@redhat.com>
This commit is contained in:
Josh Salomon 2025-04-09 21:33:22 +03:00
parent 73b63ea318
commit e244a692ed

View file

@ -41,10 +41,8 @@ def available_providers() -> List[ProviderSpec]:
adapter=AdapterSpec( adapter=AdapterSpec(
adapter_type="postgresql", adapter_type="postgresql",
pip_packages=[ pip_packages=[
# "asyncpg",
# "datasets",
"psycopg", "psycopg",
"psycopg[pool]", "psycopg_pool",
], ],
module="llama_stack.providers.remote.datasetio.postgresql", module="llama_stack.providers.remote.datasetio.postgresql",
config_class="llama_stack.providers.remote.datasetio.postgresql.PostgreSQLDatasetIOConfig", config_class="llama_stack.providers.remote.datasetio.postgresql.PostgreSQLDatasetIOConfig",