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

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:

waldiez/models/agents/extra_requirements.py

@@ -1,9 +1,9 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-"""Extra requirements for agents."""
+
 # pylint: disable=line-too-long
 
-from typing import Set
+"""Extra requirements for agents."""
 
 from .agent import WaldiezAgent
 from .rag_user_proxy import WaldiezRagUserProxy
@@ -26,7 +26,7 @@ def get_retrievechat_extra_requirements(
     """
     # https://github.com/ag2ai/ag2/blob/main/pyproject.toml
     # with chromadb and sentence_transdormers relaxed
-    rag_requirements: Set[str] = {
+    rag_requirements: set[str] = {
         "protobuf==5.29.3",
         "chromadb>=0.5.23",
         "sentence_transformers",

waldiez/models/agents/group_manager/group_manager.py

@@ -1,5 +1,8 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
+
+# pyright: reportArgumentType=false,reportIncompatibleVariableOverride=false
+
 """Group chat manager agent."""
 
 import warnings
@@ -29,23 +32,23 @@ class WaldiezGroupManager(WaldiezAgent):
         The group manager agent's data.
     """
 
-    agent_type: Annotated[  # pyright: ignore
+    agent_type: Annotated[
         Literal["group_manager", "manager"],
         Field(
-            "manager",
+            "group_manager",
             title="Agent type",
             description=(
                 "The agent type: 'group_manager' for a group manager agent"
             ),
             alias="agentType",
         ),
-    ] = "group_manager"
-    data: Annotated[  # pyright: ignore
+    ]
+    data: Annotated[
         WaldiezGroupManagerData,
         Field(
             title="Data",
             description="The group manager agent's data",
-            default_factory=WaldiezGroupManagerData,  # pyright: ignore
+            default_factory=WaldiezGroupManagerData,
         ),
     ]
 
@@ -83,10 +86,11 @@ class WaldiezGroupManager(WaldiezAgent):
                 stacklevel=2,
             )
         if v != "group_manager":  # pragma: no cover
-            raise ValueError(
+            msg = (
                 "The agent type must be 'group_manager'. "
                 "Use 'group_manager' instead."
             )
+            raise ValueError(msg)
         return "group_manager"
 
     def validate_initial_agent_id(self, all_agent_ids: list[str]) -> None:
@@ -106,10 +110,11 @@ class WaldiezGroupManager(WaldiezAgent):
         """
         initial_agent_id = self.data.initial_agent_id
         if initial_agent_id not in all_agent_ids:
-            raise ValueError(
+            msg = (
                 f"Initial agent ID '{initial_agent_id}' "
                 f"is not in the list of agent IDs: {all_agent_ids}"
             )
+            raise ValueError(msg)
 
     def get_speakers_order(self) -> list[str]:
         """Get the order of the speakers.

waldiez/models/agents/group_manager/group_manager_data.py

@@ -1,8 +1,9 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
-"""Group chat manager data."""
 
-from typing import Optional
+# pyright: reportArgumentType=false
+
+"""Group chat manager data."""
 
 from pydantic import Field
 from typing_extensions import Annotated, Literal
@@ -47,7 +48,7 @@ class WaldiezGroupManagerData(WaldiezAgentData):
             description="The human input mode, Defaults to `NEVER`",
             alias="humanInputMode",
         ),
-    ] = "NEVER"
+    ]
     max_round: Annotated[
         int,
         Field(
@@ -56,9 +57,9 @@ class WaldiezGroupManagerData(WaldiezAgentData):
             description="The maximum number of rounds to have in the group.",
             alias="maxRound",
         ),
-    ] = 20
+    ]
     admin_name: Annotated[
-        Optional[str],
+        str | None,
         Field(
             default=None,
             title="Group Admin name",
@@ -68,24 +69,24 @@ class WaldiezGroupManagerData(WaldiezAgentData):
             ),
             alias="adminName",
         ),
