添加llm call tool

main.py

@@ -314,6 +314,164 @@ class BangumiPlugin(Star):
         result = await self.subscription_service.unsubscribe(session_key, query)
         yield event.plain_result(result)
 
+    # --- LLM Tool 区 ---
+
+    @filter.llm_tool(name="bangumi_search")
+    async def llm_search(
+        self, event: AstrMessageEvent, query: str, top_k: int = 1
+    ) -> AsyncGenerator[object, None]:
+        """在 Bangumi 数据库中全类别搜索番剧、漫画、游戏等条目，展示条目简介、评分等信息。当用户询问某个作品的详情、评价、评分时调用。
+
+        Args:
+            query(string): 搜索关键词，如作品名称
+            top_k(number): 返回结果数量，默认 1，最多 5
+        """
+        if not self.search_service:
+            yield event.plain_result("❌ 搜索服务未就绪")
+            return
+        async for result in self.search_service.handle_subject_search(
+            event, query, int(top_k), subject_type=None
+        ):
+            yield result
+
+    @filter.llm_tool(name="bangumi_search_anime")
+    async def llm_search_anime(
+        self, event: AstrMessageEvent, query: str, top_k: int = 1
+    ) -> AsyncGenerator[object, None]:
+        """在 Bangumi 数据库中搜索 TV 番剧/动漫条目，展示动漫详情与评分。当用户明确询问某部 TV 动漫/番剧时调用。
+
+        Args:
+            query(string): 搜索关键词，如番剧名称
+            top_k(number): 返回结果数量，默认 1，最多 5
+        """
+        if not self.search_service:
+            yield event.plain_result("❌ 搜索服务未就绪")
+            return
+        async for result in self.search_service.handle_subject_search(
+            event, query, int(top_k), subject_type=[2], subject_tags=["TV"]
+        ):
+            yield result
+
+    @filter.llm_tool(name="bangumi_search_movie")
+    async def llm_search_movie(
+        self, event: AstrMessageEvent, query: str, top_k: int = 1
+    ) -> AsyncGenerator[object, None]:
+        """在 Bangumi 数据库中搜索剧场版动画条目。当用户询问某部剧场版/电影动画的信息时调用。
+
+        Args:
+            query(string): 搜索关键词，如剧场版名称
+            top_k(number): 返回结果数量，默认 1，最多 5
+        """
+        if not self.search_service:
+            yield event.plain_result("❌ 搜索服务未就绪")
+            return
+        async for result in self.search_service.handle_subject_search(
+            event, query, int(top_k), subject_type=[2], subject_tags=["剧场版"]
+        ):
+            yield result
+
+    @filter.llm_tool(name="bangumi_search_manga")
+    async def llm_search_manga(
+        self, event: AstrMessageEvent, query: str, top_k: int = 1
+    ) -> AsyncGenerator[object, None]:
+        """在 Bangumi 数据库中搜索漫画条目，展示漫画详情与评分。当用户询问某部漫画的信息时调用。
+
+        Args:
+            query(string): 搜索关键词，如漫画名称
+            top_k(number): 返回结果数量，默认 1，最多 5
+        """
+        if not self.search_service:
+            yield event.plain_result("❌ 搜索服务未就绪")
+            return
+        async for result in self.search_service.handle_subject_search(
+            event, query, int(top_k), subject_type=[1], subject_tags=["漫画"]
+        ):
+            yield result
+
+    @filter.llm_tool(name="bangumi_today_calendar")
+    async def llm_today_calendar(
+        self, event: AstrMessageEvent
+    ) -> AsyncGenerator[object, None]:
+        """获取今日番剧放送时刻表，展示今天有哪些动漫新番更新。当用户询问"今天有什么番"、"今天更新了什么"、"今日放送"时调用。"""
+        if not self.search_service:
+            yield event.plain_result("❌ 搜索服务未就绪")
+            return
+        async for result in self.search_service.handle_calendar(event):
+            yield result
+
+    @filter.llm_tool(name="bangumi_subscribe")
+    async def llm_subscribe(
+        self,
+        event: AstrMessageEvent,
+        query: str = "",
+        subject_id: str = "",
+    ) -> AsyncGenerator[object, None]:
+        """订阅番剧更新通知。当用户表达想要追番、订阅、收到某番剧更新提醒时调用。
+        若提供 subject_id 则直接订阅；否则按 query 关键词搜索，唯一匹配时自动订阅，多个候选时返回列表供用户确认（用户选定后再次调用并传入 subject_id）。
+
+        Args:
+            query(string): 要订阅的番剧名称关键词，与 subject_id 二选一
+            subject_id(string): 番剧的 Bangumi ID，优先级高于 query
+        """
+        if not self.subscription_service:
+            yield event.plain_result("❌ 订阅服务未就绪")
+            return
+
+        session_key = self._resolve_session_key(event)
+
+        if subject_id:
+            result = await self.subscription_service.subscribe_by_subject_id(
+                session_id=session_key,
+                subject_id=subject_id,
+            )
+            yield event.plain_result(result)
+            return
+
+        if not query:
+            yield event.plain_result("❌ 请提供番剧名称关键词或 Bangumi ID")
+            return
+
+        error_msg, candidates = await self.subscription_service.get_subscribe_candidates(
+            keyword=query,
+            limit=self.config_manager.get_max_fuzzy_results(),
+        )
+        if error_msg:
+            yield event.plain_result(error_msg)
+            return
+        if not candidates:
+            yield event.plain_result("🔍 未找到相关番剧")
+            return
+
+        if len(candidates) == 1:
+            result = await self.subscription_service.subscribe_by_subject_id(
+                session_id=session_key,
+                subject_id=candidates[0]["subject_id"],
+            )
+            yield event.plain_result(result)
+            return
+
+        lines = ["⚠️ 匹配到多个候选，请告知序号或提供 Bangumi ID 以确认订阅："]
+        for index, candidate in enumerate(candidates, start=1):
+            lines.append(f"{index}. {candidate['name']} (ID: {candidate['subject_id']})")
+        yield event.plain_result("\n".join(lines))
+
+    @filter.llm_tool(name="bangumi_unsubscribe")
+    async def llm_unsubscribe(
+        self, event: AstrMessageEvent, query: str
+    ) -> AsyncGenerator[object, None]:
+        """取消订阅番剧更新通知（弃坑）。当用户表达想要退订、弃坑、取消追某部番剧时调用。
+
+        Args:
+            query(string): 要取消订阅的番剧名称关键词或 Bangumi ID
+        """
+        if not self.subscription_service:
+            yield event.plain_result("❌ 订阅服务未就绪")
+            return
+
+        session_key = self._resolve_session_key(event)
+        result = await self.subscription_service.unsubscribe(session_key, query)
+        yield event.plain_result(result)
+
     async def terminate(self) -> None:
         logger.info("正在清理 Bangumi 插件资源...")
         if self.scheduler_manager.scheduler.running: