Update scripts/sync_bailian_models.py

scripts/sync_bailian_models.py

@@ -9,22 +9,44 @@ from datetime import datetime
 DB_PATH = "/app/server/data/freeapi.db"
 BAILIAN_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1/models"
 
-def get_table_columns(conn, table_name):
-    """返回表的列名列表"""
-    cursor = conn.execute(f"PRAGMA table_info({table_name})")
+def get_table_columns(conn, table):
+    cursor = conn.execute(f"PRAGMA table_info({table})")
     return [row[1] for row in cursor.fetchall()]
 
-def get_or_create_bailian_provider(conn):
-    """获取或创建百炼 provider"""
+def ensure_provider_table(conn):
     cursor = conn.cursor()
-    # 尝试查找已有的百炼 provider（根据 base_url 包含 dashscope）
-    cursor.execute("SELECT id FROM providers WHERE base_url LIKE '%dashscope.aliyuncs.com%' LIMIT 1")
+    cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name IN ('providers', 'channels')")
+    tables = [row[0] for row in cursor.fetchall()]
+    if 'providers' in tables:
+        return 'providers'
+    elif 'channels' in tables:
+        return 'channels'
+    else:
+        # 创建 providers 表
+        cursor.execute("""
+            CREATE TABLE providers (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                name TEXT,
+                base_url TEXT,
+                api_key TEXT,
+                created_at TEXT
+            )
+        """)
+        conn.commit()
+        return 'providers'
+
+def get_or_create_bailian_provider(conn, provider_table):
+    cursor = conn.cursor()
+    # 尝试查找已有百炼记录
+    cursor.execute(f"SELECT id FROM {provider_table} WHERE base_url LIKE '%dashscope.aliyuncs.com%' LIMIT 1")
     row = cursor.fetchone()
     if row:
         return row[0]
-    # 如果没有，创建一个新的
-    cursor.execute("INSERT INTO providers (name, base_url, api_key, created_at) VALUES (?, ?, ?, ?)",
-                   ("Bailian", BAILIAN_URL, "", datetime.utcnow().isoformat()))
+    # 插入新记录
+    cursor.execute(f"""
+        INSERT INTO {provider_table} (name, base_url, api_key, created_at)
+        VALUES (?, ?, ?, ?)
+    """, ("Bailian", BAILIAN_URL, "", datetime.utcnow().isoformat()))
     conn.commit()
     return cursor.lastrowid
 
@@ -48,37 +70,31 @@ def main():
         print(f"❌ Failed to fetch models: {e}")
         return 1
 
-    # 连接数据库
-    try:
-        conn = sqlite3.connect(DB_PATH)
-        conn.row_factory = sqlite3.Row
-    except Exception as e:
-        print(f"❌ Cannot connect to database: {e}")
-        return 1
+    conn = sqlite3.connect(DB_PATH)
+    provider_table = ensure_provider_table(conn)
+    provider_id = get_or_create_bailian_provider(conn, provider_table)
 
-    # 获取 provider_id
-    provider_id = get_or_create_bailian_provider(conn)
-    print(f"Using provider_id: {provider_id}")
+    # 获取 models 表的列
+    model_columns = get_table_columns(conn, "models")
+    print(f"models table columns: {model_columns}")
 
-    # 获取 models 表列名
-    columns = get_table_columns(conn, "models")
-    print(f"models table columns: {columns}")
+    # 准备插入语句（根据实际列名动态构建）
+    insert_cols = []
+    for col in ['id', 'provider_id', 'enabled', 'created_at', 'name']:
+        if col in model_columns:
+            insert_cols.append(col)
+    if not insert_cols:
+        print("❌ models table missing expected columns")
+        return 1
 
-    # 确定插入时使用的列
-    insert_cols = [col for col in ['id', 'provider_id', 'enabled', 'created_at', 'name'] if col in columns]
-    # 如果 'name' 列存在，则 name 使用 model_id；否则忽略
     placeholders = ','.join(['?' for _ in insert_cols])
-
     cursor = conn.cursor()
     added = 0
     for model_id in models:
-        # 检查是否已存在
         cursor.execute("SELECT 1 FROM models WHERE id = ?", (model_id,))
         if cursor.fetchone():
             print(f"⏭️ Already exists: {model_id}")
             continue
-
-        # 准备插入数据
         values = []
         for col in insert_cols:
             if col == 'id':
@@ -90,7 +106,7 @@ def main():
             elif col == 'created_at':
                 values.append(datetime.utcnow().isoformat())
             elif col == 'name':
-                values.append(model_id)  # name 与 id 相同
+                values.append(model_id)
         sql = f"INSERT INTO models ({','.join(insert_cols)}) VALUES ({placeholders})"
         cursor.execute(sql, values)
         added += 1