agentomatic 0.6.0__tar.gz → 0.8.0__tar.gz

{agentomatic-0.6.0 → agentomatic-0.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentomatic
-Version: 0.6.0
+Version: 0.8.0
 Summary: Drop agents, not code — zero-code multi-agent API platform framework
 Project-URL: Homepage, https://github.com/UnicoLab/agentomatic
 Project-URL: Documentation, https://unicolab.github.io/agentomatic
@@ -34,10 +34,12 @@ Requires-Dist: typing-extensions>=4.12
 Requires-Dist: uvicorn[standard]>=0.32
 Provides-Extra: all
 Requires-Dist: aiosqlite>=0.20; extra == 'all'
+Requires-Dist: cryptography>=42.0; extra == 'all'
 Requires-Dist: deepeval>=2.0; extra == 'all'
 Requires-Dist: holysheet>=0.1; extra == 'all'
 Requires-Dist: langchain-core>=0.3; extra == 'all'
 Requires-Dist: langchain-ollama>=0.2; extra == 'all'
+Requires-Dist: langgraph-swarm>=0.1; extra == 'all'
 Requires-Dist: langgraph>=0.4; extra == 'all'
 Requires-Dist: ollama>=0.4; extra == 'all'
 Requires-Dist: opentelemetry-api>=1.20; extra == 'all'
@@ -45,6 +47,8 @@ Requires-Dist: opentelemetry-instrumentation-fastapi>=0.40; extra == 'all'
 Requires-Dist: opentelemetry-instrumentation-httpx>=0.40; extra == 'all'
 Requires-Dist: opentelemetry-sdk>=1.20; extra == 'all'
 Requires-Dist: prometheus-client>=0.21; extra == 'all'
+Requires-Dist: pyjwt[crypto]>=2.8; extra == 'all'
+Requires-Dist: python-dotenv>=1.0; extra == 'all'
 Requires-Dist: questionary>=2.0; extra == 'all'
 Requires-Dist: rich>=13.0; extra == 'all'
 Requires-Dist: sqlalchemy[asyncio]>=2.0; extra == 'all'
@@ -74,6 +78,8 @@ Provides-Extra: docs
 Requires-Dist: mike>=2.1; extra == 'docs'
 Requires-Dist: mkdocs-material>=9.5; extra == 'docs'
 Requires-Dist: pymdown-extensions>=10.0; extra == 'docs'
+Provides-Extra: dotenv
+Requires-Dist: python-dotenv>=1.0; extra == 'dotenv'
 Provides-Extra: langchain
 Requires-Dist: langchain-community>=0.3; extra == 'langchain'
 Requires-Dist: langchain-core>=0.3; extra == 'langchain'
@@ -91,7 +97,12 @@ Requires-Dist: langchain-openai>=0.3; extra == 'openai'
 Provides-Extra: optimize
 Requires-Dist: deepeval>=2.0; extra == 'optimize'
 Requires-Dist: holysheet>=0.1; extra == 'optimize'
+Provides-Extra: security
+Requires-Dist: cryptography>=42.0; extra == 'security'
+Requires-Dist: pyjwt[crypto]>=2.8; extra == 'security'
 Provides-Extra: studio
+Provides-Extra: swarm
+Requires-Dist: langgraph-swarm>=0.1; extra == 'swarm'
 Provides-Extra: telemetry
 Requires-Dist: opentelemetry-api>=1.20; extra == 'telemetry'
 Requires-Dist: opentelemetry-instrumentation-fastapi>=0.40; extra == 'telemetry'
@@ -134,11 +145,12 @@ Build, trace, optimize, and time-travel debug production-ready AI agent APIs in
 |---|---|
 | 🎯 **Agentomatic Studio** | Embedded visual agent debugger with graph rendering, live SSE node streaming, state mutation, and historical time-travel capabilities. |
 | ⚡ **Prompt Optimizer** | Enterprise-grade prompt and configuration fitting utilizing 5 distinct optimizers with deployment recommendations. |
-| 🔍 **Zero-Code Auto-Discovery** | Drop an agent folder → 25+ fully-documented REST endpoints appear automatically. |
+| 🔍 **Zero-Code Auto-Discovery** | Drop an agent folder → 26 fully-documented REST endpoints appear automatically. |
 | 🚀 **Rich API Surface** | Natively handles `invoke`, `stream`, `chat`, `A2A`, `health`, `config`, `threads`, `memory`, and `feedback`. |
 | 🗄️ **Pluggable Storage** | Use `MemoryStore`, `SQLAlchemy`, or plug in your own custom persistence layer. |
 | 🔐 **Enterprise Middleware** | High-performance pipeline with JWT Auth, dynamic rate limiting, and Prometheus telemetry — all toggleable. |
-| 📦 **Scaffolding Templates** | Jumpstart development with 5 templates: `basic`, `full`, `rag`, `chatbot`, and `custom`. |
+| 📦 **Scaffolding Templates** | Jumpstart development with 8 templates: `basic`, `full`, `rag`, `chatbot`, `deepagent`, `custom`, `legacy_dict`, `plugin`. |
+| 🧬 **Class-Based Agents** | Define agents as Python classes with ML lifecycle: `compile()` → `fit()` → `evaluate()` → `transform()`. |
 | 🤖 **A2A Protocol** | True Agent-to-Agent communication flows integrated out of the box. |
 | 🔌 **Framework Agnostic** | Fully supports LangGraph, LangChain, or raw Python execution logic. |
 | 🩺 **Beautiful CLI** | A rich terminal experience with commands like `doctor`, `inspect`, and `test`. |
@@ -216,13 +228,12 @@ curl -X POST http://localhost:8000/api/v1/my_agent/invoke \
 
 ## 📂 Agent Structure
 
-Only `__init__.py` is required. Everything else is optional overrides:
+Only `agent.py` is required. Everything else is optional overrides:
 
 ```
 agents/my_agent/
-├── __init__.py      ← REQUIRED: manifest + node_fn
-├── graph.py         ← Optional: LangGraph StateGraph
-├── nodes.py         ← Optional: node functions
+├── __init__.py      ← Optional: Python package init
+├── agent.py         ← REQUIRED: Contains your BaseGraphAgent subclass
 ├── config.py        ← Optional: Pydantic config
 ├── schemas.py       ← Optional: custom request/response models
 ├── tools.py         ← Optional: LangChain tools
@@ -239,13 +250,16 @@ agents/my_agent/
 agentomatic init my_agent --template <template>
 ```
 
-| Template | Files | Description |
-|----------|-------|-------------|
-| `basic` | 7 | Minimal agent — quick start |
-| `full` | 11 | All override files — config, schemas, api, tools |
-| `rag` | 9 | Retrieve → Generate pipeline |
-| `chatbot` | 8 | Conversational with memory |
-| `custom` | 4 | Framework-agnostic — no LangGraph |
+| Template | Description |
+|----------|-------------|
+| `basic` | Minimal class-based agent (recommended) — quick start |
+| `full` | All override files — class agent with config, schemas, api, tools, prompts |
+| `rag` | RAG class-based agent — retrieve → generate pipeline |
+| `chatbot` | Conversational class-based agent with memory |
+| `deepagent` | Deep Agent — planning, tools, subagents (requires deepagents package) |
+| `custom` | Framework-agnostic — no LangGraph dependency |
+| `legacy_dict` | Legacy functional agent — 3 files (`__init__`, graph, nodes) |
+| `plugin` | ML Model Plugin — wrap classical ML models with REST endpoints |
 
 ## 🖥️ CLI
 
@@ -263,11 +277,59 @@ agentomatic init my_agent --template <template>
   doctor           Environment health check
   optimize <name>  Run prompt optimization
   ui               Launch Chainlit debug UI standalone
+  pipeline         Pipeline management commands
+```
+
+## 🧬 Class-Based Agents (NEW)
+
+Define agents as Python classes with built-in graph wiring and ML lifecycle:
+
+```python
+from dataclasses import dataclass, field
+from agentomatic import BaseGraphAgent
+
+@dataclass
+class MyState:
+    query: str = ""
+    output: dict = field(default_factory=dict)
+
+class MyAgent(BaseGraphAgent[MyState]):
+    agent_name = "my_agent"
+
+    def build_graph(self):
+        g = self.new_graph()
+        g.add_node("process", self.process)
+        g.add_node("format", self.format_out)
+        g.set_entry_point("process")
+        g.add_edge("process", "format")
+        g.set_finish_point("format")
+        return g.compile()
+
+    def process(self, state):
+        state.output = {"response": f"Hello! You asked: {state.query}"}
+        return state
+
+    def format_out(self, state):
+        return state
+
+    def input_to_state(self, data):
+        return MyState(query=data.get("query", ""))
+
+    def state_to_output(self, state):
+        return state.output
+
+# ML-like workflow
+agent = MyAgent()
+result = agent.transform({"query": "Hello!"})
+agent.compile(dataset, metrics)
+agent.fit(dataset)
+report = agent.evaluate(dataset.test, metrics)
+agent.save("compiled/v1")
 ```
 
 ## 🎨 Agentomatic Studio
 
-Agentomatic ships with a built-in React-based visual studio designed for time-travel debugging, real-time node streaming, and state inspection for all underlying LangGraph agents.
+Agentomatic ships with a built-in React-based visual studio designed for time-travel debugging, real-time node streaming, and state inspection. Works with class-based agents, LangGraph, LangChain, and any custom framework via the adapter system.
 
 To use the studio, install the optional package dependencies and run with the `--studio` flag:
 
@@ -276,7 +338,7 @@ pip install "agentomatic[studio]"
 agentomatic run --studio
 ```
 
-The unified server will bind to `http://localhost:8000` and mount the studio at `http://localhost:8000/studio/ui/`. 
+The unified server will bind to `http://localhost:8000` and mount the studio at `http://localhost:8000/studio/ui/`.
 
 **Key Studio Features**:
 - **Live Node Streaming**: Watch Server-Sent Events (SSE) transition node activity dynamically.
@@ -284,6 +346,36 @@ The unified server will bind to `http://localhost:8000` and mount the studio at
 - **Time-Travel History**: Rewind to any state checkpoint and replay from historical forks.
 - **Live State Editing**: Mutate graph state payloads on the fly during a breakpoint pause.
 
+## 🧠 ML Model Plugins (NEW)
+
+Agentomatic isn't just for LLMs. Wrap classical ML models (Scikit-Learn, PyTorch, PyMC) securely with auto-generated REST endpoints:
+
+```python
+from agentomatic.plugins import BaseMLPlugin
+from pydantic import BaseModel
+
+class IrisInput(BaseModel):
+    sepal_length: float
+    sepal_width: float
+    petal_length: float
+    petal_width: float
+
+class IrisPlugin(BaseMLPlugin[IrisInput, dict]):
+    async def load_model(self):
+        # Load sklearn model from disk
+        import joblib
+        self.model = joblib.load("iris_model.pkl")
+
+    async def predict(self, inputs: IrisInput) -> dict:
+        prediction = self.model.predict([[
+            inputs.sepal_length, inputs.sepal_width,
+            inputs.petal_length, inputs.petal_width
+        ]])
+        return {"species": prediction[0]}
+```
+
+Place it in `plugins/` and Agentomatic auto-discovers it alongside your AI agents!
+
 ## ⚙️ Configuration
 
 ```python

{agentomatic-0.6.0 → agentomatic-0.8.0}/README.md

@@ -29,11 +29,12 @@ Build, trace, optimize, and time-travel debug production-ready AI agent APIs in
 |---|---|
 | 🎯 **Agentomatic Studio** | Embedded visual agent debugger with graph rendering, live SSE node streaming, state mutation, and historical time-travel capabilities. |
 | ⚡ **Prompt Optimizer** | Enterprise-grade prompt and configuration fitting utilizing 5 distinct optimizers with deployment recommendations. |
-| 🔍 **Zero-Code Auto-Discovery** | Drop an agent folder → 25+ fully-documented REST endpoints appear automatically. |
+| 🔍 **Zero-Code Auto-Discovery** | Drop an agent folder → 26 fully-documented REST endpoints appear automatically. |
 | 🚀 **Rich API Surface** | Natively handles `invoke`, `stream`, `chat`, `A2A`, `health`, `config`, `threads`, `memory`, and `feedback`. |
 | 🗄️ **Pluggable Storage** | Use `MemoryStore`, `SQLAlchemy`, or plug in your own custom persistence layer. |
 | 🔐 **Enterprise Middleware** | High-performance pipeline with JWT Auth, dynamic rate limiting, and Prometheus telemetry — all toggleable. |
-| 📦 **Scaffolding Templates** | Jumpstart development with 5 templates: `basic`, `full`, `rag`, `chatbot`, and `custom`. |
+| 📦 **Scaffolding Templates** | Jumpstart development with 8 templates: `basic`, `full`, `rag`, `chatbot`, `deepagent`, `custom`, `legacy_dict`, `plugin`. |
+| 🧬 **Class-Based Agents** | Define agents as Python classes with ML lifecycle: `compile()` → `fit()` → `evaluate()` → `transform()`. |
 | 🤖 **A2A Protocol** | True Agent-to-Agent communication flows integrated out of the box. |
 | 🔌 **Framework Agnostic** | Fully supports LangGraph, LangChain, or raw Python execution logic. |
 | 🩺 **Beautiful CLI** | A rich terminal experience with commands like `doctor`, `inspect`, and `test`. |
@@ -111,13 +112,12 @@ curl -X POST http://localhost:8000/api/v1/my_agent/invoke \
 
 ## 📂 Agent Structure
 
-Only `__init__.py` is required. Everything else is optional overrides:
+Only `agent.py` is required. Everything else is optional overrides:
 
 ```
 agents/my_agent/
-├── __init__.py      ← REQUIRED: manifest + node_fn
-├── graph.py         ← Optional: LangGraph StateGraph
-├── nodes.py         ← Optional: node functions
+├── __init__.py      ← Optional: Python package init
+├── agent.py         ← REQUIRED: Contains your BaseGraphAgent subclass
 ├── config.py        ← Optional: Pydantic config
 ├── schemas.py       ← Optional: custom request/response models
 ├── tools.py         ← Optional: LangChain tools
@@ -134,13 +134,16 @@ agents/my_agent/
 agentomatic init my_agent --template <template>
 ```
 
-| Template | Files | Description |
-|----------|-------|-------------|
-| `basic` | 7 | Minimal agent — quick start |
-| `full` | 11 | All override files — config, schemas, api, tools |
-| `rag` | 9 | Retrieve → Generate pipeline |
-| `chatbot` | 8 | Conversational with memory |
-| `custom` | 4 | Framework-agnostic — no LangGraph |
+| Template | Description |
+|----------|-------------|
+| `basic` | Minimal class-based agent (recommended) — quick start |
+| `full` | All override files — class agent with config, schemas, api, tools, prompts |
+| `rag` | RAG class-based agent — retrieve → generate pipeline |
+| `chatbot` | Conversational class-based agent with memory |
+| `deepagent` | Deep Agent — planning, tools, subagents (requires deepagents package) |
+| `custom` | Framework-agnostic — no LangGraph dependency |
+| `legacy_dict` | Legacy functional agent — 3 files (`__init__`, graph, nodes) |
+| `plugin` | ML Model Plugin — wrap classical ML models with REST endpoints |
 
 ## 🖥️ CLI
 
@@ -158,11 +161,59 @@ agentomatic init my_agent --template <template>
   doctor           Environment health check
   optimize <name>  Run prompt optimization
   ui               Launch Chainlit debug UI standalone
+  pipeline         Pipeline management commands
+```
+
+## 🧬 Class-Based Agents (NEW)
+
+Define agents as Python classes with built-in graph wiring and ML lifecycle:
+
+```python
+from dataclasses import dataclass, field
+from agentomatic import BaseGraphAgent
+
+@dataclass
+class MyState:
+    query: str = ""
+    output: dict = field(default_factory=dict)
+
+class MyAgent(BaseGraphAgent[MyState]):
+    agent_name = "my_agent"
+
+    def build_graph(self):
+        g = self.new_graph()
+        g.add_node("process", self.process)
+        g.add_node("format", self.format_out)
+        g.set_entry_point("process")
+        g.add_edge("process", "format")
+        g.set_finish_point("format")
+        return g.compile()
+
+    def process(self, state):
+        state.output = {"response": f"Hello! You asked: {state.query}"}
+        return state
+
+    def format_out(self, state):
+        return state
+
+    def input_to_state(self, data):
+        return MyState(query=data.get("query", ""))
+
+    def state_to_output(self, state):
+        return state.output
+
+# ML-like workflow
+agent = MyAgent()
+result = agent.transform({"query": "Hello!"})
+agent.compile(dataset, metrics)
+agent.fit(dataset)
+report = agent.evaluate(dataset.test, metrics)
+agent.save("compiled/v1")
 ```
 
 ## 🎨 Agentomatic Studio
 
-Agentomatic ships with a built-in React-based visual studio designed for time-travel debugging, real-time node streaming, and state inspection for all underlying LangGraph agents.
+Agentomatic ships with a built-in React-based visual studio designed for time-travel debugging, real-time node streaming, and state inspection. Works with class-based agents, LangGraph, LangChain, and any custom framework via the adapter system.
 
 To use the studio, install the optional package dependencies and run with the `--studio` flag:
 
@@ -171,7 +222,7 @@ pip install "agentomatic[studio]"
 agentomatic run --studio
 ```
 
-The unified server will bind to `http://localhost:8000` and mount the studio at `http://localhost:8000/studio/ui/`. 
+The unified server will bind to `http://localhost:8000` and mount the studio at `http://localhost:8000/studio/ui/`.
 
 **Key Studio Features**:
 - **Live Node Streaming**: Watch Server-Sent Events (SSE) transition node activity dynamically.
@@ -179,6 +230,36 @@ The unified server will bind to `http://localhost:8000` and mount the studio at
 - **Time-Travel History**: Rewind to any state checkpoint and replay from historical forks.
 - **Live State Editing**: Mutate graph state payloads on the fly during a breakpoint pause.
 
+## 🧠 ML Model Plugins (NEW)
+
+Agentomatic isn't just for LLMs. Wrap classical ML models (Scikit-Learn, PyTorch, PyMC) securely with auto-generated REST endpoints:
+
+```python
+from agentomatic.plugins import BaseMLPlugin
+from pydantic import BaseModel
+
+class IrisInput(BaseModel):
+    sepal_length: float
+    sepal_width: float
+    petal_length: float
+    petal_width: float
+
+class IrisPlugin(BaseMLPlugin[IrisInput, dict]):
+    async def load_model(self):
+        # Load sklearn model from disk
+        import joblib
+        self.model = joblib.load("iris_model.pkl")
+
+    async def predict(self, inputs: IrisInput) -> dict:
+        prediction = self.model.predict([[
+            inputs.sepal_length, inputs.sepal_width,
+            inputs.petal_length, inputs.petal_width
+        ]])
+        return {"species": prediction[0]}
+```
+
+Place it in `plugins/` and Agentomatic auto-discovers it alongside your AI agents!
+
 ## ⚙️ Configuration
 
 ```python

{agentomatic-0.6.0 → agentomatic-0.8.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentomatic"
-version = "0.6.0"
+version = "0.8.0"
 description = "Drop agents, not code — zero-code multi-agent API platform framework"
 readme = "README.md"
 license = {text = "MIT"}
@@ -52,7 +52,10 @@ telemetry = [
     "opentelemetry-instrumentation-fastapi>=0.40",
     "opentelemetry-instrumentation-httpx>=0.40",
 ]
-all = ["agentomatic[langgraph,ollama,metrics,db,cli,studio,optimize,telemetry]"]
+dotenv = ["python-dotenv>=1.0"]
+security = ["pyjwt[crypto]>=2.8", "cryptography>=42.0"]
+swarm = ["langgraph-swarm>=0.1"]
+all = ["agentomatic[langgraph,ollama,metrics,db,cli,studio,optimize,telemetry,dotenv,security,swarm]"]
 docs = [
     "mkdocs-material>=9.5",
     "mike>=2.1",
@@ -86,6 +89,8 @@ Changelog = "https://github.com/UnicoLab/agentomatic/blob/main/CHANGELOG.md"
 
 
 
+
+
 [dependency-groups]
 dev = [
     "pytest>=8",
@@ -98,6 +103,8 @@ dev = [
     "build>=1.0",
     "twine>=6.0",
     "python-semantic-release>=9.0",
+    "mike>=2.2.0",
+    "mkdocs-material>=9.7.6",
 ]
 docs = [
     "mkdocs-material>=9.5",
@@ -190,12 +197,18 @@ exclude_lines = [
 # ---------------------------------------------------------------------------
 [tool.mypy]
 python_version = "3.11"
+plugins = ["pydantic.mypy"]
 warn_return_any = false
 warn_unused_configs = true
 check_untyped_defs = true
 explicit_package_bases = true
 mypy_path = "src"
 
+[tool.pydantic-mypy]
+init_forbid_extra = true
+init_typed = true
+warn_required_dynamic_aliases = true
+
 # Pydantic models with Field(default=...) trigger false call-arg errors
 [[tool.mypy.overrides]]
 module = [
@@ -205,6 +218,20 @@ module = [
     "agentomatic.studio.run_tracker",
     "agentomatic.demo.*",
     "agentomatic.optimize.*",
+    "agentomatic.providers.*",
 ]
 disable_error_code = ["call-arg", "override", "misc", "attr-defined"]
 
+# Optional dependencies that may not be installed
+[[tool.mypy.overrides]]
+module = [
+    "chainlit",
+    "chainlit.*",
+    "litellm",
+    "opentelemetry.*",
+    "langchain_openai",
+    "langchain_google_vertexai",
+    "holysheet",
+    "yaml",
+]
+ignore_missing_imports = true

{agentomatic-0.6.0 → agentomatic-0.8.0}/src/agentomatic/__init__.py

@@ -24,6 +24,13 @@ With storage::
         enable_auth=True,
         auth_api_key="secret",
     )
+
+With stacks (v0.6)::
+
+    from agentomatic import AgentPlatform
+
+    platform = AgentPlatform.from_folder("agents/", stack="local")
+    app = platform.build()
 """
 
 from __future__ import annotations
@@ -36,6 +43,7 @@ from agentomatic.core.manifest import AgentManifest, RegisteredAgent
 from agentomatic.core.memory_manager import ConversationMemoryManager
 from agentomatic.core.platform import AgentPlatform
 from agentomatic.core.registry import AgentRegistry
+from agentomatic.core.schemas import SchemaValidator
 from agentomatic.core.state import BaseAgentState
 from agentomatic.prompts import PromptManager
 
@@ -45,6 +53,24 @@ from agentomatic.protocols.decorators import APIResponse, handle_api_errors, log
 # Studio
 from agentomatic.studio import GraphInspector, RunTracker
 
+# Pipelines (lazy — avoids hard failure if yaml not installed)
+try:
+    from agentomatic.pipelines import Pipeline, PipelineConfig, PipelineResult
+except ImportError:
+    Pipeline = None  # type: ignore[assignment,misc]
+    PipelineConfig = None  # type: ignore[assignment,misc]
+    PipelineResult = None  # type: ignore[assignment,misc]
+
+# Class-owned graph agents (v0.7)
+from agentomatic.agents import (
+    AgentDataset,
+    AgentExample,
+    AgentGraph,
+    BaseGraphAgent,
+    GraphBuilder,
+    agent_node,
+)
+
 __all__ = [
     # Core
     "AgentPlatform",
@@ -53,6 +79,7 @@ __all__ = [
     "AgentRegistry",
     "BaseAgentState",
     "ConversationMemoryManager",
+    "SchemaValidator",
     # Protocols
     "APIResponse",
     "handle_api_errors",
@@ -62,6 +89,17 @@ __all__ = [
     # Studio
     "GraphInspector",
     "RunTracker",
+    # Pipelines
+    "Pipeline",
+    "PipelineConfig",
+    "PipelineResult",
+    # Class-owned graph agents (v0.7)
+    "BaseGraphAgent",
+    "AgentGraph",
+    "GraphBuilder",
+    "agent_node",
+    "AgentDataset",
+    "AgentExample",
     # Version
     "__version__",
 ]

{agentomatic-0.6.0 → agentomatic-0.8.0}/src/agentomatic/_version.py

@@ -2,4 +2,4 @@
 
 from __future__ import annotations
 
-__version__ = "0.6.0"
+__version__ = "0.8.0"

agentomatic-0.8.0/src/agentomatic/agents/__init__.py

@@ -0,0 +1,93 @@
+"""Class-owned graph agents — ML-like lifecycle for GenAI agents.
+
+Define agents as Python classes with LangGraph-style graph wiring::
+
+    from dataclasses import dataclass
+    from agentomatic.agents import BaseGraphAgent
+
+    @dataclass
+    class MyState:
+        query: str = ""
+        response: str = ""
+
+    class MyAgent(BaseGraphAgent[MyState]):
+        agent_name = "my_agent"
+
+        def __init__(self, *, llm=None):
+            super().__init__()
+            self.llm = llm
+
+        def build_graph(self):
+            g = self.new_graph()
+            g.add_node("process", self.process)
+            g.set_entry_point("process")
+            g.set_finish_point("process")
+            return g.compile()
+
+        def process(self, state):
+            state.response = "hello"
+            return state
+
+        def input_to_state(self, data):
+            return MyState(query=data.get("query", ""))
+
+        def state_to_output(self, state):
+            return {"response": state.response}
+
+    # ML-like workflow
+    agent = MyAgent(llm=my_llm)
+    result = agent.transform({"query": "hello"})
+"""
+
+from __future__ import annotations
+
+from agentomatic.agents.base import BaseGraphAgent
+from agentomatic.agents.builder import GraphBuilder
+from agentomatic.agents.decorators import agent_node
+from agentomatic.agents.graph import AgentGraph, GraphNode
+from agentomatic.agents.metrics import (
+    CallableMetric,
+    ContainsTermsMetric,
+    ExactKeyMatchMetric,
+    OptimizeMetricAdapter,
+)
+from agentomatic.agents.optimizers import (
+    GridSearchOptimizer,
+    NoOpOptimizer,
+    PromptFitterBridge,
+)
+from agentomatic.agents.types import (
+    AgentDataset,
+    AgentExample,
+    EvaluationReport,
+    ExampleResult,
+    Metric,
+    Optimizer,
+    TraceEvent,
+)
+
+__all__ = [
+    # Core
+    "BaseGraphAgent",
+    "AgentGraph",
+    "GraphNode",
+    "GraphBuilder",
+    "agent_node",
+    # Types
+    "AgentDataset",
+    "AgentExample",
+    "EvaluationReport",
+    "ExampleResult",
+    "Metric",
+    "Optimizer",
+    "TraceEvent",
+    # Metrics
+    "ExactKeyMatchMetric",
+    "ContainsTermsMetric",
+    "CallableMetric",
+    "OptimizeMetricAdapter",
+    # Optimizers
+    "NoOpOptimizer",
+    "GridSearchOptimizer",
+    "PromptFitterBridge",
+]