Update scripts/sync_bailian_models.py

scripts/sync_bailian_models.py

@@ -2,33 +2,32 @@
 import os
 import sys
 import sqlite3
-import requests
+import json
+import urllib.request
+import urllib.error
 from datetime import datetime
 
-# 设置数据库路径和百炼API端点
 DB_PATH = "/app/server/data/freeapi.db"
 BAILIAN_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1/models"
 
 def main():
-    # 获取百炼的API Key
     api_key = os.getenv("BAILIAN_API_KEY")
     if not api_key:
-        print("❌ 错误: 环境变量 BAILIAN_API_KEY 未设置")
-        return 1
+        print("⚠️ BAILIAN_API_KEY not set, skipping model sync")
+        return 0
 
-    # 调用百炼API获取模型列表
-    headers = {"Authorization": f"Bearer {api_key}"}
+    # 请求百炼 API
+    req = urllib.request.Request(BAILIAN_URL)
+    req.add_header("Authorization", f"Bearer {api_key}")
     try:
-        response = requests.get(BAILIAN_URL, headers=headers, timeout=30)
-        response.raise_for_status()
-        models_data = response.json()
-        # 假设返回的JSON结构与 OpenAI 兼容模式一致，即包含一个 'data' 列表
-        models_list = [item["id"] for item in models_data.get("data", [])]
-        if not models_list:
-            print("⚠️ 警告: 未能从API获取到任何模型。")
-            return 1
+        with urllib.request.urlopen(req, timeout=30) as response:
+            data = json.loads(response.read().decode('utf-8'))
+            models_list = [item["id"] for item in data.get("data", [])]
+            if not models_list:
+                print("⚠️ No models found from API")
+                return 1
     except Exception as e:
-        print(f"❌ 请求百炼API失败: {e}")
+        print(f"❌ Failed to fetch models: {e}")
         return 1
 
     # 连接数据库
@@ -36,25 +35,25 @@ def main():
         conn = sqlite3.connect(DB_PATH)
         cursor = conn.cursor()
     except Exception as e:
-        print(f"❌ 无法连接数据库: {e}")
+        print(f"❌ Cannot connect to database: {e}")
         return 1
 
-    # 同步模型列表
     added_count = 0
     for model_id in models_list:
-        # 检查模型是否已存在
         cursor.execute("SELECT 1 FROM models WHERE id = ?", (model_id,))
         if not cursor.fetchone():
-            cursor.execute("INSERT INTO models (id, name, provider_id, enabled, created_at) VALUES (?, ?, ?, 1, ?)",
-                           (model_id, model_id, 1, datetime.utcnow().isoformat()))
+            # 根据你的数据库实际结构调整 provider_id（可能需要查询 provider 表）
+            # 这里假设百炼渠道的 provider_id 为 1，请根据实际情况修改
+            cursor.execute("INSERT INTO models (id, name, provider_id, enabled, created_at) VALUES (?, ?, 1, 1, ?)",
+                           (model_id, model_id, datetime.utcnow().isoformat()))
             added_count += 1
-            print(f"➕ 添加模型: {model_id}")
+            print(f"➕ Added model: {model_id}")
         else:
-            print(f"⏭️ 模型已存在: {model_id}")
+            print(f"⏭️ Already exists: {model_id}")
 
     conn.commit()
     conn.close()
-    print(f"✅ 同步完成。新增模型数量: {added_count}")
+    print(f"✅ Sync completed. New models added: {added_count}")
     return 0
 
 if __name__ == "__main__":