mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-17 17:51:20 +08:00
6070da9ae2
* fix: resume ChatUI streams after refresh * fix: hide chat run service errors * fix: harden resumed chat streams
75 lines
2.0 KiB
Python
75 lines
2.0 KiB
Python
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from astrbot.core.agent.response import AgentResponse
|
|
from astrbot.core.astr_agent_run_util import run_agent
|
|
from astrbot.core.message.message_event_result import MessageChain
|
|
|
|
|
|
class _FakeEvent:
|
|
"""Minimal event surface used by the agent stream bridge."""
|
|
|
|
def is_stopped(self) -> bool:
|
|
return False
|
|
|
|
def get_extra(self, key: str):
|
|
del key
|
|
return None
|
|
|
|
def get_platform_name(self) -> str:
|
|
return "test"
|
|
|
|
|
|
class _StreamingErrorRunner:
|
|
"""Agent runner that finishes with one provider error response."""
|
|
|
|
streaming = True
|
|
req = None
|
|
|
|
def __init__(self, error_text: str) -> None:
|
|
self.error_text = error_text
|
|
self.finished = False
|
|
self.run_context = SimpleNamespace(context=SimpleNamespace(event=_FakeEvent()))
|
|
|
|
async def step(self):
|
|
self.finished = True
|
|
yield AgentResponse(
|
|
type="err",
|
|
data={"chain": MessageChain().message(self.error_text)},
|
|
)
|
|
|
|
def done(self) -> bool:
|
|
return self.finished
|
|
|
|
|
|
class _MalformedStreamingErrorRunner(_StreamingErrorRunner):
|
|
"""Agent runner that returns an invalid provider error payload."""
|
|
|
|
async def step(self):
|
|
self.finished = True
|
|
yield AgentResponse(type="err", data={})
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_agent_forwards_streaming_provider_error():
|
|
error_text = (
|
|
"LLM 响应错误: Not found the model k2.7-code-highspeed or Permission denied"
|
|
)
|
|
runner = _StreamingErrorRunner(error_text)
|
|
|
|
chains = [chain async for chain in run_agent(runner)]
|
|
|
|
assert len(chains) == 1
|
|
assert chains[0].get_plain_text() == error_text
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_agent_replaces_malformed_streaming_provider_error():
|
|
runner = _MalformedStreamingErrorRunner("unused")
|
|
|
|
chains = [chain async for chain in run_agent(runner)]
|
|
|
|
assert len(chains) == 1
|
|
assert chains[0].get_plain_text() == "Error occurred during AI execution."
|