Dont print exceptions for cancelled errors (#119)

This commit is contained in:
Jack Gerrits 2024-06-24 19:54:19 -04:00 committed by GitHub
parent a73a0330e5
commit bb4bb5bd14

View File

@ -2,7 +2,7 @@ import asyncio
import inspect import inspect
import logging import logging
import threading import threading
from asyncio import Future from asyncio import CancelledError, Future
from collections import defaultdict from collections import defaultdict
from collections.abc import Sequence from collections.abc import Sequence
from dataclasses import dataclass from dataclasses import dataclass
@ -281,11 +281,13 @@ class SingleThreadedAgentRuntime(AgentRuntime):
try: try:
_all_responses = await asyncio.gather(*responses) _all_responses = await asyncio.gather(*responses)
except BaseException: except BaseException as e:
# Ignore cancelled errors from logs
if isinstance(e, CancelledError):
return
logger.error("Error processing publish message", exc_info=True) logger.error("Error processing publish message", exc_info=True)
return finally:
self._outstanding_tasks.decrement()
self._outstanding_tasks.decrement()
# TODO if responses are given for a publish # TODO if responses are given for a publish
async def _process_response(self, message_envelope: ResponseMessageEnvelope) -> None: async def _process_response(self, message_envelope: ResponseMessageEnvelope) -> None: