refactor(conversations): Improve type handling and code formatting across conversation stores to prepare for MPHU WhatsApp Service

- Refactor type annotations in conversation store methods
- Update import statements to improve readability
- Simplify complex type definitions in callback and firebase conversation stores
- Remove unnecessary whitespace and improve code formatting
- Standardize message type handling across conversation store implementations
- Remove unused or redundant code sections
This refactoring aims to enhance type safety and code consistency in conversation store modules.

Author: arthurbrenno

agentle/agents/apis/object_schema.py

@@ -46,20 +46,20 @@ def from_json_schema(
     ) -> 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 @@ def from_json_schema(
                     }
                 }
             }
-            
+
             schema = ObjectSchema.from_json_schema(json_schema)
             ```
         """

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/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/agents/conversations/callback_conversation_store.py

@@ -81,13 +81,7 @@ def __init__(
         self,
         get_callback: Callable[
             [str],
-            Awaitable[
-                Sequence[
-                    DeveloperMessage
-                    | UserMessage
-                    | AssistantMessage
-                ]
-            ],
+            Awaitable[Sequence[DeveloperMessage | UserMessage | AssistantMessage]],
         ],
         add_callback: Callable[
             [

agentle/agents/conversations/conversation_store.py

@@ -4,7 +4,9 @@
 
 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.generated_assistant_message import (
+    GeneratedAssistantMessage,
+)
 from agentle.generations.models.messages.user_message import UserMessage
 
 
@@ -45,7 +47,12 @@ async def get_conversation_history_async(
 
     @abc.abstractmethod
     async def add_message_async[T = Any](
-        self, chat_id: str, message: DeveloperMessage | UserMessage | AssistantMessage | GeneratedAssistantMessage[T]
+        self,
+        chat_id: str,
+        message: DeveloperMessage
+        | UserMessage
+        | AssistantMessage
+        | GeneratedAssistantMessage[T],
     ) -> None: ...
 
     @abc.abstractmethod

agentle/agents/conversations/firebase_conversation_store.py

@@ -32,7 +32,12 @@ def __init__(
 
     @override
     async def add_message_async[T = Any](
-        self, chat_id: str, message: DeveloperMessage | UserMessage | AssistantMessage | GeneratedAssistantMessage[T]
+        self,
+        chat_id: str,
+        message: DeveloperMessage
+        | UserMessage
+        | AssistantMessage
+        | GeneratedAssistantMessage[T],
     ) -> None:
         from google.cloud import firestore
 

agentle/agents/whatsapp/whatsapp_bot.py

@@ -2285,7 +2285,7 @@ async def _send_response(
             ... )
         """
         response_text = ""
-        
+
         if isinstance(response, GeneratedAssistantMessage):
             # Check if we have structured output (parsed)
             if response.parsed:

agentle/utils/parse_streaming_json.py

@@ -101,12 +101,12 @@ def fix_common_json_issues(json_str: str) -> str:
         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,20 +141,22 @@ def extract_data_manually(json_str: str) -> dict[str, Any]:
         # 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*["}]|$)'
         long_matches = re.findall(long_string_pattern, json_str, re.DOTALL)
 
         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