您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

405 行
18KB

  1. # access LinkAI knowledge base platform
  2. # docs: https://link-ai.tech/platform/link-app/wechat
  3. import time
  4. import requests
  5. import config
  6. from bot.bot import Bot
  7. from bot.chatgpt.chat_gpt_session import ChatGPTSession
  8. from bot.session_manager import SessionManager
  9. from bridge.context import Context, ContextType
  10. from bridge.reply import Reply, ReplyType
  11. from common.log import logger
  12. from config import conf, pconf
  13. import threading
  14. from common import memory, utils
  15. import base64
  16. class LinkAIBot(Bot):
  17. # authentication failed
  18. AUTH_FAILED_CODE = 401
  19. NO_QUOTA_CODE = 406
  20. def __init__(self):
  21. super().__init__()
  22. self.sessions = LinkAISessionManager(LinkAISession, model=conf().get("model") or "gpt-3.5-turbo")
  23. self.args = {}
  24. def reply(self, query, context: Context = None) -> Reply:
  25. if context.type == ContextType.TEXT:
  26. return self._chat(query, context)
  27. elif context.type == ContextType.IMAGE_CREATE:
  28. if not conf().get("text_to_image"):
  29. logger.warn("[LinkAI] text_to_image is not enabled, ignore the IMAGE_CREATE request")
  30. return Reply(ReplyType.TEXT, "")
  31. ok, res = self.create_img(query, 0)
  32. if ok:
  33. reply = Reply(ReplyType.IMAGE_URL, res)
  34. else:
  35. reply = Reply(ReplyType.ERROR, res)
  36. return reply
  37. else:
  38. reply = Reply(ReplyType.ERROR, "Bot不支持处理{}类型的消息".format(context.type))
  39. return reply
  40. def _chat(self, query, context, retry_count=0) -> Reply:
  41. """
  42. 发起对话请求
  43. :param query: 请求提示词
  44. :param context: 对话上下文
  45. :param retry_count: 当前递归重试次数
  46. :return: 回复
  47. """
  48. if retry_count > 2:
  49. # exit from retry 2 times
  50. logger.warn("[LINKAI] failed after maximum number of retry times")
  51. return Reply(ReplyType.TEXT, "请再问我一次吧")
  52. try:
  53. # load config
  54. if context.get("generate_breaked_by"):
  55. logger.info(f"[LINKAI] won't set appcode because a plugin ({context['generate_breaked_by']}) affected the context")
  56. app_code = None
  57. else:
  58. plugin_app_code = self._find_group_mapping_code(context)
  59. app_code = context.kwargs.get("app_code") or plugin_app_code or conf().get("linkai_app_code")
  60. linkai_api_key = conf().get("linkai_api_key")
  61. session_id = context["session_id"]
  62. session_message = self.sessions.session_msg_query(query, session_id)
  63. logger.debug(f"[LinkAI] session={session_message}, session_id={session_id}")
  64. # image process
  65. img_cache = memory.USER_IMAGE_CACHE.get(session_id)
  66. if img_cache:
  67. messages = self._process_image_msg(app_code=app_code, session_id=session_id, query=query, img_cache=img_cache)
  68. if messages:
  69. session_message = messages
  70. model = conf().get("model")
  71. # remove system message
  72. if session_message[0].get("role") == "system":
  73. if app_code or model == "wenxin":
  74. session_message.pop(0)
  75. body = {
  76. "app_code": app_code,
  77. "messages": session_message,
  78. "model": model, # 对话模型的名称, 支持 gpt-3.5-turbo, gpt-3.5-turbo-16k, gpt-4, wenxin, xunfei
  79. "temperature": conf().get("temperature"),
  80. "top_p": conf().get("top_p", 1),
  81. "frequency_penalty": conf().get("frequency_penalty", 0.0), # [-2,2]之间,该值越大则更倾向于产生不同的内容
  82. "presence_penalty": conf().get("presence_penalty", 0.0), # [-2,2]之间,该值越大则更倾向于产生不同的内容
  83. }
  84. file_id = context.kwargs.get("file_id")
  85. if file_id:
  86. body["file_id"] = file_id
  87. logger.info(f"[LINKAI] query={query}, app_code={app_code}, model={body.get('model')}, file_id={file_id}")
  88. headers = {"Authorization": "Bearer " + linkai_api_key}
  89. # do http request
  90. base_url = conf().get("linkai_api_base", "https://api.link-ai.chat")
  91. res = requests.post(url=base_url + "/v1/chat/completions", json=body, headers=headers,
  92. timeout=conf().get("request_timeout", 180))
  93. if res.status_code == 200:
  94. # execute success
  95. response = res.json()
  96. reply_content = response["choices"][0]["message"]["content"]
  97. total_tokens = response["usage"]["total_tokens"]
  98. logger.info(f"[LINKAI] reply={reply_content}, total_tokens={total_tokens}")
  99. self.sessions.session_reply(reply_content, session_id, total_tokens, query=query)
  100. agent_suffix = self._fetch_agent_suffix(response)
  101. if agent_suffix:
  102. reply_content += agent_suffix
  103. if not agent_suffix:
  104. knowledge_suffix = self._fetch_knowledge_search_suffix(response)
  105. if knowledge_suffix:
  106. reply_content += knowledge_suffix
  107. # image process
  108. if response["choices"][0].get("img_urls"):
  109. thread = threading.Thread(target=self._send_image, args=(context.get("channel"), context, response["choices"][0].get("img_urls")))
  110. thread.start()
  111. if response["choices"][0].get("text_content"):
  112. reply_content = response["choices"][0].get("text_content")
  113. return Reply(ReplyType.TEXT, reply_content)
  114. else:
  115. response = res.json()
  116. error = response.get("error")
  117. logger.error(f"[LINKAI] chat failed, status_code={res.status_code}, "
  118. f"msg={error.get('message')}, type={error.get('type')}")
  119. if res.status_code >= 500:
  120. # server error, need retry
  121. time.sleep(2)
  122. logger.warn(f"[LINKAI] do retry, times={retry_count}")
  123. return self._chat(query, context, retry_count + 1)
  124. return Reply(ReplyType.TEXT, "提问太快啦,请休息一下再问我吧")
  125. except Exception as e:
  126. logger.exception(e)
  127. # retry
  128. time.sleep(2)
  129. logger.warn(f"[LINKAI] do retry, times={retry_count}")
  130. return self._chat(query, context, retry_count + 1)
  131. def _process_image_msg(self, app_code: str, session_id: str, query:str, img_cache: dict):
  132. try:
  133. enable_image_input = False
  134. app_info = self._fetch_app_info(app_code)
  135. if not app_info:
  136. logger.debug(f"[LinkAI] not found app, can't process images, app_code={app_code}")
  137. return None
  138. plugins = app_info.get("data").get("plugins")
  139. for plugin in plugins:
  140. if plugin.get("input_type") and "IMAGE" in plugin.get("input_type"):
  141. enable_image_input = True
  142. if not enable_image_input:
  143. return
  144. msg = img_cache.get("msg")
  145. path = img_cache.get("path")
  146. msg.prepare()
  147. logger.info(f"[LinkAI] query with images, path={path}")
  148. messages = self._build_vision_msg(query, path)
  149. memory.USER_IMAGE_CACHE[session_id] = None
  150. return messages
  151. except Exception as e:
  152. logger.exception(e)
  153. def _find_group_mapping_code(self, context):
  154. try:
  155. if context.kwargs.get("isgroup"):
  156. group_name = context.kwargs.get("msg").from_user_nickname
  157. if config.plugin_config and config.plugin_config.get("linkai"):
  158. linkai_config = config.plugin_config.get("linkai")
  159. group_mapping = linkai_config.get("group_app_map")
  160. if group_mapping and group_name:
  161. return group_mapping.get(group_name)
  162. except Exception as e:
  163. logger.exception(e)
  164. return None
  165. def _build_vision_msg(self, query: str, path: str):
  166. try:
  167. suffix = utils.get_path_suffix(path)
  168. with open(path, "rb") as file:
  169. base64_str = base64.b64encode(file.read()).decode('utf-8')
  170. messages = [{
  171. "role": "user",
  172. "content": [
  173. {
  174. "type": "text",
  175. "text": query
  176. },
  177. {
  178. "type": "image_url",
  179. "image_url": {
  180. "url": f"data:image/{suffix};base64,{base64_str}"
  181. }
  182. }
  183. ]
  184. }]
  185. return messages
  186. except Exception as e:
  187. logger.exception(e)
  188. def reply_text(self, session: ChatGPTSession, app_code="", retry_count=0) -> dict:
  189. if retry_count >= 2:
  190. # exit from retry 2 times
  191. logger.warn("[LINKAI] failed after maximum number of retry times")
  192. return {
  193. "total_tokens": 0,
  194. "completion_tokens": 0,
  195. "content": "请再问我一次吧"
  196. }
  197. try:
  198. body = {
  199. "app_code": app_code,
  200. "messages": session.messages,
  201. "model": conf().get("model") or "gpt-3.5-turbo", # 对话模型的名称, 支持 gpt-3.5-turbo, gpt-3.5-turbo-16k, gpt-4, wenxin, xunfei
  202. "temperature": conf().get("temperature"),
  203. "top_p": conf().get("top_p", 1),
  204. "frequency_penalty": conf().get("frequency_penalty", 0.0), # [-2,2]之间,该值越大则更倾向于产生不同的内容
  205. "presence_penalty": conf().get("presence_penalty", 0.0), # [-2,2]之间,该值越大则更倾向于产生不同的内容
  206. }
  207. if self.args.get("max_tokens"):
  208. body["max_tokens"] = self.args.get("max_tokens")
  209. headers = {"Authorization": "Bearer " + conf().get("linkai_api_key")}
  210. # do http request
  211. base_url = conf().get("linkai_api_base", "https://api.link-ai.chat")
  212. res = requests.post(url=base_url + "/v1/chat/completions", json=body, headers=headers,
  213. timeout=conf().get("request_timeout", 180))
  214. if res.status_code == 200:
  215. # execute success
  216. response = res.json()
  217. reply_content = response["choices"][0]["message"]["content"]
  218. total_tokens = response["usage"]["total_tokens"]
  219. logger.info(f"[LINKAI] reply={reply_content}, total_tokens={total_tokens}")
  220. return {
  221. "total_tokens": total_tokens,
  222. "completion_tokens": response["usage"]["completion_tokens"],
  223. "content": reply_content,
  224. }
  225. else:
  226. response = res.json()
  227. error = response.get("error")
  228. logger.error(f"[LINKAI] chat failed, status_code={res.status_code}, "
  229. f"msg={error.get('message')}, type={error.get('type')}")
  230. if res.status_code >= 500:
  231. # server error, need retry
  232. time.sleep(2)
  233. logger.warn(f"[LINKAI] do retry, times={retry_count}")
  234. return self.reply_text(session, app_code, retry_count + 1)
  235. return {
  236. "total_tokens": 0,
  237. "completion_tokens": 0,
  238. "content": "提问太快啦,请休息一下再问我吧"
  239. }
  240. except Exception as e:
  241. logger.exception(e)
  242. # retry
  243. time.sleep(2)
  244. logger.warn(f"[LINKAI] do retry, times={retry_count}")
  245. return self.reply_text(session, app_code, retry_count + 1)
  246. def _fetch_app_info(self, app_code: str):
  247. headers = {"Authorization": "Bearer " + conf().get("linkai_api_key")}
  248. # do http request
  249. base_url = conf().get("linkai_api_base", "https://api.link-ai.chat")
  250. params = {"app_code": app_code}
  251. res = requests.get(url=base_url + "/v1/app/info", params=params, headers=headers, timeout=(5, 10))
  252. if res.status_code == 200:
  253. return res.json()
  254. else:
  255. logger.warning(f"[LinkAI] find app info exception, res={res}")
  256. def create_img(self, query, retry_count=0, api_key=None):
  257. try:
  258. logger.info("[LinkImage] image_query={}".format(query))
  259. headers = {
  260. "Content-Type": "application/json",
  261. "Authorization": f"Bearer {conf().get('linkai_api_key')}"
  262. }
  263. data = {
  264. "prompt": query,
  265. "n": 1,
  266. "model": conf().get("text_to_image") or "dall-e-2",
  267. "response_format": "url",
  268. "img_proxy": conf().get("image_proxy")
  269. }
  270. url = conf().get("linkai_api_base", "https://api.link-ai.chat") + "/v1/images/generations"
  271. res = requests.post(url, headers=headers, json=data, timeout=(5, 90))
  272. t2 = time.time()
  273. image_url = res.json()["data"][0]["url"]
  274. logger.info("[OPEN_AI] image_url={}".format(image_url))
  275. return True, image_url
  276. except Exception as e:
  277. logger.error(format(e))
  278. return False, "画图出现问题,请休息一下再问我吧"
  279. def _fetch_knowledge_search_suffix(self, response) -> str:
  280. try:
  281. if response.get("knowledge_base"):
  282. search_hit = response.get("knowledge_base").get("search_hit")
  283. first_similarity = response.get("knowledge_base").get("first_similarity")
  284. logger.info(f"[LINKAI] knowledge base, search_hit={search_hit}, first_similarity={first_similarity}")
  285. plugin_config = pconf("linkai")
  286. if plugin_config and plugin_config.get("knowledge_base") and plugin_config.get("knowledge_base").get("search_miss_text_enabled"):
  287. search_miss_similarity = plugin_config.get("knowledge_base").get("search_miss_similarity")
  288. search_miss_text = plugin_config.get("knowledge_base").get("search_miss_suffix")
  289. if not search_hit:
  290. return search_miss_text
  291. if search_miss_similarity and float(search_miss_similarity) > first_similarity:
  292. return search_miss_text
  293. except Exception as e:
  294. logger.exception(e)
  295. def _fetch_agent_suffix(self, response):
  296. try:
  297. plugin_list = []
  298. logger.debug(f"[LinkAgent] res={response}")
  299. if response.get("agent") and response.get("agent").get("chain") and response.get("agent").get("need_show_plugin"):
  300. chain = response.get("agent").get("chain")
  301. suffix = "\n\n- - - - - - - - - - - -"
  302. i = 0
  303. for turn in chain:
  304. plugin_name = turn.get('plugin_name')
  305. suffix += "\n"
  306. need_show_thought = response.get("agent").get("need_show_thought")
  307. if turn.get("thought") and plugin_name and need_show_thought:
  308. suffix += f"{turn.get('thought')}\n"
  309. if plugin_name:
  310. plugin_list.append(turn.get('plugin_name'))
  311. suffix += f"{turn.get('plugin_icon')} {turn.get('plugin_name')}"
  312. if turn.get('plugin_input'):
  313. suffix += f":{turn.get('plugin_input')}"
  314. if i < len(chain) - 1:
  315. suffix += "\n"
  316. i += 1
  317. logger.info(f"[LinkAgent] use plugins: {plugin_list}")
  318. return suffix
  319. except Exception as e:
  320. logger.exception(e)
  321. def _send_image(self, channel, context, image_urls):
  322. if not image_urls:
  323. return
  324. try:
  325. for url in image_urls:
  326. reply = Reply(ReplyType.IMAGE_URL, url)
  327. channel.send(reply, context)
  328. except Exception as e:
  329. logger.error(e)
  330. class LinkAISessionManager(SessionManager):
  331. def session_msg_query(self, query, session_id):
  332. session = self.build_session(session_id)
  333. messages = session.messages + [{"role": "user", "content": query}]
  334. return messages
  335. def session_reply(self, reply, session_id, total_tokens=None, query=None):
  336. session = self.build_session(session_id)
  337. if query:
  338. session.add_query(query)
  339. session.add_reply(reply)
  340. try:
  341. max_tokens = conf().get("conversation_max_tokens", 2500)
  342. tokens_cnt = session.discard_exceeding(max_tokens, total_tokens)
  343. logger.debug(f"[LinkAI] chat history, before tokens={total_tokens}, now tokens={tokens_cnt}")
  344. except Exception as e:
  345. logger.warning("Exception when counting tokens precisely for session: {}".format(str(e)))
  346. return session
  347. class LinkAISession(ChatGPTSession):
  348. def calc_tokens(self):
  349. if not self.messages:
  350. return 0
  351. return len(str(self.messages))
  352. def discard_exceeding(self, max_tokens, cur_tokens=None):
  353. cur_tokens = self.calc_tokens()
  354. if cur_tokens > max_tokens:
  355. for i in range(0, len(self.messages)):
  356. if i > 0 and self.messages[i].get("role") == "assistant" and self.messages[i - 1].get("role") == "user":
  357. self.messages.pop(i)
  358. self.messages.pop(i - 1)
  359. return self.calc_tokens()
  360. return cur_tokens