pilot.linkstec 0.0.22__py3-none-any.whl → 0.0.23__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of pilot.linkstec might be problematic. Click here for more details.

Files changed (6) hide show
  1. pilot/generater/vertexai.py +5 -9
  2. {pilot_linkstec-0.0.22.dist-info → pilot_linkstec-0.0.23.dist-info}/METADATA +1 -1
  3. {pilot_linkstec-0.0.22.dist-info → pilot_linkstec-0.0.23.dist-info}/RECORD +6 -6
  4. {pilot_linkstec-0.0.22.dist-info → pilot_linkstec-0.0.23.dist-info}/WHEEL +0 -0
  5. {pilot_linkstec-0.0.22.dist-info → pilot_linkstec-0.0.23.dist-info}/licenses/LICENSE +0 -0
  6. {pilot_linkstec-0.0.22.dist-info → pilot_linkstec-0.0.23.dist-info}/top_level.txt +0 -0
pilot/generater/vertexai.py CHANGED
@@ -30,13 +30,14 @@ class VertexAISingleton:
30
30
  if not self._initialized:
31
31
  with self._lock:
32
32
  if not self._initialized:
33
- self.model = GenerativeModel(model_name)
34
- self.encoding = tiktoken.get_encoding("cl100k_base")
35
- # system_promptにSQL_SYSTEM_PROMPT_ENを追加
36
33
  if system_prompt:
37
34
  self.system_prompt = f"{system_prompt.rstrip()}\n\n{self.SQL_SYSTEM_PROMPT_EN}"
38
35
  else:
39
36
  self.system_prompt = self.SQL_SYSTEM_PROMPT_EN
37
+ self.model = GenerativeModel(model_name=model_name,system_instruction = self.system_prompt)
38
+ self.encoding = tiktoken.get_encoding("cl100k_base")
39
+ # system_promptにSQL_SYSTEM_PROMPT_ENを追加
40
+
40
41
  self._initialized = True
41
42
  else:
42
43
  # 既存インスタンスでもsystem_promptを更新可能に
@@ -47,14 +48,9 @@ class VertexAISingleton:
47
48
  """複数スレッドから安全に呼び出し可能"""
48
49
  try:
49
50
  # システムプロンプトをconfigとして渡す
50
-
51
- generation_config = types.GenerateContentConfig(
52
- system_instruction = self.system_prompt
53
- )
54
51
  prompt = self.exchange_prompt(prompt)
55
52
  response = self.model.generate_content(
56
- contents=prompt, # 引数名を明示
57
- generation_config = generation_config # 正しい名前で渡す
53
+ contents=prompt # 引数名を明示
58
54
  )
59
55
  return {
60
56
  "prompt": prompt,
{pilot_linkstec-0.0.22.dist-info → pilot_linkstec-0.0.23.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: pilot.linkstec
3
- Version: 0.0.22
3
+ Version: 0.0.23
4
4
  Summary: pilot of the ship, a tool for managing and deploying Python projects.
5
5
  Author-email: wanglr <wanglr1980@gmail.com>
6
6
  License-Expression: MIT
{pilot_linkstec-0.0.22.dist-info → pilot_linkstec-0.0.23.dist-info}/RECORD RENAMED
@@ -9,7 +9,7 @@ pilot/conver/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
9
9
  pilot/conver/converfileEncodding.py,sha256=UqjcWO0bzkuTRHLEWrWJkeo3p-P7WuYE7jFKveyPekA,2781
10
10
  pilot/conver/nkf_converter.py,sha256=JqgThmXcdnTGMsLIHUEwe8sc0VGMqDaKCIQTg-UE3WE,1148
11
11
  pilot/generater/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
12
- pilot/generater/vertexai.py,sha256=rEy8_i-DSm_7MbsRlKdPrDBsf8ECk0EyiAaIny3FRj8,6215
12
+ pilot/generater/vertexai.py,sha256=83cPCiXZ01s1Mmaxewe1O4h0X-mWZVT_A9Se6TJ7pVs,6047
13
13
  pilot/job/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
14
14
  pilot/job/job_interface.py,sha256=LL0hfuFfnKnkpQD99jv1hkaAIAFM-JJPrX3PFxN6O_A,120
15
15
  pilot/job/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -27,8 +27,8 @@ pilot/unit/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
27
27
  pilot/unit/impl/base_unit.py,sha256=LsFPpL28aSNv5rsZhfKv6CWhAw1XR4n-A6FOn2RBrZo,1272
28
28
  pilot/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
29
29
  pilot/util/files.py,sha256=v9uzfzo3Aq4xgnUIASEZeBJoA2nD9Qz_EA3P-FwzGFQ,1896
30
- pilot_linkstec-0.0.22.dist-info/licenses/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
31
- pilot_linkstec-0.0.22.dist-info/METADATA,sha256=ylEtRen-5bTEybxoLmCBFD3gQZzB_aQHRy_Wab4x-BU,679
32
- pilot_linkstec-0.0.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
33
- pilot_linkstec-0.0.22.dist-info/top_level.txt,sha256=BijnVJdXnIPxxx3s60M848seL4Z12gNUPod6KPJxK9c,6
34
- pilot_linkstec-0.0.22.dist-info/RECORD,,
30
+ pilot_linkstec-0.0.23.dist-info/licenses/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
31
+ pilot_linkstec-0.0.23.dist-info/METADATA,sha256=iK17nWXkGTpU8RKhNmMhQ7AKBYBocun1G1asx7CtDLQ,679
32
+ pilot_linkstec-0.0.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
33
+ pilot_linkstec-0.0.23.dist-info/top_level.txt,sha256=BijnVJdXnIPxxx3s60M848seL4Z12gNUPod6KPJxK9c,6
34
+ pilot_linkstec-0.0.23.dist-info/RECORD,,