Fixing migration to handle existing column

openchatai · Dec 12, 2023 · 6c7430d · 6c7430d
1 parent c765339
commit 6c7430d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 14 deletions.
diff --git a/llm-server/models/alembic.ini b/llm-server/models/alembic.ini
@@ -2,7 +2,7 @@
 script_location = migrations
 prepend_sys_path = .
 version_path_separator = os  # Use os.pathsep. Default configuration used for new projects.
-sqlalchemy.url = mysql+pymysql://dbuser:dbpass@localhost:3307/opencopilot
+sqlalchemy.url = mysql+pymysql://dbuser:dbpass@mysql:3306/opencopilot
 
 [post_write_hooks]
 

diff --git a/llm-server/models/migrations/versions/d845330c4432_add_system_summary_prompt_to_chatbot.py b/llm-server/models/migrations/versions/d845330c4432_add_system_summary_prompt_to_chatbot.py
@@ -23,19 +23,18 @@ def upgrade():
         .execute(sa.text("SHOW COLUMNS FROM chatbots LIKE 'summary_prompt'"))
         .fetchone()
     ):
-        op.add_column(
-            "chatbots",
-            sa.Column(
-                "summary_prompt",
-                sa.Text(),
-                nullable=False,
-                server_default=(
-                    "Given a JSON response, summarize the key information in a concise manner. "
-                    "Include relevant details, references, and links if present. "
-                    "Format the summary in Markdown for clarity and readability."
-                ),
-            ),
-        )
+        op.add_column("chatbots", sa.Column("summary_prompt", sa.Text(), nullable=True))
+
+    op.execute(
+        """
+        UPDATE chatbots
+        SET summary_prompt = "Given a JSON response, summarize the key information in a concise manner. Include relevant details, references, and links if present. Format the summary in Markdown for clarity and readability."
+        """
+    )
+
+    op.alter_column(
+        "chatbots", "summary_prompt", existing_type=sa.TEXT(), nullable=False
+    )
 
 
 def downgrade():