waldiez 0.5.10__py3-none-any.whl → 0.6.1__py3-none-any.whl

waldiez/models/agents/agent/agent.py

@@ -1,6 +1,8 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
+
 # pylint: disable=too-many-public-methods
+# pyright:  reportArgumentType=false
 """Base agent class to be inherited by all agents."""
 
 import warnings
@@ -21,10 +23,11 @@ from .code_execution import WaldiezAgentCodeExecutionConfig
 from .nested_chat import WaldiezAgentNestedChat, WaldiezAgentNestedChatMessage
 
 if TYPE_CHECKING:
+    # noinspection PyUnusedImports
     from ...chat import WaldiezChat
 
 
-# noinspection PyUnresolvedReferences
+# noinspection PyUnresolvedReferences,PyNestedDecorators
 class WaldiezAgent(WaldiezBase):
     """Waldiez Agent to be inherited by all other agents.
 
@@ -70,7 +73,7 @@ class WaldiezAgent(WaldiezBase):
             title="Type",
             description="The type of the 'node' in a graph.",
         ),
-    ] = "agent"
+    ]
     agent_type: Annotated[
         WaldiezAgentType,
         Field(
@@ -92,7 +95,7 @@ class WaldiezAgent(WaldiezBase):
             title="Description",
             description="The description of the agent",
         ),
-    ] = "Agent's description"
+    ]
     tags: Annotated[
         list[str],
         Field(
@@ -100,7 +103,7 @@ class WaldiezAgent(WaldiezBase):
             description="Tags of the agent",
             default_factory=list,
         ),
-    ] = []
+    ]
     requirements: Annotated[
         list[str],
         Field(
@@ -108,7 +111,7 @@ class WaldiezAgent(WaldiezBase):
             description="Python requirements for the agent",
             default_factory=list,
         ),
-    ] = []
+    ]
     created_at: Annotated[
         str,
         Field(
@@ -130,7 +133,7 @@ class WaldiezAgent(WaldiezBase):
         Field(
             title="Data",
             description="The data (properties) of the agent",
-            default_factory=WaldiezAgentData,  # pyright: ignore
+            default_factory=WaldiezAgentData,
         ),
     ]
 
@@ -138,7 +141,7 @@ class WaldiezAgent(WaldiezBase):
         list[WaldiezHandoff],
         Field(
             init=False,  # this is not a field in the constructor
-            default_factory=list[WaldiezHandoff],
+            default_factory=list,
             title="Handoffs",
             description=(
                 "A list of handoffs (target ids) to register. "
@@ -147,10 +150,10 @@ class WaldiezAgent(WaldiezBase):
         ),
     ] = []
 
-    _checked_nested_chats: Annotated[bool, Field(init=False, default=False)] = (
+    _checked_nested_chats: Annotated[bool, Field(default=False, init=False)] = (
         False
     )
-    _checked_handoffs: Annotated[bool, Field(init=False, default=False)] = False
+    _checked_handoffs: Annotated[bool, Field(default=False, init=False)] = False
 
     @property
     def args_to_skip(self) -> list[str]:
@@ -189,10 +192,11 @@ class WaldiezAgent(WaldiezBase):
             If handoffs have not been gathered yet.
         """
         if not self._checked_handoffs:
-            raise RuntimeError(
+            msg = (
                 "Handoffs have not been gathered yet. "
                 "Call gather_handoffs() first."
             )
+            raise RuntimeError(msg)
         return self._handoffs
 
     @field_validator("agent_type")
@@ -391,7 +395,7 @@ class WaldiezAgent(WaldiezBase):
         return class_name  # pragma: no cover
 
     @property
-    def ag2_imports(self) -> set[str]:
+    def ag2_imports(self) -> set[str]:  # pylint: disable=too-complex
         """Return the AG2 imports of the agent.
 
         Returns
@@ -412,25 +416,28 @@ class WaldiezAgent(WaldiezBase):
             case "UserProxyAgent":
                 imports.add("from autogen import UserProxyAgent")
             case "RetrieveUserProxyAgent":
-                imports.add(
+                _imp = (
                     "from autogen.agentchat.contrib.retrieve_user_proxy_agent "
                     "import RetrieveUserProxyAgent"
                 )
+                imports.add(_imp)
             case "MultimodalConversableAgent":
-                imports.add(
+                _imp = (
                     "from "
                     "autogen.agentchat.contrib.multimodal_conversable_agent "
                     "import MultimodalConversableAgent"
                 )
+                imports.add(_imp)
             case "ReasoningAgent":
                 imports.add(
                     "from autogen.agents.experimental import ReasoningAgent"
                 )
             case "CaptainAgent":
-                imports.add(
+                _imp = (
                     "from autogen.agentchat.contrib.captainagent "
                     "import CaptainAgent"
                 )
+                imports.add(_imp)
             case "GroupChatManager":  # pragma: no branch
                 imports.add("from autogen import GroupChat")
                 imports.add("from autogen.agentchat import GroupChatManager")
@@ -442,10 +449,11 @@ class WaldiezAgent(WaldiezBase):
             case "ConversableAgent":
                 imports.add("from autogen import ConversableAgent")
             case _:  # pragma: no cover
-                raise ValueError(
+                msg = (
                     f"Unknown agent class: {agent_class}. "
                     "Please implement the imports for this class."
                 )
+                raise ValueError(msg)
         return imports
 
     def validate_linked_tools(
@@ -677,8 +685,8 @@ class WaldiezAgent(WaldiezBase):
             nested_chat = WaldiezAgentNestedChat(
                 triggered_by=triggered_by,
                 messages=messages,
-                condition=chat_with_condition.condition,  # pyright: ignore
-                available=chat_with_condition.available,  # pyright: ignore
+                condition=chat_with_condition.condition,
+                available=chat_with_condition.available,
             )
             self.data.nested_chats.append(nested_chat)
 

waldiez/models/agents/agent/agent_data.py

@@ -1,8 +1,11 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
+
+# pyright: reportArgumentType=false
+
 """Common data structures for agents."""
 
-from typing import Any, Optional, Union
+from typing import Any
 
 from pydantic import ConfigDict, Field, model_validator
 from pydantic.alias_generators import to_camel
@@ -67,14 +70,14 @@ class WaldiezAgentData(WaldiezBase):
     )
 
     system_message: Annotated[
-        Optional[str],
+        str | None,
         Field(
             None,
             title="System message",
             description="The agent's system message.",
             alias="systemMessage",
         ),
-    ] = None
+    ]
     human_input_mode: Annotated[
         WaldiezAgentHumanInputMode,
         Field(
@@ -83,9 +86,9 @@ class WaldiezAgentData(WaldiezBase):
             description="The human input mode to use for the agent.",
             alias="humanInputMode",
         ),
-    ] = "NEVER"
+    ]
     code_execution_config: Annotated[
-        Union[WaldiezAgentCodeExecutionConfig, Literal[False]],
+        WaldiezAgentCodeExecutionConfig | Literal[False],
         Field(
             False,
             title="Code execution config",
@@ -95,9 +98,9 @@ class WaldiezAgentData(WaldiezBase):
             ),
             alias="codeExecutionConfig",
         ),
-    ] = False
+    ]
     agent_default_auto_reply: Annotated[
-        Optional[str],
+        str | None,
         Field(
             None,
             title="Agent's default auto reply",
@@ -106,9 +109,9 @@ class WaldiezAgentData(WaldiezBase):
             ),
             alias="agentDefaultAutoReply",
         ),
-    ] = None
+    ]
     max_consecutive_auto_reply: Annotated[
-        Optional[int],
+        int | None,
         Field(
             None,
             title="Max consecutive auto reply",
@@ -118,7 +121,7 @@ class WaldiezAgentData(WaldiezBase):
             ),
             alias="maxConsecutiveAutoReply",
         ),
-    ] = None
+    ]
     termination: Annotated[
         WaldiezAgentTerminationMessage,
         Field(
@@ -140,7 +143,7 @@ class WaldiezAgentData(WaldiezBase):
             ),
             alias="modelIds",
         ),
-    ] = []
+    ]
     tools: Annotated[
         list[WaldiezAgentLinkedTool],
         Field(
@@ -148,7 +151,7 @@ class WaldiezAgentData(WaldiezBase):
             title="Tools",
             description="A list of tools (id and executor) to register.",
         ),
-    ] = []
+    ]
     nested_chats: Annotated[
         list[WaldiezAgentNestedChat],
         Field(
@@ -158,11 +161,11 @@ class WaldiezAgentData(WaldiezBase):
             ),
             alias="nestedChats",
         ),
-    ] = []
+    ]
     context_variables: Annotated[
         dict[str, Any],
         Field(
-            default_factory=dict,  # pyright: ignore
+            default_factory=dict,
             title="Context variables",
             description=(
                 "Context variables that provide a persistent context "
@@ -172,10 +175,10 @@ class WaldiezAgentData(WaldiezBase):
             ),
             alias="contextVariables",
         ),
-    ] = {}
+    ]
 
     update_agent_state_before_reply: Annotated[
-        list[Union[str, WaldiezAgentUpdateSystemMessage]],
+        list[str | WaldiezAgentUpdateSystemMessage],
         Field(
             title="Update Agent State Before Reply",
             alias="updateAgentStateBeforeReply",
@@ -188,7 +191,7 @@ class WaldiezAgentData(WaldiezBase):
             ),
             default_factory=list,
         ),
-    ] = []
+    ]
     handoffs: Annotated[
         list[str],
         Field(
@@ -196,9 +199,9 @@ class WaldiezAgentData(WaldiezBase):
             title="Handoffs",
             description="A list of handoffs (target ids) to register.",
         ),
-    ] = []
+    ]
     after_work: Annotated[
-        Optional[WaldiezTransitionTarget],
+        WaldiezTransitionTarget | None,
         Field(
             None,
             title="After work",
@@ -208,10 +211,10 @@ class WaldiezAgentData(WaldiezBase):
             ),
             alias="afterWork",
         ),
-    ] = None
+    ]
 
     parent_id: Annotated[
-        Optional[str],
+        str | None,
         Field(
             None,
             title="Parent ID",
@@ -220,7 +223,7 @@ class WaldiezAgentData(WaldiezBase):
             ),
             alias="parentId",
         ),
-    ] = None
+    ]
 
     @model_validator(mode="after")
     def update_context_variables(self) -> Self:

waldiez/models/agents/agent/code_execution.py

@@ -2,8 +2,6 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Waldiez Agent Code Execution Configuration."""
 
-from typing import Optional
-
 from pydantic import Field
 from typing_extensions import Annotated
 
@@ -28,25 +26,25 @@ class WaldiezAgentCodeExecutionConfig(WaldiezBase):
     """
 
     work_dir: Annotated[
-        Optional[str],
+        str | None,
         Field(
             default=None,
             title="Working directory",
             description="The working directory for the code execution.",
             alias="workDir",
         ),
-    ] = None
+    ]
     use_docker: Annotated[
-        Optional[bool],
+        bool | None,
         Field(
             default=None,
             title="Use docker",
             description="Run the code in a docker container.",
             alias="useDocker",
         ),
-    ] = None
+    ]
     timeout: Annotated[
-        Optional[float],
+        float | None,
         Field(
             default=None,
             title="Timeout",
@@ -54,15 +52,15 @@ class WaldiezAgentCodeExecutionConfig(WaldiezBase):
                 "The timeout for the code execution.Default: No timeout"
             ),
         ),
-    ] = None
+    ]
     last_n_messages: Annotated[
-        Optional[int],
+        int | None,
         Field(
             default=None,
             title="Last N Messages",
             description="The number of previous messages in the chat to use.",
         ),
-    ] = None
+    ]
     functions: Annotated[
         list[str],
         Field(
@@ -70,4 +68,4 @@ class WaldiezAgentCodeExecutionConfig(WaldiezBase):
             title="Functions",
             description="If not using docker, the function ids to use",
         ),
-    ] = []
+    ]

waldiez/models/agents/agent/termination_message.py

@@ -2,8 +2,6 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Waldiez Agent Termination Message Check."""
 
-from typing import Optional
-
 from pydantic import Field, model_validator
 from typing_extensions import Annotated, Literal, Self
 
@@ -45,7 +43,7 @@ class WaldiezAgentTerminationMessage(WaldiezBase):
                 "none, keyword, method"
             ),
         ),
-    ] = "none"
+    ]
     keywords: Annotated[
         list[str],
         Field(
@@ -56,9 +54,9 @@ class WaldiezAgentTerminationMessage(WaldiezBase):
                 " the keywords to search in the message."
             ),
         ),
-    ] = []
+    ]
     criterion: Annotated[
-        Optional[Literal["found", "ending", "starting", "exact"]],
+        Literal["found", "ending", "starting", "exact"] | None,
         Field(
             "exact",
             title="Criterion",
@@ -67,9 +65,9 @@ class WaldiezAgentTerminationMessage(WaldiezBase):
                 "the criterion to use (e.g.: in, endswith, ==)"
             ),
         ),
-    ] = "exact"
+    ]
     method_content: Annotated[
-        Optional[str],
+        str | None,
         Field(
             None,
             title="Method content",
@@ -80,22 +78,22 @@ class WaldiezAgentTerminationMessage(WaldiezBase):
                 "bool (whether the message is a termination message or not.)"
             ),
         ),
-    ] = None
+    ]
 
     _string: str = "None"
 
     def get_termination_function(
         self,
-        name_prefix: Optional[str] = None,
-        name_suffix: Optional[str] = None,
+        name_prefix: str | None = None,
+        name_suffix: str | None = None,
     ) -> tuple[str, str]:
         """Get the termination function.
 
         Parameters
         ----------
-        name_prefix : str
+        name_prefix : str | None
             The function name prefix.
-        name_suffix : str
+        name_suffix : str | None
             The function name suffix.
 
         Returns

waldiez/models/agents/agent/update_system_message.py

@@ -2,8 +2,6 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Update the agent's system message before they reply."""
 
-from typing import Optional
-
 from pydantic import Field, model_validator
 from typing_extensions import Annotated, Literal, Self
 
@@ -78,8 +76,8 @@ class WaldiezAgentUpdateSystemMessage(WaldiezBase):
 
     def get_content(
         self,
-        name_prefix: Optional[str] = None,
-        name_suffix: Optional[str] = None,
+        name_prefix: str | None = None,
+        name_suffix: str | None = None,
     ) -> tuple[str, str]:
         """Get the update function content.
 

waldiez/models/agents/agents.py

@@ -2,7 +2,7 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Waldiez agents model."""
 
-from typing import Iterator
+from collections.abc import Iterator
 
 from pydantic import Field, model_validator
 from typing_extensions import Annotated, Self
@@ -44,7 +44,7 @@ class WaldiezAgents(WaldiezBase):
             description="The User proxy agents in the flow.",
             default_factory=list,
         ),
-    ] = []
+    ]
     assistantAgents: Annotated[
         list[WaldiezAssistant],
         Field(
@@ -52,7 +52,7 @@ class WaldiezAgents(WaldiezBase):
             description="The assistant agents in the flow.",
             default_factory=list,
         ),
-    ] = []
+    ]
     ragUserProxyAgents: Annotated[
         list[WaldiezRagUserProxy],
         Field(
@@ -60,7 +60,7 @@ class WaldiezAgents(WaldiezBase):
             description="The RAG user proxy agents in the flow.",
             default_factory=list,
         ),
-    ] = []
+    ]
     reasoningAgents: Annotated[
         list[WaldiezReasoningAgent],
         Field(
@@ -68,7 +68,7 @@ class WaldiezAgents(WaldiezBase):
             description="The Reasoning agents in the flow.",
             default_factory=list,
         ),
-    ] = []
+    ]
     captainAgents: Annotated[
         list[WaldiezCaptainAgent],
         Field(
@@ -76,7 +76,7 @@ class WaldiezAgents(WaldiezBase):
             description="The Captain agents in the flow.",
             default_factory=list,
         ),
-    ] = []
+    ]
     groupManagerAgents: Annotated[
         list[WaldiezGroupManager],
         Field(
@@ -84,7 +84,7 @@ class WaldiezAgents(WaldiezBase):
             description="The Group manager agents in the flow.",
             default_factory=list,
         ),
-    ] = []
+    ]
     docAgents: Annotated[
         list[WaldiezDocAgent],
         Field(
@@ -92,7 +92,7 @@ class WaldiezAgents(WaldiezBase):
             description="The Document agents in the flow.",
             default_factory=list,
         ),
-    ] = []
+    ]
 
     @property
     def members(self) -> Iterator[WaldiezAgent]:

