waldiez 0.1.4__tar.gz → 0.1.6__tar.gz

{waldiez-0.1.4 → waldiez-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: waldiez
-Version: 0.1.4
+Version: 0.1.6
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/py/
 Project-URL: repository, https://github.com/waldiez/py.git
@@ -19,7 +19,7 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Python: <3.13,>=3.10
 Requires-Dist: autogen-agentchat==0.2.37
 Requires-Dist: jupytext
-Requires-Dist: twisted==24.7.0
+Requires-Dist: twisted==24.10.0
 Provides-Extra: autogen-extras
 Requires-Dist: autogen-agentchat[anthropic]==0.2.37; extra == 'autogen-extras'
 Requires-Dist: autogen-agentchat[bedrock]==0.2.37; extra == 'autogen-extras'

{waldiez-0.1.4 → waldiez-0.1.6}/pyproject.toml

@@ -23,7 +23,7 @@ classifiers = [
 dependencies = [
     "jupytext",
     "autogen-agentchat==0.2.37",
-    "twisted==24.7.0",
+    "twisted==24.10.0",
 ]
 
 [project.urls]

{waldiez-0.1.4 → waldiez-0.1.6}/waldiez/_version.py

@@ -1,3 +1,3 @@
 """Version information for Waldiez."""
 
-__version__ = "0.1.4"
+__version__ = "0.1.6"

{waldiez-0.1.4 → waldiez-0.1.6}/waldiez/exporting/chats/helpers.py

@@ -76,6 +76,7 @@ def export_single_chat_string(
     tab = "    " * tabs
     chat, sender, recipient = flow
     chat_args = chat.get_chat_args(sender=sender)
+    chat_args = escape_summary_args_quotes(chat_args)
     if not chat_args:
         return _get_empty_simple_chat_string(
             tab,
@@ -191,6 +192,28 @@ def export_multiple_chats_string(
     return content, additional_methods_string
 
 
+def escape_summary_args_quotes(chat_args: Dict[str, Any]) -> Dict[str, Any]:
+    """Escape quotes in the summary args if they are strings.
+
+    Parameters
+    ----------
+    chat_args : Dict[str, Any]
+        The chat arguments.
+
+    Returns
+    -------
+    Dict[str, Any]
+        The chat arguments with the summary prompt escaped.
+    """
+    if "summary_args" in chat_args and isinstance(
+        chat_args["summary_args"], dict
+    ):
+        for key, value in chat_args["summary_args"].items():
+            if isinstance(value, str):
+                chat_args["summary_args"][key] = get_escaped_string(value)
+    return chat_args
+
+
 def _get_chat_message_string(
     chat: WaldiezChat,
     chat_names: Dict[str, str],
@@ -219,7 +242,7 @@ def _get_chat_message_string(
     ):
         return "None", None
     if chat.message.type == "string":
-        return chat.message.content, None
+        return get_escaped_string(chat.message.content), None
     chat_name = chat_names[chat.id]
     original_function_name = "callable_message"
     method_args = "sender, recipient, context"
@@ -264,6 +287,7 @@ def _get_chat_dict_string(
     """
     tab = "    " * tabs
     chat_args = chat.get_chat_args(sender=sender)
+    chat_args = escape_summary_args_quotes(chat_args)
     chat_string = "{"
     chat_string += "\n" + f'{tab}    "sender": {agent_names[sender.id]},'
     chat_string += "\n" + f'{tab}    "recipient": {agent_names[recipient.id]},'

{waldiez-0.1.4 → waldiez-0.1.6}/waldiez/exporting/chats/nested.py

@@ -10,6 +10,7 @@ from waldiez.models import (
 )
 
 from ..utils import get_escaped_string, get_object_string
+from .helpers import escape_summary_args_quotes
 
 
 def get_nested_chat_trigger_agent_names(
@@ -81,6 +82,7 @@ def get_nested_chat_message_string(
         sender_name = agent_names[sender_id]
     recipient_name = agent_names[recipient_id]
     chat_dict = waldiez_chat.get_chat_args()
+    chat_dict = escape_summary_args_quotes(chat_dict)
     chat_dict["recipient"] = recipient_name
     if sender_name:
         chat_dict["sender"] = sender_name