fix: Make it more robust to parse postgresql version string (#1415)

lablup · Jul 25, 2023 · be16015 · be16015
1 parent 78586d6
commit be16015
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/changes/1415.fix.md b/changes/1415.fix.md
@@ -0,0 +1 @@
+Make the parsing routine of PostgreSQL version strings more robust with additional build tags
diff --git a/src/ai/backend/manager/models/utils.py b/src/ai/backend/manager/models/utils.py
@@ -183,7 +183,8 @@ async def connect_database(
     version_check_db = create_async_engine(url)
     async with version_check_db.begin() as conn:
         result = await conn.execute(sa.text("show server_version"))
-        major, minor, *_ = map(int, result.scalar().split("."))
+        version_str = result.scalar()
+        major, minor, *_ = map(int, version_str.partition(" ")[0].split("."))
         if (major, minor) < (11, 0):
             pgsql_connect_opts["server_settings"].pop("jit")
     await version_check_db.dispose()