pilot.linkstec 0.0.106__py3-none-any.whl → 0.0.108__py3-none-any.whl

pilot/control/impl/base_controller.py

@@ -7,6 +7,7 @@ from concurrent.futures import ThreadPoolExecutor
 from pilot.control.control_interface import ControlInterface
 from pilot.unit.impl.base_unit import BaseUnit
 from pilot.config.config_reader import ConfigReader
+from pilot.generater.vera import VeraSingleton
 
 
 class BaseController(ControlInterface):
@@ -14,6 +15,9 @@ class BaseController(ControlInterface):
     config_dto = None
 
     def __init__(self):
+        #try:
+        #    VeraSingleton.get_instance()
+        #except FileNotFoundError:
         pass
 
     def _init_unit(self):

pilot/generater/ai_base.py

@@ -0,0 +1,7 @@
+from typing import Dict, Any
+
+from pilot.generater.ai_interface import AIInterface
+
+class AIBase(AIInterface):
+    def generate_content(self, prompt: str) -> Dict[str, Any]:
+        pass

pilot/generater/ai_interface.py

@@ -0,0 +1,8 @@
+from abc import ABC, abstractmethod
+from typing import Dict, Any
+
+
+class AIInterface(ABC):
+    @abstractmethod
+    def generate_content(self, prompt: str) -> Dict[str, Any]:
+        pass

pilot/generater/lmstudioai.py

@@ -0,0 +1,121 @@
+import threading
+from typing import Dict, Any, Optional
+from pilot.generater.ai_base import AIBase
+
+import requests
+
+class lmstudioAISingleton(AIBase):
+    _instance: Optional['lmstudioAISingleton'] = None
+    _lock = threading.Lock()
+
+    def __new__(cls):
+        if cls._instance is None:
+            with cls._lock:
+                if cls._instance is None:
+                    cls._instance = super(lmstudioAISingleton, cls).__new__(cls)
+                    cls._instance._initialized = False
+        return cls._instance
+
+    def __init__(self, model_name: str = "openai/gpt-oss-20b",base_url:str="http://127.0.0.1:3000/v1"):
+        if not self._initialized:
+            with self._lock:
+                if not self._initialized:
+                    self.model_name = model_name
+                    self.base_url = base_url
+                    self._session = requests.Session()
+                    self._initialized = True
+
+    def generate_content(self, prompt: str) -> Dict[str, Any]:
+        """複数スレッドから安全に呼び出し可能"""
+        try:
+            payload = {
+                "model": self.model_name,
+                "messages": [
+                    {"role": "user", "content": prompt}
+                ]
+            }
+
+            resp = self._session.post(
+                f"{self.base_url}/chat/completions",
+                json=payload,
+                timeout=600
+            )
+            resp.raise_for_status()
+            data = resp.json()
+
+            content = data["choices"][0]["message"]["content"]
+
+            return {
+                "prompt": prompt,
+                "response": self._remove_code_fence(content),
+                "success": True,
+                "error": None
+            }
+
+        except Exception as e:
+            return {
+                "prompt": prompt,
+                "response": None,
+                "success": False,
+                "error": str(e)
+            }
+
+    def start_chat(self):
+        """
+        VertexAI の ChatSession と完全互換は不可能だが、
+        既存コードを壊さないために「退化実装」を提供
+        """
+        return _LMStudioChatSession(self)
+
+    def count_tokens(self, text: str) -> int:
+        return 1
+        #try:
+        #    return len(self.encoding.encode(text))
+        #except Exception as e:
+        #    print(f"トークン計算失敗: {e}")
+        #    return 0
+
+    def _remove_code_fence(self, text: str) -> str:
+        lines = text.splitlines()
+        if lines and lines[0].startswith("```"):
+            lines = lines[1:]
+        if lines and lines[-1].startswith("```"):
+            lines = lines[:-1]
+        return "\n".join(lines)
+
+    @classmethod
+    def get_instance(cls, model_name,base_url) -> 'lmstudioAISingleton':
+        return cls(model_name,base_url)
+
+class _LMStudioChatSession:
+    """
+    VertexAI ChatSession の「最低限互換」
+    """
+    def __init__(self, client: lmstudioAISingleton):
+        self._client = client
+        self._messages = []
+
+    def send_message(self, message: str):
+        self._messages.append({"role": "user", "content": message})
+
+        payload = {
+            "model": self._client.model_name,
+            "messages": self._messages
+        }
+
+        resp = self._client._session.post(
+            f"{self._client.base_url}/chat/completions",
+            json=payload,
+            timeout=60
+        )
+        resp.raise_for_status()
+        data = resp.json()
+
+        reply = data["choices"][0]["message"]["content"]
+        self._messages.append({"role": "assistant", "content": reply})
+
+        class _Resp:
+            def __init__(self, text):
+                self.text = text
+
+        return _Resp(reply)

