Update actions/actions.py

actions/actions.py

@@ -11,7 +11,8 @@ from rasa_sdk.executor import CollectingDispatcher
 import random
 import os
 import sys
-import openai
+# import openai
+import cohere
 
 # Add "/app/actions" to the sys.path
 actions_path = os.path.abspath("/app/actions")
@@ -31,6 +32,39 @@ secret_value_0 = os.environ.get("openai")
 openai.api_key = secret_value_0
 # Provide your OpenAI API key
 
+# def generate_openai_response(user_queries, model_engine="gpt-3.5-turbo", max_tokens=100, temperature=0.5):
+#     """Generate a response using the OpenAI API.""" 
+    
+#     # Send last two user queries for vector search
+#     if len(user_queries) >= 2:
+#         results = main_search(user_queries[-1]+user_queries[-2])
+#     else:
+#         results = main_search(user_queries[-1])
+  
+#     # Create context from the results
+#     context = "".join([f"#{str(i)}" for i in results])[:2014] # Trim the context to 2014 characters - Modify as necessory
+    
+#     messages=[
+#     {"role": "system", "content": f"You are a helpful assistant tasked to answer user queries using the following context: {context}"}
+#              ] 
+#     max_user_queries_to_include = min(1,len(user_queries))
+#     # The latest query is at the end of the list
+#     for i in range(len(user_queries)):
+#         if i<max_user_queries_to_include:
+#             messages.append({"role": "user", "content": user_queries[-max_user_queries_to_include+i]})
+    
+#     response = openai.ChatCompletion.create(
+#     model="gpt-3.5-turbo",
+#     messages= messages,
+#     max_tokens=124,
+#     temperature=0,
+#     top_p=1,
+#     frequency_penalty=0,
+#     presence_penalty=0
+#     )
+#     return response.choices[0]['message']['content'].strip()#+"Debug Info:            Openai-input_1:   "+str(messages_1)+"         Openai_input_0"+str( messages_0)+"  user_queries: "+str(user_queries)
+
+
 def generate_openai_response(user_queries, model_engine="gpt-3.5-turbo", max_tokens=100, temperature=0.5):
     """Generate a response using the OpenAI API.""" 
     
@@ -52,16 +86,15 @@ def generate_openai_response(user_queries, model_engine="gpt-3.5-turbo", max_tok
         if i<max_user_queries_to_include:
             messages.append({"role": "user", "content": user_queries[-max_user_queries_to_include+i]})
     
-    response = openai.ChatCompletion.create(
-    model="gpt-3.5-turbo",
-    messages= messages,
-    max_tokens=124,
-    temperature=0,
-    top_p=1,
-    frequency_penalty=0,
-    presence_penalty=0
-    )
-    return response.choices[0]['message']['content'].strip()#+"Debug Info:            Openai-input_1:   "+str(messages_1)+"         Openai_input_0"+str( messages_0)+"  user_queries: "+str(user_queries)
+    
+    co = cohere.Client(secret_value_0)
+
+    response = co.generate(
+                              prompt=messages,
+                            )
+    print(response)
+    return response[0].text.strip()#+"Debug Info: 
+
 
 class GetOpenAIResponse(Action):