waldiez 0.1.13__py3-none-any.whl → 0.1.14__py3-none-any.whl

waldiez/_version.py

@@ -1,3 +1,3 @@
 """Version information for Waldiez."""
 
-__version__ = "0.1.13"
+__version__ = "0.1.14"

waldiez/cli.py

@@ -25,6 +25,7 @@ app = typer.Typer(
     context_settings={
         "help_option_names": ["-h", "--help"],
         "allow_extra_args": True,
+        "ignore_unknown_options": True,
     },
     add_completion=False,
     no_args_is_help=True,

waldiez/conflict_checker.py

@@ -12,9 +12,10 @@ def check_conflicts() -> None:  # pragma: no cover
     try:
         version("autogen-agentchat")
         print(
-            "Conflict detected: 'autogen-agentchat' is installed, which conflicts with 'ag2'.\n"
+            "Conflict detected: 'autogen-agentchat' is installed, "
+            "which conflicts with 'ag2' / 'pyautogen'.\n"
             "Please uninstall 'autogen-agentchat': pip uninstall -y autogen-agentchat \n"
-            "And install 'ag2' (or 'waldiez') again: pip install --force ag2"
+            "And install 'pyautogen' (or 'waldiez') again: pip install --force pyautogen waldiez"
         )
         sys.exit(1)
     except PackageNotFoundError:

waldiez/models/agents/group_manager/speakers.py

@@ -148,8 +148,8 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
             default_factory=dict,
             title="Allowed or disallowed transitions",
             description=(
-                "A mapping (agent.id => List[agent.ids])",
-                "with the allowed or disallowed transitions.",
+                "A mapping (agent.id => List[agent.ids])"
+                "with the allowed or disallowed transitions."
             ),
             alias="allowedOrDisallowedTransitions",
         ),

waldiez/models/common/base.py

@@ -14,7 +14,7 @@ class WaldiezBase(BaseModel):
     """
 
     model_config = ConfigDict(
-        extra="forbid",
+        extra="ignore",
         # treat `skillId` as `skill_id`
         alias_generator=to_camel,
         # allow passing either `skill_id` or `skillId`

waldiez/models/flow/flow_data.py

@@ -42,15 +42,27 @@ class WaldiezFlowData(WaldiezBase):
     # (they for graph connections, positions, etc.)
     nodes: Annotated[
         List[Dict[str, Any]],
-        Field(default_factory=list),
+        Field(
+            default_factory=list,
+            title="Nodes",
+            description="The nodes of the flow",
+        ),
     ]
     edges: Annotated[
         List[Dict[str, Any]],
-        Field(default_factory=list),
+        Field(
+            default_factory=list,
+            title="Edges",
+            description="The edges of the flow",
+        ),
     ]
     viewport: Annotated[
         Dict[str, Any],
-        Field(default_factory=dict),
+        Field(
+            default_factory=dict,
+            title="Viewport",
+            description="The viewport of the flow",
+        ),
     ]
     # these are the ones we use.
     agents: Annotated[

waldiez/models/model/model_data.py

@@ -74,13 +74,70 @@ class WaldiezModelData(WaldiezBase):
             alias="baseUrl",
         ),
     ]
-    api_key: Annotated[Optional[str], Field(None, alias="apiKey")]
-    api_type: Annotated[WaldiezModelAPIType, Field("other", alias="apiType")]
-    api_version: Annotated[Optional[str], Field(None, alias="apiVersion")]
-    temperature: Annotated[Optional[float], Field(None, alias="temperature")]
-    top_p: Annotated[Optional[float], Field(None, alias="topP")]
-    max_tokens: Annotated[Optional[int], Field(None, alias="maxTokens")]
+    api_key: Annotated[
+        Optional[str],
+        Field(
+            None,
+            alias="apiKey",
+            title="API Key",
+            description="The api key to use with the model",
+        ),
+    ]
+    api_type: Annotated[
+        WaldiezModelAPIType,
+        Field(
+            "other",
+            alias="apiType",
+            title="API Type",
+            description="The api type of the model",
+        ),
+    ]
+    api_version: Annotated[
+        Optional[str],
+        Field(
+            None,
+            alias="apiVersion",
+            title="API Version",
+            description="The api version of the model",
+        ),
+    ]
+    temperature: Annotated[
+        Optional[float],
+        Field(
+            None,
+            alias="temperature",
+            title="Temperature",
+            description="The temperature of the model",
+        ),
+    ]
+    top_p: Annotated[
+        Optional[float],
+        Field(
+            None,
+            alias="topP",
+            title="Top P",
+            description="The top p of the model",
+        ),
+    ]
+    max_tokens: Annotated[
+        Optional[int],
+        Field(
+            None,
+            alias="maxTokens",
+            title="Max Tokens",
+            description="The max tokens of the model",
+        ),
+    ]
     default_headers: Annotated[
-        Dict[str, str], Field(alias="defaultHeaders", default_factory=dict)
+        Dict[str, str],
+        Field(
+            alias="defaultHeaders",
+            default_factory=dict,
+            title="Default Headers",
+            description="The default headers of the model",
+        ),
+    ]
+    price: Annotated[
+        Optional[WaldiezModelPrice],
+        Field(None, title="Price", description="The price of the model"),
     ]
-    price: Annotated[Optional[WaldiezModelPrice], Field(None)]

waldiez/models/waldiez.py

@@ -211,16 +211,17 @@ class Waldiez:
             lambda requirement: not (
                 requirement.startswith("pyautogen")
                 or requirement.startswith("ag2")
+                or requirement.startswith("autogen")
             ),
             self.flow.requirements,
         )
         requirements = set(requirements_list)
         if self.has_rag_agents:
-            requirements.add(f"ag2[retrievechat]=={autogen_version}")
+            requirements.add(f"pyautogen[retrievechat]=={autogen_version}")
         else:
-            requirements.add(f"ag2=={autogen_version}")
+            requirements.add(f"pyautogen=={autogen_version}")
         if self.has_multimodal_agents:
-            requirements.add(f"ag2[lmm]=={autogen_version}")
+            requirements.add(f"pyautogen[lmm]=={autogen_version}")
         # ref: https://github.com/ag2ai/ag2/blob/main/setup.py
         models_with_additional_requirements = [
             "together",
@@ -234,7 +235,7 @@ class Waldiez:
         for model in self.models:
             if model.data.api_type in models_with_additional_requirements:
                 requirements.add(
-                    f"ag2[{model.data.api_type}]==" f"{autogen_version}"
+                    f"pyautogen[{model.data.api_type}]==" f"{autogen_version}"
                 )
         return list(requirements)
 

{waldiez-0.1.13.dist-info → waldiez-0.1.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: waldiez
-Version: 0.1.13
+Version: 0.1.14
 Summary: waldiez
 Project-URL: homepage, https://waldiez.github.io/waldiez/
 Project-URL: repository, https://github.com/waldiez/waldiez.git
@@ -16,26 +16,26 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Python: <3.13,>=3.10
-Requires-Dist: ag2==0.4.1
 Requires-Dist: jupytext
+Requires-Dist: pyautogen==0.5.0
 Requires-Dist: typer<0.13,>=0.9
 Provides-Extra: ag2-extras
-Requires-Dist: ag2[anthropic]==0.4.1; extra == 'ag2-extras'
-Requires-Dist: ag2[bedrock]==0.4.1; extra == 'ag2-extras'
-Requires-Dist: ag2[captainagent]==0.4.1; extra == 'ag2-extras'
-Requires-Dist: ag2[gemini]==0.4.1; extra == 'ag2-extras'
-Requires-Dist: ag2[groq]==0.4.1; extra == 'ag2-extras'
-Requires-Dist: ag2[lmm]==0.4.1; extra == 'ag2-extras'
-Requires-Dist: ag2[mistral]==0.4.1; extra == 'ag2-extras'
-Requires-Dist: ag2[retrievechat-mongodb]==0.4.1; extra == 'ag2-extras'
-Requires-Dist: ag2[retrievechat-pgvector]==0.4.1; extra == 'ag2-extras'
-Requires-Dist: ag2[retrievechat-qdrant]==0.4.1; extra == 'ag2-extras'
-Requires-Dist: ag2[retrievechat]==0.4.1; extra == 'ag2-extras'
-Requires-Dist: ag2[together]==0.4.1; extra == 'ag2-extras'
-Requires-Dist: ag2[websurfer]==0.4.1; extra == 'ag2-extras'
 Requires-Dist: fastembed==0.4.2; extra == 'ag2-extras'
 Requires-Dist: pgvector>=0.3.6; extra == 'ag2-extras'
 Requires-Dist: psycopg[binary]>=3.2.3; extra == 'ag2-extras'
+Requires-Dist: pyautogen[anthropic]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[bedrock]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[captainagent]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[gemini]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[groq]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[lmm]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[mistral]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[retrievechat-mongodb]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[retrievechat-pgvector]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[retrievechat-qdrant]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[retrievechat]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[together]==0.5.0; extra == 'ag2-extras'
+Requires-Dist: pyautogen[websurfer]==0.5.0; extra == 'ag2-extras'
 Requires-Dist: pymongo==4.10.1; extra == 'ag2-extras'
 Requires-Dist: qdrant-client==1.12.1; extra == 'ag2-extras'
 Provides-Extra: dev
@@ -49,7 +49,7 @@ Requires-Dist: pre-commit==4.0.1; extra == 'dev'
 Requires-Dist: pydocstyle==6.3.0; extra == 'dev'
 Requires-Dist: pylint==3.3.2; extra == 'dev'
 Requires-Dist: python-dotenv==1.0.1; extra == 'dev'
-Requires-Dist: ruff==0.8.1; extra == 'dev'
+Requires-Dist: ruff==0.8.2; extra == 'dev'
 Requires-Dist: types-pyyaml==6.0.12.20240917; extra == 'dev'
 Requires-Dist: yamllint==1.35.1; extra == 'dev'
 Provides-Extra: docs
@@ -123,7 +123,7 @@ $CONTAINER_COMMAND run \
   --rm \
   -v /path/to/a/flow.waldiez:/flow.waldiez \
   -v /path/to/an/output:/output \
-  waldiez/waldiez --export /flow.waldiez --output /output/flow[.py|.ipynb]
+  waldiez/waldiez convert --file /flow.waldiez --output /output/flow[.py|.ipynb] [--force]
 
 # with selinux and/or podman, you might get permission (or file not found) errors, so you can try:
 $CONTAINER_COMMAND run \
@@ -132,7 +132,7 @@ $CONTAINER_COMMAND run \
   -v /path/to/an/output:/output \
   --userns=keep-id \
   --security-opt label=disable \
-  waldiez/waldiez convert --file /flow.waldiez --output /output/flow[.py|.ipynb]
+  waldiez/waldiez convert --file /flow.waldiez --output /output/flow[.py|.ipynb] [--force]
 ```
 
 ```shell
@@ -144,6 +144,13 @@ $CONTAINER_COMMAND run \
   waldiez/waldiez run --file /flow.waldiez --output /output/output[.py]
 ```
 
+### UI
+
+For creating-only (no exporting or running) waldiez flows, you can use the playground at <https://waldiez.github.io>.  
+The repo for the js library is [here](https://github.com/waldiez/react).  
+We are currently working on waldiez-studio to provide a visual interface for creating and running Waldiez flows (you can find more [here](https://github.com/waldiez/studio)).  
+Until then, you can use our [Jupyter](https://github.com/waldiez/jupyter) or the [VSCode](https://github.com/waldiez/vscode) extension to create and run Waldiez flows.
+
 ### As a library
 
 #### Export a flow
@@ -244,7 +251,7 @@ with WaldiezIOStream.set_default(io_stream):
 
 ## Known Conflicts
 
-- **autogen-agentchat**: This package conflicts with `ag2`. Ensure that `autogen-agentchat` is uninstalled before installing `waldiez`. If you have already installed `autogen-agentchat`, you can uninstall it with the following command:
+- **autogen-agentchat**: This package conflicts with `ag2` / `pyautogen`. Ensure that `autogen-agentchat` is uninstalled before installing `waldiez`. If you have already installed `autogen-agentchat`, you can uninstall it with the following command:
 
   ```shell
   pip uninstall autogen-agentchat -y
@@ -253,7 +260,7 @@ with WaldiezIOStream.set_default(io_stream):
   If already installed waldiez you might need to reinstall it after uninstalling `autogen-agentchat`:
   
     ```shell
-    pip install --force --no-cache waldiez
+    pip install --force --no-cache waldiez pyautogen
     ```
 
 ## License

{waldiez-0.1.13.dist-info → waldiez-0.1.14.dist-info}/RECORD

@@ -1,8 +1,8 @@
 waldiez/__init__.py,sha256=ASGmxP3JcASIv89_tMYhYxvW0FlHK1jM0Nfb1hMN-z0,722
 waldiez/__main__.py,sha256=mUQWu4CF03Jbbscvcfb_9EFTnIMliJJJmAuWf0sRRZU,110
-waldiez/_version.py,sha256=N8g2SvYhwFeB_7w5OlUWgVZ5b8OUXI3arJvloUBkg9U,63
-waldiez/cli.py,sha256=6BefTHXqib-L_FTdYOAaUkv4DLYYpjBePdlCn2BEYi0,6307
-waldiez/conflict_checker.py,sha256=T5FA7hsxpuW0Byb4YyZQ3-uVxzNnoTQV199YRBKjvmE,707
+waldiez/_version.py,sha256=GH_14gJJkcToI_wPH-frZ3GujMLUSIN6iAMKLTY-6xA,63
+waldiez/cli.py,sha256=Ayp6Xu2HSBBFIouYh8cArjOjgXv5kbtZd_PYewcDklU,6347
+waldiez/conflict_checker.py,sha256=HWDth7kF31DKdJ3VzUpx2Zz8Q4UUTnbCwWon8BYJC0E,756
 waldiez/exporter.py,sha256=iKe-l_Me8NRWsXHIdBcrOrnLT9XIyp4iYi4HLuuj2jA,9342
 waldiez/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 waldiez/runner.py,sha256=v4A4NHdSnh96TQ5Vx1vs8UrTsshc2oAwDZrEHmq4RkU,11537
@@ -41,7 +41,7 @@ waldiez/exporting/utils/object_string.py,sha256=2kdIu4in3iUV92a2KbLWwp9POhvY-fzF
 waldiez/exporting/utils/path_check.py,sha256=aO49sbk6hUHEr65UjNNUSKO5WCGnQitiT733W-kGVtI,1062
 waldiez/io/__init__.py,sha256=-w6ywIjG8ByxGtpSeRt1bjhnP4CZzpHf8WRWckjs4ng,5172
 waldiez/models/__init__.py,sha256=IMq8vzuAgmv92kHSSuZQLF38vVd31ojgOHonuHqWYj0,2888
-waldiez/models/waldiez.py,sha256=gNZYlXVW3ONHl9snwtRruFOCa0mX23ffUOiF104ctRQ,9351
+waldiez/models/waldiez.py,sha256=yI38WEXPJa-FmRF6Try-MabWoPCeqlxq8VUt0sD-fJg,9428
 waldiez/models/agents/__init__.py,sha256=3ZyVYBHMFzZjRMIdPrBF6HLg82LPAlEubL6utL6KhfU,1856
 waldiez/models/agents/agents.py,sha256=zIqihnoBjzaQLL_P6FcVoHddcusRNYsWPIFLZD091bE,3641
 waldiez/models/agents/agent/__init__.py,sha256=inA0zV3dnwmcQlcObH_FLaZSURjFG31E_XUampJAnJU,746
@@ -58,7 +58,7 @@ waldiez/models/agents/assistant/assistant_data.py,sha256=VGdF1IZBBqBvZwe6BP2s4Li
 waldiez/models/agents/group_manager/__init__.py,sha256=To97X5vdRTcTSylWUH2hlgkNH2pdn4TRUJqvLiU5-Bk,592
 waldiez/models/agents/group_manager/group_manager.py,sha256=0Bg3rEGL414M8gIMlgg6q3et7RFgGheHbvHUcONc-qw,2687
 waldiez/models/agents/group_manager/group_manager_data.py,sha256=NHA_uRC3KxqquhW4sEoNVhT8XWpXD1W_8EYaRjzJuG8,2785
-waldiez/models/agents/group_manager/speakers.py,sha256=tlz15MO2veK_v895MsIa2zeOzEGamkdLg3W5Ctn8nV8,6648
+waldiez/models/agents/group_manager/speakers.py,sha256=OXYEEexGeQUf9hllixJTCE3atp2-_hMmdSsEW5XCg8g,6646
 waldiez/models/agents/rag_user/__init__.py,sha256=_Ge6ekCPHGuDuebbP3unUKbWrjeN8Hx3ST68_DapE7w,678
 waldiez/models/agents/rag_user/rag_user.py,sha256=l4a_IzlNPtNb-GTx22r15XIVmHxvhuM5KbXjkUwS8JU,1558
 waldiez/models/agents/rag_user/rag_user_data.py,sha256=4WjG8UcQ8ltLAxvZgsZUUlaEqHrX3KewrVIH7UJeUYo,870
@@ -74,19 +74,19 @@ waldiez/models/chat/chat_message.py,sha256=OtA65nNm4J0N3G5EHz7ga9dKsKqzClyhsVNm7
 waldiez/models/chat/chat_nested.py,sha256=OFeytlQ1Rgt6hx9_-xe47PdNnmgztokxldEDPJuGKsc,4685
 waldiez/models/chat/chat_summary.py,sha256=fiF0X6nk5dLoZFfwKBHoytk2ArhvIpVHETyCIH7uKd4,2823
 waldiez/models/common/__init__.py,sha256=1WhzhGYYUWMuHgxjiT1UralMClutO3_5BhFtnkhp8Yk,686
-waldiez/models/common/base.py,sha256=aEZhUEy46HeivemX851bVcwAAovxgXNq_rtkvkniBIk,1687
+waldiez/models/common/base.py,sha256=Aef91uGtbDfpUBGpY0m49L0tBHSaX23xvHDekIq-3H0,1687
 waldiez/models/common/method_utils.py,sha256=K7dNyvqHwvwuULLgn1_uK72Bfnm9d0CYHP3rYW5IgjM,5502
 waldiez/models/flow/__init__.py,sha256=oy_G58xDkZk_LZEvqmr-0MJrpYy1JRf-U-F5-bI1944,162
 waldiez/models/flow/flow.py,sha256=Zxcl0Z9qty4DDRLSCaMSPOBcUZrd6pcWcuCFf_vDGPE,9372
-waldiez/models/flow/flow_data.py,sha256=7dLMa0RHQSN7y8xnIqZviS9EKelCkk22ZYliNDO7qMM,2391
+waldiez/models/flow/flow_data.py,sha256=m-RYaqlWjVpjyUzX9iAtXn1GOH0pFGKd2QD3e_e7yrk,2694
 waldiez/models/model/__init__.py,sha256=32Xox2Qw1jsf7ZCcoi0vMgZ5yxulLmtBrRRxq3FBAks,246
 waldiez/models/model/model.py,sha256=aG-9YAyLDYPqF-GWRTfuI7hP1Z7tfWWfj7Dhy1Y2GSc,4997
-waldiez/models/model/model_data.py,sha256=pDPKUbltaXWjCuDArgwTOEHw_igfk_DkxzFzdn6zo14,2599
+waldiez/models/model/model_data.py,sha256=VLPb60rJeZEgVZCjjkQGiwTrKz7OddVLrXCrEuteMcc,3679
 waldiez/models/skill/__init__.py,sha256=rU88bajKOGMYoHFcE8MP0jW9H0MswbQmvz5wxS35BYE,169
 waldiez/models/skill/skill.py,sha256=fhsAI413an2_d4DBIkf7dzEuWk6rGs2t4sl97a4dj20,3473
 waldiez/models/skill/skill_data.py,sha256=RTWn8Od6w7g-nRIpsS29sqZ8sPm5dCPiK7-qXmU-KD4,815
-waldiez-0.1.13.dist-info/METADATA,sha256=opCfL797TnrS7AqpTvdz_YreuVRhKte5sA03NIzCrOg,9353
-waldiez-0.1.13.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-waldiez-0.1.13.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
-waldiez-0.1.13.dist-info/licenses/LICENSE,sha256=VQEHM6WMQLRu1qaGl3GWsoOknDwro-69eGo4NLIJPIM,1064
-waldiez-0.1.13.dist-info/RECORD,,
+waldiez-0.1.14.dist-info/METADATA,sha256=_2D_zK5D9Ud4Qct2F56jSiAnuK7zi6ZqAFHLKyBGU3M,10036
+waldiez-0.1.14.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+waldiez-0.1.14.dist-info/entry_points.txt,sha256=9MQ8Y1rD19CU7UwjNPwoyTRpQsPs2QimjrtwTD0bD6k,44
+waldiez-0.1.14.dist-info/licenses/LICENSE,sha256=VQEHM6WMQLRu1qaGl3GWsoOknDwro-69eGo4NLIJPIM,1064
+waldiez-0.1.14.dist-info/RECORD,,