Update speech_graph.py
Some checks are pending
/ build (push) Waiting to run

This commit is contained in:
Marco Vinciguerra 2024-08-01 14:48:19 +02:00
parent 9ac74de622
commit 53dc06a2cb

View File

@ -111,7 +111,7 @@ class SpeechGraph(AbstractGraph):
Returns:
str: The answer to the prompt.
"""
inputs = {"user_prompt": self.prompt, self.input_key: self.source}
self.final_state, self.execution_info = self.graph.execute(inputs)
@ -122,4 +122,4 @@ class SpeechGraph(AbstractGraph):
"output_path", "output.mp3"))
print(f"Audio saved to {self.config.get('output_path', 'output.mp3')}")
return self.final_state.get("answer", "No answer found.")
return self.final_state.get("answer", "No answer found.")