Add support for max_reponse_output_tokens in options

src/realtime_ai/aio/realtime_ai_service_manager.py

@@ -174,7 +174,8 @@ async def update_session(self, options: RealtimeAIOptions) -> dict:
                 "turn_detection": options.turn_detection,
                 "tools": options.tools,
                 "tool_choice": options.tool_choice,
-                "temperature": options.temperature
+                "temperature": options.temperature,
+                "max_response_output_tokens": options.max_output_tokens
             }
         }
         await self.send_event(event)

src/realtime_ai/audio_stream_manager.py

@@ -32,12 +32,13 @@ def _start_stream(self):
                 logger.info("Audio streaming started.")
 
     def stop_stream(self):
-        if self._is_streaming:
-            self._is_streaming = False
-            self._stop_event.set()  # Signal to the thread to stop
-            if self._stream_thread:
-                self._stream_thread.join()
-            logger.info("Audio streaming stopped.")
+        with self._lock:
+            if self._is_streaming:
+                self._is_streaming = False
+                self._stop_event.set()  # Signal to the thread to stop
+                if self._stream_thread:
+                    self._stream_thread.join()
+                logger.info("Audio streaming stopped.")
 
     def write_audio_buffer_sync(self, audio_data: bytes):
         with self._lock:

src/realtime_ai/models/realtime_ai_options.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass, field
-from typing import List, Optional, Dict, Any
+from typing import List, Optional, Dict, Any, Union
 
 
 @dataclass
@@ -26,7 +26,7 @@ class RealtimeAIOptions:
     tools: List[Dict[str, Any]] = field(default_factory=list)
     tool_choice: str = "auto"
     temperature: float = 0.8
-    max_output_tokens: Optional[int] = None
+    max_output_tokens: Union[int, str] = "inf"
     enable_auto_reconnect: bool = False
 
     def __post_init__(self):

src/realtime_ai/realtime_ai_service_manager.py

@@ -175,7 +175,8 @@ def update_session(self, options: RealtimeAIOptions) -> dict:
                 "turn_detection": options.turn_detection,
                 "tools": options.tools,
                 "tool_choice": options.tool_choice,
-                "temperature": options.temperature
+                "temperature": options.temperature,
+                "max_response_output_tokens": options.max_output_tokens
             }
         }
         self.send_event(event)