agentpool 2.2.2__tar.gz → 2.2.4__tar.gz

{agentpool-2.2.2 → agentpool-2.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentpool
-Version: 2.2.2
+Version: 2.2.4
 Summary: Pydantic-AI based Multi-Agent Framework with YAML-based Agents, Teams, Workflows & Extended ACP / AGUI integration
 Keywords: 
 Author: Philipp Temminghoff
@@ -37,6 +37,7 @@ Requires-Dist: jinjarope
 Requires-Dist: llmling-models>=1.4.1
 Requires-Dist: logfire
 Requires-Dist: mcp>=1.2.0
+Requires-Dist: pillow>=11.3.0
 Requires-Dist: platformdirs
 Requires-Dist: promptantic>=0.4.5
 Requires-Dist: psygnal>=0.11.1
@@ -64,7 +65,7 @@ Requires-Dist: ag-ui-protocol>=0.1.10 ; extra == 'ag-ui'
 Requires-Dist: braintrust ; extra == 'braintrust'
 Requires-Dist: autoevals ; extra == 'braintrust'
 Requires-Dist: claude-agent-sdk>=0.1.18 ; extra == 'claude'
-Requires-Dist: clipman ; extra == 'clipboard'
+Requires-Dist: copykitten ; extra == 'clipboard'
 Requires-Dist: rustworkx>=0.17.1 ; extra == 'coding'
 Requires-Dist: grep-ast ; extra == 'coding'
 Requires-Dist: ast-grep-py>=0.40.0 ; extra == 'coding'

{agentpool-2.2.2 → agentpool-2.2.4}/pyproject.toml

@@ -2,7 +2,7 @@
 
 [project]
 name = "agentpool"
-version = "2.2.2"
+version = "2.2.4"
 description = "Pydantic-AI based Multi-Agent Framework with YAML-based Agents, Teams, Workflows & Extended ACP / AGUI integration"
 readme = "README.md"
 requires-python = ">=3.13"
@@ -45,6 +45,7 @@ dependencies = [
   "llmling-models>=1.4.1",
   "logfire",
   "mcp>=1.2.0",
+  "pillow>=11.3.0",
   "platformdirs",
   "promptantic>=0.4.5",
   "psygnal>=0.11.1",
@@ -96,7 +97,7 @@ a2a = ["fasta2a", "starlette"]  # A2A Server
 ag-ui = ["ag-ui-protocol>=0.1.10"]  # Ag-UI Agents
 braintrust = ["braintrust", "autoevals"]  # Braintrust Prompt Hub
 claude = ["claude-agent-sdk>=0.1.18"]
-clipboard = ["clipman"]  # Clipboard functionality
+clipboard = ["copykitten"]  # Clipboard functionality
 coding = [
   "rustworkx>=0.17.1",
   "grep-ast",

{agentpool-2.2.2 → agentpool-2.2.4}/src/acp/__init__.py

@@ -3,7 +3,6 @@
 from acp.client import DefaultACPClient, HeadlessACPClient, NoOpClient, ClientSideConnection
 from acp.agent import AgentSideConnection
 from acp.bridge import ACPBridge, BridgeSettings
-from acp.filesystem import ACPFileSystem, ACPPath
 from acp.agent.protocol import Agent
 from acp.client.protocol import Client
 from acp.terminal_handle import TerminalHandle
@@ -173,9 +172,6 @@ __all__ = [  # noqa: RUF022
     "FileSystemCapability",
     # stdio helper
     "stdio_streams",
-    # filesystem
-    "ACPFileSystem",
-    "ACPPath",
     # transport
     "serve",
     "StdioTransport",

{agentpool-2.2.2 → agentpool-2.2.4}/src/acp/acp_requests.py

@@ -5,6 +5,8 @@ from __future__ import annotations
 import asyncio
 from typing import TYPE_CHECKING
 
+import structlog
+
 from acp.schema import (
     CreateTerminalRequest,
     EnvVariable,
@@ -19,7 +21,6 @@ from acp.schema import (
     WriteTextFileRequest,
 )
 from acp.terminal_handle import TerminalHandle
-from agentpool.log import get_logger
 
 
 if TYPE_CHECKING:
@@ -30,7 +31,7 @@ if TYPE_CHECKING:
         WaitForTerminalExitResponse,
     )
 
-logger = get_logger(__name__)
+logger = structlog.get_logger(__name__)
 
 
 RULES_FILE_NAMES = [

{agentpool-2.2.2 → agentpool-2.2.4}/src/acp/agent/implementations/debug_server/debug_server.py

@@ -25,6 +25,7 @@ import anyio
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import HTMLResponse
 from pydantic import BaseModel, Field
+import structlog
 import uvicorn
 
 from acp import AgentSideConnection
@@ -57,7 +58,6 @@ from acp.schema import (
     WriteTextFileResponse,
 )
 from acp.stdio import stdio_streams
-from agentpool.log import get_logger
 
 
 if TYPE_CHECKING:
@@ -81,7 +81,7 @@ if TYPE_CHECKING:
     )
 
 
-logger = get_logger(__name__)
+logger = structlog.get_logger(__name__)
 MOCK_FILE = """\
 # Mock file: {path}
 # Generated by ACP Debug Server

{agentpool-2.2.2 → agentpool-2.2.4}/src/acp/client/implementations/default_client.py

@@ -9,9 +9,10 @@ from __future__ import annotations
 from pathlib import Path
 from typing import TYPE_CHECKING, Any
 
+import structlog
+
 from acp.client import Client
 from acp.schema import ReadTextFileResponse, RequestPermissionResponse, WriteTextFileResponse
-from agentpool import log
 
 
 if TYPE_CHECKING:
@@ -34,7 +35,7 @@ if TYPE_CHECKING:
         WriteTextFileRequest,
     )
 
-logger = log.get_logger(__name__)
+logger = structlog.get_logger(__name__)
 
 
 class DefaultACPClient(Client):

{agentpool-2.2.2 → agentpool-2.2.4}/src/acp/client/implementations/headless_client.py

@@ -12,6 +12,7 @@ from typing import TYPE_CHECKING, Any
 import uuid
 
 from anyenv import ProcessManager
+import structlog
 
 from acp.client.protocol import Client
 from acp.schema import (
@@ -24,7 +25,6 @@ from acp.schema import (
     WaitForTerminalExitResponse,
     WriteTextFileResponse,
 )
-from agentpool.log import get_logger
 
 
 if TYPE_CHECKING:
@@ -40,7 +40,7 @@ if TYPE_CHECKING:
         WriteTextFileRequest,
     )
 
-logger = get_logger(__name__)
+logger = structlog.get_logger(__name__)
 
 
 class HeadlessACPClient(Client):

{agentpool-2.2.2 → agentpool-2.2.4}/src/acp/connection.py

@@ -15,6 +15,7 @@ import anyenv
 import anyio
 from anyio.streams.text import TextReceiveStream
 from pydantic import BaseModel, ValidationError
+import structlog
 
 from acp.exceptions import RequestError
 from acp.task import (
@@ -31,7 +32,6 @@ from acp.task import (
     RpcTaskKind,
     TaskSupervisor,
 )
-from agentpool import log
 
 
 if TYPE_CHECKING:
@@ -49,7 +49,7 @@ DispatcherFactory = Callable[
 ]
 
 
-logger = log.get_logger(__name__)
+logger = structlog.get_logger(__name__)
 
 
 StreamDirection = Literal["incoming", "outgoing"]

{agentpool-2.2.2 → agentpool-2.2.4}/src/acp/notifications.py

@@ -6,6 +6,7 @@ from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any, assert_never
 
 from pydantic_ai import ModelRequest, ModelResponse, ToolReturnPart, UserPromptPart
+import structlog
 
 from acp.schema import (
     AgentMessageChunk,
@@ -33,7 +34,6 @@ from acp.schema import (
 from acp.schema.tool_call import ToolCallLocation
 from acp.tool_call_reporter import ToolCallReporter
 from acp.utils import generate_tool_title, infer_tool_kind, to_acp_content_blocks
-from agentpool.log import get_logger
 from agentpool.utils.pydantic_ai_helpers import safe_args_as_dict
 
 
@@ -53,7 +53,7 @@ if TYPE_CHECKING:
 
     ContentType = Sequence[ToolCallContent | str]
 
-logger = get_logger(__name__)
+logger = structlog.get_logger(__name__)
 
 
 class ACPNotifications:

{agentpool-2.2.2 → agentpool-2.2.4}/src/acp/task/supervisor.py

@@ -7,7 +7,7 @@ from collections.abc import Callable
 from contextlib import suppress
 from typing import TYPE_CHECKING, Any
 
-from agentpool import log
+import structlog
 
 
 if TYPE_CHECKING:
@@ -16,7 +16,7 @@ if TYPE_CHECKING:
 
 __all__ = ["TaskSupervisor"]
 
-logger = log.get_logger(__name__)
+logger = structlog.get_logger(__name__)
 ErrorHandler = Callable[[asyncio.Task[Any], BaseException], None]
 
 

{agentpool-2.2.2 → agentpool-2.2.4}/src/acp/utils.py

@@ -6,6 +6,7 @@ import base64
 from typing import TYPE_CHECKING, Any
 
 from pydantic_ai import BinaryContent, FileUrl, ToolReturn
+import structlog
 
 from acp.schema import (
     AudioContentBlock,
@@ -16,7 +17,6 @@ from acp.schema import (
     ResourceContentBlock,
     TextContentBlock,
 )
-from agentpool.log import get_logger
 
 
 if TYPE_CHECKING:
@@ -27,7 +27,7 @@ if TYPE_CHECKING:
     from acp.schema import ContentBlock, ToolCallKind
 
 
-logger = get_logger(__name__)
+logger = structlog.get_logger(__name__)
 
 
 DEFAULT_PERMISSION_OPTIONS = [
@@ -131,7 +131,7 @@ def generate_tool_title(tool_name: str, tool_input: dict[str, Any]) -> str:  # n
         return f"Editing: {path}"
 
     # Command/script execution - show the command
-    if any(k in name_lower for k in ["command", "execute", "run", "shell", "script"]) and (
+    if any(k in name_lower for k in ["command", "execute", "run", "shell", "script", "bash"]) and (
         cmd := tool_input.get("command")
     ):
         # Truncate long commands
@@ -244,7 +244,7 @@ def infer_tool_kind(tool_name: str) -> ToolCallKind:  # noqa: PLR0911
         return "move"
     if any(i in name_lower for i in ["search", "find", "query", "lookup"]):
         return "search"
-    if any(i in name_lower for i in ["execute", "run", "exec", "command", "shell"]):
+    if any(i in name_lower for i in ["execute", "run", "exec", "command", "shell", "bash"]):
         return "execute"
     if any(i in name_lower for i in ["think", "plan", "reason", "analyze"]):
         return "think"

{agentpool-2.2.2 → agentpool-2.2.4}/src/agentpool/agents/acp_agent/acp_agent.py

@@ -68,7 +68,7 @@ if TYPE_CHECKING:
     from types import TracebackType
 
     from anyio.abc import Process
-    from evented.configs import EventConfig
+    from evented_config import EventConfig
     from exxec import ExecutionEnvironment
     from pydantic_ai import FinishReason
     from slashed import BaseCommand

{agentpool-2.2.2 → agentpool-2.2.4}/src/agentpool/agents/agent.py

@@ -12,27 +12,28 @@ from uuid import uuid4
 
 from anyenv import method_spawner
 import anyio
-from llmling_models import function_to_model, infer_model
 import logfire
 from psygnal import Signal
 from pydantic import ValidationError
 from pydantic._internal import _typing_extra
 from pydantic_ai import (
     Agent as PydanticAgent,
-    AgentRunResultEvent,
     BaseToolCallPart,
+    CallToolsNode,
     FunctionToolCallEvent,
     FunctionToolResultEvent,
-    PartDeltaEvent,
+    ModelRequestNode,
     PartStartEvent,
     RunContext,
-    TextPart,
-    TextPartDelta,
     ToolReturnPart,
 )
 
 from agentpool.agents.base_agent import BaseAgent
-from agentpool.agents.events import RunStartedEvent, StreamCompleteEvent, ToolCallCompleteEvent
+from agentpool.agents.events import (
+    RunStartedEvent,
+    StreamCompleteEvent,
+    ToolCallCompleteEvent,
+)
 from agentpool.agents.modes import ModeInfo
 from agentpool.log import get_logger
 from agentpool.messaging import ChatMessage, MessageHistory, MessageNode
@@ -51,6 +52,34 @@ from agentpool.utils.streams import FileTracker, merge_queue_into_iterator
 TResult = TypeVar("TResult")
 
 
+def _extract_text_from_messages(
+    messages: list[Any], include_interruption_note: bool = False
+) -> str:
+    """Extract text content from pydantic-ai messages.
+
+    Args:
+        messages: List of ModelRequest/ModelResponse messages
+        include_interruption_note: Whether to append interruption notice
+
+    Returns:
+        Concatenated text content from all ModelResponse TextParts
+    """
+    from pydantic_ai.messages import ModelResponse, TextPart as PydanticTextPart
+
+    content = "".join(
+        part.content
+        for msg in messages
+        if isinstance(msg, ModelResponse)
+        for part in msg.parts
+        if isinstance(part, PydanticTextPart)
+    )
+    if include_interruption_note:
+        if content:
+            content += "\n\n"
+        content += "[Request interrupted by user]"
+    return content
+
+
 if TYPE_CHECKING:
     from collections.abc import AsyncIterator, Coroutine, Sequence
     from datetime import datetime
@@ -225,6 +254,8 @@ class Agent[TDeps = None, OutputDataT = str](BaseAgent[TDeps, OutputDataT]):
                 Defaults to ["models.dev"] if not specified.
             commands: Slash commands
         """
+        from llmling_models import infer_model
+
         from agentpool.agents.interactions import Interactions
         from agentpool.agents.sys_prompts import SystemPrompts
         from agentpool.models.agents import NativeAgentConfig
@@ -438,6 +469,8 @@ class Agent[TDeps = None, OutputDataT = str](BaseAgent[TDeps, OutputDataT]):
             name: Optional name for the agent
             kwargs: Additional arguments for agent
         """
+        from llmling_models import function_to_model
+
         name = name or callback.__name__ or "processor"
         model = function_to_model(callback)
         return_type = _typing_extra.get_function_type_hints(callback).get("return")
@@ -580,6 +613,8 @@ class Agent[TDeps = None, OutputDataT = str](BaseAgent[TDeps, OutputDataT]):
         """Create pydantic-ai agent from current state."""
         # Monkey patch pydantic-ai to recognize AgentContext
 
+        from llmling_models import infer_model
+
         from agentpool.agents.tool_wrapping import wrap_tool
 
         tools = await self.tools.get_tools(state="enabled", names=tool_choice)
@@ -792,9 +827,6 @@ class Agent[TDeps = None, OutputDataT = str](BaseAgent[TDeps, OutputDataT]):
         self._cancelled = False
         self._current_stream_task = asyncio.current_task()
 
-        # Track accumulated content for partial message on cancellation
-        accumulated_text: list[str] = []
-
         # Execute pre-run hooks
         if self.hooks:
             pre_run_result = await self.hooks.run_pre_run_hooks(
@@ -815,6 +847,8 @@ class Agent[TDeps = None, OutputDataT = str](BaseAgent[TDeps, OutputDataT]):
         await handler(None, run_started)
         yield run_started
         try:
+            from pydantic_graph import End
+
             agentlet = await self.get_agentlet(tool_choice, model, output_type, input_provider)
             content = await convert_prompts(prompts)
             response_msg: ChatMessage[Any] | None = None
@@ -823,104 +857,104 @@ class Agent[TDeps = None, OutputDataT = str](BaseAgent[TDeps, OutputDataT]):
 
             # Use provided usage_limits or fall back to default
             effective_limits = usage_limits or self._default_usage_limits
+            history = [m for run in history_list for m in run.to_pydantic_ai()]
 
-            stream_events = agentlet.run_stream_events(
+            # Track tool call starts to combine with results later
+            pending_tcs: dict[str, BaseToolCallPart] = {}
+            file_tracker = FileTracker()
+
+            async with agentlet.iter(
                 converted,
                 deps=deps,  # type: ignore[arg-type]
-                message_history=[m for run in history_list for m in run.to_pydantic_ai()],
+                message_history=history,
                 usage_limits=effective_limits,
                 instructions=instructions,
-            )
-
-            # Stream events through merge_queue for progress events
-            file_tracker = FileTracker()
-            async with merge_queue_into_iterator(stream_events, self._event_queue) as events:
-                # Track tool call starts to combine with results later
-                pending_tcs: dict[str, BaseToolCallPart] = {}
+            ) as agent_run:
                 try:
-                    async for event in file_tracker.track(events):
-                        # Check for cancellation
+                    async for node in agent_run:
                         if self._cancelled:
                             self.log.info("Stream cancelled by user")
                             break
+                        if isinstance(node, End):
+                            break
 
-                        # Call event handlers for all events
-                        await handler(None, event)
-                        yield event  # type: ignore[misc]
-
-                        # Accumulate text content for partial message
-                        match event:
-                            case PartDeltaEvent(delta=TextPartDelta(content_delta=delta)):
-                                accumulated_text.append(delta)
-                            case PartStartEvent(part=TextPart(content=text)) if text:
-                                accumulated_text.append(text)
-
-                        match event:
-                            case (
-                                PartStartEvent(part=BaseToolCallPart() as tool_part)
-                                | FunctionToolCallEvent(part=tool_part)
+                        # Stream events from model request node
+                        if isinstance(node, ModelRequestNode):
+                            async with (
+                                node.stream(agent_run.ctx) as agent_stream,
+                                merge_queue_into_iterator(
+                                    agent_stream,  # type: ignore[arg-type]
+                                    self._event_queue,
+                                ) as merged,
+                            ):
+                                async for event in file_tracker.track(merged):
+                                    if self._cancelled:
+                                        break
+                                    await handler(None, event)
+                                    yield event
+                                    combined = self._process_tool_event(
+                                        event, pending_tcs, message_id
+                                    )
+                                    if combined:
+                                        await handler(None, combined)
+                                        yield combined
+
+                        # Stream events from tool call node
+                        elif isinstance(node, CallToolsNode):
+                            async with (
+                                node.stream(agent_run.ctx) as tool_stream,
+                                merge_queue_into_iterator(tool_stream, self._event_queue) as merged,
                             ):
-                                # Store tool call start info for later combination with result
-                                pending_tcs[tool_part.tool_call_id] = tool_part
-                            case FunctionToolResultEvent(tool_call_id=call_id) as result_event:
-                                # Check if we have a pending tool call to combine with
-                                if call_info := pending_tcs.pop(call_id, None):
-                                    # Create and yield combined event
-                                    combined_event = ToolCallCompleteEvent(
-                                        tool_name=call_info.tool_name,
-                                        tool_call_id=call_id,
-                                        tool_input=safe_args_as_dict(call_info),
-                                        tool_result=result_event.result.content
-                                        if isinstance(result_event.result, ToolReturnPart)
-                                        else result_event.result,
-                                        agent_name=self.name,
-                                        message_id=message_id,
+                                async for event in file_tracker.track(merged):
+                                    if self._cancelled:
+                                        break
+                                    await handler(None, event)
+                                    yield event
+                                    combined = self._process_tool_event(
+                                        event, pending_tcs, message_id
                                     )
-                                    await handler(None, combined_event)
-                                    yield combined_event
-                            case AgentRunResultEvent():
-                                # Capture final result data, Build final response message
-                                response_time = time.perf_counter() - start_time
-                                response_msg = await ChatMessage.from_run_result(
-                                    event.result,
-                                    agent_name=self.name,
-                                    message_id=message_id,
-                                    conversation_id=conversation_id or user_msg.conversation_id,
-                                    parent_id=user_msg.message_id,
-                                    response_time=response_time,
-                                    metadata=file_tracker.get_metadata(),
-                                )
+                                    if combined:
+                                        await handler(None, combined)
+                                        yield combined
                 except asyncio.CancelledError:
                     self.log.info("Stream cancelled via task cancellation")
                     self._cancelled = True
 
-            # Handle cancellation - emit partial message
-            if self._cancelled:
+                # Build response message
                 response_time = time.perf_counter() - start_time
-                partial_content = "".join(accumulated_text)
-                if partial_content:
-                    partial_content += "\n\n"
-                partial_content += "[Request interrupted by user]"
-                response_msg = ChatMessage(
-                    content=partial_content,
-                    role="assistant",
-                    name=self.name,
-                    message_id=message_id,
-                    conversation_id=conversation_id or user_msg.conversation_id,
-                    parent_id=user_msg.message_id,
-                    response_time=response_time,
-                    finish_reason="stop",
-                )
-                complete_event = StreamCompleteEvent(message=response_msg)
-                await handler(None, complete_event)
-                yield complete_event
-                self._current_stream_task = None
-                return
-
-            # Only finalize if we got a result (stream may exit early on error)
-            if response_msg is None:
-                msg = "Stream completed without producing a result"
-                raise RuntimeError(msg)  # noqa: TRY301
+                if self._cancelled:
+                    partial_content = _extract_text_from_messages(
+                        agent_run.all_messages(), include_interruption_note=True
+                    )
+                    response_msg = ChatMessage(
+                        content=partial_content,
+                        role="assistant",
+                        name=self.name,
+                        message_id=message_id,
+                        conversation_id=conversation_id or user_msg.conversation_id,
+                        parent_id=user_msg.message_id,
+                        response_time=response_time,
+                        finish_reason="stop",
+                    )
+                    complete_event = StreamCompleteEvent(message=response_msg)
+                    await handler(None, complete_event)
+                    yield complete_event
+                    self._current_stream_task = None
+                    return
+
+                if agent_run.result:
+                    response_msg = await ChatMessage.from_run_result(
+                        agent_run.result,
+                        agent_name=self.name,
+                        message_id=message_id,
+                        conversation_id=conversation_id or user_msg.conversation_id,
+                        parent_id=user_msg.message_id,
+                        response_time=response_time,
+                        metadata=file_tracker.get_metadata(),
+                    )
+                else:
+                    msg = "Stream completed without producing a result"
+                    raise RuntimeError(msg)  # noqa: TRY301
 
             # Execute post-run hooks
             if self.hooks:
@@ -954,6 +988,41 @@ class Agent[TDeps = None, OutputDataT = str](BaseAgent[TDeps, OutputDataT]):
         finally:
             self._current_stream_task = None
 
+    def _process_tool_event(
+        self,
+        event: RichAgentStreamEvent[Any],
+        pending_tool_calls: dict[str, BaseToolCallPart],
+        message_id: str,
+    ) -> ToolCallCompleteEvent | None:
+        """Process tool-related events and return combined event when complete.
+
+        Args:
+            event: The streaming event to process
+            pending_tool_calls: Dict tracking in-progress tool calls by ID
+            message_id: Message ID for the combined event
+
+        Returns:
+            ToolCallCompleteEvent if a tool call completed, None otherwise
+        """
+        match event:
+            case PartStartEvent(part=BaseToolCallPart() as tool_part):
+                pending_tool_calls[tool_part.tool_call_id] = tool_part
+            case FunctionToolCallEvent(part=tool_part):
+                pending_tool_calls[tool_part.tool_call_id] = tool_part
+            case FunctionToolResultEvent(tool_call_id=call_id) as result_event:
+                if call_info := pending_tool_calls.pop(call_id, None):
+                    return ToolCallCompleteEvent(
+                        tool_name=call_info.tool_name,
+                        tool_call_id=call_id,
+                        tool_input=safe_args_as_dict(call_info),
+                        tool_result=result_event.result.content
+                        if isinstance(result_event.result, ToolReturnPart)
+                        else result_event.result,
+                        agent_name=self.name,
+                        message_id=message_id,
+                    )
+        return None
+
     async def run_iter(
         self,
         *prompt_groups: Sequence[PromptCompatible],
@@ -1185,6 +1254,8 @@ class Agent[TDeps = None, OutputDataT = str](BaseAgent[TDeps, OutputDataT]):
             model: New model to use (name or instance)
 
         """
+        from llmling_models import infer_model
+
         self._model = infer_model(model) if isinstance(model, str) else model
 
     async def set_tool_confirmation_mode(self, mode: ToolConfirmationMode) -> None:
@@ -1240,6 +1311,8 @@ class Agent[TDeps = None, OutputDataT = str](BaseAgent[TDeps, OutputDataT]):
             pause_routing: Whether to pause message routing
             model: Temporary model override
         """
+        from llmling_models import infer_model
+
         old_model = self._model
         if output_type:
             old_type = self._output_type

{agentpool-2.2.2 → agentpool-2.2.4}/src/agentpool/agents/agui_agent/agui_agent.py

@@ -45,7 +45,7 @@ if TYPE_CHECKING:
     from types import TracebackType
 
     from ag_ui.core import Message, ToolMessage
-    from evented.configs import EventConfig
+    from evented_config import EventConfig
     from slashed import BaseCommand
     from tokonomics.model_discovery.model_info import ModelInfo