chore: move configuration to alembic package

timescale · Jan 7, 2025 · 7b90575 · 7b90575
1 parent 1c43792
commit 7b90575
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 5 deletions.
diff --git a/docs/python-integration.md b/docs/python-integration.md
@@ -197,7 +197,7 @@ Then you can use the `create_vectorizer` operation to create a vectorizer for yo
 
 ```python
 from alembic import op
-from pgai.vectorizer.configuration import (
+from pgai.alembic.configuration import (
     OpenAIConfig,
     CharacterTextSplitterConfig,
     PythonTemplateConfig

diff --git a/projects/extension/Dockerfile b/projects/extension/Dockerfile
@@ -52,6 +52,10 @@ WORKDIR /pgai
 COPY . .
 RUN just build install
 
+RUN mkdir -p /docker-entrypoint-initdb.d && \
+    echo "#!/bin/bash" > /docker-entrypoint-initdb.d/configure-timescaledb.sh && \
+    echo "echo \"shared_preload_libraries = 'timescaledb'\" >> \${PGDATA}/postgresql.conf" >> /docker-entrypoint-initdb.d/configure-timescaledb.sh && \
+    chmod +x /docker-entrypoint-initdb.d/configure-timescaledb.sh
 
 ###############################################################################
 # image for use in extension development

diff --git a/...cts/pgai/pgai/vectorizer/configuration.py → projects/pgai/pgai/alembic/configuration.py b/...cts/pgai/pgai/vectorizer/configuration.py → projects/pgai/pgai/alembic/configuration.py
diff --git a/projects/pgai/pgai/alembic/operations.py b/projects/pgai/pgai/alembic/operations.py
@@ -4,7 +4,7 @@
 from sqlalchemy import text
 from typing_extensions import override
 
-from pgai.vectorizer.configuration import (
+from pgai.alembic.configuration import (
     CharacterTextSplitterConfig,
     ChunkValueConfig,
     CreateVectorizerParams,

diff --git a/projects/pgai/tests/vectorizer/extensions/fixtures/migrations/generic_vectorizer.py.template b/projects/pgai/tests/vectorizer/extensions/fixtures/migrations/generic_vectorizer.py.template
@@ -7,7 +7,7 @@ Create Date: {create_date}
 from alembic import op
 from sqlalchemy import Column, Integer, String, Text
 from sqlalchemy import text
-from pgai.vectorizer.configuration import (
+from pgai.alembic.configuration import (
     OpenAIConfig,
     OllamaConfig,
     VoyageAIConfig,

diff --git a/projects/pgai/tests/vectorizer/extensions/test_alembic.py b/projects/pgai/tests/vectorizer/extensions/test_alembic.py
@@ -7,7 +7,7 @@
 
 If you add new configuration options, add a test for them here.
 Either in an existing test case or a new one.
-The config classes for migrations lie in pgai/vectorizer/configuration.py
+The config classes for migrations lie in pgai/alembic/configuration.py
 If the migration and the validation definitions overlap,
 it is possible to define a parent class in base.py
 see e.g. CharacterTextSplitterConfig as an example
@@ -22,7 +22,7 @@
 from alembic.config import Config
 from sqlalchemy import Engine, text
 
-from pgai.vectorizer.configuration import (
+from pgai.alembic.configuration import (
     CharacterTextSplitterConfig,
     HNSWIndexingConfig,
     OllamaConfig,