lite-agent 0.8.0__py3-none-any.whl → 0.9.0__py3-none-any.whl

lite_agent/agent.py

@@ -429,39 +429,76 @@ class Agent:
             role = message_dict.get("role")
 
             if role == "assistant":
-                # Extract tool_calls from content if present
+                # For NewAssistantMessage, extract directly from the message object
                 tool_calls = []
-                content = message_dict.get("content", [])
-
-                # Handle both string and array content
-                if isinstance(content, list):
-                    # Extract tool_calls from content array and filter out non-text content
-                    filtered_content = []
-                    for item in content:
-                        if isinstance(item, dict):
-                            if item.get("type") == "tool_call":
-                                tool_call = {
-                                    "id": item.get("call_id", ""),
-                                    "type": "function",
-                                    "function": {
-                                        "name": item.get("name", ""),
-                                        "arguments": item.get("arguments", "{}"),
-                                    },
-                                    "index": len(tool_calls),
-                                }
-                                tool_calls.append(tool_call)
-                            elif item.get("type") == "text":
-                                filtered_content.append(item)
-                            # Skip tool_call_result - they should be handled by separate function_call_output messages
-
-                    # Update content to only include text items
-                    if filtered_content:
-                        message_dict = message_dict.copy()
-                        message_dict["content"] = filtered_content
-                    elif tool_calls:
-                        # If we have tool_calls but no text content, set content to None per OpenAI API spec
-                        message_dict = message_dict.copy()
-                        message_dict["content"] = None
+                tool_results = []
+
+                if isinstance(message, NewAssistantMessage):
+                    # Process content directly from NewAssistantMessage
+                    for item in message.content:
+                        if item.type == "tool_call":
+                            tool_call = {
+                                "id": item.call_id,
+                                "type": "function",
+                                "function": {
+                                    "name": item.name,
+                                    "arguments": item.arguments,
+                                },
+                                "index": len(tool_calls),
+                            }
+                            tool_calls.append(tool_call)
+                        elif item.type == "tool_call_result":
+                            # Collect tool call results to be added as separate tool messages
+                            tool_results.append({
+                                "call_id": item.call_id,
+                                "output": item.output,
+                            })
+
+                    # Create assistant message with only text content and tool calls
+                    text_content = " ".join([item.text for item in message.content if item.type == "text"])
+                    message_dict = {
+                        "role": "assistant",
+                        "content": text_content if text_content else None,
+                    }
+                    if tool_calls:
+                        message_dict["tool_calls"] = tool_calls
+                else:
+                    # Legacy handling for dict messages
+                    content = message_dict.get("content", [])
+                    # Handle both string and array content
+                    if isinstance(content, list):
+                        # Extract tool_calls and tool_call_results from content array and filter out non-text content
+                        filtered_content = []
+                        for item in content:
+                            if isinstance(item, dict):
+                                if item.get("type") == "tool_call":
+                                    tool_call = {
+                                        "id": item.get("call_id", ""),
+                                        "type": "function",
+                                        "function": {
+                                            "name": item.get("name", ""),
+                                            "arguments": item.get("arguments", "{}"),
+                                        },
+                                        "index": len(tool_calls),
+                                    }
+                                    tool_calls.append(tool_call)
+                                elif item.get("type") == "tool_call_result":
+                                    # Collect tool call results to be added as separate tool messages
+                                    tool_results.append({
+                                        "call_id": item.get("call_id", ""),
+                                        "output": item.get("output", ""),
+                                    })
+                                elif item.get("type") == "text":
+                                    filtered_content.append(item)
+
+                        # Update content to only include text items
+                        if filtered_content:
+                            message_dict = message_dict.copy()
+                            message_dict["content"] = filtered_content
+                        elif tool_calls:
+                            # If we have tool_calls but no text content, set content to None per OpenAI API spec
+                            message_dict = message_dict.copy()
+                            message_dict["content"] = None
 
                 # Look ahead for function_call messages (legacy support)
                 j = i + 1
@@ -484,19 +521,33 @@ class Agent:
                     else:
                         break
 
-                # Create assistant message with tool_calls if any
-                assistant_msg = message_dict.copy()
-                if tool_calls:
-                    assistant_msg["tool_calls"] = tool_calls  # type: ignore
+                # For legacy dict messages, create assistant message with tool_calls if any
+                if not isinstance(message, NewAssistantMessage):
+                    assistant_msg = message_dict.copy()
+                    if tool_calls:
+                        assistant_msg["tool_calls"] = tool_calls  # type: ignore
+
+                    # Convert content format for OpenAI API compatibility
+                    content = assistant_msg.get("content", [])
+                    if isinstance(content, list):
+                        # Extract text content and convert to string using list comprehension
+                        text_parts = [item.get("text", "") for item in content if isinstance(item, dict) and item.get("type") == "text"]
+                        assistant_msg["content"] = " ".join(text_parts) if text_parts else None
 
-                # Convert content format for OpenAI API compatibility
-                content = assistant_msg.get("content", [])
-                if isinstance(content, list):
-                    # Extract text content and convert to string using list comprehension
-                    text_parts = [item.get("text", "") for item in content if isinstance(item, dict) and item.get("type") == "text"]
-                    assistant_msg["content"] = " ".join(text_parts) if text_parts else None
+                    message_dict = assistant_msg
+
+                converted_messages.append(message_dict)
+
+                # Add tool messages for any tool_call_results found in the assistant message
+                converted_messages.extend([
+                    {
+                        "role": "tool",
+                        "tool_call_id": tool_result["call_id"],
+                        "content": tool_result["output"],
+                    }
+                    for tool_result in tool_results
+                ])
 
-                converted_messages.append(assistant_msg)
                 i = j  # Skip the function_call messages we've processed
 
             elif message_type == "function_call_output":

lite_agent/chat_display.py

@@ -454,7 +454,7 @@ def _display_single_message_compact(
     show_timestamp: bool = False,
     local_timezone: timezone | None = None,
 ) -> None:
-    """以紧凑格式打印单个消息。"""
+    """以列式格式打印单个消息，类似 rich log。"""
 
     def truncate_content(content: str, max_length: int) -> str:
         """截断内容并添加省略号。"""
@@ -462,20 +462,259 @@ def _display_single_message_compact(
             return content
         return content[: max_length - 3] + "..."
 
-    # 创建消息上下文
-    context_config = {
-        "console": console,
-        "index": index,
-        "message": message,
-        "max_content_length": max_content_length,
-        "truncate_content": truncate_content,
-        "show_timestamp": show_timestamp,
-        "local_timezone": local_timezone,
-    }
-    context = _create_message_context(context_config)
+    # 获取时间戳
+    timestamp = None
+    if show_timestamp:
+        message_time = _extract_message_time(message)
+        timestamp = _format_timestamp(message_time, local_timezone=local_timezone)
 
-    # 根据消息类型分发处理
-    _dispatch_message_display(message, context)
+    # 创建列式显示
+    _display_message_in_columns(message, console, index, timestamp, max_content_length, truncate_content)
+
+
+def _display_message_in_columns(
+    message: FlexibleRunnerMessage,
+    console: Console,
+    index: int | None,
+    timestamp: str | None,
+    max_content_length: int,
+    truncate_content: Callable[[str, int], str],
+) -> None:
+    """以列式格式显示消息，类似 rich log。"""
+
+    # 构建时间和索引列
+    time_str = timestamp or ""
+    index_str = f"#{index:2d}" if index is not None else ""
+
+    # 根据消息类型处理内容
+    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)
+    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(
+    message: NewUserMessage,
+    console: Console,
+    time_str: str,
+    index_str: str,
+    max_content_length: int,
+    truncate_content: Callable[[str, int], str],
+) -> None:
+    """使用列布局显示用户消息。"""
+    content_parts = []
+    for item in message.content:
+        if item.type == "text":
+            content_parts.append(item.text)
+        elif item.type == "image":
+            if item.image_url:
+                content_parts.append(f"[Image: {item.image_url}]")
+            elif item.file_id:
+                content_parts.append(f"[Image: {item.file_id}]")
+        elif item.type == "file":
+            file_name = item.file_name or item.file_id
+            content_parts.append(f"[File: {file_name}]")
+
+    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")  # 序号列
+    table.add_column(min_width=0)  # 内容列
+
+    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"[blue]User:[/blue] {line}",
+            )
+        else:
+            # 续行只在内容列显示
+            table.add_row("", "", line)
+
+    console.print(table)
+
+
+def _display_system_message_with_columns(
+    message: NewSystemMessage,
+    console: Console,
+    time_str: str,
+    index_str: str,
+    max_content_length: int,
+    truncate_content: Callable[[str, int], str],
+) -> None:
+    """使用列布局显示系统消息。"""
+    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")  # 序号列
+    table.add_column(min_width=0)  # 内容列
+
+    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}",
+            )
+        else:
+            # 续行只在内容列显示
+            table.add_row("", "", line)
+
+    console.print(table)
+
+
+def _display_assistant_message_with_columns(
+    message: NewAssistantMessage,
+    console: Console,
+    time_str: str,
+    index_str: str,
+    max_content_length: int,
+    truncate_content: Callable[[str, int], str],
+) -> None:
+    """使用列布局显示助手消息。"""
+    # 提取内容
+    text_parts = []
+    tool_calls = []
+    tool_results = []
+
+    for item in message.content:
+        if item.type == "text":
+            text_parts.append(item.text)
+        elif item.type == "tool_call":
+            tool_calls.append(item)
+        elif item.type == "tool_call_result":
+            tool_results.append(item)
+
+    # 构建元信息
+    meta_info = ""
+    if message.meta:
+        meta_parts = []
+        if message.meta.model is not None:
+            meta_parts.append(f"Model:{message.meta.model}")
+        if message.meta.latency_ms is not None:
+            meta_parts.append(f"Latency:{message.meta.latency_ms}ms")
+        if message.meta.total_time_ms is not None:
+            meta_parts.append(f"Output:{message.meta.total_time_ms}ms")
+        if message.meta.usage and message.meta.usage.input_tokens is not None and message.meta.usage.output_tokens is not None:
+            total_tokens = message.meta.usage.input_tokens + message.meta.usage.output_tokens
+            meta_parts.append(f"Tokens:↑{message.meta.usage.input_tokens}↓{message.meta.usage.output_tokens}={total_tokens}")
+
+        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")  # 序号列
+    table.add_column(min_width=0)  # 内容列
+
+    # 处理文本内容
+    first_row_added = False
+    if text_parts:
+        content = " ".join(text_parts)
+        content = truncate_content(content, max_content_length)
+        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"[green]Assistant:[/green]{meta_info} {line}",
+                )
+                first_row_added = True
+            else:
+                # 续行只在内容列显示
+                table.add_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}",
+        )
+
+    # 添加工具调用
+    for tool_call in tool_calls:
+        args_str = ""
+        if tool_call.arguments:
+            try:
+                parsed_args = json.loads(tool_call.arguments) if isinstance(tool_call.arguments, str) else tool_call.arguments
+                args_str = f" {parsed_args}"
+            except (json.JSONDecodeError, TypeError):
+                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}")
+
+    # 添加工具结果
+    for tool_result in tool_results:
+        output = truncate_content(str(tool_result.output), max_content_length)
+        time_info = ""
+        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}")
+        lines = output.split("\n")
+        for line in lines:
+            table.add_row("", "", line)
+
+    console.print(table)
+
+
+def _display_legacy_message_with_columns(
+    message: FlexibleRunnerMessage,
+    console: Console,
+    time_str: str,
+    index_str: str,
+    max_content_length: int,
+    truncate_content: Callable[[str, int], str],
+) -> None:
+    """使用列布局显示旧格式消息。"""
+    # 这里可以处理旧格式消息，暂时简单显示
+    try:
+        content = str(message.model_dump()) if hasattr(message, "model_dump") else str(message)  # type: ignore[attr-defined]
+    except Exception:
+        content = str(message)
+
+    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")  # 序号列
+    table.add_column(min_width=0)  # 内容列
+
+    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"[red]Legacy:[/red] {line}",
+            )
+        else:
+            # 续行只在内容列显示
+            table.add_row("", "", line)
+
+    console.print(table)
 
 
 def _create_message_context(context_config: dict[str, FlexibleRunnerMessage | Console | int | bool | timezone | Callable[[str, int], str] | None]) -> MessageContext:
@@ -581,6 +820,8 @@ def _display_assistant_message_compact_v2(message: AgentAssistantMessage, contex
     meta_info = ""
     if message.meta:
         meta_parts = []
+        if message.meta.model is not None:
+            meta_parts.append(f"Model:{message.meta.model}")
         if message.meta.latency_ms is not None:
             meta_parts.append(f"Latency:{message.meta.latency_ms}ms")
         if message.meta.output_time_ms is not None:
@@ -656,7 +897,11 @@ def _display_dict_function_call_compact(message: dict, context: MessageContext)
 def _display_dict_function_output_compact(message: dict, context: MessageContext) -> None:
     """显示字典类型的函数输出消息。"""
     output = context.truncate_content(str(message.get("output", "")), context.max_content_length)
-    context.console.print(f"{context.timestamp_str}{context.index_str}[cyan]Output:[/cyan]")
+    # Add execution time if available
+    time_info = ""
+    if message.get("execution_time_ms") is not None:
+        time_info = f" [dim]({message['execution_time_ms']}ms)[/dim]"
+    context.console.print(f"{context.timestamp_str}{context.index_str}[cyan]Output:[/cyan]{time_info}")
     context.console.print(f"{output}")
 
 
@@ -676,6 +921,8 @@ def _display_dict_assistant_compact(message: dict, context: MessageContext) -> N
     meta = message.get("meta")
     if meta and isinstance(meta, dict):
         meta_parts = []
+        if meta.get("model") is not None:
+            meta_parts.append(f"Model:{meta['model']}")
         if meta.get("latency_ms") is not None:
             meta_parts.append(f"Latency:{meta['latency_ms']}ms")
         if meta.get("output_time_ms") is not None:
@@ -743,30 +990,34 @@ def _display_new_assistant_message_compact(message: NewAssistantMessage, context
         elif item.type == "tool_call_result":
             tool_results.append(item)
 
-    # Display text content first if available
-    if text_parts:
-        content = " ".join(text_parts)
-        content = context.truncate_content(content, context.max_content_length)
-
-        # Add meta data information (使用英文标签)
-        meta_info = ""
-        if message.meta:
-            meta_parts = []
-            if message.meta.latency_ms is not None:
-                meta_parts.append(f"Latency:{message.meta.latency_ms}ms")
-            if message.meta.total_time_ms is not None:
-                meta_parts.append(f"Output:{message.meta.total_time_ms}ms")
-            if message.meta.usage and message.meta.usage.input_tokens is not None and message.meta.usage.output_tokens is not None:
-                total_tokens = message.meta.usage.input_tokens + message.meta.usage.output_tokens
-                meta_parts.append(f"Tokens:↑{message.meta.usage.input_tokens}↓{message.meta.usage.output_tokens}={total_tokens}")
-
-            if meta_parts:
-                meta_info = f" [dim]({' | '.join(meta_parts)})[/dim]"
+    # Add meta data information (使用英文标签)
+    meta_info = ""
+    if message.meta:
+        meta_parts = []
+        if message.meta.model is not None:
+            meta_parts.append(f"Model:{message.meta.model}")
+        if message.meta.latency_ms is not None:
+            meta_parts.append(f"Latency:{message.meta.latency_ms}ms")
+        if message.meta.total_time_ms is not None:
+            meta_parts.append(f"Output:{message.meta.total_time_ms}ms")
+        if message.meta.usage and message.meta.usage.input_tokens is not None and message.meta.usage.output_tokens is not None:
+            total_tokens = message.meta.usage.input_tokens + message.meta.usage.output_tokens
+            meta_parts.append(f"Tokens:↑{message.meta.usage.input_tokens}↓{message.meta.usage.output_tokens}={total_tokens}")
 
+        if meta_parts:
+            meta_info = f" [dim]({' | '.join(meta_parts)})[/dim]"
+
+    # Always show Assistant header if there's any content (text, tool calls, or results)
+    if text_parts or tool_calls or tool_results:
         context.console.print(f"{context.timestamp_str}{context.index_str}[green]Assistant:[/green]{meta_info}")
-        context.console.print(f"{content}")
 
-    # Display tool calls
+        # Display text content if available
+        if text_parts:
+            content = " ".join(text_parts)
+            content = context.truncate_content(content, context.max_content_length)
+            context.console.print(f"{content}")
+
+    # Display tool calls with proper indentation
     for tool_call in tool_calls:
         args_str = ""
         if tool_call.arguments:
@@ -777,11 +1028,17 @@ def _display_new_assistant_message_compact(message: NewAssistantMessage, context
                 args_str = f" {tool_call.arguments}"
 
         args_display = context.truncate_content(args_str, context.max_content_length - len(tool_call.name) - 10)
-        context.console.print(f"{context.timestamp_str}{context.index_str}[magenta]Call:[/magenta]")
-        context.console.print(f"{tool_call.name}{args_display}")
+        # Always use indented format for better hierarchy
+        context.console.print(f"  [magenta]Call:[/magenta] {tool_call.name}{args_display}")
 
-    # Display tool results
+    # Display tool results with proper indentation
     for tool_result in tool_results:
         output = context.truncate_content(str(tool_result.output), context.max_content_length)
-        context.console.print(f"{context.timestamp_str}{context.index_str}[cyan]Output:[/cyan]")
-        context.console.print(f"{output}")
+        # Add execution time if available
+        time_info = ""
+        if tool_result.execution_time_ms is not None:
+            time_info = f" [dim]({tool_result.execution_time_ms}ms)[/dim]"
+
+        # Always use indented format for better hierarchy
+        context.console.print(f"  [cyan]Output:[/cyan]{time_info}")
+        context.console.print(f"  {output}")

lite_agent/message_transfers.py

@@ -5,6 +5,8 @@ This module provides common message transfer functions that can be used
 with agents to preprocess messages before sending them to the API.
 """
 
+import json
+
 from lite_agent.types import NewUserMessage, RunnerMessages, UserTextContent
 
 
@@ -67,8 +69,25 @@ def _process_message_to_xml(message: dict | object) -> list[str]:
 
         # Handle new message format where content is a list
         if isinstance(content, list):
-            # Extract text from content items
-            text_parts = [item.text for item in content if (hasattr(item, "type") and item.type == "text") or hasattr(item, "text")]
+            # Process each content item
+            text_parts = []
+            for item in content:
+                if hasattr(item, "type"):
+                    if item.type == "text":
+                        text_parts.append(item.text)
+                    elif item.type == "tool_call":
+                        # Handle tool call content
+                        arguments = item.arguments
+                        if isinstance(arguments, dict):
+                            arguments = json.dumps(arguments, ensure_ascii=False)
+                        xml_lines.append(f"  <function_call name='{item.name}' arguments='{arguments}' />")
+                    elif item.type == "tool_call_result":
+                        # Handle tool call result content
+                        xml_lines.append(f"  <function_result call_id='{item.call_id}'>{item.output}</function_result>")
+                elif hasattr(item, "text"):
+                    text_parts.append(item.text)
+
+            # Add text content as message if any
             content_text = " ".join(text_parts)
             if content_text:
                 xml_lines.append(f"  <message role='{role}'>{content_text}</message>")

lite_agent/processors/response_event_processor.py

@@ -119,8 +119,10 @@ class ResponseEventProcessor:
                     # Extract model information from event
                     model_name = getattr(event, "model", None)
                     # Debug: check if event has model info in different location
-                    if hasattr(event, "response") and hasattr(event.response, "model"):
-                        model_name = getattr(event.response, "model", None)
+                    if hasattr(event, "response"):
+                        response = getattr(event, "response", None)
+                        if response and hasattr(response, "model"):
+                            model_name = getattr(response, "model", None)
                     # Create usage information
                     usage = MessageUsage(
                         input_tokens=self._usage_data.get("input_tokens"),

lite_agent/runner.py

@@ -28,7 +28,7 @@ from lite_agent.types import (
     UserInput,
     UserTextContent,
 )
-from lite_agent.types.events import AssistantMessageEvent, FunctionCallOutputEvent
+from lite_agent.types.events import AssistantMessageEvent, FunctionCallOutputEvent, TimingEvent
 from lite_agent.utils.message_builder import MessageBuilder
 
 
@@ -43,9 +43,14 @@ class Runner:
 
     def _start_assistant_message(self, content: str = "", meta: AssistantMessageMeta | None = None) -> None:
         """Start a new assistant message."""
+        # Create meta with model information if not provided
+        if meta is None:
+            meta = AssistantMessageMeta()
+            if hasattr(self.agent.client, "model"):
+                meta.model = self.agent.client.model
         self._current_assistant_message = NewAssistantMessage(
             content=[AssistantTextContent(text=content)],
-            meta=meta or AssistantMessageMeta(),
+            meta=meta,
         )
 
     def _ensure_current_assistant_message(self) -> NewAssistantMessage:
@@ -90,9 +95,16 @@ class Runner:
             # Add to existing assistant message
             last_message = cast("NewAssistantMessage", self.messages[-1])
             last_message.content.append(result)
+            # Ensure model information is set if not already present
+            if last_message.meta.model is None and hasattr(self.agent.client, "model"):
+                last_message.meta.model = self.agent.client.model
         else:
             # Create new assistant message with just the tool result
-            assistant_message = NewAssistantMessage(content=[result])
+            # Include model information if available
+            meta = AssistantMessageMeta()
+            if hasattr(self.agent.client, "model"):
+                meta.model = self.agent.client.model
+            assistant_message = NewAssistantMessage(content=[result], meta=meta)
             self.messages.append(assistant_message)
 
         # For completion API compatibility, create a separate assistant message
@@ -309,10 +321,14 @@ class Runner:
                 case _:
                     msg = f"Unknown API type: {self.api}"
                     raise ValueError(msg)
-            logger.debug(f"Received response from agent: {type(resp)}")
+            logger.debug("Received response stream from agent, processing chunks...")
             async for chunk in resp:
+                # Only log important chunk types to reduce noise
+                if chunk.type not in ["response_raw", "content_delta"]:
+                    logger.debug(f"Processing chunk: {chunk.type}")
                 match chunk.type:
                     case "assistant_message":
+                        logger.debug(f"Assistant message chunk: {len(chunk.message.content) if chunk.message.content else 0} content items")
                         # Start or update assistant message in new format
                         # If we already have a current assistant message, just update its metadata
                         if self._current_assistant_message is not None:
@@ -348,6 +364,7 @@ class Runner:
                         if chunk.type in includes:
                             yield chunk
                     case "function_call":
+                        logger.debug(f"Function call: {chunk.name}({chunk.arguments or '{}'})")
                         # Add tool call to current assistant message
                         # Keep arguments as string for compatibility with funcall library
                         tool_call = AssistantToolCall(
@@ -360,6 +377,7 @@ class Runner:
                         if chunk.type in includes:
                             yield chunk
                     case "usage":
+                        logger.debug(f"Usage: {chunk.usage.input_tokens} input, {chunk.usage.output_tokens} output tokens")
                         # Update the current or last assistant message with usage data and output_time_ms
                         usage_time = datetime.now(timezone.utc)
 
@@ -402,6 +420,19 @@ class Runner:
                         # Always yield usage chunk if it's in includes
                         if chunk.type in includes:
                             yield chunk
+                    case "timing":
+                        # Update timing information in current assistant message
+                        if self._current_assistant_message is not None:
+                            self._current_assistant_message.meta.latency_ms = chunk.timing.latency_ms
+                            self._current_assistant_message.meta.total_time_ms = chunk.timing.output_time_ms
+                        # Also try to update the last assistant message if no current message
+                        elif self.messages and isinstance(self.messages[-1], NewAssistantMessage):
+                            last_message = cast("NewAssistantMessage", self.messages[-1])
+                            last_message.meta.latency_ms = chunk.timing.latency_ms
+                            last_message.meta.total_time_ms = chunk.timing.output_time_ms
+                        # Always yield timing chunk if it's in includes
+                        if chunk.type in includes:
+                            yield chunk
                     case _ if chunk.type in includes:
                         yield chunk
 
@@ -832,8 +863,6 @@ class Runner:
             logger.info("Transferring conversation from %s to %s", self.agent.name, target_agent_name)
             self.agent = target_agent
 
-            return tool_call.id, output
-
         except Exception as e:
             logger.exception("Failed to execute transfer_to_agent tool call")
             output = f"Transfer failed: {e!s}"
@@ -843,6 +872,8 @@ class Runner:
                 output=output,
             )
             return tool_call.id, output
+        else:
+            return tool_call.id, output
 
     async def _handle_parent_transfer(self, tool_call: ToolCall) -> tuple[str, str]:
         """Handle parent transfer when transfer_to_parent tool is called.
@@ -883,8 +914,6 @@ class Runner:
             logger.info("Transferring conversation from %s back to parent %s", self.agent.name, self.agent.parent.name)
             self.agent = self.agent.parent
 
-            return tool_call.id, output
-
         except Exception as e:
             logger.exception("Failed to execute transfer_to_parent tool call")
             output = f"Transfer to parent failed: {e!s}"
@@ -894,3 +923,5 @@ class Runner:
                 output=output,
             )
             return tool_call.id, output
+        else:
+            return tool_call.id, output

lite_agent/utils/message_builder.py

@@ -180,7 +180,9 @@ class MessageBuilder:
                                     execution_time_ms=item.get("execution_time_ms"),
                                 ),
                             )
-                        # Add more content types as needed
+                        else:
+                            # Unknown dict type - convert to text
+                            assistant_content_items.append(AssistantTextContent(text=str(item)))
                     else:
                         # Fallback for unknown item format
                         assistant_content_items.append(AssistantTextContent(text=str(item)))

{lite_agent-0.8.0.dist-info → lite_agent-0.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lite-agent
-Version: 0.8.0
+Version: 0.9.0
 Summary: A lightweight, extensible framework for building AI agent.
 Author-email: Jianqi Pan <jannchie@gmail.com>
 License: MIT
@@ -18,7 +18,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.10
 Requires-Dist: aiofiles>=24.1.0
-Requires-Dist: funcall>=0.10.0
+Requires-Dist: funcall>=0.11.0
 Requires-Dist: openai<=1.99.5
 Requires-Dist: prompt-toolkit>=3.0.51
 Requires-Dist: rich>=14.0.0

{lite_agent-0.8.0.dist-info → lite_agent-0.9.0.dist-info}/RECORD

@@ -1,15 +1,15 @@
 lite_agent/__init__.py,sha256=Swuefee0etSiaDnn30K2hBNV9UI3hIValW3A-pRE7e0,338
-lite_agent/agent.py,sha256=XEBgUrR9yqSnqipA70hGVjkrEFy29pl-7lcABL_cDE4,29345
-lite_agent/chat_display.py,sha256=cb8tNCqSoTqyyDc8HXQhEUV9g53FelFMMZ9Pt5WvTn4,31131
+lite_agent/agent.py,sha256=Jls9fwGgMLpFunT8Tr3Cp8P9AyeuvrHklWyFD3mhvtM,32061
+lite_agent/chat_display.py,sha256=IG2oM3FtpcfcduG7b6yt2mLFpG-Z6zHxXUN_J1XNVp8,40563
 lite_agent/client.py,sha256=PTsic12TVYklUKJzb9gk8-FWl2yhNkSYRS_Cs5dosEU,8677
 lite_agent/constants.py,sha256=_xIDdQwaJrWk8N_62o-KYEo3jj1waPJ0ZOd3hHybKNo,718
 lite_agent/loggers.py,sha256=XkNkdqwD_nQGfhQJ-bBWT7koci_mMkNw3aBpyMhOICw,57
-lite_agent/message_transfers.py,sha256=wdr9ABr_V4TfXAFxO1RBe2uYIj0sFo6xc77SOcGbvVk,4504
+lite_agent/message_transfers.py,sha256=N9ViK7Gxqqa1sd3V_hkNuQ9fUipg7M95l-sVBBG2Id4,5357
 lite_agent/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lite_agent/runner.py,sha256=fHahn060SE5aiEPHFNCLvlyKANpeVO4FgeJ-BylfxHw,42872
+lite_agent/runner.py,sha256=iv4xqRJKH9PXyg7goQIBd6IfNODUo5GfjBdlZdxv3kw,45150
 lite_agent/processors/__init__.py,sha256=ybpAzpMBIE9v5I24wIBZRXeaOaPNTmoKH13aofgNI6Q,234
 lite_agent/processors/completion_event_processor.py,sha256=zoWvs8dfrIkCSITGtS-4Hpve3WFCA0UUsMvYifL2fw0,13010
-lite_agent/processors/response_event_processor.py,sha256=-Q5USpv2ZNp4zoHfE9QpwHR2yX3SRkJFjaiXfvchVE8,8417
+lite_agent/processors/response_event_processor.py,sha256=Jr3cj1ItJ8aq9UBhEEjDwWDnPNOZ2ZXjWJ3-g4ghkhM,8514
 lite_agent/response_handlers/__init__.py,sha256=za1pV3DwYhiX4lqo4TafpKlZSBGz81TViWsYebo6Qo4,386
 lite_agent/response_handlers/base.py,sha256=rSs3ImmInBG7aaGv47pYnfHblEs3DIKkT58y7nCUgOc,1682
 lite_agent/response_handlers/completion.py,sha256=e14etxhunRo-T68HgWOE-9JcmMXi5TPvPzyVwE8H7_s,3230
@@ -24,8 +24,8 @@ lite_agent/types/events.py,sha256=mFMqV55WWJbPDyb_P61nd3qMLpEnwZgVY6NTKFkINkg,23
 lite_agent/types/messages.py,sha256=QjWL2McEWckVy7AIogl2HQkUy-XXdNAAcB0oCNnojbg,9922
 lite_agent/types/tool_calls.py,sha256=Xnut8-2-Ld9vgA2GKJY6BbFlBaAv_n4W7vo7Jx21A-E,260
 lite_agent/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lite_agent/utils/message_builder.py,sha256=3O80MVCAxifDv1BVh4YNg-7XA4udR8m-z8fIcTxfmh8,8008
+lite_agent/utils/message_builder.py,sha256=L15e4-qSlCYUNNtJ1zB-EPviuoxH9bTEVNlkvaH0q3U,8142
 lite_agent/utils/metrics.py,sha256=RzOEhCWxbLmmIEkzaxOJ6tAdthI8dv2Foc98Lq8afOQ,1915
-lite_agent-0.8.0.dist-info/METADATA,sha256=K_kN9VfzXcHRmHk4KEeYi3G9CZB4kRU7arZ-EcThmK0,3486
-lite_agent-0.8.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lite_agent-0.8.0.dist-info/RECORD,,
+lite_agent-0.9.0.dist-info/METADATA,sha256=e1eaVu37mmGYrHI_KofiLTOHCerY6fXY5BPZoIXbTO4,3486
+lite_agent-0.9.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lite_agent-0.9.0.dist-info/RECORD,,