From 81e94de3185d42dba4e7bb72cf836f683f28b03f Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Tue, 10 Oct 2023 14:44:20 +0800
Subject: Add warning when meet emb name conflicting

Choose standalone embedding (in /embeddings folder) first
---
 extensions-builtin/Lora/lora_logger.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 extensions-builtin/Lora/lora_logger.py

(limited to 'extensions-builtin/Lora/lora_logger.py')

diff --git a/extensions-builtin/Lora/lora_logger.py b/extensions-builtin/Lora/lora_logger.py
new file mode 100644
index 00000000..d50e90f0
--- /dev/null
+++ b/extensions-builtin/Lora/lora_logger.py
@@ -0,0 +1,33 @@
+import sys
+import copy
+import logging
+
+
+class ColoredFormatter(logging.Formatter):
+    COLORS = {
+        "DEBUG": "\033[0;36m",  # CYAN
+        "INFO": "\033[0;32m",  # GREEN
+        "WARNING": "\033[0;33m",  # YELLOW
+        "ERROR": "\033[0;31m",  # RED
+        "CRITICAL": "\033[0;37;41m",  # WHITE ON RED
+        "RESET": "\033[0m",  # RESET COLOR
+    }
+
+    def format(self, record):
+        colored_record = copy.copy(record)
+        levelname = colored_record.levelname
+        seq = self.COLORS.get(levelname, self.COLORS["RESET"])
+        colored_record.levelname = f"{seq}{levelname}{self.COLORS['RESET']}"
+        return super().format(colored_record)
+
+
+logger = logging.getLogger("lora")
+logger.propagate = False
+
+
+if not logger.handlers:
+    handler = logging.StreamHandler(sys.stdout)
+    handler.setFormatter(
+        ColoredFormatter("[%(name)s]-%(levelname)s: %(message)s")
+    )
+    logger.addHandler(handler)
\ No newline at end of file
-- 
cgit v1.2.1


From 891ccb767c3815db48a124677d1cd0f204018ad4 Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Tue, 10 Oct 2023 15:07:25 +0800
Subject: Fix lint

---
 extensions-builtin/Lora/lora_logger.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'extensions-builtin/Lora/lora_logger.py')

diff --git a/extensions-builtin/Lora/lora_logger.py b/extensions-builtin/Lora/lora_logger.py
index d50e90f0..d51de297 100644
--- a/extensions-builtin/Lora/lora_logger.py
+++ b/extensions-builtin/Lora/lora_logger.py
@@ -30,4 +30,4 @@ if not logger.handlers:
     handler.setFormatter(
         ColoredFormatter("[%(name)s]-%(levelname)s: %(message)s")
     )
-    logger.addHandler(handler)
\ No newline at end of file
+    logger.addHandler(handler)
-- 
cgit v1.2.1