agentle 0.9.21__py3-none-any.whl → 0.9.23__py3-none-any.whl

agentle/agents/apis/__init__.py

@@ -22,14 +22,14 @@ from agentle.agents.apis.object_serialization_style import ObjectSerializationSt
 from agentle.agents.apis.parameter_location import ParameterLocation
 from agentle.agents.apis.request_config import (
     CacheStrategy,
-    CircuitBreaker,
-    CircuitBreakerError,
-    RateLimiter,
     RequestConfig,
-    ResponseCache,
     RetryStrategy,
-    RateLimitError,
 )
+from agentle.agents.apis.circuit_breaker import CircuitBreaker
+from agentle.agents.apis.rate_limiter import RateLimiter
+from agentle.agents.apis.response_cache import ResponseCache
+from agentle.agents.apis.circuit_breaker_error import CircuitBreakerError
+from agentle.agents.apis.rate_limit_error import RateLimitError
 from agentle.agents.apis.request_hook import RequestHook
 from agentle.agents.apis.object_schema import ObjectSchema
 from agentle.agents.apis.array_schema import ArraySchema

agentle/agents/apis/request_config.py

@@ -12,33 +12,16 @@ from rsb.models.base_model import BaseModel
 from rsb.models.field import Field
 
 from agentle.agents.apis.cache_strategy import CacheStrategy
-from agentle.agents.apis.circuit_breaker import CircuitBreaker
-from agentle.agents.apis.circuit_breaker_error import CircuitBreakerError
-from agentle.agents.apis.rate_limiter import RateLimiter
-from agentle.agents.apis.rate_limit_error import RateLimitError
-from agentle.agents.apis.response_cache import ResponseCache
 from agentle.agents.apis.retry_strategy import RetryStrategy
 
-# Re-export for backward compatibility
-__all__ = [
-    "RequestConfig",
-    "RetryStrategy",
-    "CacheStrategy",
-    "CircuitBreaker",
-    "RateLimiter",
-    "ResponseCache",
-    "CircuitBreakerError",
-    "RateLimitError",
-]
-
 
 class RequestConfig(BaseModel):
     """
     Enhanced configuration for HTTP requests.
-    
+
     This configuration can be set at both API-level and per-endpoint level.
     Endpoint-level configs override API-level configs.
-    
+
     Example:
         ```python
         # API-level config (applies to all endpoints)
@@ -47,7 +30,7 @@ class RequestConfig(BaseModel):
             max_retries=3,
             enable_caching=True
         )
-        
+
         # Per-endpoint override
         endpoint_config = RequestConfig(
             timeout=60.0,  # Override for this specific endpoint

agentle/generations/providers/google/adapters/generate_generate_content_response_to_generation_adapter.py

@@ -253,16 +253,19 @@ class GenerateGenerateContentResponseToGenerationAdapter[T](
                 _all_parts.extend(_parts)
 
                 if _optional_model is not None:
-                    optional_model = parse_streaming_json(
-                        "".join([str(p.text) for p in _all_parts]),
+                    # Parse streaming JSON and update final_parsed
+                    accumulated_json_text = "".join([str(p.text) for p in _all_parts])
+                    parsed_optional_model = parse_streaming_json(
+                        accumulated_json_text,
                         model=_optional_model,
                     )
-                    chunk.parsed = optional_model
+                    # Cast the optional model back to T for use in the generation
+                    final_parsed = cast(T, parsed_optional_model)
                 else:
-                    chunk.parsed = None
+                    final_parsed = None
 
-            # Extract parsed data (usually only available in final chunk)
-            if hasattr(chunk, "parsed") and chunk.parsed is not None:
+            # Also check if chunk has parsed attribute from Google API
+            elif hasattr(chunk, "parsed") and chunk.parsed is not None:
                 final_parsed = cast(T | None, chunk.parsed)
 
             # Extract usage (usually only in final chunk)

agentle/generations/providers/google/google_generation_provider.py

@@ -182,6 +182,37 @@ class GoogleGenerationProvider(GenerationProvider):
         """
         return "google"
 