waldiez/models/agents/assistant/assistant.py

@@ -1,5 +1,8 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
+
+# pyright: reportArgumentType=false,reportIncompatibleVariableOverride=false
+
 """Assistant agent model."""
 
 from pydantic import Field
@@ -24,7 +27,7 @@ class WaldiezAssistant(WaldiezAgent):
         The assistant agent's data
     """
 
-    agent_type: Annotated[  # pyright: ignore
+    agent_type: Annotated[
         Literal["assistant"],
         Field(
             "assistant",
@@ -33,11 +36,11 @@ class WaldiezAssistant(WaldiezAgent):
             alias="agentType",
         ),
     ] = "assistant"
-    data: Annotated[  # pyright: ignore
+    data: Annotated[
         WaldiezAssistantData,
         Field(
             title="Data",
             description="The assistant agent's data",
-            default_factory=WaldiezAssistantData,  # pyright: ignore
+            default_factory=WaldiezAssistantData,
         ),
     ]

waldiez/models/agents/assistant/assistant_data.py

@@ -31,7 +31,7 @@ class WaldiezAssistantData(WaldiezAgentData):
             description="The human input mode, Defaults to `NEVER`",
             alias="humanInputMode",
         ),
-    ] = "NEVER"
+    ]
     is_multimodal: Annotated[
         bool,
         Field(
@@ -40,4 +40,4 @@ class WaldiezAssistantData(WaldiezAgentData):
             description="A flag to indicate if the agent is multimodal.",
             alias="isMultimodal",
         ),
-    ] = False
+    ]

waldiez/models/agents/captain/captain_agent.py

@@ -1,5 +1,8 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
+
+# pyright: reportArgumentType=false,reportIncompatibleVariableOverride=false
+
 """Waldiez captain agent model."""
 
 from typing import Literal
@@ -26,7 +29,7 @@ class WaldiezCaptainAgent(WaldiezAgent):
         The captain agent's data.
     """
 
-    agent_type: Annotated[  # pyright: ignore
+    agent_type: Annotated[
         Literal["captain"],
         Field(
             default="captain",
@@ -34,12 +37,12 @@ class WaldiezCaptainAgent(WaldiezAgent):
             description="The agent type: 'captain' for a captain agent",
             alias="agentType",
         ),
-    ] = "captain"
-    data: Annotated[  # pyright: ignore
+    ]
+    data: Annotated[
         WaldiezCaptainAgentData,
         Field(
             title="Data",
             description="The captain agent's data",
-            default_factory=WaldiezCaptainAgentData,  # pyright: ignore
+            default_factory=WaldiezCaptainAgentData,
         ),
     ]

waldiez/models/agents/captain/captain_agent_data.py

@@ -2,8 +2,6 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Waldiez captain agent data."""
 
-from typing import Optional
-
 from pydantic import Field
 from typing_extensions import Annotated, Literal
 
@@ -32,16 +30,16 @@ class WaldiezCaptainAgentData(WaldiezAgentData):
             description="The agent lib",
             alias="agentLib",
         ),
-    ] = []
+    ]
     tool_lib: Annotated[
-        Optional[Literal["default"]],
+        Literal["default"] | None,
         Field(
             default=None,
             title="Tool lib",
             description="Whether to use the default tool lib",
             alias="toolLib",
         ),
-    ] = None
+    ]
     max_round: Annotated[
         int,
         Field(
@@ -50,7 +48,7 @@ class WaldiezCaptainAgentData(WaldiezAgentData):
             description="The maximum number of rounds in a group chat",
             alias="maxRound",
         ),
-    ] = 10
+    ]
     max_turns: Annotated[
         int,
         Field(
@@ -59,4 +57,4 @@ class WaldiezCaptainAgentData(WaldiezAgentData):
             description="The maximum number of turns for a chat",
             alias="maxTurns",
         ),
-    ] = 5
+    ]