diff --git a/llama_stack/providers/registry/datasetio.py b/llama_stack/providers/registry/datasetio.py index d9f23a4ef..fee96d81c 100644 --- a/llama_stack/providers/registry/datasetio.py +++ b/llama_stack/providers/registry/datasetio.py @@ -41,10 +41,8 @@ def available_providers() -> List[ProviderSpec]: adapter=AdapterSpec( adapter_type="postgresql", pip_packages=[ -# "asyncpg", -# "datasets", - "psycopg", - "psycopg[pool]", + "psycopg", + "psycopg_pool", ], module="llama_stack.providers.remote.datasetio.postgresql", config_class="llama_stack.providers.remote.datasetio.postgresql.PostgreSQLDatasetIOConfig",