+    @overload
+    def stream_async[T](
+        self,
+        *,
+        model: str | ModelKind | None = None,
+        messages: Sequence[Message],
+        response_schema: type[T],
+        generation_config: GenerationConfig | GenerationConfigDict | None = None,
+    ) -> AsyncGenerator[Generation[T], None]: ...
+
+    @overload
+    def stream_async(
+        self,
+        *,
+        model: str | ModelKind | None = None,
+        messages: Sequence[Message],
+        response_schema: None = None,
+        generation_config: GenerationConfig | GenerationConfigDict | None = None,
+        tools: Sequence[Tool],
+    ) -> AsyncGenerator[Generation[WithoutStructuredOutput], None]: ...
+
+    @overload
+    def stream_async(
+        self,
+        *,
+        model: str | ModelKind | None = None,
+        messages: Sequence[Message],
+        response_schema: None = None,
+        generation_config: GenerationConfig | GenerationConfigDict | None = None,
+    ) -> AsyncGenerator[Generation[WithoutStructuredOutput], None]: ...
+
     async def stream_async[T = WithoutStructuredOutput](
         self,
         *,
@@ -190,7 +221,7 @@ class GoogleGenerationProvider(GenerationProvider):
         response_schema: type[T] | None = None,
         generation_config: GenerationConfig | GenerationConfigDict | None = None,
         tools: Sequence[Tool] | None = None,
-    ) -> AsyncGenerator[Generation[WithoutStructuredOutput], None]:
+    ) -> AsyncGenerator[Generation[T], None]:
         from google.genai import types
 
         if self._normalize_generation_config(generation_config).n > 1:
@@ -260,6 +291,7 @@ class GoogleGenerationProvider(GenerationProvider):
             tools=_tools,
             max_output_tokens=_generation_config.max_output_tokens,
             response_schema=response_schema if bool(response_schema) else None,
