feat: 新增考试,可选(kimi,豆包,deepseek)模型进行答题,自己设置 API_KEY, 请不要相信 AI 的结果,最后交卷会提示你

This commit is contained in:
2025-12-12 19:17:36 +08:00
parent 6dfeac0d24
commit 14d4bc014d
4 changed files with 284 additions and 20 deletions

View File

@@ -3,6 +3,7 @@ import os
import json
from openai import OpenAI
from dotenv import load_dotenv
from httpx import Client, Proxy
load_dotenv()
@@ -24,9 +25,17 @@ class DOUBAO:
self.key = os.environ.get("DOUBAO_API_KEY")
if not self.key:
raise Exception("找不到 DOUBAO_API_KEY 请设置环境变量在运行代码")
httpx_client = Client(
proxy=Proxy("socks5://127.0.0.1:9000"), # 传入SOCKS代理地址
timeout=30.0, # 可选:设置超时时间
verify=False,
)
self.client = OpenAI(
api_key=self.key,
base_url="https://ark.cn-beijing.volces.com/api/v3",
http_client=httpx_client,
)
def chat(self, question_json: dict) -> str: