promptbuilder 0.4.25__py3-none-any.whl → 0.4.26__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 +4 -2
  2. {promptbuilder-0.4.25.dist-info → promptbuilder-0.4.26.dist-info}/METADATA +1 -1
  3. {promptbuilder-0.4.25.dist-info → promptbuilder-0.4.26.dist-info}/RECORD +6 -6
  4. {promptbuilder-0.4.25.dist-info → promptbuilder-0.4.26.dist-info}/WHEEL +0 -0
  5. {promptbuilder-0.4.25.dist-info → promptbuilder-0.4.26.dist-info}/licenses/LICENSE +0 -0
  6. {promptbuilder-0.4.25.dist-info → promptbuilder-0.4.26.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(
@@ -436,7 +437,8 @@ class BaseLLMClientAsync(ABC, utils.InheritDecoratorsMixin):
436
437
 
437
438
  total_count = BaseLLMClient._response_out_tokens(response)
438
439
 
439
- while autocomplete and response.candidates and response.candidates[0].finish_reason == FinishReason.MAX_TOKENS:
440
+ finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
441
+ while autocomplete and response.candidates and finish_reason == FinishReason.MAX_TOKENS.value:
440
442
  BaseLLMClient._append_generated_part(messages, response)
441
443
 
442
444
  response = await self._create(
{promptbuilder-0.4.25.dist-info → promptbuilder-0.4.26.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.26
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.26.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=fFaSuEvSY_bu6CJigpa5aFwAfv3x_3W0Y8x22rtC5no,28085
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.26.dist-info/licenses/LICENSE,sha256=fqXmInzgsvEOIaKSBgcrwKyYCGYF0MKErJ0YivtODcc,1096
23
+ promptbuilder-0.4.26.dist-info/METADATA,sha256=auEvAYB6we0dVXrJX6lZX24o9bV0UHCFZBZ8J9rHK5U,3729
24
+ promptbuilder-0.4.26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
25
+ promptbuilder-0.4.26.dist-info/top_level.txt,sha256=UBVcYn4UgrPy3O3fmmnPEU_kieuplBMgheetIMei4EI,14
26
+ promptbuilder-0.4.26.dist-info/RECORD,,