add streamer_mode setting and write last wingman message to file

Author: Shackless

api/interface.py

@@ -776,7 +776,8 @@ class SettingsConfig(BaseModel):
     voice_activation: VoiceActivationSettings
     wingman_pro: WingmanProSettings
     xvasynth: XVASynthSettings
-    debug_mode: bool = False
+    debug_mode: bool
+    streamer_mode: bool
 
 
 class BenchmarkResult(BaseModel):

services/config_manager.py

@@ -508,6 +508,24 @@ def get_wingmen_configs(self, config_dir: ConfigDirInfo):
                     wingmen.append(wingman_file)
         return wingmen
 
+    def save_last_wingman_message(
+        self,
+        config_dir: ConfigDirInfo,
+        wingman_file: WingmanConfigFileInfo,
+        last_message: str,
+    ):
+        message_file = wingman_file.file.replace(".yaml", ".last-message.txt")
+        message_path = path.join(self.config_dir, config_dir.directory, message_file)
+        try:
+            with open(message_path, "w", encoding="utf-8") as file:
+                file.write(last_message)
+                return True
+        except:
+            self.printr.toast_error(
+                f"Unable to save last message for Wingman '{wingman_file.name}'."
+            )
+        return False
+
     def save_wingman_config(
         self,
         config_dir: ConfigDirInfo,

services/config_migration_service.py

@@ -365,7 +365,11 @@ def migrate_settings(old: dict, new: dict) -> dict:
             self.log("- added new fasterwhisper settings and config")
 
             old["voice_activation"]["stt_provider"] = "fasterwhisper"
-            self.log("- set FasterWhisper as new default VA STT provider.")
+            self.log("- set FasterWhisper as new default VA STT provider")
+
+            old["streamer_mode"] = False
+            self.log("- added new property streamer_mode")
+
             return old
 
         def migrate_defaults(old: dict, new: dict) -> dict:

services/settings_service.py

@@ -134,6 +134,7 @@ async def save_settings(self, settings: SettingsConfig):
         # rest
         self.config_manager.settings_config.wingman_pro = settings.wingman_pro
         self.config_manager.settings_config.debug_mode = settings.debug_mode
+        self.config_manager.settings_config.streamer_mode = settings.streamer_mode
 
         # save the config file
         self.config_manager.save_settings_config()

services/tower.py

@@ -246,3 +246,17 @@ def save_wingman(self, wingman_name: str):
             source=LogSource.SYSTEM,
         )
         return False
+
+    def save_last_message(self, wingman_name: str, last_message: str):
+        for wingman in self.wingmen:
+            if wingman.name == wingman_name:
+                for wingman_file in self.config_manager.get_wingmen_configs(
+                    self.config_dir
+                ):
+                    if wingman_file.name == wingman_name:
+                        return self.config_manager.save_last_wingman_message(
+                            config_dir=self.config_dir,
+                            wingman_file=wingman_file,
+                            last_message=last_message,
+                        )
+        return False

templates/configs/settings.yaml

@@ -1,5 +1,6 @@
 debug_mode: false
 audio: {}
+streamer_mode: false
 voice_activation:
   enabled: false
   mute_toggle_key: "shift+x"

wingmen/open_ai_wingman.py

@@ -66,17 +66,17 @@ def __init__(self, *args, **kwargs):
 
         # validate will set these:
         self.openai: OpenAi | None = None
-        self.mistral: OpenAi | None  = None
-        self.groq: OpenAi | None  = None
-        self.cerebras: OpenAi | None  = None
-        self.openrouter: OpenAi | None  = None
+        self.mistral: OpenAi | None = None
+        self.groq: OpenAi | None = None
+        self.cerebras: OpenAi | None = None
+        self.openrouter: OpenAi | None = None
         self.openrouter_model_supports_tools = False
-        self.local_llm: OpenAi | None  = None
-        self.openai_azure: OpenAiAzure | None  = None
-        self.elevenlabs: ElevenLabs | None  = None
-        self.wingman_pro: WingmanPro | None  = None
-        self.google: GoogleGenAI | None  = None
-        self.perplexity: OpenAi | None  = None
+        self.local_llm: OpenAi | None = None
+        self.openai_azure: OpenAiAzure | None = None
+        self.elevenlabs: ElevenLabs | None = None
+        self.wingman_pro: WingmanPro | None = None
+        self.google: GoogleGenAI | None = None
+        self.perplexity: OpenAi | None = None
 
         # tool queue
         self.pending_tool_calls = []
@@ -556,14 +556,16 @@ async def _transcribe(self, audio_input_wav: str) -> str | None:
             )
             printr.print(traceback.format_exc(), color=LogType.ERROR, server_only=True)
 
-        if not transcript:
-            return None
-
-        # Wingman Pro might returns a serialized dict instead of a real Azure Speech transcription object
-        if isinstance(transcript, dict):
-            return transcript.get("_text")
+        result = None
+        if transcript:
+            # Wingman Pro might returns a serialized dict instead of a real Azure Speech transcription object
+            result = (
+                transcript.get("_text")
+                if isinstance(transcript, dict)
+                else transcript.text
+            )
 
-        return transcript.text
+        return result
 
     async def _get_response_for_transcript(
         self, transcript: str, benchmark: Benchmark
@@ -588,7 +590,9 @@ async def _get_response_for_transcript(
         if instant_response:
             await self.add_assistant_message(instant_response)
             benchmark.finish_snapshot()
-            if instant_response == '.': # thats for the "The UI should not give a response" option in commands
+            if (
+                instant_response == "."
+            ):  # thats for the "The UI should not give a response" option in commands
                 instant_response = None
             return instant_response, instant_response, None, True
         benchmark.finish_snapshot()
@@ -896,7 +900,8 @@ async def add_forced_assistant_command_calls(self, commands: list[CommandConfig]
         if (
             self.config.features.conversation_provider == ConversationProvider.OPENAI
         ) or (
-            self.config.features.conversation_provider == ConversationProvider.WINGMAN_PRO
+            self.config.features.conversation_provider
+            == ConversationProvider.WINGMAN_PRO
             and "gpt" in self.config.wingman_pro.conversation_deployment.lower()
         ):
             # generate tool calls in openai style
@@ -1312,7 +1317,7 @@ async def execute_command_by_function_call(
                     traceback.format_exc(), color=LogType.ERROR, server_only=True
                 )
                 function_response = (
-                    "ERROR DURING PROCESSING" # hints to AI that there was an error
+                    "ERROR DURING PROCESSING"  # hints to AI that there was an error
                 )
                 instant_response = None
             finally:

wingmen/wingman.py

@@ -329,8 +329,11 @@ async def process(self, audio_input_wav: str = None, transcript: str = None):
                         benchmark_result=benchmark_llm.finish(),
                     )
 
-            # the last step in the chain. You'll probably want to play the response to the user as audio using a TTS provider or mechanism of your choice.
             if process_result:
+                if self.settings.streamer_mode:
+                    self.tower.save_last_message(self.name, process_result)
+
+                # the last step in the chain. You'll probably want to play the response to the user as audio using a TTS provider or mechanism of your choice.
                 await self.play_to_user(str(process_result), not interrupt)
         except Exception as e:
             await printr.print_async(