waldiez 0.4.7__py3-none-any.whl → 0.4.9__py3-none-any.whl

waldiez/models/agents/{rag_user → rag_user_proxy}/vector_db_config.py

@@ -5,7 +5,7 @@
 """The vector db config for the RAG user agent."""
 
 from pathlib import Path
-from typing import Any, Dict, Optional
+from typing import Any, Optional
 
 from pydantic import ConfigDict, Field, model_validator
 from pydantic.alias_generators import to_camel
@@ -15,7 +15,7 @@ from ...common import WaldiezBase
 
 
 # pylint: disable=line-too-long
-class WaldiezRagUserVectorDbConfig(WaldiezBase):
+class WaldiezRagUserProxyVectorDbConfig(WaldiezBase):
     """The config for the vector db.
 
     Attributes
@@ -36,7 +36,7 @@ class WaldiezRagUserVectorDbConfig(WaldiezBase):
     wait_until_document_ready : Optional[float]
         Blocking call to wait until the database documents are ready (if `mongodb` is used).
         None, the default, means no wait.
-    metadata : Optional[Dict[str, Any]]
+    metadata : Optional[dict[str, Any]]
         The metadata to use for the vector db.
         Example: {"hnsw:space": "ip", "hnsw:construction_ef": 30, "hnsw:M": 32}
 
@@ -61,7 +61,7 @@ class WaldiezRagUserVectorDbConfig(WaldiezBase):
             title="Model",
             description="The model to use for the vector db embeddings.",
         ),
-    ]
+    ] = None
     use_memory: Annotated[
         bool,
         Field(
@@ -71,7 +71,7 @@ class WaldiezRagUserVectorDbConfig(WaldiezBase):
                 "Whether to use memory for the vector db (if qdrant is used)."
             ),
         ),
-    ]
+    ] = True
     use_local_storage: Annotated[
         bool,
         Field(
@@ -82,7 +82,7 @@ class WaldiezRagUserVectorDbConfig(WaldiezBase):
                 "(if qdrant or chroma is used)."
             ),
         ),
-    ]
+    ] = False
     local_storage_path: Annotated[
         Optional[str],
         Field(
@@ -93,7 +93,7 @@ class WaldiezRagUserVectorDbConfig(WaldiezBase):
                 "(if qdrant is used)."
             ),
         ),
-    ]
+    ] = None
     connection_url: Annotated[
         Optional[str],
         Field(
@@ -101,7 +101,7 @@ class WaldiezRagUserVectorDbConfig(WaldiezBase):
             title="Connection URL",
             description="The connection url for the vector db.",
         ),
-    ]
+    ] = None
     wait_until_index_ready: Annotated[
         Optional[float],
         Field(
@@ -112,7 +112,7 @@ class WaldiezRagUserVectorDbConfig(WaldiezBase):
                 "None, the default, means no wait."
             ),
         ),
-    ]
+    ] = None
     wait_until_document_ready: Annotated[
         Optional[float],
         Field(
@@ -123,9 +123,9 @@ class WaldiezRagUserVectorDbConfig(WaldiezBase):
                 "are ready. None, the default, means no wait."
             ),
         ),
-    ]
+    ] = None
     metadata: Annotated[
-        Optional[Dict[str, Any]],
+        Optional[dict[str, Any]],
         Field(
             None,
             title="Metadata",
@@ -136,7 +136,7 @@ class WaldiezRagUserVectorDbConfig(WaldiezBase):
                 '"hnsw:M": 32}'
             ),
         ),
-    ]
+    ] = None
 
     @model_validator(mode="after")
     def validate_vector_db_config(self) -> Self:
@@ -147,7 +147,7 @@ class WaldiezRagUserVectorDbConfig(WaldiezBase):
 
         Returns
         -------
-        WaldiezRagUserVectorDbConfig
+        WaldiezRagUserProxyVectorDbConfig
             The vector db config.
 
         Raises

waldiez/models/agents/reasoning/reasoning_agent.py

@@ -2,7 +2,7 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Reasoning agent model."""
 
-from typing import Any, Dict
+from typing import Any
 
 from pydantic import Field
 from typing_extensions import Annotated, Literal
@@ -14,7 +14,7 @@ from .reasoning_agent_data import WaldiezReasoningAgentData
 class WaldiezReasoningAgent(WaldiezAgent):
     """Reasoning agent model."""
 
-    agent_type: Annotated[
+    agent_type: Annotated[  # pyright: ignore
         Literal["reasoning"],
         Field(
             "reasoning",
@@ -22,17 +22,17 @@ class WaldiezReasoningAgent(WaldiezAgent):
             description="The agent type in a graph: 'reasoning'",
             alias="agentType",
         ),
-    ]
-    data: Annotated[
+    ] = "reasoning"
+    data: Annotated[  # pyright: ignore
         WaldiezReasoningAgentData,
         Field(
             title="Data",
             description="The reasoning agent's data",
-            default_factory=WaldiezReasoningAgentData,
+            default_factory=WaldiezReasoningAgentData,  # pyright: ignore
         ),
     ]
 
-    def get_reasoning_config(self) -> Dict[str, Any]:
+    def get_reasoning_config(self) -> dict[str, Any]:
         """Get the reasoning configuration based on its method.
 
         Returns

waldiez/models/agents/reasoning/reasoning_agent_data.py

@@ -2,48 +2,27 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Reasoning agent data model."""
 
-from typing import Any, Dict
+from typing import Any
 
 from pydantic import Field
 from typing_extensions import Annotated, Literal
 
-from ..assistant import WaldiezAssistantData
+from ..agent import WaldiezAgentData
 from .reasoning_agent_reason_config import WaldiezReasoningAgentReasonConfig
 
 
-class WaldiezReasoningAgentData(WaldiezAssistantData):
+class WaldiezReasoningAgentData(WaldiezAgentData):
     """Reasoning agent data model."""
 
-    max_depth: Annotated[
-        int,
+    human_input_mode: Annotated[
+        Literal["ALWAYS", "NEVER", "TERMINATE"],
         Field(
-            4,
-            title="Maximum depth",
-            description="Maximum depth of the reasoning tree",
-            alias="maxDepth",
-            deprecated=True,
+            "NEVER",
+            title="Human input mode",
+            description="The human input mode, Defaults to `NEVER`",
+            alias="humanInputMode",
         ),
-    ] = 4
-    beam_size: Annotated[
-        int,
-        Field(
-            3,
-            title="Beam size",
-            description="Number of parallel reasoning paths to maintain",
-            alias="beamSize",
-            deprecated=True,
-        ),
-    ] = 3
-    answer_approach: Annotated[
-        Literal["pool", "best"],
-        Field(
-            "pool",
-            title="Answer approach",
-            description="How to generate final answer",
-            alias="answerApproach",
-            deprecated=True,
-        ),
-    ] = "pool"
+    ] = "NEVER"
     verbose: Annotated[
         bool,
         Field(
@@ -51,7 +30,7 @@ class WaldiezReasoningAgentData(WaldiezAssistantData):
             title="Verbose",
             description="Whether to show intermediate steps",
         ),
-    ]
+    ] = True
     reason_config: Annotated[
         WaldiezReasoningAgentReasonConfig,
         Field(
@@ -62,16 +41,112 @@ class WaldiezReasoningAgentData(WaldiezAssistantData):
         ),
     ]
 
-    def get_reasoning_config(self) -> Dict[str, Any]:
+    @property
+    def method(self) -> str:
+        """Get the method of the reasoning agent.
+
+        Returns
+        -------
+        str
+            The method of the reasoning agent.
+
+        """
+        return self.reason_config.method
+
+    @property
+    def max_depth(self) -> int:
+        """Get the maximum depth of the reasoning agent.
+
+        Returns
+        -------
+        int
+            The maximum depth of the reasoning agent.
+
+        """
+        return self.reason_config.max_depth
+
+    @property
+    def beam_size(self) -> int:
+        """Get the beam size of the reasoning agent.
+
+        Returns
+        -------
+        int
+            The beam size of the reasoning agent.
+
+        """
+        return self.reason_config.beam_size
+
+    @property
+    def answer_approach(self) -> str:
+        """Get the answer approach of the reasoning agent.
+
+        Returns
+        -------
+        str
+            The answer approach of the reasoning agent.
+
+        """
+        return self.reason_config.answer_approach
+
+    @property
+    def forest_size(self) -> int:
+        """Get the forest size of the reasoning agent.
+
+        Returns
+        -------
+        int
+            The forest size of the reasoning agent.
+
+        """
+        return self.reason_config.forest_size
+
+    @property
+    def rating_scale(self) -> int:
+        """Get the rating scale of the reasoning agent.
+
+        Returns
+        -------
+        int
+            The rating scale of the reasoning agent.
+
+        """
+        return self.reason_config.rating_scale
+
+    @property
+    def exploration_constant(self) -> float:
+        """Get the exploration constant of the reasoning agent.
+
+        Returns
+        -------
+        float
+            The exploration constant of the reasoning agent.
+
+        """
+        return self.reason_config.exploration_constant
+
+    @property
+    def nsim(self) -> int:
+        """Get the number of simulations of the reasoning agent.
+
+        Returns
+        -------
+        int
+            The number of simulations of the reasoning agent.
+
+        """
+        return self.reason_config.nsim
+
+    def get_reasoning_config(self) -> dict[str, Any]:
         """Get the reasoning configuration based on the reason_config method.
 
         Returns
         -------
-        Dict[str, Any]
+        dict[str, Any]
             The reasoning configuration.
 
         """
