feat: add pgvector vectio_io provider

Author: Ygnas

distribution/README.md

@@ -38,3 +38,4 @@ You can see an overview of the APIs and Providers the image ships with in the ta
 | vector_io | inline::faiss | No | ❌ | Set the `ENABLE_FAISS` environment variable |
 | vector_io | inline::milvus | No | ✅ | N/A |
 | vector_io | remote::milvus | No | ❌ | Set the `MILVUS_ENDPOINT` environment variable |
+| vector_io | remote::pgvector | No | ❌ | Set the `PGVECTOR_DB` environment variable |

distribution/build.yaml

@@ -14,6 +14,7 @@ distribution_spec:
     - provider_type: inline::milvus
     - provider_type: remote::milvus
     - provider_type: inline::faiss
+    - provider_type: remote::pgvector
     safety:
     - provider_type: remote::trustyai_fms
       module: llama_stack_provider_trustyai_fms==0.3.1

distribution/run.yaml

@@ -93,6 +93,17 @@ providers:
       persistence:
         backend: kv_milvus_remote
         namespace: vector_io::milvus_remote
+  - provider_id: ${env.PGVECTOR_DB:+pgvector}
+    provider_type: remote::pgvector
+    config:
+      host: ${env.PGVECTOR_HOST:=localhost}
+      port: ${env.PGVECTOR_PORT:=5432}
+      db: ${env.PGVECTOR_DB:=}
+      user: ${env.PGVECTOR_USER:=}
+      password: ${env.PGVECTOR_PASSWORD:=}
+      persistence:
+        backend: kv_pgvector
+        namespace: vector_io::pgvector
   safety:
     - provider_id: trustyai_fms
       provider_type: remote::trustyai_fms
@@ -224,6 +235,9 @@ storage:
     kv_milvus_remote:
       type: kv_sqlite
       db_path: /opt/app-root/src/.llama/distributions/rh/milvus_remote_registry.db
+    kv_pgvector:
+      type: kv_sqlite
+      db_path: /opt/app-root/src/.llama/distributions/rh/pgvector_registry.db
     kv_datasetio_huggingface:
       type: kv_sqlite
       db_path: /opt/app-root/src/.llama/distributions/rh/huggingface_datasetio.db