68 lines
2.5 KiB
Python
68 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Модуль самокритики: оценивает качество сгенерированного ответа.
|
|
Проверяет соответствие источнику, отсутствие галлюцинаций, соблюдение стиля и логику.
|
|
|
|
АДАПТАЦИЯ: теперь функция принимает промпт как аргумент.
|
|
"""
|
|
|
|
import logging
|
|
import re
|
|
from core.services.giga_client import GigaClient
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
async def critique_answer(
|
|
giga: GigaClient,
|
|
query: str,
|
|
context: str,
|
|
answer: str,
|
|
prompt_text: str,
|
|
bot_config,
|
|
) -> bool:
|
|
"""
|
|
Оценивает ответ, возвращает True, если ответ приемлем, иначе False.
|
|
|
|
Аргументы:
|
|
giga (GigaClient): клиент GigaChat
|
|
query (str): исходный вопрос пользователя
|
|
context (str): контекст из базы знаний
|
|
answer (str): сгенерированный ответ
|
|
prompt_text (str): содержимое промпта самокритики
|
|
bot_config (BotConfig): объект конфигурации
|
|
|
|
Возвращает:
|
|
bool: True если ответ хороший, False если есть замечания
|
|
"""
|
|
if not prompt_text:
|
|
logger.debug("Промпт самокритики не загружен, пропускаем проверку")
|
|
return True
|
|
|
|
crit_cfg = getattr(bot_config, 'critique', {})
|
|
temperature = crit_cfg.get('temperature', 0.1)
|
|
if isinstance(temperature, dict):
|
|
temperature = 0.1
|
|
|
|
full_prompt = f"{prompt_text}\n\nВопрос: {query}\nКонтекст:\n{context}\nОтвет:\n{answer}"
|
|
|
|
try:
|
|
response = await giga.chat(
|
|
history=[],
|
|
query=full_prompt,
|
|
system_prompt=None,
|
|
file_id=None,
|
|
temperature=temperature
|
|
)
|
|
verdict = response.strip()
|
|
|
|
if re.search(r'\[OK\]|ПРИЕМЛЕМ', verdict, re.IGNORECASE):
|
|
return True
|
|
elif re.search(r'\[ISSUES\]', verdict, re.IGNORECASE):
|
|
return False
|
|
else:
|
|
logger.warning(f"Самокритика вернула неопределённый ответ: {verdict[:200]}")
|
|
return True
|
|
except Exception as e:
|
|
logger.error(f"Ошибка самокритики: {e}", exc_info=True)
|
|
return True # При ошибке пропускаем проверку |