waldiez 0.4.1__tar.gz → 0.4.3__tar.gz

{waldiez-0.4.1 → waldiez-0.4.3}/PKG-INFO

@@ -1,9 +1,11 @@
 Metadata-Version: 2.4
 Name: waldiez
-Version: 0.4.1
+Version: 0.4.3
 Summary: waldiez
-Project-URL: homepage, https://waldiez.github.io/waldiez/python
+Project-URL: homepage, https://waldiez.io
 Project-URL: repository, https://github.com/waldiez/python.git
+Project-URL: documentation, https://waldiez.github.io/python/
+Project-URL: issues, https://github.com/waldiez/python/issues
 Author-email: Panagiotis Kasnesis <pkasnesis@waldiez.io>, Lazaros Toumanidis <laztoum@waldiez.io>, Stella Ioannidou <stella@humancentered.gr>
 License-File: LICENSE
 License-File: NOTICE.md
@@ -17,7 +19,8 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Python: <3.13,>=3.10
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: <3.14,>=3.10
 Requires-Dist: aiocsv==1.3.2
 Requires-Dist: aiofiles==24.1.0
 Requires-Dist: aiosqlite==0.21.0
@@ -25,85 +28,88 @@ Requires-Dist: asyncer==0.0.8
 Requires-Dist: graphviz==0.20.3
 Requires-Dist: httpx<1
 Requires-Dist: jupytext
-Requires-Dist: numpy<2.2.0
+Requires-Dist: numpy<=2.2.4
 Requires-Dist: pandas>=2
 Requires-Dist: parso==0.8.4
-Requires-Dist: pyautogen==0.7.4
-Requires-Dist: pydantic<3,>=2.6.1
+Requires-Dist: pyautogen[openai]==0.8.6
+Requires-Dist: pydantic<3,>=2.10.2
 Requires-Dist: typer<0.16,>=0.9
 Provides-Extra: ag2-extras
 Requires-Dist: beautifulsoup4; extra == 'ag2-extras'
-Requires-Dist: chromadb>=0.5.23; extra == 'ag2-extras'
-Requires-Dist: crewai-tools==0.33.0; extra == 'ag2-extras'
-Requires-Dist: crewai>0.98.0; extra == 'ag2-extras'
-Requires-Dist: embedchain==0.1.126; extra == 'ag2-extras'
+Requires-Dist: chromadb>=0.5.10; extra == 'ag2-extras'
+Requires-Dist: embedchain; extra == 'ag2-extras'
 Requires-Dist: huggingface-hub; extra == 'ag2-extras'
 Requires-Dist: ipython; extra == 'ag2-extras'
 Requires-Dist: langchain-community<1,>=0.3.12; extra == 'ag2-extras'
 Requires-Dist: markdownify; extra == 'ag2-extras'
-Requires-Dist: pgvector>=0.3.6; extra == 'ag2-extras'
+Requires-Dist: pgvector>=0.4.0; extra == 'ag2-extras'
 Requires-Dist: protobuf>=4.25.3; extra == 'ag2-extras'
