145 lines
5.0 KiB
Python
145 lines
5.0 KiB
Python
import requests
|
||
import json
|
||
import logging
|
||
import toml
|
||
import sqlite3
|
||
|
||
class AiCat:
|
||
def __init__(self,message,qid,group_openid):
|
||
self.message = message
|
||
self.qid = qid
|
||
self.group_openid = group_openid
|
||
self.query = f"<qid>{self.qid}</qid>{self.message}"
|
||
|
||
|
||
|
||
def main(self):
|
||
with open('./config.toml', 'r', encoding='utf-8') as f:
|
||
config = toml.load(f)
|
||
|
||
# 获取所需字段
|
||
ai_service = config.get("ai_service")
|
||
# dify
|
||
dify_ip = config.get("dify_ip")
|
||
dify_token = config.get("dify_token")
|
||
# xyit
|
||
xyit_ip = config.get("xyit_ip")
|
||
xyit_appID = config.get("xyit_appID")
|
||
xyit_appKEY = config.get("xyit_appKEY")
|
||
xyit_model = config.get("xyit_model")
|
||
|
||
self.init_db()
|
||
|
||
uuid = self.get_uuid()
|
||
|
||
if uuid == "":
|
||
logging.info("未找到 UUID")
|
||
else:
|
||
logging.info(f"找到 UUID: {uuid}")
|
||
|
||
if ai_service == "dify":
|
||
# API URL
|
||
url = f"https://{dify_ip}/v1/chat-messages" # 替换为实际的 API 地址
|
||
|
||
# 请求头
|
||
headers = {
|
||
"Content-Type": "application/json",
|
||
"Authorization": dify_token # 替换为你的 API 密钥
|
||
}
|
||
|
||
# 请求体
|
||
payload = {
|
||
"query": self.query, # 用户输入/提问内容
|
||
"inputs": {}, # App 定义的变量值（默认为空）
|
||
"response_mode": "blocking", # 流式模式或阻塞模式
|
||
"user": "QBotAPI", # 用户标识，需保证唯一性
|
||
"conversation_id": uuid, # （选填）会话 ID，继续对话时需要传入
|
||
"files": [], # 文件列表（选填），适用于文件结合文本理解
|
||
"auto_generate_name": True # （选填）自动生成标题，默认为 True
|
||
}
|
||
|
||
logging.info("请求平台：dify")
|
||
|
||
elif ai_service == "xyit":
|
||
url = f"http://{xyit_ip}/{xyit_model}/" # 替换为实际的 API 地址
|
||
|
||
# 请求头
|
||
headers = {
|
||
"Content-Type": "application/json",
|
||
"appID": xyit_appID , # 替换为你的 API 密钥
|
||
"appKEY": xyit_appKEY
|
||
}
|
||
|
||
# 请求体
|
||
payload = {
|
||
"query": self.query, # 用户输入/提问内容
|
||
"conversation_id": uuid, # （选填）会话 ID，继续对话时需要传入
|
||
}
|
||
|
||
logging.info("请求平台：xyit")
|
||
|
||
else:
|
||
logging.error("未配置ai平台")
|
||
return "服务器繁忙，请稍后再逝"
|
||
|
||
# 发送 POST 请求
|
||
try:
|
||
response = requests.post(url, headers=headers, data=json.dumps(payload))
|
||
|
||
# 检查响应状态码
|
||
if response.status_code == 200:
|
||
logging.info("请求成功！返回结果：")
|
||
response_data = response.json()
|
||
print(response_data) # test
|
||
# 存储uuid
|
||
if uuid == "":
|
||
try:
|
||
with sqlite3.connect("uuid.db") as conn:
|
||
cursor = conn.cursor()
|
||
cursor.execute("INSERT INTO groups (group_openid, uuid) VALUES (?, ?)",
|
||
(self.group_openid, response_data.get("conversation_id")))
|
||
except sqlite3.Error as e:
|
||
logging.error(f"数据库插入错误: {e}")
|
||
|
||
# 提取 answer 值
|
||
answer = response_data.get("answer","服务器繁忙，请稍后再逝")
|
||
return answer
|
||
else:
|
||
logging.info(f"请求失败！状态码: {response.status_code}")
|
||
logging.info(f"错误信息: {response.text}") # 打印错误信息
|
||
return "服务器繁忙，请稍后再逝"
|
||
|
||
except Exception as e:
|
||
logging.info(f"请求过程中出现异常: {e}")
|
||
return "服务器繁忙，请稍后再逝"
|
||
|
||
|
||
def get_uuid(self):
|
||
try:
|
||
with sqlite3.connect("uuid.db") as conn:
|
||
cursor = conn.cursor()
|
||
cursor.execute("SELECT uuid FROM groups WHERE group_openid = ?", (self.group_openid,))
|
||
result = cursor.fetchone()
|
||
return result[0] if result else ""
|
||
except sqlite3.Error as e:
|
||
logging.error(f"数据库查询错误: {e}")
|
||
return ""
|
||
|
||
def init_db(self):
|
||
try:
|
||
with sqlite3.connect("uuid.db") as conn:
|
||
cursor = conn.cursor()
|
||
cursor.execute('''
|
||
CREATE TABLE IF NOT EXISTS groups (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
group_openid TEXT UNIQUE NOT NULL,
|
||
uuid TEXT NOT NULL
|
||
)
|
||
''')
|
||
conn.commit()
|
||
except sqlite3.Error as e:
|
||
logging.error(f"数据库初始化失败: {e}")
|
||
|
||
|
||
|