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

promptbuilder/llm_client/base_client.py

@@ -5,6 +5,7 @@ import hashlib
 import logging
 from abc import ABC, abstractmethod
 from typing import Iterator, AsyncIterator, Literal, overload
+from pydantic import BaseModel
 
 from promptbuilder.llm_client.types import Response, Content, Part, Tool, ToolConfig, FunctionCall, FunctionCallingConfig, Json, ThinkingConfig, ApiKey, PydanticStructure, ResultType, FinishReason
 import promptbuilder.llm_client.utils as utils
@@ -605,7 +606,12 @@ class CachedLLMClient(BaseLLMClient):
                 with open(cache_path, "rt") as f:
                     cache_data = json.load(f)
                     if cache_data["full_model_name"] == llm_client.full_model_name and json.dumps(cache_data["request"]) == json.dumps(messages_dump):
-                        return Response(**cache_data["response"]), messages_dump, cache_path
+                        response = Response(**cache_data["response"])
+                        result_type = kwargs.get("result_type", None)
+                        if result_type is not None and isinstance(result_type, type(BaseModel)):
+                            response.parsed = result_type.model_validate(response.parsed)
+
+                        return response, messages_dump, cache_path
                     else:
                         logger.debug(f"Cache mismatch for {key}")
             except (json.JSONDecodeError, KeyError) as e:
@@ -637,4 +643,3 @@ class CachedLLMClientAsync(BaseLLMClientAsync):
         response = await self.llm_client.create(messages, **kwargs)
         CachedLLMClient.save_cache(cache_path, self.llm_client.full_model_name, messages_dump, response)
         return response
-

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: promptbuilder
-Version: 0.4.21
+Version: 0.4.22
 Summary: Library for building prompts for LLMs
 Home-page: https://github.com/kapulkin/promptbuilder
 Author: Kapulkin Stanislav
@@ -14,7 +14,7 @@ License-File: LICENSE
 Requires-Dist: pydantic
 Requires-Dist: pytest
 Requires-Dist: aisuite-async
-Requires-Dist: google-genai<=1.21.0,>=1.4.0
+Requires-Dist: google-genai>=1.4.0
 Requires-Dist: anthropic
 Requires-Dist: openai
 Requires-Dist: aioboto3

{promptbuilder-0.4.21.dist-info → promptbuilder-0.4.22.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=GS-Qb20WtZnljmEUD2ibhTHDet7exoyhQ0_mGNAEKlg,24219
+promptbuilder/llm_client/base_client.py,sha256=xDDCIAIuLVf6ouItLVzGhP0a3hkr71oTH9jH790zkWE,24556
 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
@@ -19,8 +19,8 @@ promptbuilder/llm_client/main.py,sha256=5r_MhKVTD4cS90AHR89JJRKiWYBk35Y3JvhvmOxk
 promptbuilder/llm_client/openai_client.py,sha256=GdyTbUPsbACXZYF0BnCRyLVw24_WM1R_MMr6pDpiiV4,24787
 promptbuilder/llm_client/types.py,sha256=2E-aPRb5uAkLFJocmjF1Lh2aQRq9r8a5JRIw-duHfjA,7460
 promptbuilder/llm_client/utils.py,sha256=79lvSppjrrItHB5MIozbp_5Oq7TsOK4Qzt9Ae3XMLFw,7624
-promptbuilder-0.4.21.dist-info/licenses/LICENSE,sha256=fqXmInzgsvEOIaKSBgcrwKyYCGYF0MKErJ0YivtODcc,1096
-promptbuilder-0.4.21.dist-info/METADATA,sha256=A44iObCMOJxcKjF6fYxU-O-QugjMPrgkJJJDu2Rm4Vc,3738
-promptbuilder-0.4.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-promptbuilder-0.4.21.dist-info/top_level.txt,sha256=UBVcYn4UgrPy3O3fmmnPEU_kieuplBMgheetIMei4EI,14
-promptbuilder-0.4.21.dist-info/RECORD,,
+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,,