testing and linting

Author: tschellenbach

agents-core/vision_agents/core/deepseek.py

@@ -509,7 +509,7 @@ async def _run_handler(self, handler, event):
     async def _process_single_event(self, event):
         """Process a single event."""
         for handler in self._handlers.get(event.type, []):
-            module_name = getattr(handler, '__module__', 'unknown')
+            #module_name = getattr(handler, '__module__', 'unknown')
             if event.type not in self._silent_events:
                 pass
                 #logger.info(f"Called handler {handler.__name__} from {module_name} for event {event.type}")

agents-core/vision_agents/core/events/manager.py

@@ -12,7 +12,6 @@
 import logging
 import uuid
 
-from vision_agents.core.events import PluginClosedEvent
 
 from . import events, LLM
 

agents-core/vision_agents/core/llm/realtime.py

@@ -5,7 +5,6 @@
 import uuid
 from typing import Optional, List, Dict, Any
 from getstream.video.rtc.audio_track import AudioStreamTrack
-from hatch.cli import self
 
 from vision_agents.core.llm import realtime
 from aws_sdk_bedrock_runtime.client import BedrockRuntimeClient, InvokeModelWithBidirectionalStreamOperationInput
@@ -14,7 +13,6 @@
 from smithy_aws_core.identity.environment import EnvironmentCredentialsResolver
 
 from vision_agents.core.utils.video_forwarder import VideoForwarder
-from . import events
 from vision_agents.core.processors import Processor
 from vision_agents.core.edge.types import Participant
 from vision_agents.core.edge.types import PcmData
@@ -382,27 +380,25 @@ async def _handle_events(self):
                                             pass
                                 elif 'textOutput' in json_data['event']:
                                     text_content = json_data['event']['textOutput']['content']
-                                    role = json_data['event']['textOutput']['role']
+                                    #role = json_data['event']['textOutput']['role']
                                     logger.info(f"Text output from Bedrock: {text_content}")
                                 elif 'completionStart' in json_data['event']:
                                     logger.info("Completion start from Bedrock", json_data['event']['completionStart'])
                                 elif 'audioOutput' in json_data['event']:
                                     logger.info("Audio output from Bedrock")
-                                    try:
-                                        audio_content = json_data['event']['audioOutput']['content']
-                                        audio_bytes = base64.b64decode(audio_content)
-                                        #await self.audio_output_queue.put(audio_bytes)
-
-                                        audio_event = RealtimeAudioOutputEvent(
-                                            plugin_name="gemini",
-                                            audio_data=audio_bytes,
-                                            sample_rate=24000
-                                        )
-                                        self.events.send(audio_event)
-
-                                        await self.output_track.write(audio_bytes)
-                                    except Exception as e:
-                                        import pdb; pdb.set_trace()
+                                    audio_content = json_data['event']['audioOutput']['content']
+                                    audio_bytes = base64.b64decode(audio_content)
+                                    #await self.audio_output_queue.put(audio_bytes)
+
+                                    audio_event = RealtimeAudioOutputEvent(
+                                        plugin_name="gemini",
+                                        audio_data=audio_bytes,
+                                        sample_rate=24000
+                                    )
+                                    self.events.send(audio_event)
+
+                                    await self.output_track.write(audio_bytes)
+
 
                                 elif 'toolUse' in json_data['event']:
                                     logger.info(f"Tool use from Bedrock: {json_data['event']['toolUse']}")

plugins/bedrock/vision_agents/plugins/bedrock/bedrock_realtime.py

@@ -1,5 +1,6 @@
 import asyncio
 import logging
+from asyncio import CancelledError
 from typing import Optional, List, Dict, Any
 from getstream.video.rtc.audio_track import AudioStreamTrack
 from getstream.video.rtc.track_util import PcmData
@@ -241,6 +242,9 @@ async def _receive_loop(self):
                         await self._handle_tool_call(server_message.tool_call)
                     else:
                         self.logger.warning("Unrecognized event structure for gemini %s", server_message)
+        except CancelledError:
+            logger.error("Stop async iteration exception")
+            return
         except Exception as e:
             # reconnect here for some errors
             self.logger.error(f"_receive_loop error: {e}")