first commit
This commit is contained in:
55
model/AiCat.py
Normal file
55
model/AiCat.py
Normal file
@@ -0,0 +1,55 @@
|
||||
import requests
|
||||
import json
|
||||
import logging
|
||||
|
||||
class AiCat:
|
||||
def __init__(self,message,qid):
|
||||
self.message = message
|
||||
self.qid = qid
|
||||
self.query = f"<qid>{self.qid}</qid>{self.message}"
|
||||
print(self.query) # test
|
||||
def main(self):
|
||||
# API URL
|
||||
url = "https://ai.mcunc.cn/v1/chat-messages" # 替换为实际的 API 地址
|
||||
|
||||
# 请求头
|
||||
headers = {
|
||||
"Content-Type": "application/json",
|
||||
"Authorization": "Bearer app-pqib8uic8oBP95XmXuBi5ANq" # 替换为你的 API 密钥
|
||||
}
|
||||
|
||||
# 请求体
|
||||
payload = {
|
||||
"query": self.query, # 用户输入/提问内容
|
||||
"inputs": {}, # App 定义的变量值(默认为空)
|
||||
"response_mode": "blocking", # 流式模式或阻塞模式
|
||||
"user": "QBotAPI", # 用户标识,需保证唯一性
|
||||
"conversation_id": "09cc6545-b0e0-4611-aad1-cf9bf51600e1", # (选填)会话 ID,继续对话时需要传入
|
||||
"files": [], # 文件列表(选填),适用于文件结合文本理解
|
||||
"auto_generate_name": True # (选填)自动生成标题,默认为 True
|
||||
}
|
||||
|
||||
# 发送 POST 请求
|
||||
try:
|
||||
response = requests.post(url, headers=headers, data=json.dumps(payload))
|
||||
|
||||
# 检查响应状态码
|
||||
if response.status_code == 200:
|
||||
logging.info("请求成功!返回结果:")
|
||||
response_data = response.json()
|
||||
logging.info(json.dumps(response_data, indent=4, ensure_ascii=False)) # 格式化输出 JSON
|
||||
|
||||
# 提取 answer 值
|
||||
answer = response_data.get("answer", "answer 字段不存在")
|
||||
print(response_data.get("conversation_id")) # test
|
||||
return answer
|
||||
else:
|
||||
logging.info(f"请求失败!状态码: {response.status_code}")
|
||||
logging.info(f"错误信息: {response.text}") # 打印错误信息
|
||||
return "诶呀! 服务器好像出现了一点点错误呐~稍等一会再重试哦喵~"
|
||||
|
||||
except Exception as e:
|
||||
logging.info(f"请求过程中出现异常: {e}")
|
||||
return "诶呀! 服务器好像出现了一点点错误呐~稍等一会再重试哦喵~"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user