pilot/generater/vera.py

@@ -0,0 +1,101 @@
+import configparser
+import os
+import threading
+from dataclasses import dataclass
+from typing import Dict, Any, Optional
+
+from pilot.generater.lmstudioai import lmstudioAISingleton
+from pilot.generater.ai_base import AIBase
+
+@dataclass
+class AIConfigDTO:
+    veraApi: str
+    veraModel: str
+    verUrl: str
+    timeOut: int
+
+class VeraSingleton:
+    _instance: Optional['VeraSingleton'] = None
+    _lock = threading.Lock()
+
+    def __new__(cls):
+        if cls._instance is None:
+            with cls._lock:
+                if cls._instance is None:
+                    cls._instance = super(VeraSingleton, cls).__new__(cls)
+                    cls._instance._initialized = False
+        return cls._instance
+
+    def __init__(self):
+        if not self._initialized:
+            with self._lock:
+
+                cwd = os.getcwd()
+                filepath = os.path.join(cwd, 'config', 'vera.properties')
+                self.config = configparser.ConfigParser()
+                self.config.optionxform = str
+
+                with open(filepath, 'r', encoding='utf-8') as f:
+                    content = f.read()
+                if not content.lstrip().startswith('['):
+                    content = '[DEFAULT]\n' + content
+                self.config.read_string(content)
+                self.config_dto = self.create_ai_dto()
+                self.veraApi = self.config_dto.veraApi
+                self.model_name = self.config_dto.veraModel
+                self.base_url = self.config_dto.verUrl
+                self.timeout = self.config_dto.timeOut
+                self.ai_instance: AIBase  = self.get_ai_instance()
+                self._initialized = True
+
+    def get_ai_instance(self) :
+        match self.veraApi:
+            case "lmstudio":
+                return lmstudioAISingleton.get_instance(model_name=self.model_name,base_url= self.base_url)
+            case "ollama":
+                return AIBase()
+            case "gemini":
+                return AIBase()
+            case "gemini":
+                return AIBase()
+            case "gemini":
+                return AIBase()
+            case "qwen":
+                return AIBase()
+            case "azure":
+                return AIBase()
+        return AIBase()
+
+    def generate_content(self, prompt: str) -> Dict[str, Any]:
+        return self.ai_instance.generate_content(prompt)
+
+    @classmethod
+    def get_instance(cls) -> 'VeraSingleton':
+        return cls()
+
+    def create_ai_dto(self) -> AIConfigDTO:
+        vera_api = self.get('DEFAULT', 'vera_api', fallback='.')
+        vera_model = self.get('DEFAULT', 'vera_model', fallback='.')
+        ver_url = self.get('DEFAULT', 'ver_url', fallback='.')
+        time_out = int(self.get('DEFAULT', 'time_out', fallback=1))
+
+        return AIConfigDTO(
+            veraApi = vera_api,
+            veraModel = vera_model,
+            verUrl = ver_url,
+            timeOut = time_out
+        )
+
+    def get(self, section, option, fallback=None, cast_type=str):
+        try:
+            if cast_type == bool:
+                return self.config.getboolean(section, option)
+            elif cast_type == int:
+                return self.config.getint(section, option)
+            elif cast_type == float:
+                return self.config.getfloat(section, option)
+            else:
+                return self.config.get(section, option)
+        except (configparser.NoSectionError, configparser.NoOptionError):
+            return fallback
+

pilot/job/base/generate/generateJsonBaseJob.py

@@ -5,7 +5,7 @@ import time
 
 from pilot.job.impl.base_job import BaseJob
 
-from pilot.generater.vertexai import VertexAISingleton
+from pilot.generater.vera import VeraSingleton
 
 class GenerateJsonBaseJob(BaseJob):
 
