Update pdf_bot.py

pdf_bot.py

@@ -15,14 +15,20 @@ groq_api_key = os.getenv("GROQ_API_KEY")
 
 # ✅ Custom wrapper
 class ChatGroq(BaseChatModel):
+    model: str = "llama3-8b-8192"
+    temperature: float = 0.3
+    groq_api_key: str = None
+
     def __init__(self, model="llama3-8b-8192", temperature=0.3, api_key=None):
-        self.client = Groq(api_key=api_key)
+        super().__init__()
         self.model = model
         self.temperature = temperature
+        self.groq_api_key = api_key
+        self._client = Groq(api_key=api_key)  # ✅ use _client instead of client
 
     def _generate(self, messages, stop=None):
         prompt = [{"role": "user", "content": self._get_message_text(messages)}]
-        response = self.client.chat.completions.create(
+        response = self._client.chat.completions.create(
             model=self.model,
             messages=prompt,
             temperature=self.temperature,
@@ -37,7 +43,7 @@ class ChatGroq(BaseChatModel):
         return messages.content
 
     @property
-    def _llm_type(self):
+    def _llm_type(self) -> str:
         return "chat-groq"
 
 # ✅ Function to return a QA chain