orchestrator-core 4.5.0a8__tar.gz → 4.5.0a9__tar.gz

{orchestrator_core-4.5.0a8 → orchestrator_core-4.5.0a9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orchestrator-core
-Version: 4.5.0a8
+Version: 4.5.0a9
 Summary: This is the orchestrator workflow engine.
 Author-email: SURF <automation-beheer@surf.nl>
 Requires-Python: >=3.11,<3.14

{orchestrator_core-4.5.0a8 → orchestrator_core-4.5.0a9}/orchestrator/__init__.py

@@ -13,7 +13,7 @@
 
 """This is the orchestrator workflow engine."""
 
-__version__ = "4.5.0a8"
+__version__ = "4.5.0a9"
 
 
 from structlog import get_logger

{orchestrator_core-4.5.0a8 → orchestrator_core-4.5.0a9}/orchestrator/llm_settings.py

@@ -43,6 +43,9 @@ class LLMSettings(BaseSettings):
     LLM_MAX_RETRIES: int = 3
     LLM_TIMEOUT: int = 30
 
+    # Toggle creation of extensions
+    LLM_FORCE_EXTENTION_MIGRATION: bool = False
+
     @field_validator("EMBEDDING_MODEL")
     def validate_embedding_model_format(cls, v: str) -> str:
         """Validate that embedding model is in 'vendor/model' format."""

{orchestrator_core-4.5.0a8 → orchestrator_core-4.5.0a9}/orchestrator/search/llm_migration.py

@@ -17,6 +17,7 @@ from sqlalchemy import text
 from sqlalchemy.engine import Connection
 from structlog import get_logger
 
+from orchestrator.llm_settings import llm_settings
 from orchestrator.search.core.types import FieldType
 
 logger = get_logger(__name__)
@@ -30,11 +31,16 @@ def run_migration(connection: Connection) -> None:
     logger.info("Running LLM migration")
 
     try:
-        # Create PostgreSQL extensions
-        connection.execute(text("CREATE EXTENSION IF NOT EXISTS ltree;"))
-        connection.execute(text("CREATE EXTENSION IF NOT EXISTS unaccent;"))
-        connection.execute(text("CREATE EXTENSION IF NOT EXISTS pg_trgm;"))
-        connection.execute(text("CREATE EXTENSION IF NOT EXISTS vector;"))
+        # Test to see if the extenstion exists and then skip the migration; Needed for certain situations where db user
+        # has insufficient priviledges to run the `CREATE EXTENSION ...` command.
+        res = connection.execute(text("SELECT * FROM pg_extension where extname = 'vector';"))
+        if llm_settings.LLM_FORCE_EXTENTION_MIGRATION or res.rowcount == 0:
+            # Create PostgreSQL extensions
+            logger.info("Attempting to run the extention creation;")
+            connection.execute(text("CREATE EXTENSION IF NOT EXISTS ltree;"))
+            connection.execute(text("CREATE EXTENSION IF NOT EXISTS unaccent;"))
+            connection.execute(text("CREATE EXTENSION IF NOT EXISTS pg_trgm;"))
+            connection.execute(text("CREATE EXTENSION IF NOT EXISTS vector;"))
 
         # Create field_type enum
         field_type_values = "', '".join([ft.value for ft in FieldType])