pilot.linkstec 0.0.21__tar.gz → 0.0.23__tar.gz

{pilot_linkstec-0.0.21 → pilot_linkstec-0.0.23}/PKG-INFO

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

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

{pilot_linkstec-0.0.21 → pilot_linkstec-0.0.23}/src/pilot/generater/vertexai.py

@@ -3,9 +3,8 @@ import threading
 from typing import Dict, Any, Optional
 
 import tiktoken
-from vertexai.generative_models import GenerativeModel, ChatSession
+from vertexai.generative_models import GenerativeModel, ChatSession, GenerationConfig
 from google.genai import types
-import os
 
 class VertexAISingleton:
     _instance: Optional['VertexAISingleton'] = None
@@ -31,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を更新可能に
@@ -48,14 +48,9 @@ class VertexAISingleton:
         """複数スレッドから安全に呼び出し可能"""
         try:
             # システムプロンプトをconfigとして渡す
-
-            generation_config = {
-                "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.21 → pilot_linkstec-0.0.23}/src/pilot.linkstec.egg-info/PKG-INFO

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