pilot.linkstec 0.0.2__tar.gz → 0.0.4__tar.gz

{pilot_linkstec-0.0.2 → pilot_linkstec-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pilot.linkstec
-Version: 0.0.2
+Version: 0.0.4
 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.2 → pilot_linkstec-0.0.4}/pyproject.toml

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

{pilot_linkstec-0.0.2 → pilot_linkstec-0.0.4}/src/pilot/generater/vertexai.py

@@ -24,7 +24,6 @@ class VertexAISingleton:
             with self._lock:
                 if not self._initialized:
                     self.model = GenerativeModel(model_name)
-                    os.environ["TIKTOKEN_CACHE_DIR"] = r"D:\api"
                     self.encoding = tiktoken.get_encoding("cl100k_base")
                     self._initialized = True
 

{pilot_linkstec-0.0.2 → pilot_linkstec-0.0.4}/src/pilot/job/impl/base_job.py

@@ -260,11 +260,17 @@ class BaseJob(JobInterface):
         """
         return self.config_dto.work_space
 
-    def copy_file_todo_trg_to_next_step(self, result_file):
+    def copy_file_and_todo_trg_to_next_step(self, result_file):
         self.create_current_step_end_trg_file_from_input(result_file)
         next_step_file = self.copy_input_file_to_next_step(result_file)
         self.create_next_step_todo_trg_file_from_input(next_step_file)
 
+
+    def copy_file_and_end_trg_to_next_step(self, result_file):
+        self.create_current_step_end_trg_file_from_input(result_file)
+        next_step_file = self.copy_input_file_to_next_step(result_file)
+        self.create_next_step_end_trg_file_from_input(next_step_file)
+
     def change_trg_file_to_end_in_file(self, file_name):
         """
         file_pathのディレクトリ配下に指定されたファイル名の.trgまたは.beginファイルが存在すれば、.endにリネームする。
@@ -286,3 +292,37 @@ class BaseJob(JobInterface):
                 print(f"change_trg_file_to_end_in_file error: {trg_file}")
                 return False
         return False
+
+
+
+    def create_next_step_end_trg_file_from_input(self, input_file_path):
+        """
+        input_file_pathで指定されたファイルをnext stepフォルダにコピーし、.trgファイルを生成する。
+        ファイルが既に存在する場合でも.trgファイルの作成を試行する。
+        """
+        if not input_file_path or not self.next_step:
+            return None
+
+        # ファイルをコピー（既存の場合はNoneが返される）
+        copied_file_path = self.copy_input_file_to_next_step(input_file_path)
+
+        # コピーが成功しなかった場合でも、対象ファイルパスを取得
+        if not copied_file_path:
+            base_dir = os.path.join(self.config_dto.work_space, self.current_step)
+            step_dir = os.path.join(self.config_dto.work_space, self.next_step)
+            rel_path = os.path.relpath(os.path.dirname(input_file_path), base_dir)
+            dest_dir = os.path.join(step_dir, rel_path)
+            base_file_name = os.path.basename(input_file_path)
+            copied_file_path = os.path.join(dest_dir, base_file_name)
+
+        # .trgファイルを作成
+        trg_file = copied_file_path + ".trg"
+        end_file = copied_file_path + ".end"
+        begin_file = copied_file_path + ".begin"
+
+        if not (os.path.exists(trg_file) or os.path.exists(end_file) or os.path.exists(begin_file)):
+            # 空ファイルとして.trgファイルを生成
+            open(end_file, 'w', encoding='utf-8').close()
+            return end_file
+
+        return None

{pilot_linkstec-0.0.2 → pilot_linkstec-0.0.4}/src/pilot.linkstec.egg-info/PKG-INFO

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