feat:添加视频识别

test_gemini_video.py

@@ -0,0 +1,83 @@
+"""
+测试 Gemini API 视频识别
+"""
+import base64
+import requests
+import json
+
+# 配置
+API_URL = "https://api.functen.cn/v1beta/models/gemini-3-pro-preview:generateContent"
+API_KEY = "sk-NeOtq0kOU39x3LMqY09aKYLoOBJIgFkgGuDwVGgGEstXPn3M"
+VIDEO_PATH = r"D:\project\shrobot\WechatHookBot\plugins\AIChat\265d0df9ea89578bcb86f824c5255a42.mp4"
+
+def test_video():
+    # 读取视频并编码为 base64
+    print(f"读取视频: {VIDEO_PATH}")
+    with open(VIDEO_PATH, "rb") as f:
+        video_data = f.read()
+
+    video_base64 = base64.b64encode(video_data).decode()
+    print(f"视频大小: {len(video_data) / 1024 / 1024:.2f} MB")
+    print(f"Base64 长度: {len(video_base64)}")
+
+    # 构建 Gemini 原生格式请求
+    payload = {
+        "contents": [
+            {
+                "parts": [
+                    {"text": "请描述这个视频的内容"},
+                    {
+                        "inline_data": {
+                            "mime_type": "video/mp4",
+                            "data": video_base64
+                        }
+                    }
+                ]
+            }
+        ],
+        "generationConfig": {
+            "maxOutputTokens": 4096
+        }
+    }
+
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {API_KEY}"
+    }
+
+    print(f"\n发送请求到: {API_URL}")
+    print("请求中...")
+
+    try:
+        response = requests.post(
+            API_URL,
+            headers=headers,
+            json=payload,
+            timeout=180
+        )
+
+        print(f"\n状态码: {response.status_code}")
+
+        if response.status_code == 200:
+            result = response.json()
+            print("\n=== 响应内容 ===")
+            print(json.dumps(result, ensure_ascii=False, indent=2))
+
+            # 提取文本
+            if "candidates" in result:
+                for candidate in result["candidates"]:
+                    content = candidate.get("content", {})
+                    for part in content.get("parts", []):
+                        if "text" in part:
+                            print("\n=== AI 回复 ===")
+                            print(part["text"])
+        else:
+            print(f"\n错误响应: {response.text}")
+
+    except Exception as e:
+        print(f"\n请求失败: {e}")
+        import traceback
+        traceback.print_exc()
+
+if __name__ == "__main__":
+    test_video()