|
4 | 4 | import re |
5 | 5 | import uuid |
6 | 6 | from collections.abc import AsyncGenerator |
7 | | -from typing import Any |
| 7 | +from typing import Any, cast |
8 | 8 |
|
9 | 9 | from astrbot import logger |
10 | 10 | from astrbot.core.db.po import Conversation |
@@ -52,7 +52,7 @@ def __init__( |
52 | 52 | """是否是 At 机器人或者带有唤醒词或者是私聊(插件注册的事件监听器会让 is_wake 设为 True, 但是不会让这个属性置为 True)""" |
53 | 53 | self._extras: dict[str, Any] = {} |
54 | 54 | self.session = MessageSesion( |
55 | | - platform_name=platform_meta.id, |
| 55 | + platform_name=cast(str, platform_meta.id), |
56 | 56 | message_type=message_obj.type, |
57 | 57 | session_id=session_id, |
58 | 58 | ) |
@@ -152,7 +152,9 @@ def get_sender_id(self) -> str: |
152 | 152 |
|
153 | 153 | def get_sender_name(self) -> str: |
154 | 154 | """获取消息发送者的名称。(可能会返回空字符串)""" |
155 | | - return self.message_obj.sender.nickname |
| 155 | + if isinstance(self.message_obj.sender.nickname, str): |
| 156 | + return self.message_obj.sender.nickname |
| 157 | + return "" |
156 | 158 |
|
157 | 159 | def set_extra(self, key, value): |
158 | 160 | """设置额外的信息。""" |
@@ -269,7 +271,7 @@ def should_call_llm(self, call_llm: bool): |
269 | 271 | """ |
270 | 272 | self.call_llm = call_llm |
271 | 273 |
|
272 | | - def get_result(self) -> MessageEventResult: |
| 274 | + def get_result(self) -> MessageEventResult | None: |
273 | 275 | """获取消息事件的结果。""" |
274 | 276 | return self._result |
275 | 277 |
|
@@ -319,11 +321,11 @@ def request_llm( |
319 | 321 | self, |
320 | 322 | prompt: str, |
321 | 323 | func_tool_manager=None, |
322 | | - session_id: str = None, |
| 324 | + session_id: str = "", |
323 | 325 | image_urls: list[str] = [], |
324 | 326 | contexts: list = [], |
325 | 327 | system_prompt: str = "", |
326 | | - conversation: Conversation = None, |
| 328 | + conversation: Conversation | None = None, |
327 | 329 | ) -> ProviderRequest: |
328 | 330 | """创建一个 LLM 请求。 |
329 | 331 |
|
@@ -389,7 +391,7 @@ async def react(self, emoji: str): |
389 | 391 | """ |
390 | 392 | await self.send(MessageChain([Plain(emoji)])) |
391 | 393 |
|
392 | | - async def get_group(self, group_id: str = None, **kwargs) -> Group | None: |
| 394 | + async def get_group(self, group_id: str | None = None, **kwargs) -> Group | None: |
393 | 395 | """获取一个群聊的数据, 如果不填写 group_id: 如果是私聊消息,返回 None。如果是群聊消息,返回当前群聊的数据。 |
394 | 396 |
|
395 | 397 | 适配情况: |
|
0 commit comments