agentle 0.9.29__tar.gz → 0.9.31__tar.gz

agentle-0.9.31/CHANGELOG.md

@@ -0,0 +1,5 @@
+# Changelog
+
+## v0.9.31
+
+- 

{agentle-0.9.29 → agentle-0.9.31}/PKG-INFO

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

{agentle-0.9.29 → agentle-0.9.31}/agentle/agents/apis/object_schema.py

@@ -46,20 +46,20 @@ class ObjectSchema(BaseModel):
     ) -> ObjectSchema | ArraySchema | PrimitiveSchema:
         """
         Recursively convert a JSON Schema definition to Agentle schema types.
-        
+
         This method handles deeply nested objects, arrays, and primitives,
         making it easy to convert complex JSON Schema definitions.
-        
+
         Args:
             schema: JSON Schema definition (dict with 'type', 'properties', etc.)
-            
+
         Returns:
             Appropriate schema type (ObjectSchema, ArraySchema, or PrimitiveSchema)
-            
+
         Example:
             ```python
             from agentle.agents.apis.object_schema import ObjectSchema
-            
+
             json_schema = {
                 "type": "object",
                 "properties": {
@@ -79,7 +79,7 @@ class ObjectSchema(BaseModel):
                     }
                 }
             }
-            
+
             schema = ObjectSchema.from_json_schema(json_schema)
             ```
         """

{agentle-0.9.29 → agentle-0.9.31}/agentle/agents/apis/params/boolean_param.py

@@ -11,21 +11,21 @@ def boolean_param(
     location: ParameterLocation = ParameterLocation.QUERY,
 ) -> EndpointParameter:
     """Create a boolean parameter.
-    
+
     Args:
         name: Parameter name
         description: Parameter description
         required: Whether the parameter is required
         default: Default value for the parameter
         location: Where the parameter should be placed in the request
-        
+
     Returns:
         EndpointParameter configured for boolean values
-        
+
     Example:
         ```python
         from agentle.agents.apis.params.boolean_param import boolean_param
-        
+
         boolean_param(
             name="enabled",
             description="Enable feature",

{agentle-0.9.29 → agentle-0.9.31}/agentle/agents/apis/params/number_param.py

@@ -14,7 +14,7 @@ def number_param(
     format: str | None = None,
 ) -> EndpointParameter:
     """Create a number (float/decimal) parameter.
-    
+
     Args:
         name: Parameter name
         description: Parameter description
