feat: 新增考试,可选(kimi,豆包,deepseek)模型进行答题,自己设置 API_KEY, 请不要相信 AI 的结果,最后交卷会提示你
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user