Files
iOSAI/Utils/Requester.py

81 lines
2.4 KiB
Python

import requests
from Entity.Variables import prologueList
from Utils.JsonUtils import JsonUtils
from Utils.LogManager import LogManager
BaseUrl = "https://crawlclient.api.yolozs.com/api/common/"
class Requester():
comment = "comment"
prologue = "prologue"
@classmethod
def requestPrologue(cls, token):
try:
headers = {
"vvtoken": token,
}
url = BaseUrl + cls.prologue
result = requests.get(headers=headers, url=url)
json = result.json()
data = json.get("data")
for i in data:
prologueList.append(i)
except Exception as e:
LogManager.method_error(f"获取requestPrologue失败,报错的原因:{e}", "获取requestPrologue异常")
# 翻译
@classmethod
def translation(cls, msg, country="英国"):
try:
param = {
"msg": msg,
"country": country,
}
url = "https://ai.yolozs.com/translation"
result = requests.post(url=url, json=param)
if result.status_code != 200:
LogManager.error(f"翻译失败,状态码:{result.status_code},服务器返回的内容:{result.text}")
return None
json = result.json()
data = json.get("data")
return data
except Exception as e:
LogManager.method_error(f"翻译失败,报错的原因:{e}", "翻译失败异常")
# ai聊天
@classmethod
def chatToAi(cls, param):
aiConfig = JsonUtils.read_json("aiConfig")
agentName = aiConfig.get("agentName")
guildName = aiConfig.get("guildName")
contactTool = aiConfig.get("contactTool", "")
contact = aiConfig.get("contact", "")
inputs = {
"name":agentName,
"Trade_union":guildName,
"contcat_method":contactTool,
"contcat_info":contact
}
param["inputs"] = inputs
print(param)
try:
url = "https://ai.yolozs.com/chat"
result = requests.post(url=url, json=param)
json = result.json()
data = json.get("answer", {})
session_id = json.get("conversation_id", {})
return data,session_id
except Exception as e:
LogManager.method_error(f"ai聊天失败,ai聊天出现异常,报错的原因:{e}", "ai聊天接口异常")