diff --git a/ee/scripts/helm/db/init_dbs/postgresql/1.9.9/1.9.9.sql b/ee/scripts/helm/db/init_dbs/postgresql/1.9.9/1.9.9.sql
index 6cb678c96..75130e2f4 100644
--- a/ee/scripts/helm/db/init_dbs/postgresql/1.9.9/1.9.9.sql
+++ b/ee/scripts/helm/db/init_dbs/postgresql/1.9.9/1.9.9.sql
@@ -1,4 +1,9 @@
 BEGIN;
+CREATE OR REPLACE FUNCTION openreplay_version()
+    RETURNS text AS
+$$
+SELECT 'v1.9.9-ee'
+$$ LANGUAGE sql IMMUTABLE;
 
 CREATE INDEX IF NOT EXISTS user_favorite_sessions_user_id_session_id_idx ON user_favorite_sessions (user_id, session_id);
 
diff --git a/ee/scripts/helm/db/init_dbs/postgresql/init_schema.sql b/ee/scripts/helm/db/init_dbs/postgresql/init_schema.sql
index 3138e8d19..451c6b2a0 100644
--- a/ee/scripts/helm/db/init_dbs/postgresql/init_schema.sql
+++ b/ee/scripts/helm/db/init_dbs/postgresql/init_schema.sql
@@ -3,6 +3,12 @@ BEGIN;
 CREATE SCHEMA IF NOT EXISTS events_common;
 CREATE SCHEMA IF NOT EXISTS events;
 
+CREATE OR REPLACE FUNCTION openreplay_version()
+    RETURNS text AS
+$$
+SELECT 'v1.9.9-ee'
+$$ LANGUAGE sql IMMUTABLE;
+
 -- --- accounts.sql ---
 
 CREATE OR REPLACE FUNCTION generate_api_key(length integer) RETURNS text AS
diff --git a/scripts/helm/db/init_dbs/postgresql/1.9.9/1.9.9.sql b/scripts/helm/db/init_dbs/postgresql/1.9.9/1.9.9.sql
index 3aafca980..712990680 100644
--- a/scripts/helm/db/init_dbs/postgresql/1.9.9/1.9.9.sql
+++ b/scripts/helm/db/init_dbs/postgresql/1.9.9/1.9.9.sql
@@ -1,4 +1,9 @@
 BEGIN;
+CREATE OR REPLACE FUNCTION openreplay_version()
+    RETURNS text AS
+$$
+SELECT 'v1.9.9'
+$$ LANGUAGE sql IMMUTABLE;
 
 CREATE INDEX IF NOT EXISTS user_favorite_sessions_user_id_session_id_idx ON user_favorite_sessions (user_id, session_id);
 
diff --git a/scripts/helm/db/init_dbs/postgresql/init_schema.sql b/scripts/helm/db/init_dbs/postgresql/init_schema.sql
index 8f74e721e..1432af17c 100644
--- a/scripts/helm/db/init_dbs/postgresql/init_schema.sql
+++ b/scripts/helm/db/init_dbs/postgresql/init_schema.sql
@@ -3,6 +3,12 @@ BEGIN;
 CREATE SCHEMA IF NOT EXISTS events_common;
 CREATE SCHEMA IF NOT EXISTS events;
 
+CREATE OR REPLACE FUNCTION openreplay_version()
+    RETURNS text AS
+$$
+SELECT 'v1.9.9'
+$$ LANGUAGE sql IMMUTABLE;
+
 -- --- accounts.sql ---
 
 CREATE OR REPLACE FUNCTION generate_api_key(length integer) RETURNS text AS