From 72bde7b5aa67c7031387a0af059d0fd81695736a Mon Sep 17 00:00:00 2001 From: Maciej Majek Date: Fri, 7 Mar 2025 22:42:39 +0100 Subject: [PATCH] refactor: use try except block in lg callback --- src/rai_core/rai/agents/langchain/callback.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/rai_core/rai/agents/langchain/callback.py b/src/rai_core/rai/agents/langchain/callback.py index 0fa6d6a5..3ee399f9 100644 --- a/src/rai_core/rai/agents/langchain/callback.py +++ b/src/rai_core/rai/agents/langchain/callback.py @@ -22,8 +22,6 @@ from rai.communication.hri_connector import HRIConnector, HRIMessage -logger = logging.getLogger(__name__) - class HRICallbackHandler(BaseCallbackHandler): def __init__( @@ -39,14 +37,18 @@ def __init__( self.chunks_buffer = "" self.max_buffer_size = max_buffer_size self._buffer_lock = threading.Lock() + self.logger = logging.getLogger(__name__) def _should_split(self, token: str) -> bool: return token in self.splitting_chars def _send_all_targets(self, token: str): for connector_name, connector in self.connectors.items(): - connector.send_all_targets(AIMessage(content=token)) - logger.debug(f"Sent token to {connector_name}") + try: + connector.send_all_targets(AIMessage(content=token)) + self.logger.debug(f"Sent token to {connector_name}") + except Exception as e: + self.logger.error(f"Failed to send token to {connector_name}: {e}") def on_llm_new_token(self, token: str, **kwargs): if token == "":