修复依赖注入问题?

This commit is contained in:
Asankilp 2025-02-26 00:24:44 +08:00
parent 4fbe6c6366
commit f5ea844156
4 changed files with 12 additions and 9 deletions

View File

@ -133,7 +133,7 @@ class MarshoHandler:
# except:
# pass
tool_msg.append(choice.message)
for tool_call in tool_calls:
for tool_call in tool_calls: # type: ignore
try:
function_args = json.loads(tool_call.function.arguments)
except json.JSONDecodeError:
@ -160,6 +160,7 @@ class MarshoHandler:
bot=self.bot,
event=self.event,
matcher=self.matcher,
state=None,
)
).call(**function_args)
else:
@ -184,7 +185,7 @@ class MarshoHandler:
tools_list: list,
stream: bool = False,
tool_message: Optional[list] = None,
) -> Union[Tuple[UserMessage, ChatCompletionMessage], None]:
) -> Optional[Tuple[UserMessage, ChatCompletionMessage]]:
"""
处理一般聊天
"""

View File

@ -264,7 +264,7 @@ async def marsho(
context.append(context_user.as_dict(), target.id, target.private)
context.append(context_assistant.to_dict(), target.id, target.private)
else:
await UniMessage("没有回答").send()
return
except Exception as e:
await UniMessage(str(e) + suggest_solution(str(e))).send()
traceback.print_exc()

View File

@ -72,7 +72,9 @@ class Caller:
# if self.ctx.state is None:
# return False, "State is None"
if self._rule and not await self._rule(self.ctx.bot, self.ctx.event):
if self._rule and not await self._rule(
self.ctx.bot, self.ctx.event, self.ctx.state
):
return False, "告诉用户 Rule Denied 规则不匹配"
return True, ""
@ -113,9 +115,9 @@ class Caller:
# 检查函数签名,确定依赖注入参数
sig = inspect.signature(func)
for name, param in sig.parameters.items():
if param.annotation == T_State:
self.di.state = name
continue # 防止后续判断T_State子类时报错
# if param.annotation == T_State:
# self.di.state = name
# continue # 防止后续判断T_State子类时报错
if issubclass(param.annotation, Event) or isinstance(
param.annotation, Event

View File

@ -19,7 +19,7 @@ class SessionContext(BaseModel):
bot: Bot
event: Event
matcher: Matcher
# state: T_State
state: T_State | None
caller: Any = None
class Config:
@ -30,5 +30,5 @@ class SessionContextDepends(BaseModel):
bot: str | None = None
event: str | None = None
matcher: str | None = None
# state: str | None = None
state: str | None = None
caller: str | None = None