zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-15 13:14:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: return error when no valid speech text in intranscription response (#18277)

Browse Source
This commit is contained in:
Lynn
2026-08-14 16:27:34 +08:00
committed by GitHub
parent e1330423ac
commit 471070c2c8
3 changed files with 17 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/apps/restful_apis/chat_api.py
View File

@@ -1151,6 +1151,8 @@ async def transcription():
os.remove(temp_audio_path)
except Exception as e:
logging.error(f"Failed to remove temp audio file: {str(e)}")
if "**ERROR**" in text:
return get_data_error_result(message=text)
return get_json_result(data={"text": text})
async def event_stream():

16
rag/llm/sequence2txt_model.py
View File

@@ -230,7 +230,13 @@ class QWenSeq2txt(Base):
resp = dashscope.MultiModalConversation.call(model=self.model_name, messages=messages, result_format="message", asr_options={"enable_lid": True, "enable_itn": False})
try:
text = resp["output"]["choices"][0]["message"].content[0]["text"]
choices = resp["output"].get("choices") or []
if not choices:
raise ValueError("no valid speech content in response")
content = (choices[0].get("message") or {}).get("content") or []
if not content or "text" not in content[0]:
raise ValueError("no speech text in response")
text = content[0]["text"]
except Exception as e:
text = "**ERROR**: " + str(e)
return text, num_tokens_from_string(text)
@@ -369,7 +375,13 @@ class QWenSeq2txt(Base):
full = ""
for chunk in stream:
try:
piece = chunk["output"]["choices"][0]["message"].content[0]["text"]
choices = chunk["output"].get("choices") or []
if not choices:
raise ValueError("no valid speech content in response")
content = (choices[0].get("message") or {}).get("content") or []
if not content or "text" not in content[0]:
raise ValueError("no speech text in response")
piece = content[0]["text"]
full = piece
yield {"event": "delta", "text": piece}
except Exception as e:

2
test/unit_test/rag/llm/test_sequence2txt_model.py
View File

@@ -46,7 +46,7 @@ def test_fun_asr_flash_uses_native_request_format(tmp_path):
def test_qwen_audio_asr_keeps_existing_dashscope_path():
response = {"output": {"choices": [{"message": MagicMock(content=[{"text": "legacy text"}])}]}}
response = {"output": {"choices": [{"message": {"content": [{"text": "legacy text"}]}}]}}
with patch("dashscope.MultiModalConversation.call", return_value=response) as call:
model = QWenSeq2txt("test-key", "qwen-audio-asr")