BtoB_team/b2b_assistants
5
0
Fork 0
Code Issues 3 Pull requests Projects Releases Packages Wiki Activity Actions

Do quality assessment

Browse source
This commit is contained in:
Слонова Анна 2026-04-01 01:35:03 +03:00
parent 806e85fcec
commit 4f360d35b2
3 changed files with 21 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
main.py
View file

@ -44,6 +44,10 @@ QWEN_PROMPT_TEMPLATE = ""
with open("base_prompt.txt", "r") as f:
QWEN_PROMPT_TEMPLATE += f.read()
QWEN_QUALITY_PROMPT_TEMPLATE = ""
with open("quality_prompt.txt", "r") as f:
QWEN_QUALITY_PROMPT_TEMPLATE += f.read()
TEMP_DIR = tempfile.gettempdir()
GROUPING_TIMEOUT = 15.0
@ -211,6 +215,11 @@ async def generate_report(text: str, images_data: list, audios_data: list) -> Op
if report.startswith("Ошибка:"):
print(f"[QWEN] Ошибка API: {report}")
return None
quality_prompt = f"{QWEN_QUALITY_PROMPT_TEMPLATE}\n Исходные данные: {full_text} Отчёт: {report}"
quality_report = await call_qwen_api(quality_prompt)
print(f"[QWEN] Оценка качества: {quality_report}")
report = report.replace('```html', '')
report = report.replace('```', '')
@ -238,6 +247,7 @@ async def process_audio(audio_data: Dict) -> str:
if text is None:
print("[AUDIO] Распознавание не удалось.")
return ""
print(f"[AUDIO] Распознанный текст: {text[:300]}...")
return text