Update llm/utils/chat.py

llm/utils/chat.py

@@ -2,8 +2,7 @@ import time
 import logging
 
 from llm.apimodels.gemini_model import Gemini
-from llm.apimodels.hf_model import HF_Mistaril, HF_TinyLlama
-from llm.llamacpp.lc_model import LC_Phi3, LC_TinyLlama
+from llm.apimodels.hf_model import HF_Mistaril, HF_TinyLlama, HF_SmolLM135, HF_SmolLM360, HF_SmolLM, HF_Gemma2, HF_Qwen2
 
 from typing import Optional, Any
 
@@ -27,7 +26,6 @@ def prettify(raw_text: str) -> str:
     pretty = raw_text.replace("**", "")
     return pretty.strip()
 
-# option + command + F -> replace
 
 memory: ConversationBufferWindowMemory = ConversationBufferWindowMemory(k=3, ai_prefix="Chelsea")
 
@@ -92,10 +90,10 @@ class Conversation:
 
         Args:
             model_classes (list, optional): A list of LLM model classes to try in sequence.
-                Defaults to [Gemini, HF_Mistaril, HF_TinyLlama, LC_Phi3, LC_TinyLlama].
+                Defaults to [Gemini, HF_Mistaril, HF_TinyLlama, HF_SmolLM135, HF_SmolLM360, HF_SmolLM, HF_Gemma2, HF_Qwen2].
         """
 
-        self.model_classes = [Gemini, HF_Mistaril, HF_TinyLlama, LC_Phi3, LC_TinyLlama]
+        self.model_classes = [Gemini, HF_Mistaril, HF_TinyLlama, HF_SmolLM135, HF_SmolLM360, HF_SmolLM, HF_Gemma2, HF_Qwen2]
         self.current_model_index = 0
 
     def _get_conversation(self) -> Any: