mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-18 18:10:37 +08:00
5f827fffd2
- Replace LocalShellComponent's one-shot subprocess.run() with a PersistentShellSession that wraps a long-running bash process per UMO. cd/export/source now persist naturally within a conversation. - Support background task execution via nohup. - Remove unused ToolSessionManager / ToolSessionState (dead code, never wired to any consumer). - Fix performance benchmark: separate throughput and memory measurement so tracemalloc doesn't distort timing (was 7x slower). - Optimize bool validation in CommandFilter with frozenset + isinstance short-circuit.
23 lines
678 B
Python
23 lines
678 B
Python
from typing import Any, Generic
|
|
|
|
from pydantic import Field
|
|
from pydantic.dataclasses import dataclass
|
|
from typing_extensions import TypeVar
|
|
|
|
from .message import Message
|
|
|
|
TContext = TypeVar("TContext", default=Any)
|
|
|
|
|
|
@dataclass
|
|
class ContextWrapper(Generic[TContext]):
|
|
"""A context for running an agent, which can be used to pass additional data or state."""
|
|
|
|
context: TContext
|
|
messages: list[Message] = Field(default_factory=list)
|
|
"""This field stores the llm message context for the agent run, agent runners will maintain this field automatically."""
|
|
tool_call_timeout: int = 120 # Default tool call timeout in seconds
|
|
|
|
|
|
NoContext = ContextWrapper[None]
|