pyagentic-core 2.1.0a3__tar.gz → 2.2.0a1__tar.gz

{pyagentic_core-2.1.0a3 → pyagentic_core-2.2.0a1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyagentic-core
-Version: 2.1.0a3
+Version: 2.2.0a1
 Summary: Build LLM Agents in a Pythonic way
 Author-email: Ryan Mikulec <rmikulec.dev@gmail.com>
 License: MIT
@@ -20,6 +20,7 @@ Requires-Dist: typeguard>=4.4.4
 Requires-Dist: c3linearize>=0.1.0
 Requires-Dist: anthropic>=0.62.0
 Requires-Dist: google-generativeai>=0.8.0
+Requires-Dist: transitions>=0.9.3
 Dynamic: license-file
 
 # PyAgentic

{pyagentic_core-2.1.0a3 → pyagentic_core-2.2.0a1}/pyagentic/_base/_agent/_agent.py

@@ -13,6 +13,7 @@ from typing import (
     Union,
 )
 
+from transitions import Machine
 from pydantic import BaseModel, ValidationError
 
 from pyagentic.logging import get_logger
@@ -165,8 +166,10 @@ class BaseAgent(metaclass=AgentMeta):
     __description__: ClassVar[str]  # Optional: description for linked agents
     __input_template__: ClassVar[str] = None  # Optional: template for user input
     __response_format__: ClassVar[Type[BaseModel]] = None  # Optional: structured output format
+    phases: ClassVar[list[tuple[str, str, Callable]]] = None
 
     # Generated Class Attributes (built by metaclass)
+    __machine__: Machine | None = None
     __response_model__: ClassVar[Type[AgentResponse]] = None  # Pydantic response model
     __state_class__: ClassVar[Type[_AgentState]] = None  # Generated state class
     __tool_response_models__: ClassVar[dict[str, Type[ToolResponse]]]  # Tool response models
@@ -229,6 +232,34 @@ class BaseAgent(metaclass=AgentMeta):
         if self.__tool_defs__ and not self.provider.__supports_tool_calls__:
             raise Exception("Tools are not supported with this provider")
 
+    def _build_phase_machine(self) -> Machine:
+        if not self.phases:
+            return None
+
+        states = []
+        for to_, from_, _ in self.phases:
+            if to_ not in states:
+                states.append(to_)
+            if from_ not in states:
+                states.append(from_)
+
+        machine = Machine(states=states, initial=states[0])
+
+        for to_, from_, _ in self.phases:
+            machine.add_transition(
+                trigger=f"{to_}_to_{from_}",
+                source=to_,
+                dest=from_,
+            )
+
+        return machine
+
+    def _update_state_machine(self):
+        for to_, from_, condition in self.phases:
+            if condition(self.state):
+                trigger = f"{to_}_to_{from_}"
+                getattr(self.__machine__, trigger)()
+
     def __post_init__(self):
         """
         Post-initialization hook called after agent instance is created.
@@ -250,6 +281,7 @@ class BaseAgent(metaclass=AgentMeta):
             ```
         """
         self._check_llm_provider()
+        self.__machine__ = self._build_phase_machine()
 
         # Use BasicTracer as default if no tracer provided
         if not self.tracer:
@@ -547,6 +579,7 @@ class BaseAgent(metaclass=AgentMeta):
                         yield result
 
                 # Increment depth and continue loop (LLM will see tool results next iteration)
+                self._update_state_machine()
                 depth += 1
 
             # If we exhausted max_call_depth without final text, get one more response

{pyagentic_core-2.1.0a3 → pyagentic_core-2.2.0a1}/pyagentic/_base/_agent/_agent_state.py

@@ -1,9 +1,10 @@
 import asyncio
 import threading
 from typing import Any, Type, Self, Optional, ClassVar
-from pydantic import BaseModel, create_model, Field, PrivateAttr
+from pydantic import BaseModel, create_model, Field, PrivateAttr, ConfigDict
 from jinja2 import Template
-from typing import Optional, Literal
+from typing import Optional, Literal, Callable
+from transitions import Machine
 
 from pyagentic._base._exceptions import InvalidStateRefNotFoundInState
 from pyagentic._base._state import _StateDefinition

{pyagentic_core-2.1.0a3 → pyagentic_core-2.2.0a1}/pyagentic_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyagentic-core
-Version: 2.1.0a3
+Version: 2.2.0a1
 Summary: Build LLM Agents in a Pythonic way
 Author-email: Ryan Mikulec <rmikulec.dev@gmail.com>
 License: MIT
@@ -20,6 +20,7 @@ Requires-Dist: typeguard>=4.4.4
 Requires-Dist: c3linearize>=0.1.0
 Requires-Dist: anthropic>=0.62.0
 Requires-Dist: google-generativeai>=0.8.0
+Requires-Dist: transitions>=0.9.3
 Dynamic: license-file
 
 # PyAgentic

{pyagentic_core-2.1.0a3 → pyagentic_core-2.2.0a1}/pyagentic_core.egg-info/requires.txt

@@ -8,3 +8,4 @@ typeguard>=4.4.4
 c3linearize>=0.1.0
 anthropic>=0.62.0
 google-generativeai>=0.8.0
+transitions>=0.9.3

{pyagentic_core-2.1.0a3 → pyagentic_core-2.2.0a1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pyagentic-core"
-version = "2.1.0-a.3"
+version = "2.2.0-a.1"
 description = "Build LLM Agents in a Pythonic way"
 readme = "README.md"
 requires-python = ">=3.13"
@@ -23,6 +23,7 @@ dependencies = [
     "c3linearize>=0.1.0",
     "anthropic>=0.62.0",
     "google-generativeai>=0.8.0",
+    "transitions>=0.9.3",
 ]
 
 [dependency-groups]
@@ -105,4 +106,4 @@ include = ["pyagentic*"]
 compile-diagrams = "for file in docs/diagrams/source/*.d2; do d2 --layout elk \"$file\" \"docs/diagrams/$(basename \"$file\" .d2).svg\"; done"
 build-docs  = "task compile-diagrams && mkdocs build --clean"
 serve-docs  = "task compile-diagrams && mkdocs serve"
-deploy-docs = "task compile-diagrams && mkdocs gh-deploy --force"
+deploy-docs = "task compile-diagrams && mkdocs gh-deploy --force"