Update llm/llm_factory.py

llm/llm_factory.py

@@ -1,4 +1,4 @@
-from typing import Dict, List
+from typing import Dict
 
 from .huggingfacehub.hf_model import HF_Mistaril, HF_TinyLlama
 from .llamacpp.lc_model import LC_TinyLlama, LC_Phi3
@@ -11,6 +11,7 @@ class LLM_Factory:
     def create_llm(prompt_entity: str, prompt_id: int, trigger: Dict[str, str]):
 
         if trigger.keys() == "hf" and trigger.values() == "effective":
+            print(trigger)
             model = HF_Mistaril(prompt_entity=prompt_entity, prompt_id=prompt_id)
         elif trigger.keys() == "hf" and trigger.values() == "small":
             model = HF_TinyLlama(prompt_entity=prompt_entity, prompt_id=prompt_id)