aury-agent 0.0.4__py3-none-any.whl

aury/agents/cli/chat.py

@@ -0,0 +1,239 @@
+"""
+交互式对话命令
+==============
+
+提供命令行交互式对话功能。
+"""
+import asyncio
+from typing import Optional, Any
+from uuid import uuid4
+
+from rich.console import Console
+from rich.markdown import Markdown
+from rich.panel import Panel
+from rich.prompt import Prompt
+
+from aury.agents.core import Session, InvocationContext
+from aury.agents.react import ReactAgent
+
+console = Console()
+
+
+class DefaultChatAgent(ReactAgent):
+    """默认对话 Agent。"""
+    
+    name = "chat_agent"
+    description = "通用对话助手"
+    system_prompt = """你是一个友好的 AI 助手。请用中文回答用户的问题。"""
+    tools = []
+
+
+async def chat_command(
+    agent_name: Optional[str] = None,
+    session_id: Optional[str] = None,
+    config: Optional[dict] = None,
+    verbose: bool = False,
+):
+    """
+    运行交互式对话。
+    
+    Args:
+        agent_name: Agent 名称或模块路径
+        session_id: 会话 ID（可选，用于恢复会话）
+        config: 配置字典
+        verbose: 是否显示详细信息
+    """
+    config = config or {}
+    
+    # 显示欢迎信息
+    console.print(Panel(
+        "[bold blue]Aury Agent 交互式对话[/bold blue]\n\n"
+        "输入消息与 Agent 对话\n"
+        "输入 /help 查看可用命令\n"
+        "输入 /quit 或按 Ctrl+C 退出",
+        title="欢迎",
+    ))
+    
+    # 创建或加载会话
+    if session_id:
+        console.print(f"[dim]恢复会话: {session_id}[/dim]")
+        session = Session(session_id=session_id)
+        # TODO: 从存储加载会话
+    else:
+        session_id = f"chat-{uuid4().hex[:8]}"
+        session = Session(session_id=session_id)
+        console.print(f"[dim]新会话: {session_id}[/dim]")
+    
+    # 创建 Agent
+    agent = await _create_agent(agent_name, config)
+    
+    if verbose:
+        console.print(f"[dim]使用 Agent: {agent.name}[/dim]")
+    
+    # 创建上下文
+    context = InvocationContext(session=session)
+    
+    # 主循环
+    while True:
+        try:
+            # 获取用户输入
+            user_input = Prompt.ask("\n[bold cyan]你[/bold cyan]")
+            
+            if not user_input.strip():
+                continue
+            
+            # 处理命令
+            if user_input.startswith("/"):
+                should_continue = await _handle_command(
+                    user_input, session, context, verbose
+                )
+                if not should_continue:
+                    break
+                continue
+            
+            # 运行 Agent
+            console.print("[dim]思考中...[/dim]")
+            
+            try:
+                result = await agent.run(user_input, context)
+                
+                # 显示响应
+                console.print()
+                console.print(Panel(
+                    Markdown(result.output) if result.output else "[dim]无响应[/dim]",
+                    title="[bold green]Agent[/bold green]",
+                    border_style="green",
+                ))
+                
+                # 显示工具调用（详细模式）
+                if verbose and hasattr(result, 'tool_calls') and result.tool_calls:
+                    console.print("[dim]工具调用:[/dim]")
+                    for tc in result.tool_calls:
+                        console.print(f"  [dim]- {tc.name}[/dim]")
+                
+            except Exception as e:
+                console.print(f"[red]错误: {e}[/red]")
+                if verbose:
+                    import traceback
+                    console.print(f"[dim]{traceback.format_exc()}[/dim]")
+        
+        except KeyboardInterrupt:
+            console.print("\n[yellow]收到中断信号[/yellow]")
+            break
+        except EOFError:
+            break
+    
+    # 退出
+    console.print("\n[dim]再见！[/dim]")
+
+
+async def _create_agent(
+    agent_name: Optional[str],
+    config: dict,
+) -> ReactAgent:
+    """创建 Agent 实例。"""
+    
+    if agent_name is None:
+        # 使用默认 Agent
+        return DefaultChatAgent()
+    
+    # 尝试从模块加载
+    if "." in agent_name or agent_name.endswith(".py"):
+        try:
+            import importlib.util
+            from pathlib import Path
+            
+            if agent_name.endswith(".py"):
+                # 从文件加载
+                path = Path(agent_name)
+                spec = importlib.util.spec_from_file_location("agent_module", path)
+                module = importlib.util.module_from_spec(spec)
+                spec.loader.exec_module(module)
+            else:
+                # 从模块路径加载
+                module = importlib.import_module(agent_name)
+            
+            # 查找 Agent 类
+            for name in dir(module):
+                obj = getattr(module, name)
+                if (
+                    isinstance(obj, type)
+                    and issubclass(obj, ReactAgent)
+                    and obj is not ReactAgent
+                ):
+                    return obj()
+            
+            raise ValueError(f"模块中未找到 Agent 类: {agent_name}")
+        
+        except Exception as e:
+            console.print(f"[yellow]无法加载 Agent '{agent_name}': {e}[/yellow]")
+            console.print("[dim]使用默认 Agent[/dim]")
+            return DefaultChatAgent()
+    
+    # TODO: 从注册表查找已注册的 Agent
+    console.print(f"[yellow]未找到 Agent '{agent_name}'，使用默认 Agent[/yellow]")
+    return DefaultChatAgent()
+
+
+async def _handle_command(
+    command: str,
+    session: Session,
+    context: InvocationContext,
+    verbose: bool,
+) -> bool:
+    """
+    处理斜杠命令。
+    
+    Returns:
+        bool: 是否继续对话
+    """
+    cmd = command.strip().lower()
+    
+    if cmd in ["/quit", "/exit", "/q"]:
+        return False
+    
+    elif cmd == "/help":
+        console.print(Panel(
+            "[bold]可用命令:[/bold]\n\n"
+            "/help     - 显示帮助信息\n"
+            "/quit     - 退出对话\n"
+            "/clear    - 清空对话历史\n"
+            "/session  - 显示当前会话信息\n"
+            "/history  - 显示对话历史\n"
+            "/save     - 保存当前会话\n"
+            "/verbose  - 切换详细模式",
+            title="帮助",
+        ))
+    
+    elif cmd == "/clear":
+        context.history = []
+        console.print("[green]对话历史已清空[/green]")
+    
+    elif cmd == "/session":
+        console.print(Panel(
+            f"会话 ID: {session.session_id}\n"
+            f"消息数: {len(context.history) if hasattr(context, 'history') else 0}",
+            title="会话信息",
+        ))
+    
+    elif cmd == "/history":
+        if hasattr(context, 'history') and context.history:
+            for msg in context.history[-10:]:  # 最近 10 条
+                role = msg.get("role", "unknown")
+                content = msg.get("content", "")[:100]
+                console.print(f"[dim]{role}: {content}...[/dim]")
+        else:
+            console.print("[dim]暂无对话历史[/dim]")
+    
+    elif cmd == "/save":
+        # TODO: 实现会话保存
+        console.print("[green]会话已保存[/green]")
+    
+    elif cmd == "/verbose":
+        console.print(f"[dim]详细模式: {'开启' if verbose else '关闭'}[/dim]")
+    
+    else:
+        console.print(f"[yellow]未知命令: {command}[/yellow]")
+        console.print("[dim]输入 /help 查看可用命令[/dim]")
+    
+    return True

aury/agents/cli/config.py

@@ -0,0 +1,236 @@
+"""
+配置管理命令
+============
+
+管理 Aury Agent 的配置文件。
+"""
+import os
+from pathlib import Path
+from typing import Optional, Any
+
+import typer
+import yaml
+from rich.console import Console
+from rich.panel import Panel
+from rich.syntax import Syntax
+
+config_app = typer.Typer(help="配置管理命令")
+console = Console()
+
+# 默认配置目录
+DEFAULT_CONFIG_DIR = Path.home() / ".aury"
+DEFAULT_CONFIG_FILE = DEFAULT_CONFIG_DIR / "config.yaml"
+
+
+def get_default_config() -> dict:
+    """获取默认配置。"""
+    return {
+        "agent": {
+            "default": "chat_agent",
+            "max_iterations": 20,
+        },
+        "model": {
+            "provider": "openai",
+            "model": "gpt-4",
+            "temperature": 0.7,
+            "max_tokens": 4096,
+        },
+        "session": {
+            "storage_path": str(DEFAULT_CONFIG_DIR / "sessions"),
+            "auto_save": True,
+            "max_history": 100,
+        },
+        "logging": {
+            "level": "INFO",
+            "file": None,
+        },
+    }
+
+
+def load_config(config_path: Optional[Path] = None) -> dict:
+    """
+    加载配置文件。
+    
+    Args:
+        config_path: 配置文件路径，为 None 时使用默认路径
+        
+    Returns:
+        配置字典
+    """
+    if config_path is None:
+        config_path = DEFAULT_CONFIG_FILE
+    
+    config = get_default_config()
+    
+    if config_path.exists():
+        try:
+            with open(config_path, "r", encoding="utf-8") as f:
+                user_config = yaml.safe_load(f) or {}
+            
+            # 合并用户配置
+            _deep_merge(config, user_config)
+        except Exception as e:
+            console.print(f"[yellow]警告: 无法加载配置文件 {config_path}: {e}[/yellow]")
+    
+    return config
+
+
+def save_config(config: dict, config_path: Optional[Path] = None) -> None:
+    """保存配置文件。"""
+    if config_path is None:
+        config_path = DEFAULT_CONFIG_FILE
+    
+    # 确保目录存在
+    config_path.parent.mkdir(parents=True, exist_ok=True)
+    
+    with open(config_path, "w", encoding="utf-8") as f:
+        yaml.safe_dump(config, f, allow_unicode=True, default_flow_style=False)
+
+
+def _deep_merge(base: dict, override: dict) -> None:
+    """深度合并字典。"""
+    for key, value in override.items():
+        if key in base and isinstance(base[key], dict) and isinstance(value, dict):
+            _deep_merge(base[key], value)
+        else:
+            base[key] = value
+
+
+@config_app.command("show")
+def show_config(
+    config_file: Optional[Path] = typer.Option(
+        None,
+        "--file", "-f",
+        help="配置文件路径",
+    ),
+):
+    """显示当前配置。"""
+    config = load_config(config_file)
+    
+    yaml_str = yaml.safe_dump(config, allow_unicode=True, default_flow_style=False)
+    
+    console.print(Panel(
+        Syntax(yaml_str, "yaml", theme="monokai"),
+        title="当前配置",
+    ))
+
+
+@config_app.command("init")
+def init_config(
+    force: bool = typer.Option(
+        False,
+        "--force", "-f",
+        help="强制覆盖已存在的配置",
+    ),
+):
+    """初始化默认配置文件。"""
+    if DEFAULT_CONFIG_FILE.exists() and not force:
+        console.print(f"[yellow]配置文件已存在: {DEFAULT_CONFIG_FILE}[/yellow]")
+        console.print("[dim]使用 --force 覆盖[/dim]")
+        raise typer.Exit(1)
+    
+    # 创建默认配置
+    config = get_default_config()
+    save_config(config)
+    
+    console.print(f"[green]配置文件已创建: {DEFAULT_CONFIG_FILE}[/green]")
+
+
+@config_app.command("set")
+def set_config(
+    key: str = typer.Argument(..., help="配置项键名（用点号分隔，如 model.temperature）"),
+    value: str = typer.Argument(..., help="配置项值"),
+    config_file: Optional[Path] = typer.Option(
+        None,
+        "--file", "-f",
+        help="配置文件路径",
+    ),
+):
+    """设置配置项。"""
+    config = load_config(config_file)
+    
+    # 解析键路径
+    keys = key.split(".")
+    current = config
+    
+    for k in keys[:-1]:
+        if k not in current:
+            current[k] = {}
+        current = current[k]
+    
+    # 尝试解析值类型
+    try:
+        # 尝试解析为数字
+        if "." in value:
+            parsed_value = float(value)
+        else:
+            parsed_value = int(value)
+    except ValueError:
+        # 尝试解析为布尔值
+        if value.lower() in ["true", "yes"]:
+            parsed_value = True
+        elif value.lower() in ["false", "no"]:
+            parsed_value = False
+        elif value.lower() == "null":
+            parsed_value = None
+        else:
+            parsed_value = value
+    
+    current[keys[-1]] = parsed_value
+    
+    # 保存
+    save_config(config, config_file)
+    
+    console.print(f"[green]已设置 {key} = {parsed_value}[/green]")
+
+
+@config_app.command("get")
+def get_config(
+    key: str = typer.Argument(..., help="配置项键名"),
+    config_file: Optional[Path] = typer.Option(
+        None,
+        "--file", "-f",
+        help="配置文件路径",
+    ),
+):
+    """获取配置项值。"""
+    config = load_config(config_file)
+    
+    # 解析键路径
+    keys = key.split(".")
+    current = config
+    
+    try:
+        for k in keys:
+            current = current[k]
+        console.print(f"{key} = {current}")
+    except KeyError:
+        console.print(f"[red]配置项不存在: {key}[/red]")
+        raise typer.Exit(1)
+
+
+@config_app.command("path")
+def config_path():
+    """显示配置文件路径。"""
+    console.print(f"配置目录: {DEFAULT_CONFIG_DIR}")
+    console.print(f"配置文件: {DEFAULT_CONFIG_FILE}")
+    console.print(f"文件存在: {'是' if DEFAULT_CONFIG_FILE.exists() else '否'}")
+
+
+@config_app.command("edit")
+def edit_config():
+    """使用默认编辑器编辑配置文件。"""
+    import subprocess
+    
+    if not DEFAULT_CONFIG_FILE.exists():
+        console.print("[yellow]配置文件不存在，先创建默认配置[/yellow]")
+        config = get_default_config()
+        save_config(config)
+    
+    editor = os.environ.get("EDITOR", "vim")
+    
+    try:
+        subprocess.run([editor, str(DEFAULT_CONFIG_FILE)])
+    except FileNotFoundError:
+        console.print(f"[red]编辑器未找到: {editor}[/red]")
+        console.print(f"[dim]请设置 EDITOR 环境变量或手动编辑: {DEFAULT_CONFIG_FILE}[/dim]")