Skip to content

Commit 1656fda

Browse files
author
Davidson Gomes
committed
refactor(a2a_task_manager): enhance logging and file handling in streaming task processing
1 parent 6bf0ea5 commit 1656fda

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

src/services/a2a_task_manager.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -388,6 +388,7 @@ async def _stream_task_process(
388388
self, request: SendTaskStreamingRequest, agent: Agent
389389
) -> AsyncIterable[SendTaskStreamingResponse]:
390390
"""Processes a task in streaming mode using the specified agent."""
391+
# Extrair e processar arquivos da mesma forma que no método _process_task
391392
query = self._extract_user_query(request.params)
392393

393394
try:
@@ -447,6 +448,23 @@ async def _stream_task_process(
447448
),
448449
)
449450

451+
# Use os arquivos processados do _extract_user_query
452+
files = getattr(self, "_last_processed_files", None)
453+
454+
# Log sobre os arquivos processados
455+
if files:
456+
logger.info(
457+
f"Streaming: Passando {len(files)} arquivos processados para run_agent_stream"
458+
)
459+
for file_info in files:
460+
logger.info(
461+
f"Streaming: Arquivo sendo enviado: {file_info.filename} ({file_info.content_type})"
462+
)
463+
else:
464+
logger.warning(
465+
"Streaming: Nenhum arquivo processado disponível para enviar ao agente"
466+
)
467+
450468
async for chunk in run_agent_stream(
451469
agent_id=str(agent.id),
452470
external_id=external_id,
@@ -455,6 +473,7 @@ async def _stream_task_process(
455473
artifacts_service=artifacts_service,
456474
memory_service=memory_service,
457475
db=self.db,
476+
files=files, # Passar os arquivos processados para o streaming
458477
):
459478
try:
460479
chunk_data = json.loads(chunk)

0 commit comments

Comments
 (0)