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

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
+    ]

waldiez/models/agents/doc_agent/doc_agent.py

@@ -1,5 +1,8 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
+
+# pyright: reportArgumentType=false,reportIncompatibleVariableOverride=false
+
 """Document agent model."""
 
 from typing import Literal
@@ -23,7 +26,7 @@ class WaldiezDocAgent(WaldiezAgent):
     Extends `WaldiezAgent`.
     """
 
-    agent_type: Annotated[  # pyright: ignore
+    agent_type: Annotated[
         Literal["doc_agent"],
         Field(
             default="doc_agent",
@@ -31,13 +34,13 @@ class WaldiezDocAgent(WaldiezAgent):
             description="The agent type: 'doc_agent' for a document agent",
             alias="agentType",
         ),
-    ] = "doc_agent"
-    data: Annotated[  # pyright: ignore
+    ]
+    data: Annotated[
         WaldiezDocAgentData,
         Field(
             title="Data",
             description="The document agent's data",
-            default_factory=WaldiezDocAgentData,  # pyright: ignore
+            default_factory=WaldiezDocAgentData,
         ),
     ]
 

waldiez/models/agents/doc_agent/doc_agent_data.py

@@ -3,7 +3,6 @@
 """Document agent data model."""
 
 from pathlib import Path
-from typing import Optional, Union
 
 from pydantic import Field, model_validator
 from typing_extensions import Annotated, Self
@@ -25,14 +24,14 @@ class WaldiezDocAgentData(WaldiezAgentData):
     """
 
     collection_name: Annotated[
-        Optional[str],
+        str | None,
         Field(
             title="Collection Name",
             description="The name of the collection for the document agent.",
             default=None,
             alias="collectionName",
         ),
-    ] = None
+    ]
     reset_collection: Annotated[
         bool,
         Field(
@@ -43,9 +42,9 @@ class WaldiezDocAgentData(WaldiezAgentData):
             default=False,
             alias="resetCollection",
         ),
-    ] = False
+    ]
     parsed_docs_path: Annotated[
-        Optional[Union[str, Path]],
+        str | Path | None,
         Field(
             title="Parsed Documents Path",
             description=(
@@ -54,16 +53,16 @@ class WaldiezDocAgentData(WaldiezAgentData):
             default=None,
             alias="parsedDocsPath",
         ),
-    ] = None
+    ]
     query_engine: Annotated[
-        Optional[WaldiezDocAgentQueryEngine],
+        WaldiezDocAgentQueryEngine | None,
         Field(
             title="Query Engine",
             description="The query engine to use for the document agent.",
             default=None,
             alias="queryEngine",
         ),
-    ] = None
+    ]
 
     @model_validator(mode="after")
     def validate_parsed_docs_path(self) -> Self:
@@ -129,13 +128,13 @@ class WaldiezDocAgentData(WaldiezAgentData):
 
 
 def get_parsed_docs_path(
-    parsed_docs_path: Optional[Union[str, Path]] = None,
+    parsed_docs_path: str | Path | None = None,
 ) -> str:
     """Get the parsed documents path for the document agent.
 
     Parameters
     ----------
-    parsed_docs_path : Optional[Union[str, Path]]
+    parsed_docs_path : str | Path | None
         The path to the parsed documents.
 
     Returns

waldiez/models/agents/doc_agent/rag_query_engine.py

@@ -3,7 +3,6 @@
 """Document agent data model."""
 
 from pathlib import Path
-from typing import Optional, Union
 
 from platformdirs import user_data_dir
 from pydantic import Field, model_validator
@@ -19,24 +18,23 @@ class WaldiezDocAgentQueryEngine(WaldiezBase):
     """
 
     type: Annotated[
-        Optional[
-            Literal[
-                "VectorChromaQueryEngine",
-                "VectorChromaCitationQueryEngine",
-                "InMemoryQueryEngine",
-            ]
-        ],
+        Literal[
+            "VectorChromaQueryEngine",
+            "VectorChromaCitationQueryEngine",
+            "InMemoryQueryEngine",
+        ]
+        | None,
         "RAG Query Engine type",
     ] = "VectorChromaQueryEngine"
     db_path: Annotated[
-        Optional[Union[str, Path]],
+        str | Path | None,
         Field(
             title="Database Path",
             description="The path to the database for the query engine.",
             default=None,
             alias="dbPath",
         ),
-    ] = None
+    ]
     enable_query_citations: Annotated[
         bool,
         Field(
@@ -47,7 +45,7 @@ class WaldiezDocAgentQueryEngine(WaldiezBase):
             default=False,
             alias="enableQueryCitations",
         ),
-    ] = False
+    ]
     citation_chunk_size: Annotated[
         int,
         Field(
@@ -56,7 +54,7 @@ class WaldiezDocAgentQueryEngine(WaldiezBase):
             default=512,
             alias="citationChunkSize",
         ),
-    ] = 512
+    ]
 
     @model_validator(mode="after")
     def validate_db_path(self) -> Self: