Update llm/llamacpp/lc_model.py

llm/llamacpp/lc_model.py

@@ -37,14 +37,19 @@ class LC_TinyLlama(LLMInterface, ABC):
                 path_to_model = os.path.join("../models", self.model_config["model_name"])
                 with open(path_to_model, "wb") as f:
                     f.write(get_file.content)
-                logger.info("Model file successfully recorded")
-                f.close()
+                    logger.info("Model file successfully recorded")
+        except FileExistsError:
+            print(f"Model file {path_to_model} already exists. Skipping download.")
+            logger.info(f"Model file {path_to_model} already exists. Skipping download.")
         except OSError as e:
-            print(f"Error while write a file to directory : {e}")
+            print(f"Error while writing a file to directory : {e}")
             logger.error(msg="Error while write a file to directory", exc_info=e)
+        finally:
+            f.close()
 
     @staticmethod
     def __read_yaml():
+        print(os.listdir('../'))
         try:
             yaml_file = os.path.join("../", 'prompts.yaml')
             with open(yaml_file, 'r') as file:
@@ -95,7 +100,7 @@ class LC_TinyLlama(LLMInterface, ABC):
             logger.info(f"Unfortunately dictionary empty or None")
 
     def get_unused(self, current_lc):
-        models_dir = "../models"
+        models_dir = ".../models"
 
         if len(os.listdir(models_dir)) > 1:
             file_names = [os.path.basename(md) for md in os.listdir(models_dir)]
@@ -126,11 +131,14 @@ class LC_Phi3(LLMInterface, ABC):
                 path_to_model = os.path.join("../models", self.model_config["model_name"])
                 with open(path_to_model, "wb") as f:
                     f.write(get_file.content)
-                logger.info("Model file successfully recorded")
-                f.close()
+                    logger.info("Model file successfully recorded")
+        except FileExistsError:
+            logger.info(f"Model file {path_to_model} already exists. Skipping download.")
         except OSError as e:
-            print(f"Error while write a file to directory : {e}")
+            print(f"Error while writing a file to directory : {e}")
             logger.error(msg="Error while write a file to directory", exc_info=e)
+        finally:
+            f.close()
 
     @staticmethod
     def __read_yaml():