@@ -14,24 +14,20 @@ class GenerateJsonBaseJob(BaseJob):
     result_file_path: str
 
     def run(self):
-        #with self._begin_file_lock:
-        #    if not self.change_current_trg_to_begin():
-        #        return
-        #prompt = self.get_file_content()
         prompt = self.prompt_content
         # トークン数チェック
-        vertexai = VertexAISingleton.get_instance()
-        token_count = vertexai.count_tokens(prompt)
-        if token_count == 0:
-            super().run()
-            return
-        if token_count > 900000:
-            print(f"警告: promptのトークン数が900000を超えています ({token_count} tokens)")
-            super().run()
-            return
+        vera_ai = VeraSingleton.get_instance()
+        #token_count = vertexai.count_tokens(prompt)
+        #if token_count == 0:
+        #    super().run()
+        #    return
+        #if token_count > 900000:
+        #    print(f"警告: promptのトークン数が900000を超えています ({token_count} tokens)")
+        #    super().run()
+        #    return
         # VertexAI で生成
         start = time.time()
-        result = vertexai.generate_content(prompt)
+        result = vera_ai.generate_content(prompt)
         end = time.time()
         print(f"Ai 処理時間 {self.file_path}: {end - start:.2f}秒")
 

pilot/job/base/generate/generateTextBaseJob.py

@@ -5,7 +5,7 @@ import time
 
 from pilot.job.impl.base_job import BaseJob
 
-from pilot.generater.vertexai import VertexAISingleton
+from pilot.generater.vera import VeraSingleton
 
 class GenerateTextBaseJob(BaseJob):
 
@@ -14,24 +14,20 @@ class GenerateTextBaseJob(BaseJob):
     result_file_path: str
 
     def run(self):
-        #with self._begin_file_lock:
-        #    if not self.change_current_trg_to_begin():
-        #        return
-        #prompt = self.get_file_content()
         prompt = self.prompt_content
         # トークン数チェック
-        vertexai = VertexAISingleton.get_instance()
-        token_count = vertexai.count_tokens(prompt)
-        if token_count == 0:
-            super().run()
-            return
-        if token_count > 900000:
-            print(f"警告: promptのトークン数が900000を超えています ({token_count} tokens)")
-            super().run()
-            return
+        vera_ai = VeraSingleton.get_instance()
+        #token_count = vertexai.count_tokens(prompt)
+        #if token_count == 0:
+        #    super().run()
+        #    return
+        #if token_count > 900000:
+        #    print(f"警告: promptのトークン数が900000を超えています ({token_count} tokens)")
+        #    super().run()
+        #    return
         # VertexAI で生成
         start = time.time()
-        result = vertexai.generate_content(prompt)
+        result = vera_ai.generate_content(prompt)
         end = time.time()
         print(f"AI 処理時間 {self.file_path}: {end - start:.2f}秒")
         result_content = result.get('response', '')

{pilot_linkstec-0.0.106.dist-info → pilot_linkstec-0.0.108.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pilot.linkstec
-Version: 0.0.106
+Version: 0.0.108
 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.106.dist-info → pilot_linkstec-0.0.108.dist-info}/RECORD

@@ -16,7 +16,7 @@ pilot/config/config_reader.py,sha256=MchG-9j-Xfy8AquL_6tlvqQR65MFJ3c3hwWBl6IF2jw
 pilot/control/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/control/control_interface.py,sha256=zGv380oQgAKPAIHDHeFdPYzhj2Ngo2T66NWlNloA7vY,124
 pilot/control/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pilot/control/impl/base_controller.py,sha256=049WVHCVfKyAiWHD1epLO8nzDbRDDOopXe0E4K401KA,3743
+pilot/control/impl/base_controller.py,sha256=CuO6rkkc6LX0DE7-54L9Sc-0Wg8nXNletFUGmeZeksE,3885
 pilot/conver/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/conver/commentRemover.py,sha256=S8uwp9Glp0bdv4EFqf62WIcOTLiJZdracG2FAMKY1EY,3777
 pilot/conver/converfileEncodding.py,sha256=YpPPTf52ujfWn8YlKuKYhSbcToQj34Gs4PmdcUbm040,1586
