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

Fix: xinference asr (#18110)

Browse Source
This commit is contained in:
Lynn
2026-08-11 19:07:50 +08:00
committed by GitHub
parent fbcb8656ca
commit cd6996b301
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
rag/llm/sequence2txt_model.py
View File

@@ -30,7 +30,7 @@ from openai import OpenAI
from openai.lib.azure import AzureOpenAI
from common.token_utils import num_tokens_from_string
from rag.utils.url_utils import ensure_v1
from rag.utils.url_utils import append_api_path, ensure_v1
logger = logging.getLogger(__name__)
@@ -396,7 +396,7 @@ class XinferenceSeq2txt(Base):
self.model_name = model_name
self.key = key
def transcription(self, audio, language="zh", prompt=None, response_format="json"):
def transcription(self, audio, language="Chinese", prompt=None, response_format="json"):
if isinstance(audio, str):
with open(audio, "rb") as audio_file:
audio_data = audio_file.read()
@@ -410,7 +410,8 @@ class XinferenceSeq2txt(Base):
files = {"file": (audio_file_name, audio_data, "audio/wav")}
try:
response = requests.post(f"{self.base_url}/v1/audio/transcriptions", files=files, data=payload, timeout=60)
headers = {"Authorization": f"Bearer {self.key}"} if self.key else {}
response = requests.post(append_api_path(self.base_url, "audio/transcriptions"), files=files, data=payload, headers=headers, timeout=60)
response.raise_for_status()
result = response.json()