phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 09:21:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add remote::postgresql dataset provider to the registry

Browse source 
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
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/registry/datasetio.py
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", "psycopg",
# "datasets", "psycopg_pool",
"psycopg",
"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",