pilot.linkstec 0.0.110__tar.gz → 0.0.111__tar.gz

{pilot_linkstec-0.0.110 → pilot_linkstec-0.0.111}/PKG-INFO

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

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

pilot_linkstec-0.0.111/src/pilot/job/base/generate/generatePromptBaseJob.py

@@ -0,0 +1,23 @@
+import json
+import os
+
+from pilot.file_tool.create_prompt_file import process_file
+from pilot.job.impl.base_job import BaseJob
+
+class GeneratePromptBaseJob(BaseJob):
+    prompt_file_path: str = None
+    template_file_path:list = []
+    prompt_content: str
+    result_content: str
+    result_file_path: str
+    replace_map  = {
+    }
+
+    def run(self):
+        if self.prompt_file_path is None:
+            self.prompt_file_path = self.change_extension_pathlib(self.file_path, 'prompt')
+        cwd = os.getcwd()
+        self.template_file_path.insert(0, cwd)
+        template = os.path.join(*self.template_file_path)
+        process_file(template, self.prompt_file_path, self.replace_map)
+        super().run()

{pilot_linkstec-0.0.110 → pilot_linkstec-0.0.111}/src/pilot/job/base/generate/generateTextBaseJob.py

@@ -12,9 +12,14 @@ class GenerateTextBaseJob(BaseJob):
     prompt_content: str
     result_content: str
     result_file_path: str
+    result_extension = '.md'
 
     def run(self):
         prompt = self.prompt_content
+        if prompt is None or prompt == '':
+            prompt = self.get_file_content()
+        if self.result_file_path is None:
+            self.result_file_path = self.change_extension_pathlib(self.file_path, self.result_extension)
         # トークン数チェック
         vera_ai = VeraSingleton.get_instance()
         #token_count = vertexai.count_tokens(prompt)
@@ -31,6 +36,7 @@ class GenerateTextBaseJob(BaseJob):
         end = time.time()
         print(f"AI 処理時間 {self.file_path}: {end - start:.2f}秒")
         result_content = result.get('response', '')
+
         with open(self.result_file_path, 'w', encoding='utf-8') as f:
             f.write(result_content)
         super().run()

{pilot_linkstec-0.0.110 → pilot_linkstec-0.0.111}/src/pilot.linkstec.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pilot.linkstec
-Version: 0.0.110
+Version: 0.0.111
 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.110 → pilot_linkstec-0.0.111}/src/pilot.linkstec.egg-info/SOURCES.txt

@@ -64,6 +64,7 @@ src/pilot/job/base/convert/encodingTransformerJob.py
 src/pilot/job/base/convert/tabReplaceJob.py
 src/pilot/job/base/generate/__init__.py
 src/pilot/job/base/generate/generateJsonBaseJob.py
+src/pilot/job/base/generate/generatePromptBaseJob.py
 src/pilot/job/base/generate/generateTextBaseJob.py
 src/pilot/job/base/program/__init__.py
 src/pilot/job/base/program/cobol/__init__.py