@@ -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