pilot.linkstec 0.0.13__tar.gz → 0.0.15__tar.gz

{pilot_linkstec-0.0.13 → pilot_linkstec-0.0.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pilot.linkstec
-Version: 0.0.13
+Version: 0.0.15
 Summary: pilot of the ship, a tool for managing and deploying Python projects.
 Author-email: wanglr <wanglr1980@gmail.com>
 License-Expression: MIT

{pilot_linkstec-0.0.13 → pilot_linkstec-0.0.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pilot.linkstec"
-version = "0.0.13"
+version = "0.0.15"
 authors = [
   { name="wanglr", email="wanglr1980@gmail.com" },
 ]

{pilot_linkstec-0.0.13 → pilot_linkstec-0.0.15}/src/pilot/generater/vertexai.py

@@ -24,7 +24,7 @@ class VertexAISingleton:
             with self._lock:
                 if not self._initialized:
                     self.model = GenerativeModel(model_name)
-
+                    self.encoding = tiktoken.get_encoding("cl100k_base")
                     self._initialized = True
 
     def generate_content(self, prompt: str) -> Dict[str, Any]:
@@ -52,7 +52,6 @@ class VertexAISingleton:
     def count_tokens(self, text: str) -> int:
         """与えられたテキストのトークン数を返す（bert-base-uncasedのみ使用）"""
         try:
-            self.encoding = tiktoken.get_encoding("cl100k_base")
             tokens = self.encoding.encode(text)
             return len(tokens)
         except Exception as e:

{pilot_linkstec-0.0.13 → pilot_linkstec-0.0.15}/src/pilot/job/impl/base_job.py

@@ -325,4 +325,20 @@ class BaseJob(JobInterface):
             open(end_file, 'w', encoding='utf-8').close()
             return end_file
 
-        return None
+        return None
+
+    def pre_run(self):
+        """
+        ジョブ実行前の前処理を行うメソッド。
+        必要に応じてサブクラスでオーバーライドして使用する。
+        """
+        pass
+
+    def post_run(self):
+        """
+        ジョブ実行後の後処理を行うメソッド。
+        必要に応じてサブクラスでオーバーライドして使用する。
+        """
+        pass
+
+

pilot_linkstec-0.0.15/src/pilot/logging/logger.py

@@ -0,0 +1,57 @@
+import logging
+import sys
+import threading
+
+# コンソールカラーのコード定義
+RESET = "\x1b[0m"
+COLOR_MAP = {
+    logging.DEBUG: "\x1b[37m",  # 白色
+    logging.INFO: "\x1b[32m",  # 緑色
+    logging.WARNING: "\x1b[33m",  # 黄色
+    logging.ERROR: "\x1b[31m",  # 赤色
+    logging.CRITICAL: "\x1b[41m",  # 赤背景白文字
+}
+
+
+class ColoredFormatter(logging.Formatter):
+    def format(self, record):
+        color = COLOR_MAP.get(record.levelno, RESET)
+        # スレッド名を取得
+        thread_name = threading.current_thread().name
+        thread_name = thread_name.split('_', 1)[1] if '_' in thread_name else thread_name
+        # クラス名、メソッド名、行番号
+        prefix = f"{self.formatTime(record, '%Y-%m-%d %H:%M:%S')} " \
+                 f"[{record.levelname}] " \
+                 f"[{thread_name}] " \
+                 f"[{record.module}.{record.funcName}:{record.lineno}]"
+        message = super().format(record)
+        return f"{color}{prefix} {message}{RESET}"
+
+
+_global_logger_configured = False
+
+
+def setup_global_logger(log_level: int = logging.INFO):
+    global _global_logger_configured
+
+    if not _global_logger_configured:
+        logging.basicConfig(
+            level=log_level,
+            format='%(message)s'
+        )
+
+        root_logger = logging.getLogger()
+        if root_logger.handlers:
+            root_logger.handlers.clear()
+
+        handler = logging.StreamHandler(sys.stdout)
+        formatter = ColoredFormatter('%(message)s')
+        handler.setFormatter(formatter)
+        root_logger.addHandler(handler)
+        root_logger.setLevel(log_level)
+
+        _global_logger_configured = True
+
+
+def get_logger(name: str) -> logging.Logger:
+    return logging.getLogger(name)

{pilot_linkstec-0.0.13 → pilot_linkstec-0.0.15}/src/pilot.linkstec.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pilot.linkstec
-Version: 0.0.13
+Version: 0.0.15
 Summary: pilot of the ship, a tool for managing and deploying Python projects.
 Author-email: wanglr <wanglr1980@gmail.com>
 License-Expression: MIT

{pilot_linkstec-0.0.13 → pilot_linkstec-0.0.15}/src/pilot.linkstec.egg-info/SOURCES.txt

@@ -21,6 +21,8 @@ src/pilot/job/__init__.py
 src/pilot/job/job_interface.py
 src/pilot/job/impl/__init__.py
 src/pilot/job/impl/base_job.py
+src/pilot/logging/__init__.py
+src/pilot/logging/logger.py
 src/pilot/splitters/__init__.py
 src/pilot/splitters/cobolsplitter.py
 src/pilot/unit/__init__.py