pilot.linkstec 0.0.13__py3-none-any.whl → 0.0.15__py3-none-any.whl

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/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/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.dist-info → pilot_linkstec-0.0.15.dist-info}/METADATA

@@ -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.dist-info → pilot_linkstec-0.0.15.dist-info}/RECORD

@@ -9,11 +9,13 @@ pilot/conver/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/conver/converfileEncodding.py,sha256=UqjcWO0bzkuTRHLEWrWJkeo3p-P7WuYE7jFKveyPekA,2781
 pilot/conver/nkf_converter.py,sha256=JqgThmXcdnTGMsLIHUEwe8sc0VGMqDaKCIQTg-UE3WE,1148
 pilot/generater/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pilot/generater/vertexai.py,sha256=_MLmDuiVdEsAbKCTEM6KPbALnTXNTcIht2QMm-p9Kjs,2585
+pilot/generater/vertexai.py,sha256=v-hEvPwsUtrxP57yu9f5pUBj7BGImegO4lv10QWM1XA,2591
 pilot/job/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/job/job_interface.py,sha256=LL0hfuFfnKnkpQD99jv1hkaAIAFM-JJPrX3PFxN6O_A,120
 pilot/job/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pilot/job/impl/base_job.py,sha256=c8ANXdJZDuO0KEorPhO41uwr0tMTaY1jis0e_Ep3jtg,13572
+pilot/job/impl/base_job.py,sha256=rxh2WgKBf3yJ_x0aPFMGpP8aKjPiGbD4767OqseuZWg,14015
+pilot/logging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pilot/logging/logger.py,sha256=TF7eGr3w8GK5v4sf71lDt97uVoBtCgqrZuCdbMmeQBU,1815
 pilot/splitters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/splitters/cobolsplitter.py,sha256=oPwxKRjA7TyXWaWV3jdy59lJZy1mRn6yxD9ivqFYCuY,5461
 pilot/unit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -22,8 +24,8 @@ pilot/unit/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/unit/impl/base_unit.py,sha256=LsFPpL28aSNv5rsZhfKv6CWhAw1XR4n-A6FOn2RBrZo,1272
 pilot/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/util/files.py,sha256=v9uzfzo3Aq4xgnUIASEZeBJoA2nD9Qz_EA3P-FwzGFQ,1896
-pilot_linkstec-0.0.13.dist-info/licenses/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
-pilot_linkstec-0.0.13.dist-info/METADATA,sha256=qh87JuOVrhmDgpaPgiTFTT63nVI5MGcXZJtxYMAilOg,679
-pilot_linkstec-0.0.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pilot_linkstec-0.0.13.dist-info/top_level.txt,sha256=BijnVJdXnIPxxx3s60M848seL4Z12gNUPod6KPJxK9c,6
-pilot_linkstec-0.0.13.dist-info/RECORD,,
+pilot_linkstec-0.0.15.dist-info/licenses/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
+pilot_linkstec-0.0.15.dist-info/METADATA,sha256=cnelVXeTfz-4eAEBKxVqV0l9tBz2ndYw6vlPMKbZyBo,679
+pilot_linkstec-0.0.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pilot_linkstec-0.0.15.dist-info/top_level.txt,sha256=BijnVJdXnIPxxx3s60M848seL4Z12gNUPod6KPJxK9c,6
+pilot_linkstec-0.0.15.dist-info/RECORD,,