Update llm/utils/chat.py

llm/utils/chat.py

@@ -1,5 +1,4 @@
 import time
-import logging
 
 from llm.apimodels.gemini_model import Gemini
 from llm.apimodels.hf_model import HF_Mistaril, HF_TinyLlama, HF_SmolLM135, HF_SmolLM360, HF_SmolLM, HF_Gemma2, HF_Qwen2
@@ -9,19 +8,6 @@ from typing import Optional, Any
 from langchain.chains.conversation.memory import ConversationBufferWindowMemory
 from langchain.chains import ConversationChain
 
-logger = logging.getLogger(__name__)
-logger.setLevel(logging.WARNING)
-
-file_handler = logging.FileHandler(
-    "logs/chelsea_llm_chat.log")  # for all modules here template for logs file is "llm/logs/chelsea_{module_name}_{entity}.log"
-logger.setLevel(logging.INFO)  # informed
-
-formatted = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s")
-file_handler.setFormatter(formatted)
-
-logger.addHandler(file_handler)
-logger.info("Getting information from chat module")
-
 def prettify(raw_text: str) -> str:        
     pretty = raw_text.replace("**", "")
     return pretty.strip()
@@ -49,7 +35,6 @@ def has_failed(conversation, prompt) -> Optional[str]:
         result = prettify(raw_text=response)
         return result
     except Exception as e:
-        logger.error(msg="Error during prediction with conversation in has_failed function", exc_info=e)
         print(f"Error during prediction with conversation in has_failed function: {e}")
         return None
 
@@ -79,7 +64,6 @@ def has_delay(conversation, prompt) -> Optional[str]:
         return result  # Return the prettified response
 
     except Exception as e:
-        logger.error(msg="Error during prediction with conversation in has_delay function", exc_info=e)
         print(f"Error during prediction with conversation in has_delay function: {e}")
 
 
@@ -93,7 +77,7 @@ class Conversation:
                 Defaults to [Gemini, HF_SmolLM135, HF_SmolLM360, HF_TinyLlama, HF_SmolLM, HF_Gemma2, HF_Mistaril, HF_Qwen2].
         """
 
-        self.model_classes = [HF_SmolLM135, HF_SmolLM360, HF_TinyLlama, HF_SmolLM, HF_Gemma2, HF_Mistaril, HF_Qwen2]
+        self.model_classes = [Gemini, HF_Gemma2, HF_SmolLM, HF_SmolLM360, HF_Mistaril, HF_Qwen2, HF_TinyLlama, HF_SmolLM135]
         self.current_model_index = 0
 
     def _get_conversation(self) -> Any:
@@ -105,7 +89,6 @@ class Conversation:
             print("current model class is: ", current_model_class)
             return ConversationChain(llm=current_model_class().execution(), memory=memory, return_final_only=True)
         except Exception as e:
-            logger.error(msg="Error during conversation chain in get_conversation function", exc_info=e)
             print(f"Error during conversation chain in get_conversation function: {e}")
 
     def chatting(self, prompt: str) -> str: