mirror of
https://github.com/czl9707/build-your-own-openclaw.git
synced 2026-08-14 00:47:59 +00:00
Zane Chen
8e71c10f13
* chore: respecting stop reason * fix: add back template substitution * typo fix * increase timeout allow slower model
126 lines
4.0 KiB
Python
126 lines
4.0 KiB
Python
"""Chat CLI command for interactive sessions with event-driven architecture."""
|
|
|
|
import asyncio
|
|
|
|
import typer
|
|
from rich.console import Console
|
|
from rich.panel import Panel
|
|
from rich.prompt import Prompt
|
|
from rich.text import Text
|
|
|
|
from mybot.core.agent import Agent
|
|
from mybot.core.context import SharedContext
|
|
from mybot.core.events import (
|
|
OutboundEvent,
|
|
InboundEvent,
|
|
CliEventSource,
|
|
)
|
|
from mybot.server import (
|
|
AgentWorker,
|
|
Worker,
|
|
)
|
|
from mybot.utils.config import Config, ConfigReloader
|
|
from mybot.utils.logging import setup_logging
|
|
|
|
|
|
class ChatLoop:
|
|
"""Interactive chat session using event-driven architecture."""
|
|
|
|
def __init__(self, config: Config, agent_id: str | None = None):
|
|
self.config = config
|
|
self.console = Console()
|
|
self.context = SharedContext(config=config, channels=[])
|
|
self.config_reloader = ConfigReloader(config)
|
|
|
|
self.workers: list[Worker] = [
|
|
self.context.eventbus,
|
|
AgentWorker(self.context),
|
|
]
|
|
|
|
self.response_queue: asyncio.Queue[OutboundEvent] = asyncio.Queue()
|
|
self.context.eventbus.subscribe(OutboundEvent, self.handle_outbound_event)
|
|
|
|
agent_id = agent_id or config.default_agent
|
|
self.agent_def = self.context.agent_loader.load(agent_id)
|
|
|
|
async def handle_outbound_event(self, event: OutboundEvent) -> None:
|
|
"""Handle outbound events by adding to response queue."""
|
|
await self.response_queue.put(event)
|
|
self.context.eventbus.ack(event)
|
|
|
|
def get_user_input(self) -> str:
|
|
"""Get user input with styled prompt."""
|
|
prompt_text = Text("You", style="cyan")
|
|
user_input = Prompt.ask(prompt_text, console=self.console)
|
|
return user_input.strip()
|
|
|
|
def display_agent_response(self, content: str) -> None:
|
|
"""Display agent response with styled prefix."""
|
|
prefix = Text(f"{self.agent_def.id}: ", style="green")
|
|
|
|
self.console.print(prefix, end="")
|
|
self.console.print(content)
|
|
|
|
async def run(self) -> None:
|
|
"""Run the interactive chat loop."""
|
|
self.console.print(
|
|
Panel(
|
|
Text("Welcome to my-bot!", style="bold cyan"),
|
|
title="Chat",
|
|
border_style="cyan",
|
|
)
|
|
)
|
|
self.console.print("Type '/help' for commands, 'quit' or 'exit' to end.\n")
|
|
|
|
self.config_reloader.start()
|
|
|
|
for worker in self.workers:
|
|
worker.start()
|
|
|
|
session_id = (
|
|
Agent(self.agent_def, self.context).new_session(CliEventSource()).session_id
|
|
)
|
|
|
|
try:
|
|
while True:
|
|
user_input = await asyncio.to_thread(self.get_user_input)
|
|
if user_input.lower() in ("quit", "exit", "q"):
|
|
self.console.print("\n[bold yellow]Goodbye![/bold yellow]")
|
|
break
|
|
|
|
if not user_input:
|
|
continue
|
|
|
|
event = InboundEvent(
|
|
session_id=session_id,
|
|
source=CliEventSource(),
|
|
content=user_input,
|
|
)
|
|
await self.context.eventbus.publish(event)
|
|
|
|
try:
|
|
response = await asyncio.wait_for(
|
|
self.response_queue.get(), timeout=120.0
|
|
)
|
|
|
|
self.display_agent_response(response.content)
|
|
except asyncio.TimeoutError:
|
|
self.console.print("[red]Agent response timed out[/red]")
|
|
self.console.print()
|
|
|
|
except (KeyboardInterrupt, EOFError):
|
|
self.console.print("\n[bold yellow]Goodbye![/bold yellow]")
|
|
finally:
|
|
for worker in self.workers:
|
|
await worker.stop()
|
|
self.config_reloader.stop()
|
|
|
|
|
|
def chat_command(ctx: typer.Context, agent_id: str | None = None) -> None:
|
|
"""Start interactive chat session."""
|
|
config = ctx.obj.get("config")
|
|
setup_logging(config, console_output=False)
|
|
|
|
chat_loop = ChatLoop(config, agent_id=agent_id)
|
|
asyncio.run(chat_loop.run())
|