waldiez 0.1.0__tar.gz → 0.1.2__tar.gz

{waldiez-0.1.0 → waldiez-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: waldiez
-Version: 0.1.0
+Version: 0.1.2
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/py/
 Project-URL: repository, https://github.com/waldiez/py.git
@@ -74,11 +74,11 @@ Description-Content-Type: text/markdown
 
 # Waldiez
 
-![CI Build](https://github.com/waldiez/py/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/py/badge.svg)](https://coveralls.io/github/waldiez/py)
+![CI Build](https://github.com/waldiez/py/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/py/badge.svg)](https://coveralls.io/github/waldiez/py) [![PyPI version](https://badge.fury.io/py/waldiez.svg)](https://badge.fury.io/py/waldiez)
 
 Translate a Waldiez flow:
 
-[![Flow](./docs/flow.png)](./docs/flow.png)
+![Flow](https://raw.githubusercontent.com/waldiez/py/refs/heads/main/docs/flow.png)
 
 To a python script or a jupyter notebook with the corresponding [autogen](https://github.com/microsoft/autogen/) agents and chats.
 
@@ -91,14 +91,13 @@ To a python script or a jupyter notebook with the corresponding [autogen](https:
 
 ## Installation
 
-<!-- 
 On PyPI:
 
 ```bash
 python -m pip install waldiez
-``` -->
+```
 
-From this repository:
+From the repository:
 
 ```bash
 python -m pip install git+https://github.com/waldiez/py.git

{waldiez-0.1.0 → waldiez-0.1.2}/README.md

@@ -1,10 +1,10 @@
 # Waldiez
 
-![CI Build](https://github.com/waldiez/py/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/py/badge.svg)](https://coveralls.io/github/waldiez/py)
+![CI Build](https://github.com/waldiez/py/actions/workflows/main.yaml/badge.svg) [![Coverage Status](https://coveralls.io/repos/github/waldiez/py/badge.svg)](https://coveralls.io/github/waldiez/py) [![PyPI version](https://badge.fury.io/py/waldiez.svg)](https://badge.fury.io/py/waldiez)
 
 Translate a Waldiez flow:
 
-[![Flow](./docs/flow.png)](./docs/flow.png)
+![Flow](https://raw.githubusercontent.com/waldiez/py/refs/heads/main/docs/flow.png)
 
 To a python script or a jupyter notebook with the corresponding [autogen](https://github.com/microsoft/autogen/) agents and chats.
 
@@ -17,14 +17,13 @@ To a python script or a jupyter notebook with the corresponding [autogen](https:
 
 ## Installation
 
-<!-- 
 On PyPI:
 
 ```bash
 python -m pip install waldiez
-``` -->
+```
 
-From this repository:
+From the repository:
 
 ```bash
 python -m pip install git+https://github.com/waldiez/py.git

{waldiez-0.1.0 → waldiez-0.1.2}/waldiez/_version.py

@@ -1,3 +1,3 @@
 """Version information for Waldiez."""
 
-__version__ = "0.1.0"
+__version__ = "0.1.2"

{waldiez-0.1.0 → waldiez-0.1.2}/waldiez/exporting/agents/rag_user/chroma_utils.py

@@ -28,8 +28,11 @@ def _get_chroma_client_string(agent: WaldiezRagUser) -> Tuple[str, str]:
         agent.retrieve_config.db_config.use_local_storage
         and agent.retrieve_config.db_config.local_storage_path is not None
     ):
+        # on windows, we get:
+        # SyntaxError: (unicode error) 'unicodeescape' codec can't decode bytes
+        # in position 2-3: truncated \UXXXXXXXX escape
         local_path = Path(agent.retrieve_config.db_config.local_storage_path)
-        client_str += f'PersistentClient(path="{local_path}")'
+        client_str += f'PersistentClient(path=r"{local_path}")'
     else:
         client_str += "Client()"
     return client_str, to_import

{waldiez-0.1.0 → waldiez-0.1.2}/waldiez/exporting/agents/rag_user/qdrant_utils.py

@@ -30,7 +30,7 @@ def _get_qdrant_client_string(agent: WaldiezRagUser) -> Tuple[str, str]:
         and agent.retrieve_config.db_config.local_storage_path
     ):
         local_path = Path(agent.retrieve_config.db_config.local_storage_path)
-        client_str += f'location="{local_path}")'
+        client_str += f'location=r"{local_path}")'
     elif agent.retrieve_config.db_config.connection_url:
         client_str += (
             f'location="{agent.retrieve_config.db_config.connection_url}")'

{waldiez-0.1.0 → waldiez-0.1.2}/waldiez/exporting/agents/rag_user/rag_user.py

@@ -1,6 +1,6 @@
 """RAG User related exporting utils."""
 
-from typing import Dict, Set, Tuple
+from typing import Dict, List, Set, Tuple, Union
 
 from waldiez.models import (
     WaldiezAgent,
@@ -9,7 +9,7 @@ from waldiez.models import (
     WaldiezRagUserRetrieveConfig,
 )
 
-from ...utils import get_object_string
+from ...utils import get_object_string, get_path_string
 from .vector_db import get_rag_user_vector_db_string
 
 
@@ -60,6 +60,7 @@ def get_rag_user_retrieve_config_str(
             "\n\n"
         )
         args_dict["custom_text_split_function"] = text_split_arg_name
+    # docs_path = args_dict.pop("docs_path", [])
     args_content = get_object_string(args_dict)
     # get the last line (where the dict ends)
     args_parts = args_content.split("\n")
@@ -131,9 +132,9 @@ def _get_args_dict(
     agent: WaldiezRagUser,
     retrieve_config: WaldiezRagUserRetrieveConfig,
     model_names: Dict[str, str],
-) -> Dict[str, str]:
+) -> Dict[str, Union[str, List[str]]]:
     model_arg = _get_model_arg(agent, retrieve_config, model_names)
-    args_dict = {
+    args_dict: Dict[str, Union[str, List[str]]] = {
         "task": retrieve_config.task,
         "model": model_arg,
     }
@@ -142,13 +143,23 @@ def _get_args_dict(
         "context_max_tokens",
         "customized_prompt",
         "customized_answer_prefix",
-        "docs_path",
     ]
     for arg in optional_args:
         arg_value = getattr(retrieve_config, arg)
         if arg_value is not None:
             args_dict[arg] = arg_value
             args_dict[arg] = getattr(retrieve_config, arg)
+    docs_path: Union[str, List[str]] = []
+    if retrieve_config.docs_path:
+        docs_path = (
+            retrieve_config.docs_path
+            if isinstance(retrieve_config.docs_path, list)
+            else [retrieve_config.docs_path]
+        )
+        docs_path = [get_path_string(path) for path in docs_path]
+        args_dict["docs_path"] = docs_path
+    if docs_path:
+        args_dict["docs_path"] = docs_path
     non_optional_args = [
         "new_docs",
         "update_context",

{waldiez-0.1.0 → waldiez-0.1.2}/waldiez/exporting/utils/__init__.py

@@ -15,12 +15,14 @@ from .naming import (
     get_valid_python_variable_name,
 )
 from .object_string import get_object_string
+from .path_check import get_path_string
 
 __all__ = [
     "add_autogen_dot_import",
     "comment",
     "get_logging_start_string",
     "get_logging_stop_string",
+    "get_path_string",
     "get_pylint_ignore_comment",
     "get_sqlite_to_csv_string",
     "get_sqlite_to_csv_call_string",

{waldiez-0.1.0 → waldiez-0.1.2}/waldiez/exporting/utils/object_string.py

@@ -74,6 +74,8 @@ def get_object_string(obj: Any, tabs: int = 1) -> str:
         return f"[{to_return}]"
 
     if isinstance(obj, str):
+        if obj.startswith("r'") or obj.startswith('r"'):
+            return obj
         return f'"{obj}"'
 
     if obj is None:

waldiez-0.1.2/waldiez/exporting/utils/path_check.py

@@ -0,0 +1,48 @@
+"""Path check utility functions."""
+
+import os
+from pathlib import Path
+
+# pylint: disable=broad-except
+
+
+def _is_local_path(string: str) -> bool:
+    """Check if a string is a local path.
+
+    Parameters
+    ----------
+    string : str
+        The string to check.
+
+    Returns
+    -------
+    bool
+        True if the path is a local path.
+    """
+    try:
+        path = Path(string).resolve()
+        return path.exists()
+    except Exception:  # pragma: no cover
+        return False
+
+
+def get_path_string(string: str) -> str:
+    """Get the path string.
+
+    Parameters
+    ----------
+    string : str
+        The string to check.
+
+    Returns
+    -------
+    str
+        The local path string.
+    """
+    # On windows, we get paths like "C:\path\to\file"
+    # if so, let's try to avoid invalid escape sequences
+    if not _is_local_path(string):
+        return string
+    if os.name == "nt":  # pragma: no cover
+        return f"r'{string}'"
+    return f"'{string}'"

{waldiez-0.1.0 → waldiez-0.1.2}/waldiez/models/agents/agent/agent_data.py

@@ -24,9 +24,7 @@ class WaldiezAgentData(WaldiezBase):
     human_input_mode : Literal["ALWAYS", "NEVER", "TERMINATE"]
         The human input mode to use for the agent.
     code_execution_config : Union[WaldiezAgentCodeExecutionConfig, False]
-        The code execution config. Either False (no execution) or a dict
-    max_tokens : Optional[int]
-        The maximum tokens to use. Default: None (no limit).
+        The code execution config. Either False (no execution) or a dict.
     agent_default_auto_reply : Optional[str]
         The agent's default auto reply when no input is received.
     max_consecutive_auto_reply : Optional[int]
@@ -45,7 +43,7 @@ class WaldiezAgentData(WaldiezBase):
     """
 
     model_config = ConfigDict(
-        extra="forbid",
+        extra="ignore",
         alias_generator=to_camel,
         populate_by_name=True,
         # we have a field starting with "model_" (model_ids)
@@ -71,15 +69,6 @@ class WaldiezAgentData(WaldiezBase):
             alias="humanInputMode",
         ),
     ]
-    max_tokens: Annotated[
-        Optional[int],
-        Field(
-            None,
-            title="Max tokens",
-            description="The maximum tokens to use. Default: None (no limit).",
-            alias="maxTokens",
-        ),
-    ]
     code_execution_config: Annotated[
         Union[WaldiezAgentCodeExecutionConfig, Literal[False]],
         Field(