agentle 0.9.29__py3-none-any.whl → 0.9.30__py3-none-any.whl

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
 
 
@@ -78,10 +81,23 @@ class CallbackConversationStore(ConversationStore):
         self,
         get_callback: Callable[
             [str],
-            Awaitable[Sequence[DeveloperMessage | UserMessage | AssistantMessage]],
+            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 +162,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/agents/conversations/conversation_store.py

@@ -1,8 +1,10 @@
 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 +44,8 @@ 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/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,8 @@ 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/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/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/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/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.dist-info → agentle-0.9.30.dist-info}/METADATA

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

{agentle-0.9.29.dist-info → agentle-0.9.30.dist-info}/RECORD

@@ -109,13 +109,13 @@ agentle/agents/asgi/blacksheep/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 agentle/agents/asgi/blacksheep/agent_to_blacksheep_application_adapter.py,sha256=EI-0TBWlL8wewFuOtNoTeY7T1BTbj3GVQjUmkuPrv74,12080
 agentle/agents/asgi/blacksheep/agent_to_blacksheep_route_handler_adapter.py,sha256=kAX3_W6Gj8JI9yjXm6bnWvzq4YH8E4RD_jPy-Z_RtoA,25124
 agentle/agents/conversations/__init__.py,sha256=8XNgbcRauHDWbaiwKzoQkMbA4v3IOB1rtKGPfjReHPY,573
-agentle/agents/conversations/callback_conversation_store.py,sha256=2ITH1ETBIjbTxq_i9x86kQEaKtlSyi1GMR9OAw2tHcQ,7945
-agentle/agents/conversations/conversation_store.py,sha256=LeSwBrQvHjPFLTAoGrY8s2b5h5o43x_BOmrtGIiJ0qM,1824
-agentle/agents/conversations/firebase_conversation_store.py,sha256=POnO7mTkL4HLhz0l8DUXZzBluGGnOzmpASZ2k0BC-uE,4167
-agentle/agents/conversations/json_file_conversation_store.py,sha256=OWmvr7XbUlueIatFgqvMCaf3hnz11UoMtX4d_pEf1pA,9285
-agentle/agents/conversations/local_conversation_store.py,sha256=grnpHX0wc89QjLJXIIZzhMo3O4qsFcO4gGBcUFQUQA0,2112
-agentle/agents/conversations/mysql_conversation_store.py,sha256=sBrU5SGDxelWRifjFKUjcx2s4gBWccUmA_7k4Atao2s,11243
-agentle/agents/conversations/postgres_conversation_store.py,sha256=Z0zuh2-LbqjE7ft81r24iMrksdpIGNiIwQEjZjRgUNM,9695
+agentle/agents/conversations/callback_conversation_store.py,sha256=oWVRRoWhnMK_rxaOYBGT6oGF58rwIEVC8nR9QGT3jig,8391
+agentle/agents/conversations/conversation_store.py,sha256=vOJDCclf3CtiuRmt1ei9cy7KQtBRiK63MKs2ElL_S1I,1989
+agentle/agents/conversations/firebase_conversation_store.py,sha256=Ie9Q05q0Pv6AU9DC3uYbNQ9LvN0WTKxloen-IO15m84,4323
+agentle/agents/conversations/json_file_conversation_store.py,sha256=NvUGXzxKU_ozoNM4rodqKtrDR1r7WeUvTUDm3eYXkkM,9544
+agentle/agents/conversations/local_conversation_store.py,sha256=cPFQcCieZ9d0K3fc0MvfzcUucnL8KPMK9K8IhOEn28Y,2432
+agentle/agents/conversations/mysql_conversation_store.py,sha256=XKlURkjRQpHcotRyhQHneqzzZXehyR2IJtIqh4kyEgM,11501
+agentle/agents/conversations/postgres_conversation_store.py,sha256=kwB3UsME1oqKj-dYbhMw-XPe0pOm47MSoA6RFNWHrNk,9953
 agentle/agents/errors/__init__.py,sha256=g2WoSvolIhcRutVsaEctMRsKK_r02wpSquvoNEjDOPo,1694
 agentle/agents/errors/max_tool_calls_exceeded_error.py,sha256=LXyGXOSNhYyc9W8f0ZcKMuQiFdpFjLr_yM4nkpUvZF8,2817
 agentle/agents/errors/tool_suspension_error.py,sha256=_wL6URtj2eqnuSLpTvabqD8IYfz8z-1eszifzibyfoc,3096
@@ -1018,7 +1018,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.29.dist-info/METADATA,sha256=R9crja9iNTdYzuPt86thtNZGJFIkwz7wou53L--okQQ,86849
-agentle-0.9.29.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agentle-0.9.29.dist-info/licenses/LICENSE,sha256=T90S9vqRS6qP-voULxAcvwEs558wRRo6dHuZrjgcOUI,1085
-agentle-0.9.29.dist-info/RECORD,,
+agentle-0.9.30.dist-info/METADATA,sha256=OcjyI2hJ0Q5CcgJTjSfD0RhgUzfJHb2aNiXYo5Mz-VM,86849
+agentle-0.9.30.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agentle-0.9.30.dist-info/licenses/LICENSE,sha256=T90S9vqRS6qP-voULxAcvwEs558wRRo6dHuZrjgcOUI,1085
+agentle-0.9.30.dist-info/RECORD,,