pilot.linkstec 0.0.22__tar.gz → 0.0.24__tar.gz

{pilot_linkstec-0.0.22 → pilot_linkstec-0.0.24}/PKG-INFO

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

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

{pilot_linkstec-0.0.22 → pilot_linkstec-0.0.24}/src/pilot/generater/vertexai.py

@@ -13,11 +13,11 @@ class VertexAISingleton:
     encoding = None
 
     SQL_SYSTEM_PROMPT_EN = (
-        "If there is any SQL-related processing　\n"
-        "1. If there is an SQL statement like \"SELECT COUNT(X) INTO :COLUMN FROM TABLE_NAME;\"　please recognize TABLE_NAME as a table."
+        "For SQL-related processing, analysis, and modifications:\n"
+        "1. When analyzing SQL statements such as \"SELECT COUNT(X) INTO :COLUMN FROM TABLE_NAME;\", please identify TABLE_NAME as a database table.\n"
+        "2. Extract and document all table names, column references, and SQL operations for comprehensive program analysis."
     )
 
-
     def __new__(cls, model_name: str = "gemini-2.5-pro", system_prompt: Optional[str] = None):
         if cls._instance is None:
             with cls._lock:
@@ -30,13 +30,14 @@ class VertexAISingleton:
         if not self._initialized:
             with self._lock:
                 if not self._initialized:
-                    self.model = GenerativeModel(model_name)
-                    self.encoding = tiktoken.get_encoding("cl100k_base")
-                    # system_promptにSQL_SYSTEM_PROMPT_ENを追加
                     if system_prompt:
                         self.system_prompt = f"{system_prompt.rstrip()}\n\n{self.SQL_SYSTEM_PROMPT_EN}"
                     else:
                         self.system_prompt = self.SQL_SYSTEM_PROMPT_EN
+                    self.model = GenerativeModel(model_name=model_name,system_instruction = self.system_prompt)
+                    self.encoding = tiktoken.get_encoding("cl100k_base")
+                    # system_promptにSQL_SYSTEM_PROMPT_ENを追加
+
                     self._initialized = True
         else:
             # 既存インスタンスでもsystem_promptを更新可能に
@@ -47,14 +48,9 @@ class VertexAISingleton:
         """複数スレッドから安全に呼び出し可能"""
         try:
             # システムプロンプトをconfigとして渡す
-
-            generation_config = types.GenerateContentConfig(
-                system_instruction = self.system_prompt
-            )
             prompt = self.exchange_prompt(prompt)
             response = self.model.generate_content(
-                contents=prompt,  # 引数名を明示
-                generation_config = generation_config  # 正しい名前で渡す
+                contents=prompt # 引数名を明示
             )
             return {
                 "prompt": prompt,

{pilot_linkstec-0.0.22 → pilot_linkstec-0.0.24}/src/pilot.linkstec.egg-info/PKG-INFO

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