[383a81]: / production / action-server / actions / get_answer.py

Download this file

29 lines (23 with data), 886 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from decouple import config
from typing import Any, Text, Dict
from rasa_sdk import Action, Tracker
from rasa_sdk.executor import CollectingDispatcher
from .retrievers import create_retriever
from .processors import create_processor
from .dispatchers import dispatch
class GetAnswer(Action):
def __init__(self):
super().__init__()
# Creates retriever using config.yml
self.retriever = create_retriever()
# Creates processor using config.yml
self.processor = create_processor()
def name(self):
return "get_answer"
def run(self,
dispatcher: CollectingDispatcher,
tracker: Tracker,
domain: Dict[Text, Any]):
context = self.retriever.retrieve(tracker)
answer = self.processor.process(tracker, context)
dispatch(tracker, dispatcher, answer)