lite-agent 0.3.0__tar.gz → 0.4.1__tar.gz

lite_agent-0.4.1/.claude/settings.local.json

@@ -0,0 +1,21 @@
+{
+  "$schema": "https://json.schemastore.org/claude-code-settings.json",
+  "permissions": {
+    "allow": [
+      "Bash(pytest:*)",
+      "mcp__context7__resolve-library-id",
+      "mcp__context7__get-library-docs",
+      "Bash(ruff check:*)",
+      "Bash(python:*)",
+      "Bash(ruff format:*)",
+      "Bash(rm:*)",
+      "Bash(grep:*)",
+      "mcp__ide__getDiagnostics",
+      "Bash(uv run pytest:*)",
+      "Bash(timeout:*)",
+      "Bash(pyright:*)",
+      "Bash(rg:*)"
+    ],
+    "deny": []
+  }
+}

{lite_agent-0.3.0 → lite_agent-0.4.1}/.vscode/launch.json

@@ -8,7 +8,7 @@
             "name": "Python main",
             "type": "debugpy",
             "request": "launch",
-            "program": "${workspaceFolder}/src/easy_agent/__main__.py",
+            "program": "${file}",
             "console": "integratedTerminal",
             "justMyCode": false
         }

{lite_agent-0.3.0 → lite_agent-0.4.1}/CHANGELOG.md

