Welcome to RDK S600 Developer Kit repository
3. AI large model voice interaction
3. AI Large Model Voice Interaction
3. AI Large Model Voice Interaction1. Concept Introduction1.1 What is "AI Large Model Voice Interaction"?1.2 Brief Description of Implementation Principles2. Code AnalysisKey Code1. Voice Input Node (largemodel/largemodel/asr.py)2. AI Service and Voice Output Node (largemodel/largemodel/model_service.py)Code Analysis3. Practical Operations3.1 Configuring Offline Voice Interaction3.2 Startup and Testing
1. Concept Introduction
1.1 What is "AI Large Model Voice Interaction"?
In the largemodel project, AI Large Model Voice Interaction connects the previously introduced Offline ASR and Offline TTS with the Large Language Model (LLM) core, forming a complete conversational system that can listen, speak, and think.
This is no longer an isolated function, but the prototype of a true voice assistant. Users can have natural language conversations with the robot through voice, and the robot can understand questions, think about answers, and respond with voice. The entire process is completed locally without requiring network access.
The core of this function is the model_service ROS2 node. It acts as the brain and neural center, subscribing to ASR recognition results, invoking the LLM for thinking, and then publishing the LLM's text replies to the TTS node for speech synthesis.
1.2 Brief Description of Implementation Principles
The implementation principle of this function is a classic data pipeline:
- Voice -> Text (ASR): The
asrnode continuously listens to ambient sounds. Once it detects that the user has finished speaking, it converts the audio to text and publishes it to the/asr_texttopic. - Text -> Thinking -> Text (LLM): The
model_servicenode subscribes to the/asr_texttopic. When it receives text from ASR, it sends it as a prompt to the locally deployed large language model (such asQwenrunning throughOllama). The LLM generates a response text based on the context. - Text -> Voice (TTS): After the
model_servicenode gets the LLM's response, it publishes it to the/tts_texttopic. - Voice Playback: The
tts_onlynode subscribes to the/tts_texttopic. Upon receiving the text, it immediately invokes the offline TTS model to synthesize it into audio and plays it through the speaker.
This process forms a complete closed loop of Voice Input -> Text Processing -> Text Output -> Voice Output.
2. Code Analysis
Key Code
1. Voice Input Node (largemodel/largemodel/asr.py)
x# From largemodel/largemodel/asr.pyclass ASRNode(Node): def __init__(self): # ... self.asr_pub = self.create_publisher(String, "asr", 5) # ... def kws_handler(self)->None: if self.listen_for_speech(self.mic_index): asr_text = self.ASR_conversion(self.user_speechdir) if asr_text != 'error': self.asr_pub_result(asr_text) def asr_pub_result(self,asr_result:str)->None: msg=String(data=asr_result) self.asr_pub.publish(msg)2. AI Service and Voice Output Node (largemodel/largemodel/model_service.py)
xxxxxxxxxx# From largemodel/largemodel/model_service.pyclass LargeModelService(Node): def __init__(self): # ... self.asrsub = self.create_subscription(String,'asr', self.asr_callback,1) # ... def asr_callback(self,msg): """Callback function for handling ASR messages.""" # ... result = self.model_client.infer_with_text(msg.data, message=messages_to_use) self.process_model_result(result) def process_model_result(self, result, from_seewhat=False): """Process the result returned by the model.""" # ... user_friendly_response = response_json.get("response", "我正在处理...") # ... self._safe_play_audio(user_friendly_response) # ... self.execute_tools(tools_list) def _safe_play_audio(self, text_to_speak: str): """ Synthesizes and plays all non-empty messages only in non-text chat mode. """ if not self.text_chat_mode and text_to_speak: try: self.model_client.voice_synthesis(text_to_speak, self.tts_out_path) self.play_audio_async(self.tts_out_path) except Exception as e: self.get_logger().error(f"Safe audio playback failed: {e}")Code Analysis
The voice interaction functionality of the AI large model is collaboratively completed by two independent ROS nodes, asr.py and model_service.py, which communicate through the ROS topic /asr to form a complete processing loop.
Voice Input and Publishing (
asr.py):- The
ASRNodenode is responsible for processing voice input. In thekws_handlerfunction, it completes recording and text conversion. - After successful conversion, the
asr_pub_resultfunction is called. This function places the obtained text string into astd_msgs.msg.Stringmessage and publishes it to the/asrtopic through theself.asr_pubpublisher.
- The
Subscription and Response (
model_service.py):- During initialization, the
LargeModelServicenode subscribes to the/asrtopic through thecreate_subscriptionmethod, specifyingasr_callbackas its callback function. - When
ASRNodepublishes a message on the/asrtopic, the ROS system automatically callsLargeModelService'sasr_callbackfunction, passing the message as the parametermsg.
- During initialization, the
Interaction Loop:
- After the
asr_callbackfunction receives the message, it extracts the text data (msg.data) and passes it to the large model for inference (self.model_client.infer_with_text). - After inference is complete, the
process_model_resultfunction is called. This function extracts the response text from the model results. - Then, the
_safe_play_audiofunction is called, which uses TTS functionality (self.model_client.voice_synthesis) to synthesize the response text into audio and play it. - At this point, a complete interaction loop from "receiving user voice" to "responding to user with voice" is completed. The two nodes are decoupled through ROS's publish-subscribe pattern, with
asr.pyhandling input andmodel_service.pyhandling processing and output.
- After the
3. Practical Operations
3.1 Configuring Offline Voice Interaction
To implement a completely offline voice interaction system, you need to ensure that ASR, TTS, and LLM are all configured for offline mode.
Open the main configuration file
yahboom.yaml:xxxxxxxxxxgedit ~/yahboom_ws/src/largemodel/config/yahboom.yamlModify/confirm the following key configurations:
xxxxxxxxxxasrros__parametersuse_oline_asrFalse # Critical: Set to False to enable offline ASRmodel_serviceros__parametersuseolinettsFalse # Critical: Set to False to enable offline TTSllm_platform'ollama' # Critical: Set to 'ollama' to enable offline LLMOpen the model interface configuration file
large_model_interface.yaml:xxxxxxxxxxgedit ~/yahboom_ws/src/largemodel/config/large_model_interface.yamlConfirm all offline model paths are correct:
xxxxxxxxxx# large_model_interface.yaml## Offline Large Modelollama_model"qwen2.5:7b" # Ensure this model has been downloaded via ollama pull## Offline Speech Recognitionlocal_asr_model"/path/to/your/SenseVoiceSmall" # Ensure ASR model path is correct## Offline Speech Synthesiszh_tts_model"/path/to/your/zh_CN-huayan-medium.onnx" # Ensure TTS model path is correct# ...
3.2 Startup and Testing
Start the
largemodelmain program: Run the following command to start voice interaction:xxxxxxxxxxros2 launch largemodel largemodel_control.launch.pyTest:
Wake up: Say to the microphone: "Hello, yahboom."
Conversation: After the speaker responds, you can speak your question.
Observe logs: In the terminal running the
launchfile, you should see:- The ASR node recognizes your question and prints it out.
- The
model_servicenode receives the text, invokes the LLM, and prints out the LLM's response.
Listen to the answer: After a while, you should hear the robot answer your question in synthesized voice from the speaker.
Possible issue: Hello yahboom can wake up but audio cannot be captured
Solution:
Confirm USB microphone is recognized by the system
xxxxxxxxxxarecord -lCheck if "USB Audio Device" appears in the output
Confirm mic_index configuration
xxxxxxxxxxgrep mic_index ~/yahboom_ws/install/largemodel/share/largemodel/config/yahboom.yamlRecommended to use 0 (uses PulseAudio default source, automatic resampling)
Confirm PulseAudio default source is USB microphone
xxxxxxxxxxpactl get-default-sourceIf not, first check the source name:
xxxxxxxxxxpactl list short sources | grep usbThen set the default source:
xxxxxxxxxxpactl set-default-source <source_name>Start testing
xxxxxxxxxxros2 launch largemodel largemodel_control.launch.py