@@ -24,14 +24,14 @@ def number_param(
         default: Default value for the parameter
         location: Where the parameter should be placed in the request
         format: Format hint (e.g., 'float', 'double', 'decimal')
-        
+
     Returns:
         EndpointParameter configured for number values
-        
+
     Example:
         ```python
         from agentle.agents.apis.params.number_param import number_param
-        
+
         number_param(
             name="price",
             description="Product price",

{agentle-0.9.29 → agentle-0.9.31}/agentle/agents/conversations/callback_conversation_store.py

@@ -46,11 +46,14 @@ Example:
 from __future__ import annotations
 
 from collections.abc import Awaitable, Callable, Sequence
-from typing import override
+from typing import Any, override
 
 from agentle.agents.conversations.conversation_store import ConversationStore
 from agentle.generations.models.messages.assistant_message import AssistantMessage
 from agentle.generations.models.messages.developer_message import DeveloperMessage
+from agentle.generations.models.messages.generated_assistant_message import (
+    GeneratedAssistantMessage,
+)
 from agentle.generations.models.messages.user_message import UserMessage
 
 
@@ -81,7 +84,14 @@ class CallbackConversationStore(ConversationStore):
             Awaitable[Sequence[DeveloperMessage | UserMessage | AssistantMessage]],
         ],
         add_callback: Callable[
-            [str, DeveloperMessage | UserMessage | AssistantMessage], Awaitable[None]
+            [
+                str,
+                DeveloperMessage
+                | UserMessage
+                | AssistantMessage
+                | GeneratedAssistantMessage[Any],
+            ],
+            Awaitable[None],
         ],
         clear_callback: Callable[[str], Awaitable[None]],
         message_limit: int | None = None,
@@ -146,8 +156,13 @@ class CallbackConversationStore(ConversationStore):
         return await self._get_callback(chat_id)
 
     @override
-    async def add_message_async(
-        self, chat_id: str, message: DeveloperMessage | UserMessage | AssistantMessage
+    async def add_message_async[T = Any](
+        self,
+        chat_id: str,
+        message: DeveloperMessage
+        | UserMessage
+        | AssistantMessage
+        | GeneratedAssistantMessage[T],
     ) -> None:
         """
         Add a message to the conversation using the user-provided add callback.

{agentle-0.9.29 → agentle-0.9.31}/agentle/agents/conversations/conversation_store.py

@@ -1,8 +1,12 @@
 import abc
 from collections.abc import Sequence
+from typing import Any
 
 from agentle.generations.models.messages.assistant_message import AssistantMessage
 from agentle.generations.models.messages.developer_message import DeveloperMessage
+from agentle.generations.models.messages.generated_assistant_message import (
+    GeneratedAssistantMessage,
+)
 from agentle.generations.models.messages.user_message import UserMessage
 
 
@@ -42,8 +46,13 @@ class ConversationStore(abc.ABC):
     ) -> Sequence[DeveloperMessage | UserMessage | AssistantMessage]: ...
 
     @abc.abstractmethod
-    async def add_message_async(
-        self, chat_id: str, message: DeveloperMessage | UserMessage | AssistantMessage
+    async def add_message_async[T = Any](
+        self,
+        chat_id: str,
+        message: DeveloperMessage
+        | UserMessage
+        | AssistantMessage
+        | GeneratedAssistantMessage[T],
     ) -> None: ...
 
     @abc.abstractmethod

{agentle-0.9.29 → agentle-0.9.31}/agentle/agents/conversations/firebase_conversation_store.py

@@ -1,11 +1,14 @@
 from __future__ import annotations
 
 from collections.abc import Sequence
-from typing import TYPE_CHECKING, override
+from typing import TYPE_CHECKING, Any, override
 
 from agentle.agents.conversations.conversation_store import ConversationStore
 from agentle.generations.models.messages.assistant_message import AssistantMessage
 from agentle.generations.models.messages.developer_message import DeveloperMessage
+from agentle.generations.models.messages.generated_assistant_message import (
+    GeneratedAssistantMessage,
+)
 from agentle.generations.models.messages.user_message import UserMessage
 
 if TYPE_CHECKING:
@@ -28,8 +31,13 @@ class FirebaseConversationStore(ConversationStore):
         self._collection_name = collection_name
 
     @override
-    async def add_message_async(
-        self, chat_id: str, message: DeveloperMessage | UserMessage | AssistantMessage
+    async def add_message_async[T = Any](
+        self,
+        chat_id: str,
+        message: DeveloperMessage
+        | UserMessage
+        | AssistantMessage
+        | GeneratedAssistantMessage[T],
     ) -> None:
         from google.cloud import firestore
 

{agentle-0.9.29 → agentle-0.9.31}/agentle/agents/conversations/json_file_conversation_store.py

@@ -7,6 +7,9 @@ from typing import Any, override
 from agentle.agents.conversations.conversation_store import ConversationStore
 from agentle.generations.models.messages.assistant_message import AssistantMessage
 from agentle.generations.models.messages.developer_message import DeveloperMessage
+from agentle.generations.models.messages.generated_assistant_message import (
+    GeneratedAssistantMessage,
+)
 from agentle.generations.models.messages.user_message import UserMessage
 
 
@@ -87,8 +90,12 @@ class JSONFileConversationStore(ConversationStore):
         # For other non-serializable objects, convert to string
         return str(obj)
 
-    def _message_to_dict(
-        self, message: DeveloperMessage | UserMessage | AssistantMessage
+    def _message_to_dict[T](
+        self,
+        message: DeveloperMessage
+        | UserMessage
+        | AssistantMessage
+        | GeneratedAssistantMessage[T],
     ) -> dict[str, Any]:
         """Convert a message object to a dictionary for JSON serialization."""
         # Get the basic message dictionary
@@ -177,8 +184,13 @@ class JSONFileConversationStore(ConversationStore):
             return UserMessage.model_validate(message_data)
 
     @override
-    async def add_message_async(
-        self, chat_id: str, message: DeveloperMessage | UserMessage | AssistantMessage
+    async def add_message_async[T = Any](
+        self,
+        chat_id: str,
+        message: DeveloperMessage
+        | UserMessage
+        | AssistantMessage
+        | GeneratedAssistantMessage[T],
     ) -> None:
         """Add a message to the conversation."""
         messages_data = self._load_messages(chat_id)

{agentle-0.9.29 → agentle-0.9.31}/agentle/agents/conversations/local_conversation_store.py

@@ -1,14 +1,19 @@
 from collections.abc import MutableMapping, MutableSequence, Sequence
-from typing import override
+from typing import Any, override
+
 from agentle.agents.conversations.conversation_store import ConversationStore
 from agentle.generations.models.messages.assistant_message import AssistantMessage
 from agentle.generations.models.messages.developer_message import DeveloperMessage
+from agentle.generations.models.messages.generated_assistant_message import (
+    GeneratedAssistantMessage,
+)
 from agentle.generations.models.messages.user_message import UserMessage
 
 
 class LocalConversationStore(ConversationStore):
     __messages: MutableMapping[
-        str, MutableSequence[DeveloperMessage | UserMessage | AssistantMessage]
+        str,
+        MutableSequence[DeveloperMessage | UserMessage | AssistantMessage],
     ]
 
     def __init__(
@@ -20,8 +25,13 @@ class LocalConversationStore(ConversationStore):
         self.__messages = {}
 
     @override
-    async def add_message_async(
-        self, chat_id: str, message: DeveloperMessage | UserMessage | AssistantMessage
+    async def add_message_async[T = Any](
+        self,
+        chat_id: str,
+        message: DeveloperMessage
+        | UserMessage
+        | AssistantMessage
+        | GeneratedAssistantMessage[T],
     ) -> None:
         if chat_id not in self.__messages:
             self.__messages[chat_id] = []
@@ -41,6 +51,9 @@ class LocalConversationStore(ConversationStore):
                     # Don't add message if limit reached and not overriding
                     return
 
+        if isinstance(message, GeneratedAssistantMessage):
+            message = message.to_assistant_message()
+
         self.__messages[chat_id].append(message)
 
     @override

{agentle-0.9.29 → agentle-0.9.31}/agentle/agents/conversations/mysql_conversation_store.py

@@ -8,6 +8,9 @@ from typing import TYPE_CHECKING, Any, override
 from agentle.agents.conversations.conversation_store import ConversationStore
 from agentle.generations.models.messages.assistant_message import AssistantMessage
 from agentle.generations.models.messages.developer_message import DeveloperMessage
+from agentle.generations.models.messages.generated_assistant_message import (
+    GeneratedAssistantMessage,
+)
 from agentle.generations.models.messages.user_message import UserMessage
 
 if TYPE_CHECKING:
@@ -74,7 +77,11 @@ class MySQLConversationStore(ConversationStore):
                 await conn.commit()
 
     def _message_to_dict(
-        self, message: DeveloperMessage | UserMessage | AssistantMessage
+        self,
+        message: DeveloperMessage
+        | UserMessage
+        | AssistantMessage
+        | GeneratedAssistantMessage[Any],
     ) -> dict[str, Any]:
         """
         Convert a Message object to a dictionary for JSON serialization.
@@ -125,8 +132,13 @@ class MySQLConversationStore(ConversationStore):
             return UserMessage.model_validate(message_dict)
 
     @override
-    async def add_message_async(
-        self, chat_id: str, message: DeveloperMessage | UserMessage | AssistantMessage
+    async def add_message_async[T = Any](
+        self,
+        chat_id: str,
+        message: DeveloperMessage
+        | UserMessage
+        | AssistantMessage
+        | GeneratedAssistantMessage[T],
     ) -> None:
         """
         Add a message to the conversation store.

{agentle-0.9.29 → agentle-0.9.31}/agentle/agents/conversations/postgres_conversation_store.py

@@ -7,6 +7,9 @@ from typing import TYPE_CHECKING, Any, override
 from agentle.agents.conversations.conversation_store import ConversationStore
 from agentle.generations.models.messages.assistant_message import AssistantMessage
 from agentle.generations.models.messages.developer_message import DeveloperMessage
+from agentle.generations.models.messages.generated_assistant_message import (
+    GeneratedAssistantMessage,
+)
 from agentle.generations.models.messages.user_message import UserMessage
 
 if TYPE_CHECKING:
@@ -84,7 +87,11 @@ class PostgresConversationStore(ConversationStore):
             """)
 
     def _message_to_dict(
-        self, message: DeveloperMessage | UserMessage | AssistantMessage
+        self,
+        message: DeveloperMessage
+        | UserMessage
+        | AssistantMessage
+        | GeneratedAssistantMessage[Any],
     ) -> dict[str, Any]:
         """Convert a message object to a dictionary for JSON serialization."""
         message_dict = message.model_dump()
@@ -119,8 +126,13 @@ class PostgresConversationStore(ConversationStore):
             return UserMessage.model_validate(message_dict)
 
     @override
-    async def add_message_async(
-        self, chat_id: str, message: DeveloperMessage | UserMessage | AssistantMessage
+    async def add_message_async[T = Any](
+        self,
+        chat_id: str,
+        message: DeveloperMessage
+        | UserMessage
+        | AssistantMessage
+        | GeneratedAssistantMessage[T],
     ) -> None:
         """Add a message to the conversation."""
         async with self._pool.acquire() as conn:

{agentle-0.9.29 → agentle-0.9.31}/agentle/agents/whatsapp/whatsapp_bot.py

@@ -2285,7 +2285,7 @@ class WhatsAppBot[T_Schema: WhatsAppResponseBase = WhatsAppResponseBase](BaseMod
             ... )
         """
         response_text = ""
-        
+
         if isinstance(response, GeneratedAssistantMessage):
             # Check if we have structured output (parsed)
             if response.parsed:

{agentle-0.9.29 → agentle-0.9.31}/agentle/utils/parse_streaming_json.py

@@ -101,12 +101,12 @@ def parse_streaming_json[T: BaseModel](potential_json: str | None, model: type[T
         in_string = False
         escape_next = False
         last_quote_pos = -1
-        
+
         for i, char in enumerate(json_str):
             if escape_next:
                 escape_next = False
                 continue
-            if char == '\\':
+            if char == "\\":
                 escape_next = True
                 continue
             if char == '"':
@@ -141,7 +141,7 @@ def parse_streaming_json[T: BaseModel](potential_json: str | None, model: type[T
         # Pattern: "key": "value..." - capture everything until the next unescaped quote or EOF
         string_pattern = r'["\']([\w]+)["\']:\s*["\']([^"\']*?)(?:["\']|$)'
         string_matches = re.findall(string_pattern, json_str, re.DOTALL)
-        
+
         # Also try to capture very long strings that span multiple lines
         # This catches incomplete strings during streaming
         long_string_pattern = r'["\']([\w_]+)["\']:\s*["\'](.+?)(?:["\'],?\s*["}]|$)'
@@ -149,12 +149,14 @@ def parse_streaming_json[T: BaseModel](potential_json: str | None, model: type[T
 
         for key, value in string_matches:
             data[key] = value
-        
+
         # Prefer long_matches for fields that might be truncated in string_matches
         for key, value in long_matches:
             # Only override if the long match has more content
             existing = data.get(key, "")
-            if key not in data or (isinstance(existing, str) and len(value) > len(existing)):
+            if key not in data or (
+                isinstance(existing, str) and len(value) > len(existing)
+            ):
                 data[key] = value
 
         # Extract string key-value pairs with unquoted keys

{agentle-0.9.29 → agentle-0.9.31}/pyproject.toml

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

agentle-0.9.29/CHANGELOG.md

@@ -1,61 +0,0 @@
-# Changelog
-
-## v0.9.29
-refactor(whatsapp): Update Data and Key model field definitions
-
-- Modify `status` field in Data model to be optional (str | None)
-- Update `remoteJidAlt` in Key model to be optional (str | None)
-- Add default None to `id` field in Key model
-- Improve model flexibility by allowing None for certain fields
-
-## v0.9.28
-making field optional in whatsappwebhookpayload
-
-## v0.9.27
-feat(whatsapp): Add structured response base model for WhatsApp bot
-
-- Create WhatsAppResponseBase model to standardize bot response structure
-- Enhance WhatsAppBot to support generic structured output schemas
-- Update _send_response method to handle structured response parsing
-- Add type hints and documentation for new response handling mechanism
-- Improve logging for structured response processing
-- Enable more flexible and type-safe bot response generation
-This change introduces a base model for WhatsApp bot responses that ensures a consistent response structure while allowing for extensible, type-safe output schemas.
-
-
-## v0.9.26
-feat(api): Enhance API endpoint and function name generation
-
-- Improve aiohttp session handling in `make_single_request()` to prevent connection errors
-- Add robust function name generation for OpenAPI spec endpoints
-- Create new test script to validate function name generation for OpenAPI specs
-- Update example script to demonstrate edge case handling for API endpoint names
-- Ensure proper session and connector closure in async API requests
-- Add comprehensive test cases for problematic path name conversions
-Addresses potential issues with API endpoint generation and async request management, improving overall robustness of API integration capabilities.
-
-refactor(extractor): Enhance HTML processing and base64 image removal
-
-- Consolidate BeautifulSoup operations for more robust HTML processing
-- Implement comprehensive base64 image removal strategy with detailed debugging
-- Add multiple removal techniques for base64 images in img tags, anchors, and styles
-- Improve error handling and type checking during HTML manipulation
-- Update example code to use different LLM model and async extraction method
-- Add debug print statements to track base64 image removal process
-- Refactor main content extraction and tag filtering logic
-
-## v0.9.25
-
-- refactor(extractor): reorganize imports and add model_config attribute
-
-- Moved the import of run_sync to a more appropriate location
-- Introduced model_config attribute using ConfigDict for better configuration management
-
-refactor(whatsapp): streamline WhatsApp bot structure and introduce v2 components
-
-- Removed unnecessary context_manager field from WhatsAppBot class.
-- Updated AudioMessage class to improve type handling in convert_long_to_str method.
-- Added new v2 module with BotConfig, BatchProcessorManager, and message limit definitions for enhanced configuration and processing capabilities.
-- Introduced new files for in-memory batch processing and payload handling.
-- Established a new WhatsAppBot class in v2 for better organization and functionality.
-

agentle-0.9.29/examples/test_api_openapi.py

@@ -1,155 +0,0 @@
-"""
-Test OpenAPI spec loading with function name validation.
-
-This example demonstrates loading an API from an OpenAPI spec
-and verifying that all generated function names are valid.
-"""
-
-import asyncio
-import re
-
-from agentle.agents.apis.api import API
-
-# Pattern from Google's adapter - function names must match this
-FUNCTION_NAME_PATTERN = re.compile(r"^[a-zA-Z_][a-zA-Z0-9_\.\-]*$")
-
-
-async def test_openapi_function_names():
-    """Test that OpenAPI spec loading generates valid function names."""
-    print("\n🧪 Testing OpenAPI Spec Function Name Generation")
-    print("=" * 70)
-
-    try:
-        # Load PetStore API spec
-        print("\n📥 Loading PetStore OpenAPI spec...")
-        api = await API.from_openapi_spec(
-            "https://petstore3.swagger.io/api/v3/openapi.json",
-            name="PetStore",
-        )
-
-        print(f"✅ Loaded API: {api.name}")
-        print(f"   Base URL: {api.base_url}")
-        print(f"   Total Endpoints: {len(api.endpoints)}")
-
-        # Validate all endpoint names
-        print("\n🔍 Validating function names...")
-        invalid_names = []
-        
-        for endpoint in api.endpoints:
-            if not FUNCTION_NAME_PATTERN.match(endpoint.name):
-                invalid_names.append(endpoint.name)
-                print(f"   ❌ INVALID: {endpoint.name}")
-            else:
-                print(f"   ✓ {endpoint.name}")
-
-        if invalid_names:
-            print(f"\n❌ Found {len(invalid_names)} invalid function names!")
-            print("Invalid names:", invalid_names)
-            return False
-        else:
-            print(f"\n✅ All {len(api.endpoints)} function names are valid!")
-            return True
-
-    except Exception as e:
-        print(f"\n❌ Error: {e}")
-        import traceback
-        traceback.print_exc()
-        return False
-
-
-async def test_edge_case_paths():
-    """Test OpenAPI spec with edge case paths."""
-    print("\n🧪 Testing Edge Case Path Patterns")
-    print("=" * 70)
-
-    # Create a mock OpenAPI spec with edge cases
-    mock_spec = {
-        "openapi": "3.0.0",
-        "info": {"title": "Edge Case API", "version": "1.0.0"},
-        "servers": [{"url": "https://api.example.com"}],
-        "paths": {
-            "/123-resource": {
-                "get": {
-                    "operationId": None,  # Force auto-generation
-                    "summary": "Get resource starting with number",
-                }
-            },
-            "/api/v2/users": {
-                "get": {
-                    "operationId": None,
-                    "summary": "Get users",
-                }
-            },
-            "/user-profile": {
-                "get": {
-                    "operationId": None,
-                    "summary": "Get user profile",
-                }
-            },
-            "/": {
-                "get": {
-                    "operationId": None,
-                    "summary": "Root endpoint",
-                }
-            },
-        },
-    }
-
-    # Remove None operationIds (they should be missing, not None)
-    for path_item in mock_spec["paths"].values():
-        for operation in path_item.values():
-            if "operationId" in operation and operation["operationId"] is None:
-                del operation["operationId"]
-
-    try:
-        api = await API.from_openapi_spec(mock_spec, name="EdgeCaseAPI")
-
-        print(f)
-o.run(main()
-    asyncimain__":me__ == "__
-if __na70)
-
-=" * rint("
-    p")s failed!sttet("❌ Some   prin         else:
-assed!")
- tests p"✅ All    print(t2:
-     resullt1 and     if resu"=" * 70)
-\n" +    print("
- aths()
-se_ptest_edge_cait  = awat2esul
-    rEdge cases: Test 2# )
-
-    mes(_napi_functiont_openait tes1 = awasult  re  c
-API speal Openst 1: Re
-    # Te0)
-"=" * 7print(s")
-     Testtionme Valida Na Functionint("API)
-    pr" * 70"=("\n" +  print."""
-   estsl t""Run al
-    " def main():
-
-
-asyncturn False        rec()
-_ex.printtraceback       back
- race import t     {e}")
-   Error: rint(f"\n❌
-        p:s e at Exception
-    excepll_valid
- return a      )
-
- d!"aliinvn names are functio Some \n❌ print(f"    
-           else:
-    !")validn names are se functioAll edge ca"\n✅    print(f        d:
- _vali if all   se
-
-    = Fald l_vali         al      alid:
-  if not is_v        me}")
-   int.na} -> {endpondpoint.path {e  {status}f" nt(         pri"
-   else "❌if is_valid  = "✓" ustat       se)
-     nt.namatch(endpoiPATTERN.m_NAME_NCTION_valid = FU          ispoints:
-   api.endpoint in end      for True
-  lid =ll_va      aames
-   all nte# Valida      
-
-  )}")endpointslen(api.ndpoints: {(f"   Eprint       )
- .name}"api: {APIded "✅ Loa