Update llm/llm_factory.py

llm/llm_factory.py

@@ -12,16 +12,16 @@ class LLM_Factory:
 
         print(trigger)
 
-        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)
-        elif trigger.keys() == "lc" and trigger.values() == "effective":
-            model = LC_Phi3(prompt_entity=prompt_entity, prompt_id=prompt_id)
-        elif trigger.keys() == "lc" and trigger.values() == "small":
-            model = LC_TinyLlama(prompt_entity=prompt_entity, prompt_id=prompt_id)
-        else:
-            model = None
+        for key, value in trigger.items():
+            if key == "hf" and value == "effective":
+                model = HF_Mistaril(prompt_entity=prompt_entity, prompt_id=prompt_id)
+            elif key == "hf" and value == "small":
+                model = HF_TinyLlama(prompt_entity=prompt_entity, prompt_id=prompt_id)
+            elif key == "lc" and value == "effective":
+                model = LC_Phi3(prompt_entity=prompt_entity, prompt_id=prompt_id)
+            elif key == "lc" and value == "small":
+                model = LC_TinyLlama(prompt_entity=prompt_entity, prompt_id=prompt_id)
+            else:
+                model = None
 
         return model