+            response_mime_type="application/json" if bool(response_schema) else None,
             automatic_function_calling=types.AutomaticFunctionCallingConfig(
                 disable=disable_function_calling,
                 maximum_remote_calls=maximum_remote_calls,
@@ -295,10 +327,8 @@ class GoogleGenerationProvider(GenerationProvider):
             raise
 
         # Create the response
-        response = GenerateGenerateContentResponseToGenerationAdapter[
-            WithoutStructuredOutput
-        ](
-            response_schema=None,
+        response = GenerateGenerateContentResponseToGenerationAdapter[T](
+            response_schema=response_schema,
             model=used_model,
         ).adapt(generate_content_response_stream)
 

{agentle-0.9.21.dist-info → agentle-0.9.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentle
-Version: 0.9.21
+Version: 0.9.23
 Summary: ...
 Author-email: Arthur Brenno <64020210+arthurbrenno@users.noreply.github.com>
 License-File: LICENSE

{agentle-0.9.21.dist-info → agentle-0.9.23.dist-info}/RECORD

@@ -62,7 +62,7 @@ agentle/agents/a2a/tasks/task_status_update_event.py,sha256=vF9WFfQV5nNmQpSKPU91
 agentle/agents/a2a/tasks/managment/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentle/agents/a2a/tasks/managment/in_memory.py,sha256=_G5VuXqEPBMtE6XJg1d7WmqFr1qzd0-99FoqM_qMwAE,23841
 agentle/agents/a2a/tasks/managment/task_manager.py,sha256=rBCuzu4DqIs55xDnwXY0w5Rs9ybv6OJpgpugAQLhtoU,3112
-agentle/agents/apis/__init__.py,sha256=BwDEEBkHCXCiUsFFbkFFW7VbK0Wim7avsO7HgKQ7GbY,2593
+agentle/agents/apis/__init__.py,sha256=PX7oAe0hRGvyLB295DrBF1VBsqgp5ZmGI4BCZvLUozo,2811
 agentle/agents/apis/api.py,sha256=dl146HQwawq4ll8lEcWl7ZM46-FBBgmX_TXEopY-Le8,25709
 agentle/agents/apis/api_key_authentication.py,sha256=MtMA4qkCjM3ou42a1fDgKI4u3NQkB_Zr-gEA5_oysZ0,1311
 agentle/agents/apis/api_key_location.py,sha256=0pj_8rTkd0pkUJ2eP_Kur3AvT3JD8JpFIxQsWDzeg_c,188
@@ -93,7 +93,7 @@ agentle/agents/apis/parameter_location.py,sha256=k4vSvrljwsxorV_sf2Ch_OP4zvIGvhH
 agentle/agents/apis/primitive_schema.py,sha256=fmDiMrXxtixORKIDvg8dx_h7w1AqPXpFOK8q_XxS_UQ,705
 agentle/agents/apis/rate_limit_error.py,sha256=9ukRInSwf6PPEg5fPLm-eq4xfgYOExraOgGCRKzVbQk,116
 agentle/agents/apis/rate_limiter.py,sha256=EZ-39YMDNeotygdvxP3B735cJexSk0-20zO0pxaGKiU,1842
-agentle/agents/apis/request_config.py,sha256=3KwVpKIrvNQG38wTAyDbfb6EYOfAr0Y0kd3prNHjKC4,5413
+agentle/agents/apis/request_config.py,sha256=vHKtoW9uxh2lTsuuZ38B2_EMcrLmOtUvVhL8VkOxjZY,4850
 agentle/agents/apis/request_hook.py,sha256=C9p6obF0myGkHoIScWuJTMp2RZlg3eEsbn80mDlE_sw,369
 agentle/agents/apis/response_cache.py,sha256=l-Ec_YVf1phhaTKWNdAORBDWZS3pXK_BBUHGP-oQkqQ,1617
 agentle/agents/apis/retry_strategy.py,sha256=_W8ZXXmA0kPFLJ0uwwV9ZycmSnj4dsicisFrbN8FtAU,219
@@ -316,10 +316,10 @@ agentle/generations/providers/failover/__init__.py,sha256=VBv_ZZlBFwwBRD2PZUEKi2
 agentle/generations/providers/failover/failover_generation_provider.py,sha256=ixZk1zAP_Gu3HTf6nSo_ZDTl91U6x8vR0AyNHKB2b8k,24562
 agentle/generations/providers/google/__init__.py,sha256=6H9LKSrN7Xyz5ZJ3PI4DIwOsAi8A84YoLSO9ms67cYI,105
 agentle/generations/providers/google/function_calling_config.py,sha256=5ZL1FDCfeVVe9a04BpHC0_nBlLhmWg9nGoHVvCXNcI4,1226
-agentle/generations/providers/google/google_generation_provider.py,sha256=zWpD2JdaV75czsw8ngGcAiXbzYzTwXoJg3TBrUetm5w,24540
+agentle/generations/providers/google/google_generation_provider.py,sha256=9Dl8D7npzpHL7qwO7Mgqiu6fzmWzPh-7jfvB6ZuUdYI,25590
 agentle/generations/providers/google/adapters/__init__.py,sha256=sUwLR-CDzvWHn6hwqSsHYztws6DGx6ZiKp32vYeVMfk,1003
 agentle/generations/providers/google/adapters/agentle_tool_to_google_tool_adapter.py,sha256=XQM_aL5zVfrRbNwtO_lDwXqvT8iX83trDNbEPSjZLmI,28482
-agentle/generations/providers/google/adapters/generate_generate_content_response_to_generation_adapter.py,sha256=PCm0imfvMFssMENrQGbHJKSOG1hOqMyilX3Eo2PO3Ls,16960
+agentle/generations/providers/google/adapters/generate_generate_content_response_to_generation_adapter.py,sha256=Z9v7c3NnnUFb-sqXQJHe_3ERb-o0eYzO_1JIGa2bRIM,17197
 agentle/generations/providers/google/adapters/google_content_to_generated_assistant_message_adapter.py,sha256=4sOcT9VLr58YfEnkOm392odhNzzmbTspnnmrz5K6bHc,7829
 agentle/generations/providers/google/adapters/google_part_to_part_adapter.py,sha256=infqHs_LVjpzIOipIVbRHQesXTKoOHVLiTUFaPhaBV8,6520
 agentle/generations/providers/google/adapters/message_to_google_content_adapter.py,sha256=sJ6MEn_R-KkLTEsNhxmKhBDbzlKwOUYqinydag6hqi4,6071
@@ -1007,7 +1007,7 @@ agentle/web/actions/scroll.py,sha256=WqVVAORNDK3BL1oASZBPmXJYeSVkPgAOmWA8ibYO82I
 agentle/web/actions/viewport.py,sha256=KCwm88Pri19Qc6GLHC69HsRxmdJz1gEEAODfggC_fHo,287
 agentle/web/actions/wait.py,sha256=IKEywjf-KC4ni9Gkkv4wgc7bY-hk7HwD4F-OFWlyf2w,571
 agentle/web/actions/write_text.py,sha256=9mxfHcpKs_L7BsDnJvOYHQwG8M0GWe61SRJAsKk3xQ8,748
-agentle-0.9.21.dist-info/METADATA,sha256=upRbzrSFVg5heeNeBqLVlYcl9NTLrIuZEIx15PdpoK8,86849
-agentle-0.9.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agentle-0.9.21.dist-info/licenses/LICENSE,sha256=T90S9vqRS6qP-voULxAcvwEs558wRRo6dHuZrjgcOUI,1085
-agentle-0.9.21.dist-info/RECORD,,
+agentle-0.9.23.dist-info/METADATA,sha256=OQxUrAm6kxfo2dh5nIh5rkJ2ZYconQ_QzC5D7AgSFqQ,86849
+agentle-0.9.23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agentle-0.9.23.dist-info/licenses/LICENSE,sha256=T90S9vqRS6qP-voULxAcvwEs558wRRo6dHuZrjgcOUI,1085
+agentle-0.9.23.dist-info/RECORD,,