Update actions/actions.py

actions/actions.py

@@ -10,8 +10,60 @@ from rasa_sdk.events import SlotSet, FollowupAction
 from rasa_sdk.executor import CollectingDispatcher
 import random
 import os
+import sys
 import openai
 
+# Add "/actions" to the sys.path
+actions_path = os.path.abspath("/actions")
+sys.path.insert(0, actions_path)
+
+# Import search_content.py from /actions folder
+from search_content import main_search
+
+
+# Import api key from secrets
+secret_value_0 = os.environ.get("openai")
+
+openai.api_key = secret_value_0
+# Provide your OpenAI API key
+
+def generate_openai_response(query, model_engine="text-davinci-002", max_tokens=124, temperature=0.8):
+    """Generate a response using the OpenAI API."""
+    # Run the main function from search_content.py and store the results in a variable
+    results = main_search(query)
+
+    # Create context from the results
+    context = "".join([f"#{str(i)}" for i in results])[:2014] # Trim the context to 2014 characters - Modify as necessory
+    prompt_template = f"Relevant context: {context}\n\n Answer the question in detail: {query}"
+
+    # Generate a response using the OpenAI API
+    response = openai.Completion.create(
+        engine=model_engine,
+        prompt=prompt_template,
+        max_tokens=max_tokens,
+        temperature=temperature,
+        n=1,
+        stop=None,
+    )
+
+    return response.choices[0].text.strip()
+
+class GetOpenAIResponse(Action):
+
+    def name(self) -> Text:
+        return "action_get_response_openai"
+
+    def run(self,
+            dispatcher: CollectingDispatcher,
+            tracker: Tracker,
+            domain: Dict[Text, Any]) -> List[Dict[Text, Any]]:
+
+        # Use OpenAI API to generate a response
+
+        # Output the generated response to user
+        generated_text = response.choices[0].text
+        dispatcher.utter_message(text=generated_text)
+                
 class GeneralHelp(Action):
     def name(self) -> Text:
         return "action_general_help"
@@ -153,33 +205,3 @@ class SayHelloWorld(Action):
         # Output the generated response to user
         generated_text = response.choices[0].text
         dispatcher.utter_message(text=generated_text)
-
-
-class GetOpenAIResponse(Action):
-
-    def name(self) -> Text:
-        return "action_get_response_openai"
-
-    def run(self,
-            dispatcher: CollectingDispatcher,
-            tracker: Tracker,
-            domain: Dict[Text, Any]) -> List[Dict[Text, Any]]:
-
-        # Use OpenAI API to generate a response
-        secret_value_0 = os.environ.get("openai")
-        openai.api_key = secret_value_0
-        model_engine = "text-davinci-002"
-        prompt_template = tracker.latest_message.get('text')
-
-        response = openai.Completion.create(
-            engine=model_engine,
-            prompt=prompt_template,
-            max_tokens=124,
-            temperature=0.8,
-            n=1,
-            stop=None,
-        )
-
-        # Output the generated response to user
-        generated_text = response.choices[0].text
-        dispatcher.utter_message(text=generated_text)