Alina Lozovskaya commited on
Commit
434109e
·
1 Parent(s): 26ea1c7

Fix mypy errors

Browse files
src/reachy_mini_conversation_demo/openai_realtime.py CHANGED
@@ -238,14 +238,14 @@ class OpenaiRealtimeHandler(AsyncStreamHandler):
238
  await self.connection.close()
239
  self.connection = None
240
 
241
- def format_timestamp(self):
242
  """Format current timestamp with date, time, and elapsed seconds."""
243
  loop_time = asyncio.get_event_loop().time() # monotonic
244
  elapsed_seconds = loop_time - self.start_time
245
  dt = datetime.now() # wall-clock
246
  return f"[{dt.strftime('%Y-%m-%d %H:%M:%S')} | +{elapsed_seconds:.1f}s]"
247
 
248
- async def send_idle_signal(self, idle_duration) -> None:
249
  """Send an idle signal to the openai server."""
250
  logger.debug("Sending idle signal")
251
  self.is_idle_tool_call = True
 
238
  await self.connection.close()
239
  self.connection = None
240
 
241
+ def format_timestamp(self) -> str:
242
  """Format current timestamp with date, time, and elapsed seconds."""
243
  loop_time = asyncio.get_event_loop().time() # monotonic
244
  elapsed_seconds = loop_time - self.start_time
245
  dt = datetime.now() # wall-clock
246
  return f"[{dt.strftime('%Y-%m-%d %H:%M:%S')} | +{elapsed_seconds:.1f}s]"
247
 
248
+ async def send_idle_signal(self, idle_duration: float) -> None:
249
  """Send an idle signal to the openai server."""
250
  logger.debug("Sending idle signal")
251
  self.is_idle_tool_call = True
tests/test_openai_realtime.py CHANGED
@@ -1,18 +1,18 @@
1
  import asyncio
2
- from datetime import datetime
3
  from unittest.mock import MagicMock
4
 
5
  from reachy_mini_conversation_demo.tools import ToolDependencies
6
  from reachy_mini_conversation_demo.openai_realtime import OpenaiRealtimeHandler
7
 
8
 
9
- def _build_handler(loop):
10
  asyncio.set_event_loop(loop)
11
  deps = ToolDependencies(reachy_mini=MagicMock(), movement_manager=MagicMock())
12
  return OpenaiRealtimeHandler(deps)
13
 
14
 
15
- def test_format_timestamp_uses_wall_clock():
16
  """Test that format_timestamp uses wall clock time."""
17
  loop = asyncio.new_event_loop()
18
  try:
@@ -26,4 +26,4 @@ def test_format_timestamp_uses_wall_clock():
26
 
27
  # Extract year from "[YYYY-MM-DD ...]"
28
  year = int(formatted[1:5])
29
- assert year == datetime.utcnow().year
 
1
  import asyncio
2
+ from datetime import datetime, timezone
3
  from unittest.mock import MagicMock
4
 
5
  from reachy_mini_conversation_demo.tools import ToolDependencies
6
  from reachy_mini_conversation_demo.openai_realtime import OpenaiRealtimeHandler
7
 
8
 
9
+ def _build_handler(loop: asyncio.AbstractEventLoop) -> OpenaiRealtimeHandler:
10
  asyncio.set_event_loop(loop)
11
  deps = ToolDependencies(reachy_mini=MagicMock(), movement_manager=MagicMock())
12
  return OpenaiRealtimeHandler(deps)
13
 
14
 
15
+ def test_format_timestamp_uses_wall_clock() -> None:
16
  """Test that format_timestamp uses wall clock time."""
17
  loop = asyncio.new_event_loop()
18
  try:
 
26
 
27
  # Extract year from "[YYYY-MM-DD ...]"
28
  year = int(formatted[1:5])
29
+ assert year == datetime.now(timezone.utc).year