77 lines
2.8 KiB
Python
77 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
Модуль суммаризации документа.
|
||
Создаёт краткий пересказ, сохраняя ключевые факты и цифры.
|
||
|
||
АДАПТАЦИЯ: теперь функция принимает промпт как аргумент.
|
||
"""
|
||
|
||
import logging
|
||
from core.services.giga_client import GigaClient
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
async def summarize_document(
|
||
giga: GigaClient,
|
||
text: str,
|
||
title: str,
|
||
prompt_text: str,
|
||
bot_config,
|
||
) -> str:
|
||
"""
|
||
Суммаризирует документ.
|
||
|
||
Аргументы:
|
||
giga (GigaClient): клиент GigaChat
|
||
text (str): полный текст документа
|
||
title (str): название документа
|
||
prompt_text (str): содержимое промпта суммаризации
|
||
bot_config (BotConfig): объект конфигурации
|
||
|
||
Возвращает:
|
||
str: краткое изложение документа или сообщение об ошибке
|
||
"""
|
||
if not prompt_text:
|
||
logger.error("Промпт суммаризации не загружен")
|
||
return "Не удалось загрузить промпт для суммаризации."
|
||
|
||
summary_cfg = getattr(bot_config, 'summary', {})
|
||
temperature = summary_cfg.get('temperature', 0.1)
|
||
max_chars = summary_cfg.get('max_chars', 8000)
|
||
|
||
# Обрезаем текст, если он слишком длинный
|
||
if len(text) > max_chars:
|
||
# Обрезаем по границам предложений (используем razdel)
|
||
try:
|
||
from razdel import sentenize
|
||
sentences = [sent.text for sent in sentenize(text)]
|
||
except ImportError:
|
||
import re
|
||
sentences = re.split(r'(?<=[.!?])\s+', text)
|
||
|
||
truncated = []
|
||
current_len = 0
|
||
for sent in sentences:
|
||
if current_len + len(sent) <= max_chars:
|
||
truncated.append(sent)
|
||
current_len += len(sent)
|
||
else:
|
||
break
|
||
text = ' '.join(truncated)
|
||
logger.warning(f"Текст обрезан до {max_chars} символов по границам предложений")
|
||
|
||
full_prompt = f"{prompt_text}\n\nНазвание документа: {title}\nТекст:\n{text}"
|
||
|
||
try:
|
||
response = await giga.chat(
|
||
history=[],
|
||
query=full_prompt,
|
||
system_prompt=None,
|
||
file_id=None,
|
||
temperature=temperature
|
||
)
|
||
return response.strip()
|
||
except Exception as e:
|
||
logger.error(f"Ошибка суммаризации: {e}", exc_info=True)
|
||
return "Не удалось создать суммаризацию." |