phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-31 04:33:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Initial commit for postgresql dataset provider

Browse source 
Core implementation of postgresql dataset for llama stack

Signed-off-by: Josh Salomon <jsalomon@redhat.com>
This commit is contained in:
Josh Salomon 2025-03-26 12:15:40 +02:00
parent bdfe7fee92
commit 8e3b579df2
8 changed files with 512 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_stack/providers/registry/datasetio.py
View file

@ -36,4 +36,18 @@ def available_providers() -> List[ProviderSpec]:
config_class="llama_stack.providers.remote.datasetio.huggingface.HuggingfaceDatasetIOConfig",
),
),
remote_provider_spec(
api=Api.datasetio,
adapter=AdapterSpec(
adapter_type="postgresql",
pip_packages=[
# "asyncpg",
# "datasets",
"psycopg",
"psycopg[pool]",
],
module="llama_stack.providers.remote.datasetio.postgresql",
config_class="llama_stack.providers.remote.datasetio.postgresql.PostgreSQLDatasetIOConfig",
),
),
]