Update llm/llamacpp/lc_model.py

llm/llamacpp/lc_model.py

@@ -40,6 +40,7 @@ class LC_TinyLlama(LLMInterface, ABC):
                 logger.info("Model file successfully recorded")
                 f.close()
         except OSError as e:
+            print(f"Error while write a file to directory : {e}")
             logger.error(msg="Error while write a file to directory", exc_info=e)
 
     @staticmethod
@@ -50,13 +51,14 @@ class LC_TinyLlama(LLMInterface, ABC):
                 data = yaml.safe_load(file)
             return data
         except Exception as e:
+            print(f"Execution filed : {e}")
             logger.error(msg="Execution filed", exc_info=e)
 
     def execution(self):
         try:
             data = self.__read_yaml()
             prompts = data["prompts"][
-                self.prompt_id]  # get second prompt from yaml, need change id parameter to get other prompt
+                self.prompt_id]  # to get second prompt from yaml, need change id parameter to get other prompt
             template = prompts["prompt_template"]
             prompt = PromptTemplate(template=template, input_variables=["entity"])
 
@@ -76,13 +78,14 @@ class LC_TinyLlama(LLMInterface, ABC):
             output = llm_chain.invoke({"question": self.prompt_entity})
             return output
         except Exception as e:
+            print(f"Execution filed : {e}")
             logger.critical(msg="Execution filed", exc_info=e)
 
     def clear_llm(self, unused_model_dict, current_lc):
         # If unused_model_dict is not empty
-        if len(unused_model_dict) > 1 and unused_model_dict is not None:
+        if len(unused_model_dict) > 1 or unused_model_dict is not None:
             # go through key and value
-            for key, value in zip(unused_model_dict.keys(), unused_model_dict.values()):
+            for key, value in unused_model_dict.items():
                 # check if path is existing and key is not current using model
                 if os.path.exists(value) and key != current_lc:
                     # delete files from models directory except of current_lc
@@ -126,6 +129,7 @@ class LC_Phi3(LLMInterface, ABC):
                 logger.info("Model file successfully recorded")
                 f.close()
         except OSError as e:
+            print(f"Error while write a file to directory : {e}")
             logger.error(msg="Error while write a file to directory", exc_info=e)
 
     @staticmethod
@@ -136,6 +140,7 @@ class LC_Phi3(LLMInterface, ABC):
                 data = yaml.safe_load(file)
             return data
         except Exception as e:
+            print(f"Execution filed : {e}")
             logger.error(msg="Execution filed", exc_info=e)
 
     def execution(self):
@@ -162,20 +167,21 @@ class LC_Phi3(LLMInterface, ABC):
             output = llm_chain.invoke({"question": self.prompt_entity})
             return output
         except Exception as e:
+            print(f"Execution filed : {e}")
             logger.critical(msg="Execution filed", exc_info=e)
 
     def clear_llm(self, unused_model_dict, current_lc):
         # If unused_model_dict is not empty
-        if len(unused_model_dict) > 1:
+        if len(unused_model_dict) > 1 or unused_model_dict is not None:
             # go through key and value
-            for key, value in zip(unused_model_dict.keys(), unused_model_dict.values()):
+            for key, value in unused_model_dict.items():
                 # check if path is existing and key is not current using model
                 if os.path.exists(value) and key != current_lc:
                     # delete files from models directory except of current_lc
                     os.remove(value)
                     logger.info(f"Successfully deleted file {value}")
         else:
-            logger.info(f"Unfortunately dictionary empty")
+            logger.info(f"Unfortunately dictionary empty or None")
 
     def get_unused(self, current_lc):
         models_dir = "../models"