Update actions/actions.py

actions/actions.py

@@ -80,7 +80,7 @@ class GetOpenAIResponse(Action):
 
         # Use OpenAI API to generate a response
         #query = tracker.latest_message.get('text')
-        conversation_data = [FollowupAction("action_extract_history")]
+        conversation_data = extract_conversation_history()
         #response = generate_openai_response(conversation_data[0])
         response = conversation_data
                 
@@ -229,47 +229,49 @@ class SayHelloWorld(Action):
         generated_text = response.choices[0].text
         dispatcher.utter_message(text=generated_text)
 
-class ExtractConversationhistory(Action):
-    def name(self) -> Text:
-        return "action_extract_history"
-
-    def run(self, dispatcher: CollectingDispatcher, tracker: Tracker, domain: Dict[Text, Any]) -> List[Dict[Text, Any]]:
-        conversation_history = tracker.events
-
-        user_queries = []
-        bot_responses = []
-        current_user_query = ""
-        previous_user_query = None
-        previous_bot_response = None
-
-        for event in conversation_history:
-            if event.get("event") == "user":
-                user_queries.append(event.get("text"))
-            elif event.get("event") == "bot":
-                bot_responses.append(event.get("text"))
-
-        if user_queries:
-            if len(user_queries) >= 2:
-                previous_user_query = user_queries[-2]
-            else:
-                pass
-
-            try:
-                current_user_query = user_queries[-1]
-            except:
-                pass
-
-        if bot_responses:
-            if len(bot_responses) >= 2:
-                previous_bot_response = bot_responses[-2]
-            else:
-                pass
+from typing import List, Dict, Any, Text
+from rasa_sdk.executor import CollectingDispatcher
+from rasa_sdk.tracker import Tracker
+
+def extract_conversation_history(dispatcher: CollectingDispatcher, tracker: Tracker, domain: Dict[Text, Any]) -> Dict[Text, Any]:
+    conversation_history = tracker.events
+
+    user_queries = []
+    bot_responses = []
+    current_user_query = ""
+    previous_user_query = None
+    previous_bot_response = None
+
+    for event in conversation_history:
+        if event.get("event") == "user":
+            user_queries.append(event.get("text"))
+        elif event.get("event") == "bot":
+            bot_responses.append(event.get("text"))
+
+    if user_queries:
+        if len(user_queries) >= 2:
+            previous_user_query = user_queries[-2]
         else:
             pass
-        conversation_data = {
-            "previous_user_query": previous_user_query,
-            "previous_bot_response": previous_bot_response,
-            "current_user_query": current_user_query
-        }
-        # Now you can use the conversation_data dictionary as needed.
-        return conversation_data
+
+        try:
+            current_user_query = user_queries[-1]
+        except:
+            pass
+
+    if bot_responses:
+        if len(bot_responses) >= 2:
+            previous_bot_response = bot_responses[-2]
+        else:
+            pass
+    else:
+        pass
+        
+    conversation_data = {
+        "previous_user_query": previous_user_query,
+        "previous_bot_response": previous_bot_response,
+        "current_user_query": current_user_query
+    }
+    
+    # Now you can use the conversation_data dictionary as needed.
+    return conversation_data