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

promptbuilder/llm_client/base_client.py

@@ -117,9 +117,13 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
                 tools=tools,
                 tool_config=tool_config,
             )
+            finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
             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
         return response
 
     @logfire_decorators.create
@@ -450,9 +454,13 @@ class BaseLLMClientAsync(ABC, utils.InheritDecoratorsMixin):
                 tools=tools,
                 tool_config=tool_config,
             )
+            finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
             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
         return response
 
     @logfire_decorators.create_async

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

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

{promptbuilder-0.4.26.dist-info → promptbuilder-0.4.28.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=fFaSuEvSY_bu6CJigpa5aFwAfv3x_3W0Y8x22rtC5no,28085
+promptbuilder/llm_client/base_client.py,sha256=T-51WbVuMktCdoI0Z7Mk4LcaqV6GRqoDiZeZJPKoimI,28763
 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.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,,
+promptbuilder-0.4.28.dist-info/licenses/LICENSE,sha256=fqXmInzgsvEOIaKSBgcrwKyYCGYF0MKErJ0YivtODcc,1096
+promptbuilder-0.4.28.dist-info/METADATA,sha256=O94B_otCBTHQHr5G_-VqiC5a_xj8rCSydUSzG0GABIo,3729
+promptbuilder-0.4.28.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+promptbuilder-0.4.28.dist-info/top_level.txt,sha256=UBVcYn4UgrPy3O3fmmnPEU_kieuplBMgheetIMei4EI,14
+promptbuilder-0.4.28.dist-info/RECORD,,