Update hf_model.py

hf_model.py

@@ -2,7 +2,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 from datetime import datetime
 import os
-from .mew_log import log_info, log_error  # Import your custom logging methods
+from mew_log import log_info, log_error  # Import your custom logging methods
 
 class HFModel:
     def __init__(self, model_name):
@@ -17,7 +17,7 @@ class HFModel:
             self.tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
             log_info(f"=== Model Loaded Successfully: {self.friendly_name} ===")
         except Exception as e:
-            log_error(f"ERROR Loading Model: {e}")
+            log_error(f"ERROR Loading Model: {e}", e)
             raise
 
     def generate_response(self, input_text, max_length=100, skip_special_tokens=True):
@@ -28,7 +28,7 @@ class HFModel:
             log_info(f"Generated Response: {response}")
             return response
         except Exception as e:
-            log_error(f"ERROR Generating Response: {e}")
+            log_error(f"ERROR Generating Response: {e}", e)
             raise
 
     def stream_response(self, input_text, max_length=100, skip_special_tokens=True):
@@ -38,7 +38,7 @@ class HFModel:
                 response = self.tokenizer.decode(output, skip_special_tokens=skip_special_tokens).strip()
                 yield response
         except Exception as e:
-            log_error(f"ERROR Streaming Response: {e}")
+            log_error(f"ERROR Streaming Response: {e}", e)
             raise
 
     def chat(self, user_input, max_length=100, skip_special_tokens=True):
@@ -59,7 +59,7 @@ class HFModel:
 
             return model_response
         except Exception as e:
-            log_error(f"ERROR in Chat: {e}")
+            log_error(f"ERROR in Chat: {e}", e)
             raise
 
     def save_chat_log(self):
@@ -71,7 +71,7 @@ class HFModel:
                     f.write(f"**{role.capitalize()}:**\n\n{content}\n\n---\n\n")
             log_info(f"Chat log saved to {self.log_file}")
         except Exception as e:
-            log_error(f"ERROR Saving Chat Log: {e}")
+            log_error(f"ERROR Saving Chat Log: {e}", e)
             raise
 
     def clear_chat_history(self):
@@ -79,7 +79,7 @@ class HFModel:
             self.chat_history = []
             log_info("Chat history cleared.")
         except Exception as e:
-            log_error(f"ERROR Clearing Chat History: {e}")
+            log_error(f"ERROR Clearing Chat History: {e}", e)
             raise
 
     def print_chat_history(self):
@@ -90,5 +90,5 @@ class HFModel:
                 print(f"{role.capitalize()}: {content}\n")
             log_info("Printed chat history to console.")
         except Exception as e:
-            log_error(f"ERROR Printing Chat History: {e}")
+            log_error(f"ERROR Printing Chat History: {e}", e)
             raise