@@ -1,3 +1,69 @@
+## v0.4.1
+
+[v0.4.0...v0.4.1](https://github.com/Jannchie/lite-agent/compare/v0.4.0...v0.4.1)
+
+### :art: Refactors
+
+- **runner**: replace if-elif with match-case for chunk handling - By [Jannchie](mailto:jannchie@gmail.com) in [e9a8464](https://github.com/Jannchie/lite-agent/commit/e9a8464)
+
+## v0.4.0
+
+[v0.3.0...v0.4.0](https://github.com/Jannchie/lite-agent/compare/v0.3.0...v0.4.0)
+
+### :rocket: Breaking Changes
+
+- **agent**: unify message handling using new messages format - By [Jannchie](mailto:jannchie@gmail.com) in [f31769a](https://github.com/Jannchie/lite-agent/commit/f31769a)
+- **chat-display**: remove display_chat_history usage and exports - By [Jannchie](mailto:jannchie@gmail.com) in [fca6ff7](https://github.com/Jannchie/lite-agent/commit/fca6ff7)
+- **chat-display**: rename rich_helpers to chat_display and update related imports - By [Jannchie](mailto:panjianqi@preferred.jp) in [8bbd83b](https://github.com/Jannchie/lite-agent/commit/8bbd83b)
+- **chunks**: rename tool_call and tool_call_result to function_call and function_call_output - By [Jannchie](mailto:jannchie@gmail.com) in [2821478](https://github.com/Jannchie/lite-agent/commit/2821478)
+- **function-call**: rename function_call_id to call_id - By [Jannchie](mailto:jannchie@gmail.com) in [d415fce](https://github.com/Jannchie/lite-agent/commit/d415fce)
+- **messages**: remove legacy function call messages and migrate to new assistant message structure - By [Jannchie](mailto:jannchie@gmail.com) in [c632629](https://github.com/Jannchie/lite-agent/commit/c632629)
+- **messages**: remove conversion helpers and fully switch to new message format - By [Jannchie](mailto:jannchie@gmail.com) in [75e47bf](https://github.com/Jannchie/lite-agent/commit/75e47bf)
+- **messages**: introduce new structured message types and unified message format - By [Jannchie](mailto:jannchie@gmail.com) in [cb8c091](https://github.com/Jannchie/lite-agent/commit/cb8c091)
+- **processors**: rename stream_chunk_processor to completion_event_processor - By [Jannchie](mailto:jannchie@gmail.com) in [80dedb9](https://github.com/Jannchie/lite-agent/commit/80dedb9)
+- **runner**: remove final_message type and streamline tool call handling - By [Jannchie](mailto:jannchie@gmail.com) in [ba91c29](https://github.com/Jannchie/lite-agent/commit/ba91c29)
+- **types**: rename chunk types to event types and update references - By [Jannchie](mailto:jannchie@gmail.com) in [d1354b3](https://github.com/Jannchie/lite-agent/commit/d1354b3)
+- **types**: remove finalmessagechunk type and usage - By [Jannchie](mailto:jannchie@gmail.com) in [6b3281e](https://github.com/Jannchie/lite-agent/commit/6b3281e)
+
+### :sparkles: Features
+
+- **chat-display**: add support for new message format - By [Jannchie](mailto:jannchie@gmail.com) in [a3fd1c6](https://github.com/Jannchie/lite-agent/commit/a3fd1c6)
+- **chat-display**: add meta stats, local time, and message performance info to chat display and message types - By [Jannchie](mailto:jannchie@gmail.com) in [bda3346](https://github.com/Jannchie/lite-agent/commit/bda3346)
+- **responses**: support new message format for response api - By [Jannchie](mailto:jannchie@gmail.com) in [e34a2ef](https://github.com/Jannchie/lite-agent/commit/e34a2ef)
+- **responses**: add response streaming support and client methods - By [Jannchie](mailto:jannchie@gmail.com) in [f484530](https://github.com/Jannchie/lite-agent/commit/f484530)
+- **rich-helpers**: add print_messages for compact output - By [Jannchie](mailto:panjianqi@preferred.jp) in [5a49203](https://github.com/Jannchie/lite-agent/commit/5a49203)
+- **runner**: support interactive tool call confirmation in terminal && improve chunk handling - By [Jannchie](mailto:jannchie@gmail.com) in [7164284](https://github.com/Jannchie/lite-agent/commit/7164284)
+
+### :adhesive_bandage: Fixes
+
+- **agent**: use isinstance for to_llm_dict checks - By [Jannchie](mailto:jannchie@gmail.com) in [e996785](https://github.com/Jannchie/lite-agent/commit/e996785)
+- **examples**: add content checks for dict messages - By [Jannchie](mailto:panjianqi@preferred.jp) in [94afcae](https://github.com/Jannchie/lite-agent/commit/94afcae)
+- **import**: update stream_handler imports && fix response api logic - By [Jannchie](mailto:jannchie@gmail.com) in [be192da](https://github.com/Jannchie/lite-agent/commit/be192da)
+- **input-image**: ensure file_id or image_url present and improve conversion handling - By [Jannchie](mailto:jannchie@gmail.com) in [d0bcdd5](https://github.com/Jannchie/lite-agent/commit/d0bcdd5)
+- **runner**: improve wait_for_user finish detection - By [Jannchie](mailto:jannchie@gmail.com) in [98435b0](https://github.com/Jannchie/lite-agent/commit/98435b0)
+- **runner**: fix final message processing to return responses - By [Jannchie](mailto:jannchie@gmail.com) in [e813f4a](https://github.com/Jannchie/lite-agent/commit/e813f4a)
+- **stream-chunk-processor**: construct usage event with correct usage type - By [Jannchie](mailto:jannchie@gmail.com) in [89b2cbe](https://github.com/Jannchie/lite-agent/commit/89b2cbe)
+- **tests**: improve type hints and patching mocks - By [Jannchie](mailto:jannchie@gmail.com) in [e4dfd64](https://github.com/Jannchie/lite-agent/commit/e4dfd64)
+- **tests**: update patch paths for litellm.acompletion - By [Jannchie](mailto:jannchie@gmail.com) in [897a284](https://github.com/Jannchie/lite-agent/commit/897a284)
+
+### :art: Refactors
+
+- **examples**: replace final_message with assistant_message in includes - By [Jannchie](mailto:jannchie@gmail.com) in [c5431ab](https://github.com/Jannchie/lite-agent/commit/c5431ab)
+- **runner**: simplify message appending logic && tidy code style - By [Jannchie](mailto:jannchie@gmail.com) in [412a8ec](https://github.com/Jannchie/lite-agent/commit/412a8ec)
+- **runner**: replace api handling with match statement && set default api to responses - By [Jannchie](mailto:jannchie@gmail.com) in [75c823a](https://github.com/Jannchie/lite-agent/commit/75c823a)
+- **stream-handlers**: rename litellm_stream_handler to litellm_completion_stream_handler && update imports and tests - By [Jannchie](mailto:jannchie@gmail.com) in [5b7ac92](https://github.com/Jannchie/lite-agent/commit/5b7ac92)
+- **tests**: rename tool_call types to function_call && remove unused usage tests - By [Jannchie](mailto:jannchie@gmail.com) in [1e0992f](https://github.com/Jannchie/lite-agent/commit/1e0992f)
+
+### :memo: Documentation
+
+- add claude guidance documentation - By [Jannchie](mailto:panjianqi@preferred.jp) in [a741494](https://github.com/Jannchie/lite-agent/commit/a741494)
+
+### :wrench: Chores
+
+- **dependencies**: update lock file - By [Jannchie](mailto:jannchie@gmail.com) in [042f6f3](https://github.com/Jannchie/lite-agent/commit/042f6f3)
+- **dev-deps**: add pytest-asyncio to dev dependencies - By [Jannchie](mailto:jannchie@gmail.com) in [57543cd](https://github.com/Jannchie/lite-agent/commit/57543cd)
+- **lint**: remove commented ruff exclude config - By [Jannchie](mailto:jannchie@gmail.com) in [25635dd](https://github.com/Jannchie/lite-agent/commit/25635dd)
+
 ## v0.3.0
 
 [v0.2.0...v0.3.0](https://github.com/Jannchie/lite-agent/compare/v0.2.0...v0.3.0)

lite_agent-0.4.1/CLAUDE.md

@@ -0,0 +1,100 @@
+# CLAUDE.md
+
+This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
+
+## Development Commands
+
+### Testing
+
+```bash
+pytest                    # Run all tests
+pytest tests/unit/        # Run only unit tests
+pytest tests/integration/ # Run only integration tests
+pytest --cov             # Run with coverage
+```
+
+### Linting and Formatting
+
+```bash
+ruff check               # Run linter
+ruff format              # Format code
+```
+
+### Package Management
+
+```bash
+uv add lite-agent # Install from PyPI
+uv add --dev lite-agent # Install dev package
+```
+
+## Project Architecture
+
+LiteAgent is a lightweight AI agent framework built on top of LiteLLM. The core architecture consists of:
+
+### Core Components
+
+**Agent (`src/lite_agent/agent.py`)**
+
+- Central agent class that manages LLM interactions, tool calls, and message handling
+- Supports tool registration via `funcall` library for type-safe function calling
+- Handles agent handoffs (parent-child relationships) for complex workflows
+- Manages completion conditions ("stop" vs "call" for different termination behaviors)
+- Converts between OpenAI's Response API and Completion API message formats
+
+**Runner (`src/lite_agent/runner.py`)**
+
+- Orchestrates agent execution with streaming support
+- Manages conversation flow and message history
+- Handles tool call execution and agent transfers
+- Supports continuation from previous states and chat history management
+- Provides both streaming and batch execution modes
+
+**Type System (`src/lite_agent/types/`)**
+
+- Comprehensive Pydantic models for all message types and chunks
+- Supports both Response API and Completion API formats
+- Type-safe definitions for tool calls, chunks, and messages
+
+### Key Features
+
+**Tool Integration**
+
+- Uses `funcall` library for automatic tool schema generation from Python functions
+- Supports basic types, Pydantic models, and dataclasses as parameters
+- Dynamic tool registration for agent handoffs and control flow
+
+**Agent Handoffs**
+
+- Parent-child agent relationships for complex task delegation
+- Automatic `transfer_to_agent` and `transfer_to_parent` tool registration
+- Chat history tracking across agent transitions
+
+**Message Processing**
+
+- Bidirectional conversion between OpenAI API formats
+- Streaming chunk processing with configurable output filtering
+- Message transfer callbacks for preprocessing
+
+**Completion Modes**
+
+- `"stop"`: Traditional completion until model decides to stop
+- `"call"`: Completion until specific tool (`wait_for_user`) is called
+
+### Examples Directory Structure
+
+Examples demonstrate various usage patterns:
+
+- `basic.py`: Simple agent with tool calling
+- `handoffs.py`: Agent-to-agent transfers
+- `context.py`: Context passing to tools
+- `chat_display_demo.py`: Rich console output formatting
+- `channels/`: Channel-based communication patterns
+
+### Testing Architecture
+
+- **Unit tests**: Test individual components in isolation
+- **Integration tests**: Test full agent workflows with mocked LLM responses
+- **Performance tests**: Test memory usage and performance characteristics
+- **Mock system**: JSONL-based conversation recording/playback for deterministic testing
+
+The framework emphasizes simplicity and extensibility while maintaining full type safety and comprehensive streaming support.

{lite_agent-0.3.0 → lite_agent-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lite-agent
-Version: 0.3.0
+Version: 0.4.1
 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.7.0
+Requires-Dist: funcall>=0.10.0
 Requires-Dist: prompt-toolkit>=3.0.51
 Requires-Dist: rich>=14.0.0
 Description-Content-Type: text/markdown

{lite_agent-0.3.0 → lite_agent-0.4.1}/examples/basic.py

@@ -4,6 +4,7 @@ import logging
 from rich.logging import RichHandler
 
 from lite_agent.agent import Agent
+from lite_agent.chat_display import display_messages
 from lite_agent.runner import Runner
 
 logging.basicConfig(
@@ -19,7 +20,6 @@ logger.setLevel(logging.DEBUG)
 
 async def get_temperature(city: str) -> str:
     """Get the temperature for a city."""
-    await asyncio.sleep(1)
     return f"The temperature in {city} is 25°C."
 
 
@@ -35,11 +35,11 @@ async def main():
     runner = Runner(agent)
     resp = runner.run(
         "What is the temperature in New York?",
-        includes=["final_message", "usage", "tool_call", "tool_call_result"],
-        record_to="tests/mocks/basic/1.jsonl",
+        includes=["usage", "assistant_message", "function_call", "function_call_output", "timing"],
     )
     async for chunk in resp:
         logger.info(chunk)
+    display_messages(runner.messages)
     print(runner.messages)
 
 

lite_agent-0.4.1/examples/basic_agent.py

@@ -0,0 +1,54 @@
+import asyncio
+import logging
+
+from rich.logging import RichHandler
+
+from lite_agent.agent import Agent
+
+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_temperature(city: str) -> str:
+    """Get the temperature for a city."""
+    await asyncio.sleep(1)
+    return f"The temperature in {city} is 25°C."
+
+
+async def get_whether(city: str) -> str:
+    """Get the weather for a city."""
+    await asyncio.sleep(1)
+    return f"The weather in {city} is sunny with a few clouds."
+
+
+agent = Agent(
+    model="gpt-4.1-nano",
+    name="Weather Assistant",
+    instructions="You are a helpful weather assistant. Before using tools, briefly explain what you are going to do. Provide friendly and informative responses.",
+    tools=[get_temperature, get_whether],
+)
+
+
+async def main():
+    resp = await agent.completion(
+        [
+            {
+                "role": "user",
+                "content": "What is the temperature and whether in New York?",
+            },
+        ],
+    )
+    async for chunk in resp:
+        if chunk.type != "completion_raw":
+            logger.info(chunk)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

{lite_agent-0.3.0 → lite_agent-0.4.1}/examples/channels/rich_channel.py

@@ -1,31 +1,32 @@
 from rich.console import Console
 
-from lite_agent.types import AgentChunk, ContentDeltaChunk
+from lite_agent.types import AgentChunk, ContentDeltaEvent
 
 
 class RichChannel:
     def __init__(self) -> None:
         self.console = Console()
         self.map = {
-            "final_message": self.handle_final_message,
-            "tool_call": self.handle_tool_call,
-            "tool_call_result": self.handle_tool_call_result,
-            "tool_call_delta": self.handle_tool_call_delta,
+            "function_call": self.handle_tool_call,
+            "function_call_output": self.handle_tool_call_result,
+            "function_call_delta": self.handle_function_call_delta,
             "content_delta": self.handle_content_delta,
             "usage": self.handle_usage,
-            "require_confirm": self.handle_require_confirm,
         }
-        self.new_turn = True
+        self._new_turn = True
 
     async def handle(self, chunk: AgentChunk):
-        handler = self.map[chunk.type]
-        return await handler(chunk)
+        handler = self.map.get(chunk.type)
+        if handler is None:
+            return None
+        return await handler(chunk)  # type: ignore
 
-    async def handle_final_message(self, _chunk: AgentChunk):
+    def new_turn(self):
         print()
-        self.new_turn = True
+        self._new_turn = True
 
     async def handle_tool_call(self, chunk: AgentChunk):
+        print()
         name = getattr(chunk, "name", "<unknown>")
         arguments = getattr(chunk, "arguments", "")
         self.console.print(f"🛠️  [green]{name}[/green]([yellow]{arguments}[/yellow])")
@@ -35,16 +36,14 @@ class RichChannel:
         content = getattr(chunk, "content", "")
         self.console.print(f"🛠️  [green]{name}[/green] → [yellow]{content}[/yellow]")
 
-    async def handle_tool_call_delta(self, chunk: AgentChunk): ...
-    async def handle_content_delta(self, chunk: ContentDeltaChunk):
-        if self.new_turn:
+    async def handle_function_call_delta(self, chunk: AgentChunk): ...
+    async def handle_content_delta(self, chunk: ContentDeltaEvent):
+        if self._new_turn:
             self.console.print("🤖 ", end="")
-            self.new_turn = False
+            self._new_turn = False
         print(chunk.delta, end="", flush=True)
 
     async def handle_usage(self, chunk: AgentChunk):
         if False:
             usage = chunk.usage
             self.console.print(f"In: {usage.prompt_tokens}, Out: {usage.completion_tokens}, Total: {usage.total_tokens}")
-
-    async def handle_require_confirm(self, chunk: AgentChunk): ...

lite_agent-0.3.0/examples/rich_helpers_demo.py → lite_agent-0.4.1/examples/chat_display_demo.py

@@ -1,13 +1,13 @@
 """
-示例：使用 rich_helpers 美观渲染聊天记录
+示例：使用 chat_display 美观显示聊天记录
 
-这个示例展示了如何使用 rich_helpers 模块中的函数来美观地渲染聊天记录。
+这个示例展示了如何使用 chat_display 模块中的函数来美观地显示聊天记录。
 """
 
 import asyncio
 
+from lite_agent import display_messages
 from lite_agent.agent import Agent
-from lite_agent.rich_helpers import print_chat_history, print_chat_summary
 from lite_agent.runner import Runner
 
 
@@ -53,7 +53,7 @@ async def create_sample_chat_history() -> Runner:
     runner.append_message(
         {
             "type": "function_call",
-            "function_call_id": "call_123",
+            "call_id": "call_123",
             "name": "get_weather",
             "arguments": '{"city": "Tokyo"}',
             "content": "",
@@ -76,7 +76,7 @@ async def create_sample_chat_history() -> Runner:
     runner.append_message(
         {
             "type": "function_call",
-            "function_call_id": "call_456",
+            "call_id": "call_456",
             "name": "calculate",
             "arguments": '{"expression": "25 * 4 + 10"}',
             "content": "",
@@ -97,32 +97,15 @@ async def create_sample_chat_history() -> Runner:
 
 
 async def main():
-    """主函数：演示 rich_helpers 的使用。"""
-    print("🎨 Rich Chat History Renderer Demo\n")
+    """主函数：演示 chat_display 的使用。"""
+    print("🎨 Chat Display Demo\n")
 
     # 创建示例聊天历史
     runner = await create_sample_chat_history()
 
-    # 1. 展示聊天摘要
-    print("📊 Chat Summary:")
-    print_chat_summary(runner.messages)
-    print()
-
-    # 2. 渲染完整的聊天历史
-    print("💬 Full Chat History:")
-    print_chat_history(runner.messages)
-
-    # 3. 展示不同的渲染选项
-    print("\n" + "=" * 60)
-    print("🎛️  Different Rendering Options:")
-    print("=" * 60)
-
-    # 不显示时间戳和索引
-    print("\n📝 Without timestamps and indices:")
-    print_chat_history(
+    # print demo
+    display_messages(
         runner.messages,
-        show_timestamps=False,
-        show_indices=False,
     )
 
 

{lite_agent-0.3.0 → lite_agent-0.4.1}/examples/confirm_and_continue.py

@@ -43,13 +43,13 @@ async def main():
     runner = Runner(agent)
     resp = runner.run(
         "What is the weather in New York? And what is the temperature there?",
-        includes=["final_message", "usage", "tool_call", "tool_call_result"],
+        includes=["usage", "assistant_message", "function_call", "function_call_output"],
         record_to="tests/mocks/confirm_and_continue/1.jsonl",
     )
     async for chunk in resp:
         logger.info(chunk)
     resp = runner.run_continue_stream(
-        includes=["final_message", "usage", "tool_call", "tool_call_result"],
+        includes=["usage", "assistant_message", "function_call", "function_call_output"],
         record_to="tests/mocks/confirm_and_continue/2.jsonl",
     )
     async for chunk in resp:

{lite_agent-0.3.0 → lite_agent-0.4.1}/examples/consolidate_history.py

@@ -26,7 +26,7 @@ def test_consolidate_history():
     print("Consolidated messages count:", len(result))
     print("\nConsolidated content:")
     print("=" * 50)
-    if result and isinstance(result[0], dict):
+    if result and isinstance(result[0], dict) and "content" in result[0]:
         print(result[0]["content"])
     print("=" * 50)
 
@@ -38,7 +38,7 @@ def test_consolidate_history():
     single_message = [{"role": "user", "content": "Just a simple test"}]
     single_result = consolidate_history_transfer(single_message)
     print(f"\nSingle message result count: {len(single_result)}")
-    if single_result and isinstance(single_result[0], dict):
+    if single_result and isinstance(single_result[0], dict) and "content" in single_result[0]:
         print("Single message consolidated content:")
         print(single_result[0]["content"])
 

{lite_agent-0.3.0 → lite_agent-0.4.1}/examples/context.py

@@ -2,9 +2,10 @@ import asyncio
 import logging
 
 from funcall import Context
+from pydantic import BaseModel
 from rich.logging import RichHandler
 
-from lite_agent.agent import Agent, BaseModel
+from lite_agent.agent import Agent
 from lite_agent.runner import Runner
 
 
@@ -47,7 +48,7 @@ async def main():
     runner = Runner(agent)
     resp = runner.run(
         "What is the temperature in current city?",
-        includes=["final_message", "usage", "tool_call", "tool_call_result"],
+        includes=["assistant_message", "usage", "function_call", "function_call_output"],
         record_to="tests/mocks/context/1.jsonl",
         context=weather_context,
     )

{lite_agent-0.3.0 → lite_agent-0.4.1}/examples/handoffs.py

@@ -65,7 +65,7 @@ async def main():
     runner = Runner(parent)
     resp = runner.run(
         "Hello, I need to check the whether and temperature of Tokyo.",
-        includes=["final_message", "tool_call", "tool_call_result"],
+        includes=["assistant_message", "function_call", "function_call_output"],
         record_to="tests/mocks/handoffs/1.jsonl",
     )
     async for message in resp:

{lite_agent-0.3.0 → lite_agent-0.4.1}/examples/image.py

@@ -49,7 +49,7 @@ async def main():
                 ],
             },
         ],
-        includes=["final_message", "usage"],
+        includes=["assistant_message", "usage"],
     )
     async for chunk in resp:
         logger.info(chunk)

{lite_agent-0.3.0 → lite_agent-0.4.1}/examples/message_transfer_example.py

@@ -10,10 +10,12 @@ import re
 from lite_agent.agent import Agent
 from lite_agent.message_transfers import consolidate_history_transfer
 from lite_agent.types import (
-    AgentAssistantMessage,
-    AgentSystemMessage,
-    AgentUserMessage,
+    AssistantTextContent,
+    NewAssistantMessage,
+    NewSystemMessage,
+    NewUserMessage,
     RunnerMessages,
+    UserTextContent,
 )
 
 
@@ -40,11 +42,13 @@ def example_message_transfer(messages: RunnerMessages) -> RunnerMessages:
                 processed_messages.append(processed_message)
             else:
                 processed_messages.append(message)
-        elif isinstance(message, AgentUserMessage):
+        elif isinstance(message, NewUserMessage):
             # Handle user message models
             processed_message = message.model_copy()
-            if isinstance(message.content, str):
-                processed_message.content = f"[PROCESSED] {message.content}"
+            # Process text content items
+            for i, content_item in enumerate(processed_message.content):
+                if isinstance(content_item, UserTextContent):
+                    processed_message.content[i] = UserTextContent(text=f"[PROCESSED] {content_item.text}")
             processed_messages.append(processed_message)
         else:
             # Keep other messages unchanged
@@ -83,14 +87,27 @@ def privacy_filter_transfer(messages: RunnerMessages) -> RunnerMessages:
                 processed_messages.append(processed_message)
             else:
                 processed_messages.append(message)
-        elif isinstance(message, (AgentUserMessage, AgentAssistantMessage, AgentSystemMessage)):
-            # Handle message models with content
-            if isinstance(message.content, str):
-                processed_message = message.model_copy()
-                processed_message.content = filter_content(message.content)
-                processed_messages.append(processed_message)
-            else:
-                processed_messages.append(message)
+        elif isinstance(message, NewSystemMessage):
+            # Handle system message with string content
+            processed_message = message.model_copy()
+            processed_message.content = filter_content(message.content)
+            processed_messages.append(processed_message)
+        elif isinstance(message, NewUserMessage):
+            # Handle user message models with content list
+            processed_message = message.model_copy()
+            # Process text content items
+            for i, content_item in enumerate(processed_message.content):
+                if isinstance(content_item, UserTextContent):
+                    processed_message.content[i] = UserTextContent(text=filter_content(content_item.text))
+            processed_messages.append(processed_message)
+        elif isinstance(message, NewAssistantMessage):
+            # Handle assistant message models with content list
+            processed_message = message.model_copy()
+            # Process text content items
+            for i, content_item in enumerate(processed_message.content):
+                if isinstance(content_item, AssistantTextContent):
+                    processed_message.content[i] = AssistantTextContent(text=filter_content(content_item.text))
+            processed_messages.append(processed_message)
         else:
             # Keep other message types unchanged
             processed_messages.append(message)