promptbuilder 0.4.25__py3-none-any.whl → 0.4.27__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.
Files changed (6) hide show
  1. promptbuilder/llm_client/base_client.py +6 -2
  2. {promptbuilder-0.4.25.dist-info → promptbuilder-0.4.27.dist-info}/METADATA +1 -1
  3. {promptbuilder-0.4.25.dist-info → promptbuilder-0.4.27.dist-info}/RECORD +6 -6
  4. {promptbuilder-0.4.25.dist-info → promptbuilder-0.4.27.dist-info}/WHEEL +0 -0
  5. {promptbuilder-0.4.25.dist-info → promptbuilder-0.4.27.dist-info}/licenses/LICENSE +0 -0
  6. {promptbuilder-0.4.25.dist-info → promptbuilder-0.4.27.dist-info}/top_level.txt +0 -0
promptbuilder/llm_client/base_client.py CHANGED
@@ -104,7 +104,8 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
104
104
 
105
105
  total_count = BaseLLMClient._response_out_tokens(response)
106
106
 
107
- while autocomplete and response.candidates and response.candidates[0].finish_reason == FinishReason.MAX_TOKENS:
107
+ finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
108
+ while autocomplete and response.candidates and finish_reason == FinishReason.MAX_TOKENS.value:
108
109
  BaseLLMClient._append_generated_part(messages, response)
109
110
 
110
111
  response = self._create(
@@ -116,6 +117,7 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
116
117
  tools=tools,
117
118
  tool_config=tool_config,
118
119
  )
120
+ finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
119
121
  total_count += BaseLLMClient._response_out_tokens(response)
120
122
  if max_tokens is not None and total_count >= max_tokens:
121
123
  break
@@ -436,7 +438,8 @@ class BaseLLMClientAsync(ABC, utils.InheritDecoratorsMixin):
436
438
 
437
439
  total_count = BaseLLMClient._response_out_tokens(response)
438
440
 
439
- while autocomplete and response.candidates and response.candidates[0].finish_reason == FinishReason.MAX_TOKENS:
441
+ finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
442
+ while autocomplete and response.candidates and finish_reason == FinishReason.MAX_TOKENS.value:
440
443
  BaseLLMClient._append_generated_part(messages, response)
441
444
 
442
445
  response = await self._create(
@@ -448,6 +451,7 @@ class BaseLLMClientAsync(ABC, utils.InheritDecoratorsMixin):
448
451
  tools=tools,
449
452
  tool_config=tool_config,
450
453
  )
454
+ finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
451
455
  total_count += BaseLLMClient._response_out_tokens(response)
452
456
  if max_tokens is not None and total_count >= max_tokens:
453
457
  break
{promptbuilder-0.4.25.dist-info → promptbuilder-0.4.27.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: promptbuilder
3
- Version: 0.4.25
3
+ Version: 0.4.27
4
4
  Summary: Library for building prompts for LLMs
5
5
  Home-page: https://github.com/kapulkin/promptbuilder
6
6
  Author: Kapulkin Stanislav
{promptbuilder-0.4.25.dist-info → promptbuilder-0.4.27.dist-info}/RECORD RENAMED
@@ -9,7 +9,7 @@ promptbuilder/agent/utils.py,sha256=vTkphKw04v_QDIJtoB2JKK0RGY6iI1t_0LbmuStunzI,
9
9
  promptbuilder/llm_client/__init__.py,sha256=wJ33cnRtZX_YPsbcGxEu3SEZMOhPX7-fHI59MEPUe7I,517
10
10
  promptbuilder/llm_client/aisuite_client.py,sha256=_TfB1xv73kIn4n8ulV3bj8bHOVm4DOD5uIdX-pbYoXY,15563
11
11
  promptbuilder/llm_client/anthropic_client.py,sha256=HSIAZVOQoi3hinjhBVGqpt91k0x38xj6EUsPSUrlAA0,28076
12
- promptbuilder/llm_client/base_client.py,sha256=dCeXoxmFvZvZI2vNQBJ2u2bANM75gv4-D0Bg4QGwwq4,27835
12
+ promptbuilder/llm_client/base_client.py,sha256=j7v0tcL-zvW7Md83o2Vbf1_w7A4VwkJZjZsfivl9Hkw,28377
13
13
  promptbuilder/llm_client/bedrock_client.py,sha256=rJMzVV7x_sNJ1nVVqWU1sU-Pq7xlxFxIa_hTD2wtM1Y,27943
14
14
  promptbuilder/llm_client/config.py,sha256=exQEm35wp7lK5SfXNpN5H9VZEb2LVa4pyZ-cxGt1U-U,1124
15
15
  promptbuilder/llm_client/exceptions.py,sha256=t-X7r_a8B1jNu8eEavde1jXu5dz97yV3IG4YHOtgh0Y,4836
@@ -19,8 +19,8 @@ promptbuilder/llm_client/main.py,sha256=5r_MhKVTD4cS90AHR89JJRKiWYBk35Y3JvhvmOxk
19
19
  promptbuilder/llm_client/openai_client.py,sha256=lT0RCiixJBoCtzUbL_0J5NQ5G8KGONzK3dQ73_NgL78,24789
20
20
  promptbuilder/llm_client/types.py,sha256=kgbg5FRzvZwu98y1OhAZJDneXBNPnsFZueQCr9HXIY4,8063
21
21
  promptbuilder/llm_client/utils.py,sha256=79lvSppjrrItHB5MIozbp_5Oq7TsOK4Qzt9Ae3XMLFw,7624
22
- promptbuilder-0.4.25.dist-info/licenses/LICENSE,sha256=fqXmInzgsvEOIaKSBgcrwKyYCGYF0MKErJ0YivtODcc,1096
23
- promptbuilder-0.4.25.dist-info/METADATA,sha256=YnG9z3jd7CVu7AIrvFwcm7NXlPmzNMX_T_s6xoETPrY,3729
24
- promptbuilder-0.4.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
25
- promptbuilder-0.4.25.dist-info/top_level.txt,sha256=UBVcYn4UgrPy3O3fmmnPEU_kieuplBMgheetIMei4EI,14
26
- promptbuilder-0.4.25.dist-info/RECORD,,
22
+ promptbuilder-0.4.27.dist-info/licenses/LICENSE,sha256=fqXmInzgsvEOIaKSBgcrwKyYCGYF0MKErJ0YivtODcc,1096
23
+ promptbuilder-0.4.27.dist-info/METADATA,sha256=A9oBv00dCQPHEzBVpIbaM1dbb-evsTy2qOXFnvXJ-yk,3729
24
+ promptbuilder-0.4.27.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
25
+ promptbuilder-0.4.27.dist-info/top_level.txt,sha256=UBVcYn4UgrPy3O3fmmnPEU_kieuplBMgheetIMei4EI,14
26
+ promptbuilder-0.4.27.dist-info/RECORD,,