credentials

main.py

@@ -51,7 +51,7 @@ Don't try to make up an answer, if you don't know, say you can contact the suppo
 Answer in the same language the question was asked.
 Answer in a way that is easy to understand.
 Try to limit the answer to 3-4 sentences.
-Provide source url as well at the end of the answer.
+Provide source url as well at the end of the answer if possible.
 Do not say "Based on the information you provided, ..." or "I think the answer is...". Just answer the question directly in detail.
 
 History: {chat_history}
@@ -67,7 +67,7 @@ Answer:
         )
         self.db = Chroma(persist_directory="./pragetx_chroma", embedding_function=HuggingFaceEmbeddings(), collection_name="pragetx")  
         # self.llm = ChatOllama(model="phi3:3.8b", base_url="http://localhost:11434", num_gpu=16)
-        self.llm = HuggingChat(email = os.getenv("HF_EMAIL") , psw = os.getenv("HF_PASS") )
+        self.llm = HuggingChat(email = os.getenv("HF_EMAIL") , psw = os.getenv("HF_PASS") , cookie_path="./usercookies")
         self.chain = (
             {"chat_history": self.chat_history, "context": self.db.as_retriever(search_kwargs={"k":3}), "question": RunnablePassthrough()} |
             self.prompt |

utils.py

@@ -56,17 +56,22 @@ class HuggingChat(LLM):
             raise ValueError("email, psw, or cookie_path is required.")
         
         try:
-            if self.email and self.psw:
+            if self.cookie_path:
+                cookies = hugchat.ChatBot(cookie_path=self.cookie_path)
+            elif self.email and self.psw:
                 # Create a ChatBot using email and psw
                 from hugchat.login import Login
                 start_time = time.time()
                 sign = Login(self.email, self.psw)
                 cookies = sign.login()
+                # save cookies to cookie_path
+                if self.cookie_path:
+                    sign.save_cookies(self.cookie_path)
                 end_time = time.time()
                 if self.log : print(f"\n[LOG] Login successfull in {round(end_time - start_time)} seconds")
-            else:
-                # Create a ChatBot using cookie_path
-                cookies = self.cookie_path and hugchat.ChatBot(cookie_path=self.cookie_path)
+            # else:
+            #     # Create a ChatBot using cookie_path
+            #     cookies = self.cookie_path and hugchat.ChatBot(cookie_path=self.cookie_path)
             
             self.chatbot = cookies.get_dict() and hugchat.ChatBot(cookies=cookies.get_dict())
             if self.log : print(f"[LOG] LLM WRAPPER created successfully")