| from groq import Groq | |
| from dotenv import load_dotenv | |
| import os | |
| load_dotenv() | |
| api_key = os.getenv("GROQ_API_KEY") | |
| class ChatBot: | |
| def __init__(self): | |
| self.client = Groq(api_key=api_key) | |
| self.system_prompt = """You are a helpful assistant that answers questions based on website content. | |
| Always respond truthfully based on the provided context. If unsure, say you don't know.""" | |
| def generate_response(self, query, context, source_url): | |
| """Generate response using RAG""" | |
| context_text = "\n\n".join([doc.page_content for doc in context]) | |
| prompt = f"""Context: {context_text} | |
| Question: {query} | |
| Answer:""" | |
| try: | |
| response = self.client.chat.completions.create( | |
| messages=[ | |
| {"role": "system", "content": self.system_prompt}, | |
| {"role": "user", "content": prompt} | |
| ], | |
| model="mixtral-8x7b-32768", | |
| temperature=0.5 | |
| ) | |
| return f"{response.choices[0].message.content}\nSource: {source_url}" | |
| except Exception as e: | |
| return f"Error generating response: {str(e)}" |