75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# This migration script will run after the module update
|
||
|
|
# It will migrate data from openai_prompt_template to openwebui_prompt_template
|
||
|
|
|
||
|
|
import logging
|
||
|
|
_logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
def migrate(cr, version):
|
||
|
|
"""
|
||
|
|
Migrate data from openai_prompt_template to openwebui_prompt_template
|
||
|
|
"""
|
||
|
|
if not version:
|
||
|
|
return
|
||
|
|
|
||
|
|
_logger.info("Starting migration of prompt templates from openai to openwebui")
|
||
|
|
|
||
|
|
# Check if the old table exists
|
||
|
|
cr.execute("SELECT EXISTS (SELECT FROM information_schema.tables WHERE table_name = 'openai_prompt_template')")
|
||
|
|
if not cr.fetchone()[0]:
|
||
|
|
_logger.info("No openai_prompt_template table found, skipping migration")
|
||
|
|
return
|
||
|
|
|
||
|
|
# Check if there's data to migrate
|
||
|
|
cr.execute("SELECT COUNT(*) FROM openai_prompt_template")
|
||
|
|
count = cr.fetchone()[0]
|
||
|
|
_logger.info(f"Found {count} records to migrate from openai_prompt_template")
|
||
|
|
|
||
|
|
if count > 0:
|
||
|
|
# Copy data from old table to new table
|
||
|
|
cr.execute("""
|
||
|
|
INSERT INTO openwebui_prompt_template (
|
||
|
|
id, name, content, is_default, module, create_uid, create_date, write_uid, write_date
|
||
|
|
)
|
||
|
|
SELECT
|
||
|
|
id, name, content, is_default, module, create_uid, create_date, write_uid, write_date
|
||
|
|
FROM
|
||
|
|
openai_prompt_template
|
||
|
|
ON CONFLICT (id) DO UPDATE SET
|
||
|
|
name = EXCLUDED.name,
|
||
|
|
content = EXCLUDED.content,
|
||
|
|
is_default = EXCLUDED.is_default,
|
||
|
|
module = EXCLUDED.module,
|
||
|
|
write_uid = EXCLUDED.write_uid,
|
||
|
|
write_date = EXCLUDED.write_date
|
||
|
|
""")
|
||
|
|
|
||
|
|
# Update sequence if needed
|
||
|
|
cr.execute("""
|
||
|
|
SELECT MAX(id) FROM openwebui_prompt_template
|
||
|
|
""")
|
||
|
|
max_id = cr.fetchone()[0] or 0
|
||
|
|
if max_id > 0:
|
||
|
|
cr.execute(f"""
|
||
|
|
SELECT setval('openwebui_prompt_template_id_seq', {max_id})
|
||
|
|
""")
|
||
|
|
|
||
|
|
# Log the migration
|
||
|
|
_logger.info(f"Successfully migrated {count} records from openai_prompt_template to openwebui_prompt_template")
|
||
|
|
|
||
|
|
# Update any references in res_config_settings
|
||
|
|
cr.execute("""
|
||
|
|
UPDATE ir_config_parameter
|
||
|
|
SET key = REPLACE(key, 'openai.', 'openwebui.')
|
||
|
|
WHERE key LIKE 'openai.%'
|
||
|
|
""")
|
||
|
|
|
||
|
|
# Update any references to prompt template IDs in system parameters
|
||
|
|
cr.execute("""
|
||
|
|
UPDATE ir_config_parameter
|
||
|
|
SET key = REPLACE(key, 'openai_prompt_template_id', 'openwebui_prompt_template_id')
|
||
|
|
WHERE key LIKE '%openai_prompt_template_id%'
|
||
|
|
""")
|
||
|
|
|
||
|
|
_logger.info("Updated system parameters to use openwebui prefix")
|