agentle 0.9.22__tar.gz → 0.9.23__tar.gz

agentle-0.9.23/CHANGELOG.md

@@ -0,0 +1,5 @@
+# Changelog
+
+## v0.9.23
+
+- fix(google_provider): wrongly setting "parsed" attribute on generation

{agentle-0.9.22 → agentle-0.9.23}/PKG-INFO

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

{agentle-0.9.22 → agentle-0.9.23}/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-0.9.22 → agentle-0.9.23}/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.22 → agentle-0.9.23}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentle"
-version = "v0.9.22"
+version = "v0.9.23"
 description = "..."
 readme = "README.md"
 requires-python = ">=3.13"

agentle-0.9.22/CHANGELOG.md

@@ -1,5 +0,0 @@
-# Changelog
-
-## v0.9.22
-
-- fix(apis): import cycle errors