Skip to content

Commit c05acf0

Browse files
committed
style(tests): format AgentRequest initialization and list comprehensions
refactor test_invoker.py to improve code readability by formatting AgentRequest parameter initialization with proper line breaks and indentation, and reformatting list comprehensions for better readability. 改进了 AgentRequest 初始化和列表推导式的格式化,提高了代码可读性 Change-Id: I2ed9ef8cf8396105df3c17f9327379e7a2b2b925 Signed-off-by: OhYee <oyohyee@oyohyee.com>
1 parent 77b76dd commit c05acf0

File tree

1 file changed

+45
-11
lines changed

1 file changed

+45
-11
lines changed

tests/unittests/server/test_invoker.py

Lines changed: 45 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,11 @@ class TestInvokerBasic:
2020
@pytest.fixture
2121
def req(self):
2222
return AgentRequest(
23-
messages=[], tools=[], stream=False, raw_request=None, protocol="unknown"
23+
messages=[],
24+
tools=[],
25+
stream=False,
26+
raw_request=None,
27+
protocol="unknown",
2428
)
2529

2630
@pytest.mark.asyncio
@@ -45,7 +49,9 @@ async def invoke_agent(req: AgentRequest) -> AsyncGenerator[str, None]:
4549
# 应该有 2 个 TEXT 事件(不再有边界事件)
4650
assert len(items) == 2
4751

48-
content_events = [item for item in items if item.event == EventType.TEXT]
52+
content_events = [
53+
item for item in items if item.event == EventType.TEXT
54+
]
4955
assert len(content_events) == 2
5056
assert content_events[0].data["delta"] == "hello"
5157
assert content_events[1].data["delta"] == " world"
@@ -99,7 +105,11 @@ class TestInvokerStream:
99105
@pytest.fixture
100106
def req(self):
101107
return AgentRequest(
102-
messages=[], tools=[], stream=False, raw_request=None, protocol="unknown"
108+
messages=[],
109+
tools=[],
110+
stream=False,
111+
raw_request=None,
112+
protocol="unknown",
103113
)
104114

105115
@pytest.mark.asyncio
@@ -172,7 +182,9 @@ async def invoke_agent(req: AgentRequest) -> str:
172182
assert EventType.ERROR in event_types
173183

174184
# 检查错误信息
175-
error_event = next(item for item in items if item.event == EventType.ERROR)
185+
error_event = next(
186+
item for item in items if item.event == EventType.ERROR
187+
)
176188
assert "Test error" in error_event.data["message"]
177189
assert error_event.data["code"] == "ValueError"
178190

@@ -183,7 +195,11 @@ class TestInvokerSync:
183195
@pytest.fixture
184196
def req(self):
185197
return AgentRequest(
186-
messages=[], tools=[], stream=False, raw_request=None, protocol="unknown"
198+
messages=[],
199+
tools=[],
200+
stream=False,
201+
raw_request=None,
202+
protocol="unknown",
187203
)
188204

189205
@pytest.mark.asyncio
@@ -204,7 +220,9 @@ def invoke_agent(req: AgentRequest):
204220
async for item in result:
205221
items.append(item)
206222

207-
content_events = [item for item in items if item.event == EventType.TEXT]
223+
content_events = [
224+
item for item in items if item.event == EventType.TEXT
225+
]
208226
assert len(content_events) == 2
209227

210228
@pytest.mark.asyncio
@@ -232,7 +250,11 @@ class TestInvokerMixed:
232250
@pytest.fixture
233251
def req(self):
234252
return AgentRequest(
235-
messages=[], tools=[], stream=False, raw_request=None, protocol="unknown"
253+
messages=[],
254+
tools=[],
255+
stream=False,
256+
raw_request=None,
257+
protocol="unknown",
236258
)
237259

238260
@pytest.mark.asyncio
@@ -288,7 +310,9 @@ async def invoke_agent(req: AgentRequest):
288310
async for item in invoker.invoke_stream(req):
289311
items.append(item)
290312

291-
content_events = [item for item in items if item.event == EventType.TEXT]
313+
content_events = [
314+
item for item in items if item.event == EventType.TEXT
315+
]
292316
# 只有两个非空字符串
293317
assert len(content_events) == 2
294318
assert content_events[0].data["delta"] == "hello"
@@ -301,7 +325,11 @@ class TestInvokerNone:
301325
@pytest.fixture
302326
def req(self):
303327
return AgentRequest(
304-
messages=[], tools=[], stream=False, raw_request=None, protocol="unknown"
328+
messages=[],
329+
tools=[],
330+
stream=False,
331+
raw_request=None,
332+
protocol="unknown",
305333
)
306334

307335
@pytest.mark.asyncio
@@ -333,7 +361,9 @@ async def invoke_agent(req: AgentRequest):
333361
async for item in invoker.invoke_stream(req):
334362
items.append(item)
335363

336-
content_events = [item for item in items if item.event == EventType.TEXT]
364+
content_events = [
365+
item for item in items if item.event == EventType.TEXT
366+
]
337367
assert len(content_events) == 2
338368

339369

@@ -343,7 +373,11 @@ class TestInvokerToolCall:
343373
@pytest.fixture
344374
def req(self):
345375
return AgentRequest(
346-
messages=[], tools=[], stream=False, raw_request=None, protocol="unknown"
376+
messages=[],
377+
tools=[],
378+
stream=False,
379+
raw_request=None,
380+
protocol="unknown",
347381
)
348382

349383
@pytest.mark.asyncio

0 commit comments

Comments
 (0)