dream

1
import logging
2
import os
3
import pathlib
4

5
from df_engine.core import Context, Actor
6

7
from common.constants import CAN_NOT_CONTINUE
8
from common.dff.integration.context import set_confidence, set_can_continue
9
from common.programy.model import get_programy_model
10
from common.sensitive import psycho_help_spec
11

12

13
logger = logging.getLogger(__name__)
14
LANGUAGE = os.getenv("LANGUAGE", "EN")
15
model_folder = "data_ru" if LANGUAGE == "RU" else "data"
16
logger.info(f"Selected dff-program-y-skill: {LANGUAGE} language.")
17

18
try:
19
    logger.info("Start to load model")
20
    model = get_programy_model(pathlib.Path(model_folder))
21
    logger.info("Load model")
22
except Exception as e:
23
    logger.exception(e)
24
    raise (e)
25

26

27
def programy_reponse(ctx: Context, actor: Actor, *args, **kwargs) -> str:
28
    response = model(ctx.requests.values())
29
    if psycho_help_spec in response:
30
        set_confidence(ctx, actor, 0.85)
31
        set_can_continue(ctx, actor, continue_flag=CAN_NOT_CONTINUE)
32
        return response
33
    return response
34