-Requires-Dist: psycopg[binary]>=3.2.3; extra == 'ag2-extras'
-Requires-Dist: psycopg[binary]>=3.2.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[anthropic]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[bedrock]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[cohere]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[gemini]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[groq]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[interop-crewai]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[interop-langchain]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[lmm]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[mistral]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[neo4j]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[ollama]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[together]==0.7.4; extra == 'ag2-extras'
-Requires-Dist: pyautogen[websurfer]==0.7.4; extra == 'ag2-extras'
+Requires-Dist: psycopg==3.2.6; (sys_platform == 'linux') and extra == 'ag2-extras'
+Requires-Dist: psycopg==3.2.6; (sys_platform == 'win32' and platform_machine == 'AARCH64') and extra == 'ag2-extras'
+Requires-Dist: psycopg==3.2.6; (sys_platform == 'win32' and platform_machine == 'ARM64') and extra == 'ag2-extras'
+Requires-Dist: psycopg==3.2.6; (sys_platform == 'win32' and platform_machine == 'aarch64') and extra == 'ag2-extras'
+Requires-Dist: psycopg==3.2.6; (sys_platform == 'win32' and platform_machine == 'arm64') and extra == 'ag2-extras'
+Requires-Dist: psycopg[binary]==3.2.6; (sys_platform != 'linux' and platform_machine != 'arm64' and platform_machine != 'ARM64' and platform_machine != 'aarch64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
+Requires-Dist: psycopg[binary]>=3.2.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[anthropic]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[bedrock]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[cohere]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[gemini]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[groq]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[interop-crewai]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[interop-langchain]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[lmm]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[mistral]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[neo4j]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[ollama]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[together]==0.8.6; extra == 'ag2-extras'
+Requires-Dist: pyautogen[websurfer]==0.8.6; extra == 'ag2-extras'
 Requires-Dist: pydantic-ai>=0.0.21; extra == 'ag2-extras'
 Requires-Dist: pymongo>=4.11; extra == 'ag2-extras'
 Requires-Dist: pypdf; extra == 'ag2-extras'
+Requires-Dist: pysqlite3-binary==0.5.4; (sys_platform == 'linux' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64') and extra == 'ag2-extras'
 Requires-Dist: qdrant-client[fastembed]; extra == 'ag2-extras'
 Requires-Dist: sentence-transformers; extra == 'ag2-extras'
-Requires-Dist: weaviate-client>=4.10.2; extra == 'ag2-extras'
 Provides-Extra: dev
 Requires-Dist: autoflake==2.3.1; extra == 'dev'
-Requires-Dist: bandit==1.8.2; extra == 'dev'
+Requires-Dist: bandit==1.8.3; extra == 'dev'
 Requires-Dist: black[jupyter]==25.1.0; extra == 'dev'
-Requires-Dist: flake8==7.1.1; extra == 'dev'
-Requires-Dist: isort==6.0.0; extra == 'dev'
+Requires-Dist: flake8==7.2.0; extra == 'dev'
+Requires-Dist: isort==6.0.1; extra == 'dev'
 Requires-Dist: mypy==1.15.0; extra == 'dev'
-Requires-Dist: pandas-stubs; extra == 'dev'
-Requires-Dist: pre-commit==4.1.0; extra == 'dev'
+Requires-Dist: pandas-stubs==2.2.3.250308; extra == 'dev'
+Requires-Dist: pre-commit==4.2.0; extra == 'dev'
 Requires-Dist: pydocstyle==6.3.0; extra == 'dev'
-Requires-Dist: pylint==3.3.4; extra == 'dev'
-Requires-Dist: python-dotenv==1.0.1; extra == 'dev'
-Requires-Dist: ruff==0.9.6; extra == 'dev'
-Requires-Dist: toml; (python_version <= '3.10') and extra == 'dev'
-Requires-Dist: types-pyyaml==6.0.12.20241230; extra == 'dev'
+Requires-Dist: pylint==3.3.6; extra == 'dev'
+Requires-Dist: python-dotenv==1.1.0; extra == 'dev'
+Requires-Dist: ruff==0.11.5; extra == 'dev'
+Requires-Dist: toml==0.10.2; (python_version <= '3.10') and extra == 'dev'
+Requires-Dist: types-pyyaml==6.0.12.20250402; extra == 'dev'
 Requires-Dist: types-toml==0.10.8.20240310; extra == 'dev'
-Requires-Dist: yamllint==1.35.1; extra == 'dev'
+Requires-Dist: yamllint==1.37.0; extra == 'dev'
 Provides-Extra: docs
 Requires-Dist: mdx-include==1.4.2; extra == 'docs'
 Requires-Dist: mdx-truly-sane-lists==1.3; extra == 'docs'
 Requires-Dist: mkdocs-jupyter==0.25.1; extra == 'docs'
 Requires-Dist: mkdocs-macros-plugin==1.3.7; extra == 'docs'
-Requires-Dist: mkdocs-material==9.6.4; extra == 'docs'
-Requires-Dist: mkdocs-minify-html-plugin==0.2.4; extra == 'docs'
+Requires-Dist: mkdocs-material==9.6.11; extra == 'docs'
+Requires-Dist: mkdocs-minify-html-plugin==0.3.1; extra == 'docs'
 Requires-Dist: mkdocs==1.6.1; extra == 'docs'
-Requires-Dist: mkdocstrings-python==1.15.0; extra == 'docs'
-Requires-Dist: mkdocstrings[crystal,python]==0.28.1; extra == 'docs'
+Requires-Dist: mkdocstrings-python==1.16.10; extra == 'docs'
+Requires-Dist: mkdocstrings[crystal,python]==0.29.1; extra == 'docs'
 Provides-Extra: jupyter
 Requires-Dist: jupyterlab>=4.3.0; extra == 'jupyter'
-Requires-Dist: waldiez-jupyter==0.4.1; extra == 'jupyter'
+Requires-Dist: waldiez-jupyter==0.4.3; extra == 'jupyter'
 Provides-Extra: studio
-Requires-Dist: waldiez-studio==0.4.1; extra == 'studio'
+Requires-Dist: waldiez-studio==0.4.3; extra == 'studio'
 Provides-Extra: test
-Requires-Dist: pytest-asyncio==0.25.3; extra == 'test'
-Requires-Dist: pytest-cov==6.0.0; extra == 'test'
+Requires-Dist: pytest-asyncio==0.26.0; extra == 'test'
+Requires-Dist: pytest-cov==6.1.1; extra == 'test'
 Requires-Dist: pytest-html==4.1.1; extra == 'test'
 Requires-Dist: pytest-sugar==1.0.0; extra == 'test'
 Requires-Dist: pytest-timeout==2.3.1; extra == 'test'
 Requires-Dist: pytest-xdist==3.6.1; extra == 'test'
-Requires-Dist: pytest==8.3.4; extra == 'test'
+Requires-Dist: pytest==8.3.5; extra == 'test'
 Description-Content-Type: text/markdown
 
 # Waldiez

{waldiez-0.4.1 → waldiez-0.4.3}/pyproject.toml

@@ -7,7 +7,7 @@ authors = [
   { name = 'Stella Ioannidou', email = 'stella@humancentered.gr'}
 ]
 readme = "README.md"
-requires-python = '>=3.10,<3.13'
+requires-python = '>=3.10,<3.14'
 dynamic = ["version"]
 classifiers = [
   "Development Status :: 3 - Alpha",
@@ -17,6 +17,7 @@ classifiers = [
   "Programming Language :: Python :: 3.10",
   "Programming Language :: Python :: 3.11",
   "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
   "Operating System :: OS Independent",
   "Intended Audience :: Science/Research",
   "Intended Audience :: Developers",
@@ -26,20 +27,22 @@ dependencies =[
     "aiofiles==24.1.0",
     "aiosqlite==0.21.0",
     "asyncer==0.0.8",
-    "numpy<2.2.0",
+    "numpy<=2.2.4",
     "graphviz==0.20.3",
     "pandas>=2",
-    "pyautogen==0.7.4",
+    "pyautogen[openai]==0.8.6",
     "jupytext",
     "parso==0.8.4",
-    "pydantic>=2.6.1,<3",
+    "pydantic>=2.10.2,<3",
     "typer>=0.9,<0.16",
     "httpx<1",
 ]
 
 [project.urls]
-homepage = 'https://waldiez.github.io/waldiez/python'
+homepage = 'https://waldiez.io'
 repository = 'https://github.com/waldiez/python.git'
+documentation = "https://waldiez.github.io/python/"
+issues = "https://github.com/waldiez/python/issues"
 
 [tool.hatch.build.targets.sdist]
 packages = ["waldiez", "waldiez.*"]
@@ -51,71 +54,74 @@ exclude = [ "**/example.py" ]
 
 [project.optional-dependencies]
 studio = [
-    "waldiez_studio==0.4.1",
+    "waldiez_studio==0.4.3",
 ]
 jupyter = [
-    "waldiez_jupyter==0.4.1",
-    "jupyterlab>=4.3.0",
+    "waldiez_jupyter==0.4.3",
+    'jupyterlab>=4.3.0',
 ]
 dev = [
     'autoflake==2.3.1',
-    'bandit==1.8.2',
+    'bandit==1.8.3',
     'black[jupyter]==25.1.0',
-    'flake8==7.1.1',
-    'isort==6.0.0',
+    'flake8==7.2.0',
+    'isort==6.0.1',
     'mypy==1.15.0',
-    "pandas-stubs",
-    'pre-commit==4.1.0',
+    'pandas-stubs==2.2.3.250308',
+    'pre-commit==4.2.0',
     'pydocstyle==6.3.0',
-    'pylint==3.3.4',
-    'python-dotenv==1.0.1',
-    'ruff==0.9.6',
-    'types-PyYAML==6.0.12.20241230',
-    'yamllint==1.35.1',
-    'toml; python_version <= "3.10"',
-    'types-toml== 0.10.8.20240310',
+    'pylint==3.3.6',
+    'python-dotenv==1.1.0',
+    'ruff==0.11.5',
+    'types-pyyaml==6.0.12.20250402',
+    'yamllint==1.37.0',
+    'toml==0.10.2; python_version <= "3.10"',
+    'types-toml==0.10.8.20240310',
 ]
 ag2_extras =[
     # models
-    'pyautogen[anthropic]==0.7.4',
-    'pyautogen[bedrock]==0.7.4',
-    'pyautogen[cohere]==0.7.4',
-    'pyautogen[gemini]==0.7.4',
-    'pyautogen[groq]==0.7.4',
-    'pyautogen[lmm]==0.7.4',
-    'pyautogen[mistral]==0.7.4',
-    'pyautogen[ollama]==0.7.4',
-    'pyautogen[together]==0.7.4',
+    'pyautogen[anthropic]==0.8.6',
+    'pyautogen[bedrock]==0.8.6',
+    'pyautogen[cohere]==0.8.6',
+    'pyautogen[gemini]==0.8.6',
+    'pyautogen[groq]==0.8.6',
+    'pyautogen[lmm]==0.8.6',
+    'pyautogen[mistral]==0.8.6',
+    'pyautogen[ollama]==0.8.6',
+    'pyautogen[together]==0.8.6',
     # utils
-    'pyautogen[interop-crewai]==0.7.4',
-    'pyautogen[interop-langchain]==0.7.4',
-    'pyautogen[neo4j]==0.7.4',
-    'pyautogen[websurfer]==0.7.4',
-    'chromadb>=0.5.23',
-    'pgvector>=0.3.6',
-    'psycopg[binary]>=3.2.3',
+    'pyautogen[interop-crewai]==0.8.6',
+    'pyautogen[interop-langchain]==0.8.6',
+    'pyautogen[neo4j]==0.8.6',
+    'pyautogen[websurfer]==0.8.6',
+    'chromadb>=0.5.10',
+    'pgvector>=0.4.0',
+    "psycopg==3.2.6; sys_platform == 'linux'",
+    "psycopg==3.2.6; sys_platform == 'win32' and platform_machine == 'arm64'",
+    "psycopg==3.2.6; sys_platform == 'win32' and platform_machine == 'ARM64'",
+    "psycopg==3.2.6; sys_platform == 'win32' and platform_machine == 'aarch64'",
+    "psycopg==3.2.6; sys_platform == 'win32' and platform_machine == 'AARCH64'",
+    "psycopg[binary]==3.2.6; sys_platform != 'linux' and platform_machine != 'arm64' and platform_machine != 'ARM64' and platform_machine != 'aarch64' and platform_machine != 'AARCH64'",
     'protobuf>=4.25.3',
     'pypdf',
+    "pysqlite3-binary==0.5.4; sys_platform == 'linux' and platform_machine != 'arm64' and platform_machine != 'aarch64' and platform_machine != 'ARM64' and platform_machine != 'AARCH64'",
     'ipython',
     'markdownify',
     'beautifulsoup4',
     'sentence-transformers',
     'huggingface-hub',
     'qdrant-client[fastembed]',
-    'pgvector>=0.3.6',
-    'psycopg[binary]>=3.2.4',
-    'crewai-tools==0.33.0',
-    'embedchain==0.1.126',
-    'crewai>0.98.0',
-    'weaviate-client>=4.10.2',
+    'pgvector>=0.4.0',
+    'psycopg[binary]>=3.2.6',
+    'embedchain',
     'langchain-community>=0.3.12,<1',
     'pydantic-ai>=0.0.21',
     'pymongo>=4.11',
 ]
 test = [
-    'pytest==8.3.4',
-    'pytest-asyncio==0.25.3',
-    'pytest-cov==6.0.0',
+    'pytest==8.3.5',
+    'pytest-asyncio==0.26.0',
+    'pytest-cov==6.1.1',
     'pytest-html==4.1.1',
     'pytest-sugar==1.0.0',
     'pytest-timeout==2.3.1',
@@ -127,10 +133,10 @@ docs = [
   'mkdocs==1.6.1',
   'mkdocs-jupyter==0.25.1',
   'mkdocs-macros-plugin==1.3.7',
-  'mkdocs-material==9.6.4',
-  'mkdocs-minify-html-plugin==0.2.4',
-  'mkdocstrings[crystal,python]==0.28.1',
-  'mkdocstrings-python==1.15.0'
+  'mkdocs-material==9.6.11',
+  'mkdocs-minify-html-plugin==0.3.1',
+  'mkdocstrings[crystal,python]==0.29.1',
+  'mkdocstrings-python==1.16.10'
 ]
 
 [project.scripts]
@@ -194,6 +200,12 @@ exclude = [
     'waldiez_out',
 ]
 
+[[tool.mypy.overrides]]
+module = "tests.*"
+# with python < 3.13 we get seem to get 'call-arg' errors :(
+# and with python>=3.13 'unused-ignore' : (
+disable_error_code = ["call-arg"]
+
 [tool.pydantic-mypy]
 init_forbid_extra = true
 init_typed = true

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/_version.py

@@ -2,4 +2,4 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Version information for Waldiez."""
 
-__version__ = "0.4.1"
+__version__ = "0.4.3"

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/exporting/agent/utils/captain_agent.py

@@ -102,12 +102,12 @@ def generate_nested_config(
     with open(
         config_file_or_env_path, "w", encoding="utf-8", newline="\n"
     ) as f:
-        json.dump([llm_config], f, ensure_ascii=False, indent=4)
+        json.dump(llm_config, f, ensure_ascii=False, indent=4)
     nested_config = {
         "autobuild_init_config": {
             "config_file_or_env": config_file_or_env_name,
-            "builder_model": llm_config["model"],
-            "agent_model": llm_config["model"],
+            "builder_model": llm_config["config_list"][0]["model"],
+            "agent_model": llm_config["config_list"][0]["model"],
         },
         "autobuild_build_config": get_auto_build_build_config(
             agent, llm_config
@@ -140,20 +140,29 @@ def get_llm_config(
     temperature: Optional[float] = 1
     top_p: Optional[float] = 0.95
     max_tokens: Optional[int] = 2048
+    config_dict: Dict[str, Any] = {}
     if agent.data.model_ids:
         waldiez_model = get_waldiez_model(agent.data.model_ids[0], all_models)
         llm_config = waldiez_model.get_llm_config(skip_price=True)
         for key in ["temperature", "top_p", "max_tokens"]:
             if key not in llm_config:
                 llm_config[key] = None
-        return llm_config
+        temp = llm_config.pop("temperature", None)
+        config_dict = {
+            "config_list": [llm_config],
+        }
+        if temp is not None:
+            config_dict["temperature"] = temp
+            return config_dict
     config_dict = {
         "model": model_name,
-        "temperature": temperature,
         "top_p": top_p,
         "max_tokens": max_tokens,
     }
-    return config_dict
+    return {
+        "config_list": [config_dict],
+        "temperature": temperature,
+    }
 
 
 def get_auto_build_build_config(
@@ -195,8 +204,8 @@ def get_auto_build_build_config(
     return {
         "default_llm_config": {
             "temperature": llm_config["temperature"],
-            "top_p": llm_config["top_p"],
-            "max_tokens": llm_config["max_tokens"],
+            "top_p": llm_config["config_list"][0]["top_p"],
+            "max_tokens": llm_config["config_list"][0]["max_tokens"],
         },
         "code_execution_config": code_execution_config,
         "coding": coding,

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/exporting/agent/utils/rag_user/vector_db.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 # pylint: disable=line-too-long
 """Vector DB exporting utils for RAG user agents."""
 

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/exporting/base/base_exporter.py

@@ -10,7 +10,7 @@ from .export_position import ExportPosition
 from .import_position import ImportPosition
 
 
-# flake8: noqa E501
+# flake8: noqa: E501
 # pylint: disable=line-too-long
 class ExporterReturnType(TypedDict):
     """Exporter Return Type.

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/exporting/chats/chats_exporter.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 # pylint: disable=line-too-long
 """Chats exporter."""
 

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/exporting/chats/utils/single_chat.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 # pylint: disable=line-too-long
 """Utilities for exporting a single chat in a flow."""
 

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/exporting/chats/utils/swarm.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 """Swarm chat exporting functions."""
 
 from typing import Callable, Dict, List, Optional, Tuple

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/exporting/flow/flow_exporter.py

@@ -19,7 +19,7 @@
 - If the flow is async, the `run` function is async.
 """
 
-# flake8: noqa E501
+# flake8: noqa: E501
 # pylint: disable=line-too-long
 
 from functools import partial

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/exporting/flow/utils/flow_content.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 # pylint: disable=line-too-long
 """Utils to generate the content of a flow."""
 
@@ -22,7 +22,7 @@ def get_py_content_start(waldiez: Waldiez) -> str:
         The first part of the python script.
     """
     content = "#!/usr/bin/env python\n"
-    content += "# flake8: noqa E501\n"
+    content += "# flake8: noqa: E501\n"
     content += get_pylint_ignore_comment(False)
     content += "# cspell: disable\n"
     content += f'"""{waldiez.name}.' + "\n\n"
@@ -60,7 +60,7 @@ def get_ipynb_content_start(
         # fmt: off
         content += "# " + f"!{{sys.executable}} -m pip install -q {requirements}" + "\n"
         # fmt: on
-    content += "# flake8: noqa E501"
+    content += "# flake8: noqa: E501"
     content += get_pylint_ignore_comment(True)
     content += "# cspell: disable\n"
     return content
@@ -135,8 +135,8 @@ def get_after_run_content(
     str
         The content to add after the flow is run.
     """
-    # if th eflow has reasoning agents, we add
-    # visualize_tree(agent._root) for each agent
+    # if the flow has a reasoning agents, we add
+    # agent.visualize_tree() for each agent
     content = ""
     tab = "    "
     space = tab * tabs
@@ -146,7 +146,7 @@ def get_after_run_content(
             content += f"""
 {space}# pylint: disable=broad-except,too-many-try-statements
 {space}try:
-{space}{tab}visualize_tree({agent_name}._root)  # pylint: disable=protected-access
+{space}{tab}{agent_name}.visualize_tree()
 {space}{tab}if os.path.exists("tree_of_thoughts.png"):
 {space}{tab}{tab}new_name = "{agent_name}_tree_of_thoughts.png"
 {space}{tab}{tab}os.rename("tree_of_thoughts.png", new_name)

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/exporting/models/utils.py

@@ -92,9 +92,11 @@ def get_agent_llm_config_arg(
     content = f"{tab}llm_config=" + "{\n"
     content += f'{tab}    "config_list": ['
     got_at_least_one_model = False
+    temperature: Optional[float] = None
     for model_id in agent.data.model_ids:
         model = next((m for m in all_models if m.id == model_id), None)
         if model is not None:
+            temperature = model.data.temperature
             model_name = model_names[model_id]
             content += "\n" + f"{tab}        {model_name}_llm_config,"
             got_at_least_one_model = True
@@ -102,6 +104,8 @@ def get_agent_llm_config_arg(
         return f"{tab}llm_config=False," + "\n"
     content += "\n" + f"{tab}    ]," + "\n"
     content += f'{tab}    "cache_seed": {cache_seed},' + "\n"
+    if temperature is not None:
+        content += f'{tab}    "temperature": {temperature},' + "\n"
     content += tab + "},\n"
     return content
 

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/exporting/skills/utils.py

@@ -263,8 +263,8 @@ def get_skill_secrets_import(flow_name: str, skill: WaldiezSkill) -> str:
         return ""
         # fmt: on
     module_name = f"{flow_name}_{skill.name}"
-    ignore_noqa = "  # type: ignore # noqa"
-    return f"import {module_name}_secrets{ignore_noqa}" + "\n"
+    type_ignore_noqa = "  # type: ignore # noqa"
+    return f"import {module_name}_secrets{type_ignore_noqa}" + "\n"
 
 
 def get_agent_skill_registrations(

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/models/agents/agent/agent.py

@@ -177,8 +177,7 @@ class WaldiezAgent(WaldiezBase):
             )
         elif agent_class == "ReasoningAgent":
             imports.add(
-                "from autogen.agentchat.contrib.reasoning_agent "
-                "import ReasoningAgent, visualize_tree"
+                "from autogen.agents.experimental import ReasoningAgent"
             )
         elif agent_class == "CaptainAgent":
             imports.add(

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/models/agents/rag_user/vector_db_config.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 
 """The vector db config for the RAG user agent."""
 

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/models/model/model.py

@@ -206,7 +206,7 @@ class WaldiezModel(WaldiezBase):
         for attr, atr_type in [
             ("base_url", str),
             ("max_tokens", int),
-            ("temperature", float),
+            # ("temperature", float),
             ("top_p", float),
             ("api_version", str),
             ("default_headers", dict),

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/models/model/model_data.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-# flake8: noqa E501
+# flake8: noqa: E501
 """Waldiez Model Data."""
 
 from typing import Dict, Optional

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/models/waldiez.py

@@ -239,7 +239,7 @@ class Waldiez:
             self.flow.requirements,
         )
         requirements = set(requirements_list)
-        requirements.add(f"pyautogen=={autogen_version}")
+        requirements.add(f"pyautogen[openai]=={autogen_version}")
         if self.has_rag_agents:
             rag_extras = get_retrievechat_extra_requirements(self.agents)
             requirements.update(rag_extras)

{waldiez-0.4.1 → waldiez-0.4.3}/waldiez/runner.py

@@ -203,7 +203,9 @@ class WaldiezRunner:
             "NOTE: If new packages were added and you are using Jupyter, "
             "you might need to restart the kernel."
         )
-        results: Union["ChatResult", List["ChatResult"]] = []
+        results: Union[
+            "ChatResult", List["ChatResult"], Dict[int, "ChatResult"]
+        ] = []
         with chdir(to=temp_dir):
             self._exporter.export(Path(file_name))
             spec = importlib.util.spec_from_file_location(
@@ -217,6 +219,7 @@ class WaldiezRunner:
             spec.loader.exec_module(module)
             printer("<Waldiez> - Starting workflow...")
             results = module.main()
+            printer("<Waldiez> - Workflow finished")
             sys.path.pop(0)
             reset_env_vars(old_vars)
         after_run(
@@ -248,7 +251,9 @@ class WaldiezRunner:
             "NOTE: If new packages were added and you are using Jupyter, "
             "you might need to restart the kernel."
         )
-        results: Union["ChatResult", List["ChatResult"]] = []
+        results: Union[
+            "ChatResult", List["ChatResult"], Dict[int, "ChatResult"]
+        ] = []
         async with a_chdir(to=temp_dir):
             self._exporter.export(Path(file_name))
             spec = importlib.util.spec_from_file_location(
@@ -278,7 +283,7 @@ class WaldiezRunner:
         output_path: Optional[Union[str, Path]] = None,
         uploads_root: Optional[Union[str, Path]] = None,
         skip_mmd: bool = False,
-    ) -> Union["ChatResult", List["ChatResult"]]:
+    ) -> Union["ChatResult", List["ChatResult"], Dict[int, "ChatResult"]]:
         """Run the Waldiez workflow.
 
         Parameters
@@ -292,7 +297,7 @@ class WaldiezRunner:
 
         Returns
         -------
-        Union[ChatResult, List[ChatResult]]
+        Union["ChatResult", List["ChatResult"], Dict[int, "ChatResult"]]
             The result(s) of the chat(s).
 
         Raises