提示词专家

2025-02-23 09:07:52 +08:00
parent cd2e508241
commit 25739829e4
4527 changed files with 760147 additions and 0 deletions
--- a/test_db.py
+++ b/test_db.py
@@ -0,0 +1,73 @@
+from flask_prompt_master import create_app, db
+import pymysql
+from flask_prompt_master.init_db import templates
+
+def insert_all_templates():
+    """向 container_orchestration_assistant 表插入所有模板数据"""
+    try:
+        # 连接MySQL数据库
+        conn = pymysql.connect(
+            host='localhost',
+            user='root',
+            password='123456',
+            database='food_db',
+            charset='utf8mb4'
+        )
+        cursor = conn.cursor()
+        
+        # 先清空表
+        cursor.execute("TRUNCATE TABLE container_orchestration_assistant")
+        
+        # SQL 插入语句
+        sql = """
+        INSERT INTO container_orchestration_assistant 
+        (name, description, category, industry, profession, sub_category, system_prompt) 
+        VALUES (%(name)s, %(description)s, %(category)s, %(industry)s, %(profession)s, 
+                %(sub_category)s, %(system_prompt)s)
+        """
+        
+        # 遍历所有模板数据并插入
+        success_count = 0
+        for template in templates:
+            try:
+                # 准备模板数据
+                template_data = {
+                    'name': template['name'],
+                    'description': template['description'],
+                    'category': template.get('category', ''),
+                    'industry': template.get('industry', ''),
+                    'profession': template.get('profession', ''),
+                    'sub_category': template.get('sub_category', ''),
+                    'system_prompt': template['system_prompt']
+                }
+                
+                # 执行插入
+                cursor.execute(sql, template_data)
+                success_count += 1
+                
+                print(f"成功插入模板: {template['name']}")
+                
+            except Exception as e:
+                print(f"插入模板 {template['name']} 失败: {str(e)}")
+                continue
+        
+        # 提交事务
+        conn.commit()
+        
+        print("\n=== 数据插入完成 ===")
+        print(f"成功插入 {success_count} 个模板")
+        print(f"总计 {len(templates)} 个模板")
+        print("===================")
+            
+    except Exception as e:
+        print(f"数据库连接失败: {str(e)}")
+        if 'conn' in locals():
+            conn.rollback()
+    finally:
+        if 'cursor' in locals():
+            cursor.close()
+        if 'conn' in locals():
+            conn.close()
+
+if __name__ == '__main__':
+    insert_all_templates()