-        reason_dict: Dict[str, Any] = {
+        reason_dict: dict[str, Any] = {
             "method": self.reason_config.method,
             "max_depth": self.reason_config.max_depth,
             "forest_size": self.reason_config.forest_size,
@@ -86,31 +161,3 @@ class WaldiezReasoningAgentData(WaldiezAssistantData):
                 self.reason_config.exploration_constant
             )
         return reason_dict
-
-
-# reason_config (dict): Configuration for the reasoning method.
-# Supported parameters:
-#     method (str): The search strategy to use. Options:
-#         - "beam_search" (default): Uses beam search with parallel paths
-#         - "mcts": Uses Monte Carlo Tree Search for exploration
-#         - "lats": Uses Language Agent Tree Search with per-step rewards
-#         - "dfs": Uses depth-first search
-#                (equivalent to beam_search with beam_size=1)
-#     Common parameters:
-#         max_depth (int): Maximum depth of reasoning tree (default: 3)
-#         forest_size (int):
-#               Number of independent trees to maintain (default: 1)
-#         rating_scale (int):
-#               Scale for grading responses, e.g. 1-10 (default: 10)
-#     Beam Search specific:
-#         beam_size (int): Number of parallel paths to maintain (default: 3)
-#         answer_approach (str):
-#               How to select final answer, "pool" or "best" (default: "pool")
-#     MCTS/LATS specific:
-#         nsim (int): Number of simulations to run (default: 3)
-#         exploration_constant (float):
-#               UCT exploration parameter (default: 1.41)
-#     Example configs:
-#         `{"method": "beam_search", "beam_size": 5, "max_depth": 4}`
-#         `{"method": "mcts", "nsim": 10, "exploration_constant": 2.0}`
-#         `{"method": "lats", "nsim": 5, "forest_size": 3}`

waldiez/models/agents/reasoning/reasoning_agent_reason_config.py

@@ -7,7 +7,7 @@ from typing_extensions import Annotated, Literal
 
 from ...common import WaldiezBase
 
-ReasoningConfigMethod = Literal["beam_search", "mcts", "lats", "dfs"]
+ReasonConfigMethod = Literal["beam_search", "mcts", "lats", "dfs"]
 """Possible reasoning methods."""
 
 
@@ -37,13 +37,13 @@ class WaldiezReasoningAgentReasonConfig(WaldiezBase):
     """
 
     method: Annotated[
-        ReasoningConfigMethod,
+        ReasonConfigMethod,
         Field(
             "beam_search",
             title="Method",
             description="The search strategy to use.",
         ),
-    ]
+    ] = "beam_search"
     max_depth: Annotated[
         int,
         Field(
@@ -51,7 +51,7 @@ class WaldiezReasoningAgentReasonConfig(WaldiezBase):
             title="Maximum depth",
             description="Maximum depth of reasoning tree.",
         ),
-    ]
+    ] = 3
     forest_size: Annotated[
         int,
         Field(
@@ -59,7 +59,7 @@ class WaldiezReasoningAgentReasonConfig(WaldiezBase):
             title="Forest size",
             description="Number of independent trees to maintain.",
         ),
-    ]
+    ] = 1
     rating_scale: Annotated[
         int,
         Field(
@@ -67,7 +67,7 @@ class WaldiezReasoningAgentReasonConfig(WaldiezBase):
             title="Rating scale",
             description="Scale for grading responses, e.g. 1-10.",
         ),
-    ]
+    ] = 10
     beam_size: Annotated[
         int,
         Field(
@@ -75,7 +75,7 @@ class WaldiezReasoningAgentReasonConfig(WaldiezBase):
             title="Beam size",
             description="Number of parallel paths to maintain.",
         ),
-    ]
+    ] = 3
     answer_approach: Annotated[
         Literal["pool", "best"],
         Field(
@@ -83,7 +83,7 @@ class WaldiezReasoningAgentReasonConfig(WaldiezBase):
             title="Answer approach",
             description="How to select final answer.",
         ),
-    ]
+    ] = "pool"
     nsim: Annotated[
         int,
         Field(
@@ -91,7 +91,7 @@ class WaldiezReasoningAgentReasonConfig(WaldiezBase):
             title="Number of simulations",
             description="Number of simulations to run.",
         ),
-    ]
+    ] = 3
     exploration_constant: Annotated[
         float,
         Field(
@@ -99,4 +99,32 @@ class WaldiezReasoningAgentReasonConfig(WaldiezBase):
             title="Exploration constant",
             description="UCT exploration parameter.",
         ),
-    ]
+    ] = 1.41
+
+
+# reason_config (dict): Configuration for the reasoning method.
+# Supported parameters:
+#     method (str): The search strategy to use. Options:
+#         - "beam_search" (default): Uses beam search with parallel paths
+#         - "mcts": Uses Monte Carlo Tree Search for exploration
+#         - "lats": Uses Language Agent Tree Search with per-step rewards
+#         - "dfs": Uses depth-first search
+#                (equivalent to beam_search with beam_size=1)
+#     Common parameters:
+#         max_depth (int): Maximum depth of reasoning tree (default: 3)
+#         forest_size (int):
+#               Number of independent trees to maintain (default: 1)
+#         rating_scale (int):
+#               Scale for grading responses, e.g. 1-10 (default: 10)
+#     Beam Search specific:
+#         beam_size (int): Number of parallel paths to maintain (default: 3)
+#         answer_approach (str):
+#               How to select final answer, "pool" or "best" (default: "pool")
+#     MCTS/LATS specific:
+#         nsim (int): Number of simulations to run (default: 3)
+#         exploration_constant (float):
+#               UCT exploration parameter (default: 1.41)
+#     Example configs:
+#         `{"method": "beam_search", "beam_size": 5, "max_depth": 4}`
+#         `{"method": "mcts", "nsim": 10, "exploration_constant": 2.0}`
+#         `{"method": "lats", "nsim": 5, "forest_size": 3}`

waldiez/models/agents/user_proxy/user_proxy.py

@@ -18,26 +18,30 @@ class WaldiezUserProxy(WaldiezAgent):
 
     Attributes
     ----------
-    agent_type : Literal["user"]
+    agent_type : Literal["user", "user_proxy"]
         The agent type: 'user' for a user proxy agent
     data : WaldiezUserProxyData
         The user proxy agent's data
     """
 
-    agent_type: Annotated[
-        Literal["user"],
+    agent_type: Annotated[  # pyright: ignore
+        Literal["user", "user_proxy"],
         Field(
-            "user",
+            "user_proxy",
             title="Agent type",
-            description="The agent type in a graph: 'user'",
+            description=(
+                "The agent type in a graph. "
+                "`user` is deprecated and will be removed in future versions. "
+                "Use `user_proxy` instead."
+            ),
             alias="agentType",
         ),
     ]
-    data: Annotated[
+    data: Annotated[  # pyright: ignore
         WaldiezUserProxyData,
         Field(
             title="Data",
             description="The user proxy agent's data",
-            default_factory=WaldiezUserProxyData,
+            default_factory=WaldiezUserProxyData,  # pyright: ignore
         ),
     ]

waldiez/models/agents/user_proxy/user_proxy_data.py

@@ -24,9 +24,9 @@ class WaldiezUserProxyData(WaldiezAgentData):
     human_input_mode: Annotated[
         Literal["ALWAYS", "NEVER", "TERMINATE"],
         Field(
-            "ALWAYS",
+            default="ALWAYS",
             title="Human input mode",
             description="The human input mode, Defaults to `ALWAYS`",
             alias="humanInputMode",
         ),
-    ]
+    ] = "ALWAYS"

waldiez/models/chat/__init__.py

@@ -2,7 +2,7 @@
 # Copyright (c) 2024 - 2025 Waldiez and contributors.
 """Waldiez chat related models."""
 
-from .chat import WaldiezChat
+from .chat import WaldiezChat, WaldiezChatType
 from .chat_data import WaldiezChatData
 from .chat_message import (
     CALLABLE_MESSAGE,
@@ -37,4 +37,5 @@ __all__ = [
     "WaldiezChatNested",
     "WaldiezChatSummary",
     "WaldiezChatSummaryMethod",
+    "WaldiezChatType",
 ]