lite-agent 0.13.0__tar.gz → 0.14.0__tar.gz

{lite_agent-0.13.0 → lite_agent-0.14.0}/CHANGELOG.md

@@ -1,3 +1,17 @@
+## v0.14.0
+
+[v0.13.0...v0.14.0](https://github.com/Jannchie/lite-agent/compare/v0.13.0...v0.14.0)
+
+### :sparkles: Features
+
+- **chat-display**: improve compact message table layout and add flexible string output functions - By [Jannchie](mailto:jannchie@gmail.com) in [454fe91](https://github.com/Jannchie/lite-agent/commit/454fe91)
+- **chat-display**: add messages_to_string and chat_summary_to_string - By [Jannchie](mailto:jannchie@gmail.com) in [887cdf0](https://github.com/Jannchie/lite-agent/commit/887cdf0)
+- **context**: add context modification example and tests - By [Jannchie](mailto:jannchie@gmail.com) in [3e32ac1](https://github.com/Jannchie/lite-agent/commit/3e32ac1)
+
+### :art: Refactors
+
+- **chat-display**: use keyword arguments and typing hints - By [Jannchie](mailto:jannchie@gmail.com) in [3e7a6cd](https://github.com/Jannchie/lite-agent/commit/3e7a6cd)
+
 ## v0.13.0
 
 [v0.12.0...v0.13.0](https://github.com/Jannchie/lite-agent/compare/v0.12.0...v0.13.0)

{lite_agent-0.13.0 → lite_agent-0.14.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lite-agent
-Version: 0.13.0
+Version: 0.14.0
 Summary: A lightweight, extensible framework for building AI agent.
 Author-email: Jianqi Pan <jannchie@gmail.com>
 License: MIT

{lite_agent-0.13.0 → lite_agent-0.14.0}/examples/basic.py

@@ -6,7 +6,7 @@ from openai import BaseModel
 from rich.logging import RichHandler
 
 from lite_agent.agent import Agent
-from lite_agent.chat_display import display_messages
+from lite_agent.chat_display import chat_summary_to_string, messages_to_string
 from lite_agent.client import LiteLLMClient
 from lite_agent.runner import Runner
 
@@ -49,7 +49,10 @@ async def main():
         includes=["usage", "assistant_message", "function_call", "function_call_output", "timing"],
         context=ctx,
     )
-    display_messages(runner.messages)
+    messages = messages_to_string(runner.messages)
+    summary = chat_summary_to_string(runner.messages)
+    print(messages)
+    print(summary)
 
 
 if __name__ == "__main__":

lite_agent-0.14.0/examples/context_modify.py

@@ -0,0 +1,69 @@
+import asyncio
+import logging
+
+from funcall import Context
+from pydantic import BaseModel
+from rich.logging import RichHandler
+
+from lite_agent.agent import Agent
+from lite_agent.chat_display import display_messages
+from lite_agent.context import HistoryContext
+from lite_agent.runner import Runner
+
+
+class UserCtx(BaseModel):
+    name: str
+
+
+logging.basicConfig(
+    level=logging.WARNING,
+    format="%(message)s",
+    datefmt="[%X]",
+    handlers=[RichHandler(rich_tracebacks=True)],
+)
+
+logger = logging.getLogger("lite_agent")
+logger.setLevel(logging.DEBUG)
+
+
+async def get_user_name(context: Context[HistoryContext[UserCtx]]) -> str:
+    """Get the user name from the context."""
+    if not context.value.data:
+        msg = "User name must be specified in the context."
+        raise ValueError(msg)
+    return context.value.data.name
+
+
+async def set_user_name(context: Context[HistoryContext[UserCtx]], new_user_name: str) -> str:
+    """Set the user name in the context."""
+    if not context.value.data:
+        msg = "User name must be specified in the context."
+        raise ValueError(msg)
+
+    context.value.data.name = new_user_name
+    return f"User name set to: {new_user_name}"
+
+
+agent = Agent(
+    model="gpt-4.1-nano",
+    name="User Name Assistant",
+    instructions="You can get and set the user name.",
+    tools=[get_user_name, set_user_name],
+)
+
+
+async def main():
+    runner = Runner(agent)
+    user_ctx = Context(UserCtx(name="ATC"))
+    resp = runner.run(
+        "Set User name to TXA.",
+        context=user_ctx,
+    )
+    async for chunk in resp:
+        logger.info(chunk)
+    await runner.run_until_complete("get the user's name", context=user_ctx)
+    display_messages(runner.messages)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

{lite_agent-0.13.0 → lite_agent-0.14.0}/examples/stop_before_functions.py

@@ -60,7 +60,8 @@ async def main():
     print(f"\nHas require confirm tools: {await runner.has_require_confirm_tools()}")
 
     # Continue execution
-    resp = runner.run(None,
+    resp = runner.run(
+        None,
         includes=["usage", "assistant_message", "function_call", "function_call_output"],
     )
     async for chunk in resp:
@@ -82,7 +83,8 @@ async def main():
     print(f"\nHas require confirm tools: {await runner2.has_require_confirm_tools()}")
 
     # Continue execution
-    resp = runner2.run(None,
+    resp = runner2.run(
+        None,
         includes=["usage", "assistant_message", "function_call", "function_call_output"],
     )
     async for chunk in resp:
@@ -103,7 +105,8 @@ async def main():
     print(f"\nHas require confirm tools (decorator): {await runner3.has_require_confirm_tools()}")
 
     # Continue execution
-    resp = runner3.run(None,
+    resp = runner3.run(
+        None,
         includes=["usage", "assistant_message", "function_call", "function_call_output"],
     )
     async for chunk in resp:

{lite_agent-0.13.0 → lite_agent-0.14.0}/examples/terminal.py

@@ -60,11 +60,11 @@ async def main():
                     validate_while_typing=False,
                 )
                 if user_input.lower() in {"y", "yes"}:
-                    response = runner.run(None )
+                    response = runner.run(None)
                     async for chunk in response:
                         await rich_channel.handle(chunk)
                 else:
-                    response = runner.run(None )
+                    response = runner.run(None)
             rich_channel.new_turn()
         except (EOFError, KeyboardInterrupt):
             break

{lite_agent-0.13.0 → lite_agent-0.14.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lite-agent"
-version = "0.13.0"
+version = "0.14.0"
 description = "A lightweight, extensible framework for building AI agent."
 readme = "README.md"
 authors = [{ name = "Jianqi Pan", email = "jannchie@gmail.com" }]

lite_agent-0.14.0/src/lite_agent/__init__.py

@@ -0,0 +1,8 @@
+"""Lite Agent - A lightweight AI agent framework."""
+
+from .agent import Agent
+from .chat_display import chat_summary_to_string, display_chat_summary, display_messages, messages_to_string
+from .message_transfers import consolidate_history_transfer
+from .runner import Runner
+
+__all__ = ["Agent", "Runner", "chat_summary_to_string", "consolidate_history_transfer", "display_chat_summary", "display_messages", "messages_to_string"]

{lite_agent-0.13.0 → lite_agent-0.14.0}/src/lite_agent/chat_display.py

@@ -10,6 +10,7 @@ import json
 import time
 from collections.abc import Callable
 from datetime import datetime, timedelta, timezone
+from io import StringIO
 
 try:
     from zoneinfo import ZoneInfo
@@ -46,6 +47,7 @@ class DisplayConfig:
     console: Console | None = None
     show_indices: bool = True
     show_timestamps: bool = True
+    show_metadata: bool = True
     max_content_length: int = 1000
     local_timezone: timezone | str | None = None
 
@@ -57,6 +59,7 @@ class MessageContext:
     console: Console
     index_str: str
     timestamp_str: str
+    show_metadata: bool
     max_content_length: int
     truncate_content: Callable[[str, int], str]
 
@@ -441,6 +444,7 @@ def display_messages(
             console=console,
             max_content_length=config.max_content_length,
             show_timestamp=config.show_timestamps,
+            show_metadata=config.show_metadata,
             local_timezone=local_timezone,
         )
 
@@ -452,6 +456,7 @@ def _display_single_message_compact(
     console: Console,
     max_content_length: int = 100,
     show_timestamp: bool = False,
+    show_metadata: bool = True,
     local_timezone: timezone | None = None,
 ) -> None:
     """以列式格式打印单个消息，类似 rich log。"""
@@ -469,7 +474,7 @@ def _display_single_message_compact(
         timestamp = _format_timestamp(message_time, local_timezone=local_timezone)
 
     # 创建列式显示
-    _display_message_in_columns(message, console, index, timestamp, max_content_length, truncate_content)
+    _display_message_in_columns(message, console, index, timestamp, show_metadata=show_metadata, max_content_length=max_content_length, truncate_content=truncate_content)
 
 
 def _display_message_in_columns(
@@ -477,6 +482,8 @@ def _display_message_in_columns(
     console: Console,
     index: int | None,
     timestamp: str | None,
+    *,
+    show_metadata: bool,
     max_content_length: int,
     truncate_content: Callable[[str, int], str],
 ) -> None:
@@ -490,12 +497,9 @@ def _display_message_in_columns(
     if isinstance(message, NewUserMessage):
         _display_user_message_with_columns(message, console, time_str, index_str, max_content_length, truncate_content)
     elif isinstance(message, NewAssistantMessage):
-        _display_assistant_message_with_columns(message, console, time_str, index_str, max_content_length, truncate_content)
+        _display_assistant_message_with_columns(message, console, time_str, index_str, show_metadata=show_metadata, max_content_length=max_content_length, truncate_content=truncate_content)
     elif isinstance(message, NewSystemMessage):
         _display_system_message_with_columns(message, console, time_str, index_str, max_content_length, truncate_content)
-    else:
-        # 处理旧格式消息
-        _display_legacy_message_with_columns(message, console, time_str, index_str, max_content_length, truncate_content)
 
 
 def _display_user_message_with_columns(
@@ -523,27 +527,48 @@ def _display_user_message_with_columns(
     content = " ".join(content_parts)
     content = truncate_content(content, max_content_length)
 
-    # 创建表格来确保对齐
+    # 创建表格来确保对齐，根据配置动态调整列宽
     table = Table.grid(padding=0)
-    table.add_column(width=8, justify="left")  # 时间列
-    table.add_column(width=4, justify="left")  # 序号列
+
+    # 只有在显示时间戳时才添加时间列
+    time_width = 8 if time_str.strip() else 0
+    if time_width > 0:
+        table.add_column(width=time_width, justify="left")  # 时间列
+
+    # 只有在显示序号时才添加序号列
+    index_width = 4 if index_str.strip() else 0
+    if index_width > 0:
+        table.add_column(width=index_width, justify="left")  # 序号列
+
     table.add_column(min_width=0)  # 内容列
 
+    # 辅助函数：根据列数构建行
+    def build_table_row(*content_parts: str) -> tuple[str, ...]:
+        row_parts = []
+        if time_width > 0:
+            row_parts.append(content_parts[0] if len(content_parts) > 0 else "")
+        if index_width > 0:
+            row_parts.append(content_parts[1] if len(content_parts) > 1 else "")
+        row_parts.append(content_parts[-1] if content_parts else "")  # 内容列总是最后一个
+        return tuple(row_parts)
+
     lines = content.split("\n")
     for i, line in enumerate(lines):
         if i == 0:
             # 第一行显示 User: 标签
             table.add_row(
-                f"[dim]{time_str:8}[/dim]",
-                f"[dim]{index_str:4}[/dim]",
-                "[blue]User:[/blue]",
+                *build_table_row(
+                    f"[dim]{time_str:8}[/dim]",
+                    f"[dim]{index_str:4}[/dim]",
+                    "[blue]User:[/blue]",
+                ),
             )
             # 如果有内容，添加内容行
             if line:
-                table.add_row("", "", line)
+                table.add_row(*build_table_row("", "", line))
         else:
             # 续行只在内容列显示
-            table.add_row("", "", line)
+            table.add_row(*build_table_row("", "", line))
 
     console.print(table)
 
@@ -559,24 +584,45 @@ def _display_system_message_with_columns(
     """使用列布局显示系统消息。"""
     content = truncate_content(message.content, max_content_length)
 
-    # 创建表格来确保对齐
+    # 创建表格来确保对齐，根据配置动态调整列宽
     table = Table.grid(padding=0)
-    table.add_column(width=8, justify="left")  # 时间列
-    table.add_column(width=4, justify="left")  # 序号列
+
+    # 只有在显示时间戳时才添加时间列
+    time_width = 8 if time_str.strip() else 0
+    if time_width > 0:
+        table.add_column(width=time_width, justify="left")  # 时间列
+
+    # 只有在显示序号时才添加序号列
+    index_width = 4 if index_str.strip() else 0
+    if index_width > 0:
+        table.add_column(width=index_width, justify="left")  # 序号列
+
     table.add_column(min_width=0)  # 内容列
 
+    # 辅助函数：根据列数构建行
+    def build_table_row(*content_parts: str) -> tuple[str, ...]:
+        row_parts = []
+        if time_width > 0:
+            row_parts.append(content_parts[0] if len(content_parts) > 0 else "")
+        if index_width > 0:
+            row_parts.append(content_parts[1] if len(content_parts) > 1 else "")
+        row_parts.append(content_parts[-1] if content_parts else "")  # 内容列总是最后一个
+        return tuple(row_parts)
+
     lines = content.split("\n")
     for i, line in enumerate(lines):
         if i == 0:
             # 第一行显示完整信息
             table.add_row(
-                f"[dim]{time_str:8}[/dim]",
-                f"[dim]{index_str:4}[/dim]",
-                f"[yellow]System:[/yellow] {line}",
+                *build_table_row(
+                    f"[dim]{time_str:8}[/dim]",
+                    f"[dim]{index_str:4}[/dim]",
+                    f"[yellow]System:[/yellow] {line}",
+                ),
             )
         else:
             # 续行只在内容列显示
-            table.add_row("", "", line)
+            table.add_row(*build_table_row("", "", line))
 
     console.print(table)
 
@@ -586,6 +632,8 @@ def _display_assistant_message_with_columns(
     console: Console,
     time_str: str,
     index_str: str,
+    *,
+    show_metadata: bool,
     max_content_length: int,
     truncate_content: Callable[[str, int], str],
 ) -> None:
@@ -605,7 +653,7 @@ def _display_assistant_message_with_columns(
 
     # 构建元信息
     meta_info = ""
-    if message.meta:
+    if show_metadata and message.meta:
         meta_parts = []
         if message.meta.model is not None:
             meta_parts.append(f"Model:{message.meta.model}")
@@ -620,12 +668,31 @@ def _display_assistant_message_with_columns(
         if meta_parts:
             meta_info = f" [dim]({' | '.join(meta_parts)})[/dim]"
 
-    # 创建表格来确保对齐
+    # 创建表格来确保对齐，根据配置动态调整列宽
     table = Table.grid(padding=0)
-    table.add_column(width=8, justify="left")  # 时间列
-    table.add_column(width=4, justify="left")  # 序号列
+
+    # 只有在显示时间戳时才添加时间列
+    time_width = 8 if time_str.strip() else 0
+    if time_width > 0:
+        table.add_column(width=time_width, justify="left")  # 时间列
+
+    # 只有在显示序号时才添加序号列
+    index_width = 4 if index_str.strip() else 0
+    if index_width > 0:
+        table.add_column(width=index_width, justify="left")  # 序号列
+
     table.add_column(min_width=0)  # 内容列
 
+    # 辅助函数：根据列数构建行
+    def build_table_row(*content_parts: str) -> tuple[str, ...]:
+        row_parts = []
+        if time_width > 0:
+            row_parts.append(content_parts[0] if len(content_parts) > 0 else "")
+        if index_width > 0:
+            row_parts.append(content_parts[1] if len(content_parts) > 1 else "")
+        row_parts.append(content_parts[-1] if content_parts else "")  # 内容列总是最后一个
+        return tuple(row_parts)
+
     # 处理文本内容
     first_row_added = False
     if text_parts:
@@ -636,24 +703,28 @@ def _display_assistant_message_with_columns(
             if i == 0:
                 # 第一行显示 Assistant: 标签
                 table.add_row(
-                    f"[dim]{time_str:8}[/dim]",
-                    f"[dim]{index_str:4}[/dim]",
-                    f"[green]Assistant:[/green]{meta_info}",
+                    *build_table_row(
+                        f"[dim]{time_str:8}[/dim]",
+                        f"[dim]{index_str:4}[/dim]",
+                        f"[green]Assistant:[/green]{meta_info}",
+                    ),
                 )
                 # 如果有内容，添加内容行
                 if line:
-                    table.add_row("", "", line)
+                    table.add_row(*build_table_row("", "", line))
                 first_row_added = True
             else:
                 # 续行只在内容列显示
-                table.add_row("", "", line)
+                table.add_row(*build_table_row("", "", line))
 
     # 如果没有文本内容，只显示助手消息头
     if not first_row_added:
         table.add_row(
-            f"[dim]{time_str:8}[/dim]",
-            f"[dim]{index_str:4}[/dim]",
-            f"[green]Assistant:[/green]{meta_info}",
+            *build_table_row(
+                f"[dim]{time_str:8}[/dim]",
+                f"[dim]{index_str:4}[/dim]",
+                f"[green]Assistant:[/green]{meta_info}",
+            ),
         )
 
     # 添加工具调用
@@ -667,7 +738,7 @@ def _display_assistant_message_with_columns(
                 args_str = f" {tool_call.arguments}"
 
         args_display = truncate_content(args_str, max_content_length - len(tool_call.name) - 10)
-        table.add_row("", "", f"[magenta]Call:[/magenta] {tool_call.name}{args_display}")
+        table.add_row(*build_table_row("", "", f"[magenta]Call:[/magenta] {tool_call.name}{args_display}"))
 
     # 添加工具结果
     for tool_result in tool_results:
@@ -676,10 +747,10 @@ def _display_assistant_message_with_columns(
         if tool_result.execution_time_ms is not None:
             time_info = f" [dim]({tool_result.execution_time_ms}ms)[/dim]"
 
-        table.add_row("", "", f"[cyan]Output:[/cyan]{time_info}")
+        table.add_row(*build_table_row("", "", f"[cyan]Output:[/cyan]{time_info}"))
         lines = output.split("\n")
         for line in lines:
-            table.add_row("", "", line)
+            table.add_row(*build_table_row("", "", line))
 
     console.print(table)
 
@@ -689,6 +760,8 @@ def _display_legacy_message_with_columns(
     console: Console,
     time_str: str,
     index_str: str,
+    *,
+    show_metadata: bool,  # noqa: ARG001
     max_content_length: int,
     truncate_content: Callable[[str, int], str],
 ) -> None:
@@ -735,6 +808,7 @@ def _create_message_context(context_config: dict[str, FlexibleRunnerMessage | Co
     max_content_length = max_content_length_val
     truncate_content = context_config["truncate_content"]
     show_timestamp = context_config.get("show_timestamp", False)
+    show_metadata = bool(context_config.get("show_metadata", True))
     local_timezone = context_config.get("local_timezone")
 
     # 类型检查
@@ -773,6 +847,7 @@ def _create_message_context(context_config: dict[str, FlexibleRunnerMessage | Co
         console=console,
         index_str=index_str,
         timestamp_str=timestamp_str,
+        show_metadata=show_metadata,
         max_content_length=max_content_length,
         truncate_content=truncate_content,  # type: ignore[arg-type]
     )
@@ -1048,3 +1123,103 @@ def _display_new_assistant_message_compact(message: NewAssistantMessage, context
         # Always use indented format for better hierarchy
         context.console.print(f"  [cyan]Output:[/cyan]{time_info}")
         context.console.print(f"  {output}")
+
+
+def messages_to_string(
+    messages: RunnerMessages,
+    *,
+    show_indices: bool = False,
+    show_timestamps: bool = False,
+    show_metadata: bool = False,
+    max_content_length: int = 1000,
+    local_timezone: timezone | str | None = None,
+) -> str:
+    """
+    将消息列表转换为纯文本字符串，默认简洁格式（不显示时间、序号、元数据）。
+
+    Args:
+        messages: 要转换的消息列表
+        show_indices: 是否显示消息序号（默认False）
+        show_timestamps: 是否显示时间戳（默认False）
+        show_metadata: 是否显示元数据（如模型、延迟、token使用等，默认False）
+        max_content_length: 内容最大长度限制（默认1000）
+        local_timezone: 本地时区设置（可选）
+
+    Returns:
+        包含所有消息的纯文本字符串
+    """
+    # 创建一个没有颜色的 Console 来捕获输出
+    string_buffer = StringIO()
+    plain_console = Console(file=string_buffer, force_terminal=False, no_color=True, width=120)
+
+    # 使用配置
+    config = DisplayConfig(
+        console=plain_console,
+        show_indices=show_indices,
+        show_timestamps=show_timestamps,
+        show_metadata=show_metadata,
+        max_content_length=max_content_length,
+        local_timezone=local_timezone,
+    )
+
+    # 调用现有的 display_messages 函数，但输出到字符串缓冲区
+    display_messages(messages, config=config)
+
+    # 获取结果并清理尾随空格
+    result = string_buffer.getvalue()
+    string_buffer.close()
+
+    # 清理每行的尾随空格
+    lines = result.split("\n")
+    cleaned_lines = [line.rstrip() for line in lines]
+    return "\n".join(cleaned_lines)
+
+
+def chat_summary_to_string(messages: RunnerMessages, *, include_performance: bool = False) -> str:
+    """
+    将聊天摘要转换为纯文本字符串，默认只显示基本统计信息。
+
+    Args:
+        messages: 要分析的消息列表
+        include_performance: 是否包含性能统计信息（默认False）
+
+    Returns:
+        包含聊天摘要的纯文本字符串
+    """
+    string_buffer = StringIO()
+    plain_console = Console(file=string_buffer, force_terminal=False, no_color=True, width=120)
+
+    if include_performance:
+        # 调用现有的 display_chat_summary 函数，包含所有信息
+        display_chat_summary(messages, console=plain_console)
+    else:
+        # 只显示基本的消息统计信息
+        _display_basic_message_stats(messages, plain_console)
+
+    # 获取结果并清理
+    result = string_buffer.getvalue()
+    string_buffer.close()
+
+    return result
+
+
+def _display_basic_message_stats(messages: RunnerMessages, console: Console) -> None:
+    """显示基本的消息统计信息，不包含性能数据。"""
+    message_counts, _ = _analyze_messages(messages)
+
+    # 创建简化的统计表格
+    table = Table(title="Message Summary", show_header=True, header_style="bold blue")
+    table.add_column("Message Type", justify="left")
+    table.add_column("Count", justify="right")
+
+    # 添加消息类型统计
+    for msg_type, count in message_counts.items():
+        if msg_type != "Total":  # 跳过总计，单独处理
+            table.add_row(msg_type, str(count))
+
+    # 添加总计行
+    if "Total" in message_counts:
+        table.add_row("", "")  # 空行分隔
+        table.add_row("[bold]Total[/bold]", f"[bold]{message_counts['Total']}[/bold]")
+
+    console.print(table)

{lite_agent-0.13.0 → lite_agent-0.14.0}/src/lite_agent/context.py

@@ -2,10 +2,13 @@
 
 from __future__ import annotations
 
-from typing import Any, Generic, TypeVar
+from typing import TYPE_CHECKING, Generic, TypeVar
 
 from pydantic import BaseModel
 
+if TYPE_CHECKING:
+    from lite_agent.types import NewMessage
+
 T = TypeVar("T")
 
 
@@ -33,5 +36,5 @@ class HistoryContext(BaseModel, Generic[T]):
         ...     return f"用户 {user_id} 有 {len(messages)} 条消息"
     """
 
-    history_messages: list[Any]
+    history_messages: list[NewMessage]
     data: T | None = None