Update rasa-assistant-2/actions/actions.py

rasa-assistant-2/actions/actions.py

@@ -1,3 +1,47 @@
+import requests
+from typing import Any, Text, Dict, List
+from rasa_sdk import Action, Tracker
+from rasa_sdk.executor import CollectingDispatcher
+
+GROQ_API_KEY = "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"  # Replace with your Groq API Key
+GROQ_MODEL = "llama3-8b-8192"  # Change the model if needed
+
+class ActionGroqChat(Action):
+    def name(self) -> Text:
+        return "action_groq_chat"
+
+    def run(self, dispatcher: CollectingDispatcher,
+            tracker: Tracker,
+            domain: Dict[Text, Any]) -> List[Dict[Text, Any]]:
+
+        user_message = tracker.latest_message.get('text')
+        
+        url = "https://api.groq.com/v1/chat/completions"
+        headers = {
+            "Authorization": f"Bearer {GROQ_API_KEY}",
+            "Content-Type": "application/json"
+        }
+        payload = {
+            "model": GROQ_MODEL,
+            "messages": [{"role": "user", "content": user_message}]
+        }
+
+        try:
+            response = requests.post(url, json=payload, headers=headers)
+            response_json = response.json()
+            bot_reply = response_json["choices"][0]["message"]["content"]
+        except Exception as e:
+            bot_reply = "Sorry, I couldn't process your request."
+
+        dispatcher.utter_message(text=bot_reply)
+        return []
+
+
+
+
+
+
+
 # This files contains your custom actions which can be used to run
 # custom Python code.
 #