promptbuilder 0.4.24__py3-none-any.whl → 0.4.26__py3-none-any.whl

promptbuilder/llm_client/base_client.py

@@ -74,9 +74,6 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
         except json.JSONDecodeError as e:
             raise ValueError(f"Failed to parse LLM response as JSON:\n{text}")
 
-    @logfire_decorators.create
-    @utils.retry_cls
-    @utils.rpm_limit_cls
     def create(
         self,
         messages: list[Content],
@@ -107,7 +104,8 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
 
         total_count = BaseLLMClient._response_out_tokens(response)
 
-        while autocomplete and response.candidates and response.candidates[0].finish_reason == FinishReason.MAX_TOKENS:
+        finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
+        while autocomplete and response.candidates and finish_reason == FinishReason.MAX_TOKENS.value:
             BaseLLMClient._append_generated_part(messages, response)
 
             response = self._create(
@@ -124,6 +122,9 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
                 break
         return response
 
+    @logfire_decorators.create
+    @utils.retry_cls
+    @utils.rpm_limit_cls
     @abstractmethod
     def _create(
         self,
@@ -406,9 +407,6 @@ class BaseLLMClientAsync(ABC, utils.InheritDecoratorsMixin):
         """Return the model identifier"""
         return self.provider + ":" + self.model
     
-    @logfire_decorators.create_async
-    @utils.retry_cls_async
-    @utils.rpm_limit_cls_async
     async def create(
         self,
         messages: list[Content],
@@ -439,7 +437,8 @@ class BaseLLMClientAsync(ABC, utils.InheritDecoratorsMixin):
 
         total_count = BaseLLMClient._response_out_tokens(response)
 
-        while autocomplete and response.candidates and response.candidates[0].finish_reason == FinishReason.MAX_TOKENS:
+        finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
+        while autocomplete and response.candidates and finish_reason == FinishReason.MAX_TOKENS.value:
             BaseLLMClient._append_generated_part(messages, response)
 
             response = await self._create(
@@ -456,6 +455,9 @@ class BaseLLMClientAsync(ABC, utils.InheritDecoratorsMixin):
                 break
         return response
 
+    @logfire_decorators.create_async
+    @utils.retry_cls_async
+    @utils.rpm_limit_cls_async
     @abstractmethod
     async def _create(
         self,

{promptbuilder-0.4.24.dist-info → promptbuilder-0.4.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: promptbuilder
-Version: 0.4.24
+Version: 0.4.26
 Summary: Library for building prompts for LLMs
 Home-page: https://github.com/kapulkin/promptbuilder
 Author: Kapulkin Stanislav

{promptbuilder-0.4.24.dist-info → promptbuilder-0.4.26.dist-info}/RECORD

@@ -9,7 +9,7 @@ promptbuilder/agent/utils.py,sha256=vTkphKw04v_QDIJtoB2JKK0RGY6iI1t_0LbmuStunzI,
 promptbuilder/llm_client/__init__.py,sha256=wJ33cnRtZX_YPsbcGxEu3SEZMOhPX7-fHI59MEPUe7I,517
 promptbuilder/llm_client/aisuite_client.py,sha256=_TfB1xv73kIn4n8ulV3bj8bHOVm4DOD5uIdX-pbYoXY,15563
 promptbuilder/llm_client/anthropic_client.py,sha256=HSIAZVOQoi3hinjhBVGqpt91k0x38xj6EUsPSUrlAA0,28076
-promptbuilder/llm_client/base_client.py,sha256=0U449lHLjl3llyjQlB4hkMGOWl8cE-BtdT7L14AtIco,27835
+promptbuilder/llm_client/base_client.py,sha256=fFaSuEvSY_bu6CJigpa5aFwAfv3x_3W0Y8x22rtC5no,28085
 promptbuilder/llm_client/bedrock_client.py,sha256=rJMzVV7x_sNJ1nVVqWU1sU-Pq7xlxFxIa_hTD2wtM1Y,27943
 promptbuilder/llm_client/config.py,sha256=exQEm35wp7lK5SfXNpN5H9VZEb2LVa4pyZ-cxGt1U-U,1124
 promptbuilder/llm_client/exceptions.py,sha256=t-X7r_a8B1jNu8eEavde1jXu5dz97yV3IG4YHOtgh0Y,4836
@@ -19,8 +19,8 @@ promptbuilder/llm_client/main.py,sha256=5r_MhKVTD4cS90AHR89JJRKiWYBk35Y3JvhvmOxk
 promptbuilder/llm_client/openai_client.py,sha256=lT0RCiixJBoCtzUbL_0J5NQ5G8KGONzK3dQ73_NgL78,24789
 promptbuilder/llm_client/types.py,sha256=kgbg5FRzvZwu98y1OhAZJDneXBNPnsFZueQCr9HXIY4,8063
 promptbuilder/llm_client/utils.py,sha256=79lvSppjrrItHB5MIozbp_5Oq7TsOK4Qzt9Ae3XMLFw,7624
-promptbuilder-0.4.24.dist-info/licenses/LICENSE,sha256=fqXmInzgsvEOIaKSBgcrwKyYCGYF0MKErJ0YivtODcc,1096
-promptbuilder-0.4.24.dist-info/METADATA,sha256=pHr-63JQFZe_KY1u-uDkq41wu4FnBgyxGHZMAuRddfk,3729
-promptbuilder-0.4.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-promptbuilder-0.4.24.dist-info/top_level.txt,sha256=UBVcYn4UgrPy3O3fmmnPEU_kieuplBMgheetIMei4EI,14
-promptbuilder-0.4.24.dist-info/RECORD,,
+promptbuilder-0.4.26.dist-info/licenses/LICENSE,sha256=fqXmInzgsvEOIaKSBgcrwKyYCGYF0MKErJ0YivtODcc,1096
+promptbuilder-0.4.26.dist-info/METADATA,sha256=auEvAYB6we0dVXrJX6lZX24o9bV0UHCFZBZ8J9rHK5U,3729
+promptbuilder-0.4.26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+promptbuilder-0.4.26.dist-info/top_level.txt,sha256=UBVcYn4UgrPy3O3fmmnPEU_kieuplBMgheetIMei4EI,14
+promptbuilder-0.4.26.dist-info/RECORD,,