-    ] = None
+    ]
     speakers: Annotated[
         WaldiezGroupManagerSpeakers,
         Field(
             title="Speakers",
             description="The rules for the speaker selection and repetition",
-            default_factory=WaldiezGroupManagerSpeakers,  # pyright: ignore
+            default_factory=WaldiezGroupManagerSpeakers,
         ),
     ]
     enable_clear_history: Annotated[
-        Optional[bool],
+        bool | None,
         Field(
             default=None,
             title="Enable clear history",
             description="Enable clearing the history in the chat group.",
             alias="enableClearHistory",
         ),
-    ] = None
+    ]
     send_introductions: Annotated[
         bool,
         Field(
@@ -94,16 +95,16 @@ class WaldiezGroupManagerData(WaldiezAgentData):
             description="Send the group members' introductions.",
             alias="sendIntroductions",
         ),
-    ] = False
+    ]
     group_name: Annotated[
-        Optional[str],
+        str | None,
         Field(
             default=None,
             title="Group name",
             description="The name of the group.",
             alias="groupName",
         ),
-    ] = None
+    ]
     initial_agent_id: Annotated[
         str,
         Field(

waldiez/models/agents/group_manager/speakers.py

@@ -2,8 +2,6 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Group chat speakers."""
 
-from typing import Optional, Union
-
 from pydantic import Field, model_validator
 from typing_extensions import Annotated, Literal, Self
 
@@ -61,7 +59,7 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
     ----------
     selection_method : WaldiezGroupManagerSpeakersSelectionMethod
         The next speaker selection method.
-    selection_custom_method : Optional[str]
+    selection_custom_method : str | None
         Method for custom selection.
     max_retries_for_selecting : Optional[int]
         Max retries for selecting a speaker.
@@ -85,9 +83,9 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
             description="The next speaker selection method",
             alias="selectionMethod",
         ),
-    ] = "auto"
+    ]
     selection_custom_method: Annotated[
-        Optional[str],
+        str | None,
         Field(
             default=None,
             title="Method for custom selection.",
@@ -102,9 +100,9 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
             ),
             alias="selectionCustomMethod",
         ),
-    ] = None
+    ]
     max_retries_for_selecting: Annotated[
-        Optional[int],
+        int | None,
         Field(
             default=None,
             title="Max retries for a selecting",
@@ -114,7 +112,7 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
             ),
             alias="maxRetriesForSelecting",
         ),
-    ] = None
+    ]
     selection_mode: Annotated[
         WaldiezGroupManagerSpeakersSelectionMode,
         Field(
@@ -127,9 +125,9 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
             ),
             alias="selectionMode",
         ),
-    ] = "repeat"
+    ]
     allow_repeat: Annotated[
-        Union[bool, list[str]],
+        bool | list[str] | None,
         Field(
             default=True,
             title="Allow repeat",
@@ -139,7 +137,7 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
             ),
             alias="allowRepeat",
         ),
-    ] = True
+    ]
     allowed_or_disallowed_transitions: Annotated[
         dict[str, list[str]],
         Field(
@@ -151,7 +149,7 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
             ),
             alias="allowedOrDisallowedTransitions",
         ),
-    ] = {}
+    ]
     transitions_type: Annotated[
         WaldiezGroupManagerSpeakersTransitionsType,
         Field(
@@ -164,7 +162,7 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
             ),
             alias="transitionsType",
         ),
-    ] = "allowed"
+    ]
     order: Annotated[
         list[str],
         Field(
@@ -176,12 +174,12 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
                 "will be determined by the order of the agents in the flow."
             ),
         ),
-    ] = []
-    _custom_method_string: Optional[str] = None
-    _order: Optional[list[str]] = None
+    ]
+    _custom_method_string: str | None = None
+    _order: list[str] | None = None
 
     @property
-    def custom_method_string(self) -> Optional[str]:
+    def custom_method_string(self) -> str | None:
         """Get the custom method string.
 
         Returns
@@ -193,8 +191,8 @@ class WaldiezGroupManagerSpeakers(WaldiezBase):
 
     def get_custom_method_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 custom method function.
 

waldiez/models/agents/rag_user_proxy/rag_user_proxy.py

@@ -1,5 +1,8 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
+
+# pyright: reportArgumentType=false,reportIncompatibleVariableOverride=false
+
 """RAG user agent.
 
 It extends a user agent and has RAG related parameters (`retrieve_config`).
@@ -27,7 +30,7 @@ class WaldiezRagUserProxy(WaldiezAgent):
         See `WaldiezRagUserProxyData` for more info.
     """
 
-    agent_type: Annotated[  # pyright: ignore
+    agent_type: Annotated[
         Literal["rag_user", "rag_user_proxy"],
         Field(
             "rag_user_proxy",
@@ -39,14 +42,14 @@ class WaldiezRagUserProxy(WaldiezAgent):
             ),
             alias="agentType",
         ),
-    ] = "rag_user_proxy"
+    ]
 
-    data: Annotated[  # pyright: ignore
+    data: Annotated[
         WaldiezRagUserProxyData,
         Field(
             title="Data",
             description="The RAG user agent's data",
-            default_factory=WaldiezRagUserProxyData,  # pyright: ignore
+            default_factory=WaldiezRagUserProxyData,
         ),
     ]
 

waldiez/models/agents/rag_user_proxy/rag_user_proxy_data.py

@@ -1,5 +1,8 @@
 # SPDX-License-Identifier: Apache-2.0.
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
+
+# pyright: reportArgumentType=false
+
 """Waldiez RAG user agent data."""
 
 from pydantic import Field
@@ -27,7 +30,7 @@ class WaldiezRagUserProxyData(WaldiezUserProxyData):
         Field(
             title="Retrieve Config",
             description="The RAG user agent's retrieve config",
-            default_factory=WaldiezRagUserProxyRetrieveConfig,  # pyright: ignore  # noqa: E501
+            default_factory=WaldiezRagUserProxyRetrieveConfig,
             alias="retrieveConfig",
         ),
     ]