Fresh install of agent studio and RivaASR not working #78

Using a Jetson Orin 64GB with a fresh install of agent studio.

RIVA is running as a separate container on the system and is listening on the correct port (riva_quickstart_arm64_v2.19.0 from ngc)

When trying to use the Riva ASR plugin i get the following output:

18:09:34 | ERROR | Exception occurred handling websocket message:

{ 'add_plugin': { 'asr_threshold': -2.5,
'audio_chunk': 0.1,
'automatic_punctuation': 'true',
'boosted_score': 4,
'inverse_text_normalization': 'false',
'language_code': 'en-US',
'layout_node': {'x': 10, 'y': 10},
'name': 'RivaASR',
'profanity_filter': 'false',
'riva_server': 'localhost:50051',
'sample_rate_hz': 16000,
'type': 'RivaASR'}}
Traceback (most recent call last):
File "/opt/NanoLLM/nano_llm/web/server.py", line 193, in on_message
callback(payload, payload_size=payload_size, msg_type=msg_type, msg_id=msg_id,
File "/opt/NanoLLM/nano_llm/agents/dynamic_agent.py", line 442, in on_websocket
on_message(self, message)
File "/opt/NanoLLM/nano_llm/agents/dynamic_agent.py", line 432, in on_message
if invoke_handler(obj, key, msg):
File "/opt/NanoLLM/nano_llm/agents/dynamic_agent.py", line 414, in invoke_handler
response = func(**msg)
File "/opt/NanoLLM/nano_llm/agents/dynamic_agent.py", line 58, in add_plugin
threading.Thread(target=self.add_plugin, kwargs={'type': type, 'wait': True, 'state_dict': state_dict, 'layout_node': layout_node, **kwargs}).run()
File "/usr/lib/python3.10/threading.py", line 953, in run
self._target(*self._args, **self._kwargs)
File "/opt/NanoLLM/nano_llm/agents/dynamic_agent.py", line 87, in add_plugin
self.webserver.send_message({'plugin_added': [plugin.state_dict(config=True)]})
TypeError: RivaASR.state_dict() got an unexpected keyword argument 'config'