@@ -43,6 +43,10 @@ pilot/file_tool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/file_tool/create_prompt_file.py,sha256=pvmBm_iRWRvm5Qr8gaZwSAzJ_1vrqG9Xi_UJ-NgdwV8,1994
 pilot/file_tool/json_file_tool.py,sha256=v-qVcyKVhFfwEg4uJWDFMZU0EWvOwkYhrwhm-obMlUU,3637
 pilot/generater/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pilot/generater/ai_base.py,sha256=V-NSC--BfW8Q5Kn56B_kfyAQL3wL7n1rR6srKgnNWJA,192
+pilot/generater/ai_interface.py,sha256=NxoXWa9NWAKkN5kGOIZsK4dCCJlOLii1d-v8PcJ9BYk,193
+pilot/generater/lmstudioai.py,sha256=fxPChYuyJYjGdGHEFpfzaXigThyhInXL32RCqyuHUqU,3857
+pilot/generater/vera.py,sha256=5TTaRwKQQMZUA-4OMuvAGdPIeOAHc2eDTsh7tluFORc,3566
 pilot/generater/vertexai.py,sha256=UBVGPGjGb63wPeo8SLpj2X5g_wIHlPVgaewIOaT2ISw,3902
 pilot/job/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/job/job_interface.py,sha256=EKtuj0IcdolP494aAgTtctgamyQIoFXVwRORwOQck7A,124
@@ -52,8 +56,8 @@ pilot/job/base/convert/delimiterSwitcherJob.py,sha256=m5Y92XVbCjVWGBKzYJYX8wgLMn
 pilot/job/base/convert/encodingTransformerJob.py,sha256=P6txCqAcQ4w999ttA_gmA4VKr6klvTz4SrQIYT1gv6U,279
 pilot/job/base/convert/tabReplaceJob.py,sha256=FtPLbWwDPgH1jGsK3y5_sY4lAZ4Lhsv7GQ-1A-QgAFE,743
 pilot/job/base/generate/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pilot/job/base/generate/generateJsonBaseJob.py,sha256=gFWOOYDkdylW-Ph7R7xFdGr-TSzZycFrW8r-8i10DmA,1377
-pilot/job/base/generate/generateTextBaseJob.py,sha256=PcHXG0-mtAkiQi3Srkdk-4e3ThoSGnwzQ5QY_Nxxn84,1307
+pilot/job/base/generate/generateJsonBaseJob.py,sha256=0sWNqvfV-1DgsNjyD_GUBHogjQPNRdcqx0siNNFnSsE,1208
+pilot/job/base/generate/generateTextBaseJob.py,sha256=Pvxk2HNkQt9saBYmoRA8lhDjffMAxLMqSYvsyIlYKjg,1138
 pilot/job/base/program/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/job/base/program/cobol/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/job/base/program/cobol/convert/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -73,8 +77,8 @@ pilot/unit/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/unit/impl/base_unit.py,sha256=h6PMtd4ja0wa6mttFeI7rsv92yy96UgfLIU0QSs6Dmk,2108
 pilot/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pilot/util/files.py,sha256=v9uzfzo3Aq4xgnUIASEZeBJoA2nD9Qz_EA3P-FwzGFQ,1896
-pilot_linkstec-0.0.106.dist-info/licenses/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
-pilot_linkstec-0.0.106.dist-info/METADATA,sha256=ht4Bs-HFXqZGwoF5tZB2hOhmz1w6HtgXwSymvDbP9Aw,680
-pilot_linkstec-0.0.106.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-pilot_linkstec-0.0.106.dist-info/top_level.txt,sha256=BijnVJdXnIPxxx3s60M848seL4Z12gNUPod6KPJxK9c,6
-pilot_linkstec-0.0.106.dist-info/RECORD,,
+pilot_linkstec-0.0.108.dist-info/licenses/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
+pilot_linkstec-0.0.108.dist-info/METADATA,sha256=bVJA6FvtR9-r1jbCvKHVe-pq3_Y6XQp-9VdqYh4mfXg,680
+pilot_linkstec-0.0.108.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+pilot_linkstec-0.0.108.dist-info/top_level.txt,sha256=BijnVJdXnIPxxx3s60M848seL4Z12gNUPod6KPJxK9c,6
+pilot_linkstec-0.0.108.dist-info/RECORD,,