promptbuilder 0.4.29__py3-none-any.whl → 0.4.30__py3-none-any.whl

promptbuilder/llm_client/base_client.py

@@ -121,9 +121,10 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
             total_count += BaseLLMClient._response_out_tokens(response)
             if max_tokens is not None and total_count >= max_tokens:
                 break
-        if response.candidates:
-            BaseLLMClient._append_generated_part(messages, response)
-            response.candidates[0].content = messages[-1] if len(messages) > 0 else None
+        if response.candidates and response.candidates[0].content:
+            appended_message = BaseLLMClient._append_generated_part(messages, response)
+            if appended_message is not None:
+                response.candidates[0].content = appended_message
         return response
 
     @logfire_decorators.create
@@ -245,7 +246,7 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
     
 
     @staticmethod
-    def _append_generated_part(messages: list[Content], response: Response):
+    def _append_generated_part(messages: list[Content], response: Response) -> Content | None:
         assert(response.candidates and response.candidates[0].content), "Response must contain at least one candidate with content."
 
         text_parts = [
@@ -262,7 +263,7 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
                 response_text = "".join(part.text for part in thought_parts)
                 is_thought = True
             else:
-                raise ValueError("No text or thought found in the response parts.")
+                return None
         
         if len(messages) > 0 and messages[-1].role == "model":
             message_to_append = messages[-1]
@@ -274,6 +275,7 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
                 message_to_append.parts.append(Part(text=response_text, thought=is_thought))
         else:
             messages.append(Content(parts=[Part(text=response_text, thought=is_thought)], role="model"))
+        return messages[-1]
 
     @staticmethod
     def _response_out_tokens(response: Response):
@@ -458,9 +460,10 @@ class BaseLLMClientAsync(ABC, utils.InheritDecoratorsMixin):
             total_count += BaseLLMClient._response_out_tokens(response)
             if max_tokens is not None and total_count >= max_tokens:
                 break
-        if response.candidates:
-            BaseLLMClient._append_generated_part(messages, response)
-            response.candidates[0].content = messages[-1] if len(messages) > 0 else None
+        if response.candidates and response.candidates[0].content:
+            appended_message = BaseLLMClient._append_generated_part(messages, response)
+            if appended_message is not None:
+                response.candidates[0].content = appended_message
         return response
 
     @logfire_decorators.create_async

{promptbuilder-0.4.29.dist-info → promptbuilder-0.4.30.dist-info}/METADATA

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

{promptbuilder-0.4.29.dist-info → promptbuilder-0.4.30.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=T-51WbVuMktCdoI0Z7Mk4LcaqV6GRqoDiZeZJPKoimI,28763
+promptbuilder/llm_client/base_client.py,sha256=2VXc3Jw1ugu8VTTeK72Ntvh2Zobg23uxw1a6qY6W6HI,28908
 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
@@ -20,8 +20,8 @@ promptbuilder/llm_client/main.py,sha256=m-9jM2IYMFy6aZBUmPb52wpFlIK0H1aRj293oFmx
 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.29.dist-info/licenses/LICENSE,sha256=fqXmInzgsvEOIaKSBgcrwKyYCGYF0MKErJ0YivtODcc,1096
-promptbuilder-0.4.29.dist-info/METADATA,sha256=rfxQaTbrKl5se6wrV6b2QfMDhtSrxG7WbkG06fGtNCo,3729
-promptbuilder-0.4.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-promptbuilder-0.4.29.dist-info/top_level.txt,sha256=UBVcYn4UgrPy3O3fmmnPEU_kieuplBMgheetIMei4EI,14
-promptbuilder-0.4.29.dist-info/RECORD,,
+promptbuilder-0.4.30.dist-info/licenses/LICENSE,sha256=fqXmInzgsvEOIaKSBgcrwKyYCGYF0MKErJ0YivtODcc,1096
+promptbuilder-0.4.30.dist-info/METADATA,sha256=wKv1yEjLK2Zm8jOyXaY0_7mNcug1ZorhfHlPDFw3sA4,3729
+promptbuilder-0.4.30.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+promptbuilder-0.4.30.dist-info/top_level.txt,sha256=UBVcYn4UgrPy3O3fmmnPEU_kieuplBMgheetIMei4EI,14
+promptbuilder-0.4.30.dist-info/RECORD,,