Update to Rasa 3.x and Action code updates

.gitignore

@@ -127,3 +127,18 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+
+# VSCode
+.vscode/
+.history/
+*.DS_Store
+*.wpr
+*.wpu
+
+# Rasa stuff
+models/
+*.db*
+*.dot
+terms/
+results/
+.rasa/

Dockerfile

@@ -0,0 +1,11 @@
+FROM rasa/rasa-sdk:3.3.0
+ENV TZ=Etc/UTC
+
+COPY actions /app/actions
+
+USER root
+RUN /opt/venv/bin/python -m pip install --upgrade pip
+RUN pip install --no-cache-dir -r /app/actions/requirements.txt
+
+USER 1001
+CMD ["start", "--actions", "actions"] 

actions/actions.py

@@ -9,27 +9,45 @@
 
 from rasa_sdk import Action, Tracker
 from rasa_sdk.executor import CollectingDispatcher
-import requests
+import aiohttp
+import json
+import logging
 
+logger = logging.getLogger(__name__)
 
 class ActionHaystack(Action):
 
     def name(self) -> Text:
         return "call_haystack"
 
-    def run(self, dispatcher: CollectingDispatcher,
+    async def run(self, dispatcher: CollectingDispatcher,
             tracker: Tracker,
             domain: Dict[Text, Any]) -> List[Dict[Text, Any]]:
 
         url = "http://localhost:8000/query"
-        payload = {"query": str(tracker.latest_message["text"])}
+        payload = {
+            "query": str(tracker.latest_message["text"]),
+            "params":{"filters": {}, "Retriever": {"top_k": 3}, "Reader": {"top_k": 3}, "Query": {"debug": False}}
+        }
         headers = {
             'Content-Type': 'application/json'
         }
-        response = requests.request("POST", url, headers=headers, json=payload).json()
+
+        logger.debug(f"Calling {url}, ask: {str(tracker.latest_message['text'])}")
+        try:
+            async with aiohttp.ClientSession() as session:
+                async with session.post(url, headers=headers, json=payload) as response:
+                    response = await response.json()
+        except aiohttp.ClientConnectorError as e:
+            logger.error(f"Haystack service not responding: {str(e)}")
+            response = {"answers":[{"answer":f"Haystack service not responding: {str(e)}"}]}
 
         if response["answers"]:
+            logger.debug(f"Answers returned by haystack:\n{response}")
             answer = response["answers"][0]["answer"]
+            if not answer:
+                answer = response["answers"][1]["answer"]
+            logger.debug(f"selected answer: {answer}")
         else:
             answer = "No Answer Found!"
 

actions/requirements.txt

@@ -0,0 +1 @@
+aiohttp==3.8.1

build.sh

@@ -0,0 +1,8 @@
+#!/bin/sh
+# Build action server image
+VERS=3.3.0
+REGISTRY=<Docker organization>
+IMAGE=haystack-action
+
+time docker buildx build --platform linux/amd64,linux/arm64 --output=type=registry --tag ${REGISTRY}/${IMAGE}:${VERS} --tag ${REGISTRY}/${IMAGE}:latest .
+docker push ${REGISTRY}/${IMAGE} --all-tags

data/nlu.yml

@@ -1,4 +1,4 @@
-version: "2.0"
+version: "3.1"
 
 nlu:
 - intent: knowledge_question
@@ -30,7 +30,6 @@ nlu:
 
 - intent: goodbye
   examples: |
-    - good afternoon
     - cu
     - good by
     - cee you later

data/rules.yml

@@ -1,4 +1,4 @@
-version: "2.0"
+version: "3.1"
 
 rules:
 

data/stories.yml

@@ -1,4 +1,4 @@
-version: "2.0"
+version: "3.1"
 
 stories:
 

domain.yml

@@ -1,4 +1,4 @@
-version: '2.0'
+version: "3.1"
 config:
   store_entities_as_slots: true
 session_config: