diff --git a/main.py b/main.py index eefdf8c..136742d 100644 --- a/main.py +++ b/main.py @@ -337,19 +337,27 @@ class Jx3ApiPlugin(Star): @wz.command("战绩") async def wz_zhanji(self, event: AstrMessageEvent,name: str = "飞翔大野猪",option: str = "0"): """王者 战绩 营地ID 对局类型""" + try: data = await self.wzry.zhanji(name,option) # logger.info(f"王者荣耀战绩查询结果{data}") + prompt = f"{data['data']}\n" + prompt += f"以上数据是获取到的王者荣耀战绩信息,请根据这些数据生成一两句话的战绩总结,突出玩家的优势和特点,语言风格轻松幽默,适合在游戏群内分享。" if data["code"] == 200: url = await self.html_render(data["temp"],{"data": data["data"]}, options={}) yield event.image_result(url) else: yield event.plain_result(data["msg"]) - return + return + provider_id = await self.context.get_current_chat_provider_id(umo=event.unified_msg_origin) + llm_resp = await self.context.llm_generate(chat_provider_id=provider_id, prompt=prompt,) + out = llm_resp.completion_text + yield event.plain_result(f"{out}") except Exception as e: logger.error(f"功能函数执行错误: {e}") yield event.plain_result("猪脑过载,请稍后再试") + @wz.command("资料") async def wz_ziliao(self, event: AstrMessageEvent,name: str = "489048724"): """王者 战绩 营地ID 对局类型""" @@ -366,6 +374,19 @@ class Jx3ApiPlugin(Star): logger.error(f"功能函数执行错误: {e}") yield event.plain_result("猪脑过载,请稍后再试") + @wz.command("测试") + async def wz_test(self, event: AstrMessageEvent,name: str = "489048724"): + """王者 战绩 营地ID 对局类型""" + try: + provider_id = await self.context.get_current_chat_provider_id(umo=event.unified_msg_origin) + llm_resp = await self.context.llm_generate(chat_provider_id=provider_id, prompt="Hello, world!",) + out = llm_resp.completion_text + yield event.plain_result(f"{out}") + except Exception as e: + logger.error(f"功能函数执行错误: {e}") + yield event.plain_result("猪脑过载,请稍后再试") + + async def terminate(self): """可选择实现异步的插件销毁方法,当插件被卸载/停用时会调用。""" await self.db.close_pool()