promptbuilder 0.4.22__py3-none-any.whl → 0.4.23__py3-none-any.whl

promptbuilder/llm_client/base_client.py

@@ -179,7 +179,7 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
             tool_config=ToolConfig(function_calling_config=FunctionCallingConfig(mode=tool_choice_mode)),
         )
 
-        while autocomplete and response.candidates and response.candidates[0].finish_reason not in [FinishReason.STOP, FinishReason.MAX_TOKENS]:
+        while autocomplete and response.candidates and response.candidates[0].finish_reason in [FinishReason.STOP, FinishReason.MAX_TOKENS]:
             BaseLLMClient._append_generated_part(messages, response)
 
             response = self.create(

promptbuilder/llm_client/types.py

@@ -1,9 +1,9 @@
 import logging
 from abc import ABC, abstractmethod
-from typing import Optional, Any, Callable, Literal, TypeVar, Self
+from typing import Optional, Any, Callable, Literal, TypeVar, Self, Protocol, runtime_checkable
 from enum import Enum
 
-from pydantic import BaseModel, model_validator
+from pydantic import BaseModel, model_validator, ConfigDict
 
 
 logger = logging.getLogger(__name__)
@@ -17,6 +17,16 @@ PydanticStructure = TypeVar("PydanticStructure", bound=BaseModel)
 type ResultType = Literal["json"] | type[PydanticStructure] | None
 
 
+@runtime_checkable
+class PartLike(Protocol):
+    """Protocol for Part-like objects that have the same interface as Part."""
+    text: Optional[str]
+    function_call: Optional[Any]  # Using Any to allow different FunctionCall types
+    function_response: Optional[Any]  # Using Any to allow different FunctionResponse types
+    thought: Optional[bool]
+    inline_data: Optional[Any]  # Using Any to allow different Blob types
+
+
 class CustomApiKey(ABC):
     @abstractmethod
     def __hash__(self):
@@ -80,14 +90,16 @@ class Part(BaseModel):
         return cls(inline_data=inline_data)
 
 class Content(BaseModel):
-    parts: Optional[list[Part]] = None
-    role: Optional[Role] = None
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+    
+    parts: list[Part | PartLike] | None = None
+    role: Role | None = None
     
     def as_str(self) -> str:
         if self.parts is None:
             return ""
         else:
-            return "\n".join([part.as_str() for part in self.parts])
+            return "\n".join([(part.text or "") for part in self.parts])
 
 
 class FinishReason(Enum):

{promptbuilder-0.4.22.dist-info → promptbuilder-0.4.23.dist-info}/METADATA

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

{promptbuilder-0.4.22.dist-info → promptbuilder-0.4.23.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=aMqg05zefzck9Lz7pm7jZoKFdzr_ymFYhrAjZtzdHlQ,15561
 promptbuilder/llm_client/anthropic_client.py,sha256=vWuyFZL_LohOE0UYjB1-zTr4tJZMUcGk8H10gpjzdkk,28074
-promptbuilder/llm_client/base_client.py,sha256=xDDCIAIuLVf6ouItLVzGhP0a3hkr71oTH9jH790zkWE,24556
+promptbuilder/llm_client/base_client.py,sha256=tOTdv_dQOKNVcPaK_O-6aDCsK93TpXh_uqUXDIFHo4c,24552
 promptbuilder/llm_client/bedrock_client.py,sha256=e9vUClbybQb32028oDBW6IbyPYqj1ZSSv9y36ZqUWxM,27941
 promptbuilder/llm_client/config.py,sha256=exQEm35wp7lK5SfXNpN5H9VZEb2LVa4pyZ-cxGt1U-U,1124
 promptbuilder/llm_client/exceptions.py,sha256=t-X7r_a8B1jNu8eEavde1jXu5dz97yV3IG4YHOtgh0Y,4836
@@ -17,10 +17,10 @@ promptbuilder/llm_client/google_client.py,sha256=y1_CFXBijUiRTyAJsh-8a6CGIwwlZBs
 promptbuilder/llm_client/logfire_decorators.py,sha256=un_QnIekypOEcqTZ5v1y9pwijGnF95xwnwKO5rFSHVY,9667
 promptbuilder/llm_client/main.py,sha256=5r_MhKVTD4cS90AHR89JJRKiWYBk35Y3JvhvmOxkYHc,8110
 promptbuilder/llm_client/openai_client.py,sha256=GdyTbUPsbACXZYF0BnCRyLVw24_WM1R_MMr6pDpiiV4,24787
-promptbuilder/llm_client/types.py,sha256=2E-aPRb5uAkLFJocmjF1Lh2aQRq9r8a5JRIw-duHfjA,7460
+promptbuilder/llm_client/types.py,sha256=P7lM-Q1e0z6r5y_X3Vy107TR__b59f26ALCUOs-ap58,8015
 promptbuilder/llm_client/utils.py,sha256=79lvSppjrrItHB5MIozbp_5Oq7TsOK4Qzt9Ae3XMLFw,7624
-promptbuilder-0.4.22.dist-info/licenses/LICENSE,sha256=fqXmInzgsvEOIaKSBgcrwKyYCGYF0MKErJ0YivtODcc,1096
-promptbuilder-0.4.22.dist-info/METADATA,sha256=GtIz1H1kWvHz_1nyCYbkzmkUwsGLYE5jtaKCpt59eHc,3729
-promptbuilder-0.4.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-promptbuilder-0.4.22.dist-info/top_level.txt,sha256=UBVcYn4UgrPy3O3fmmnPEU_kieuplBMgheetIMei4EI,14
-promptbuilder-0.4.22.dist-info/RECORD,,
+promptbuilder-0.4.23.dist-info/licenses/LICENSE,sha256=fqXmInzgsvEOIaKSBgcrwKyYCGYF0MKErJ0YivtODcc,1096
+promptbuilder-0.4.23.dist-info/METADATA,sha256=tsEFAL-TTW24bAybDiFX9BmbOO8AFvk8ZO4D1870vEQ,3729
+promptbuilder-0.4.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+promptbuilder-0.4.23.dist-info/top_level.txt,sha256=UBVcYn4UgrPy3O3fmmnPEU_kieuplBMgheetIMei4EI,14
+promptbuilder-0.4.23.dist-info/RECORD,,