lite-agent 0.1.0__tar.gz → 0.2.0__tar.gz

lite_agent-0.2.0/.github/workflows/ci.yml

@@ -0,0 +1,33 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Set up uv
+        uses: astral-sh/setup-uv@v6
+        with:
+          python-version: "3.12"
+          activate-environment: true
+
+      - name: Install dependencies
+        run: uv sync
+
+      - name: Run tests with coverage
+        run: |
+          pytest --cov=src --cov-report=xml
+      - name: Upload coverage to Codecov
+        uses: codecov/codecov-action@v4
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+          files: ./coverage.xml
+          fail_ci_if_error: true

{lite_agent-0.1.0 → lite_agent-0.2.0}/.gitignore

@@ -8,3 +8,4 @@ wheels/
 
 # Virtual environments
 .venv
+.coverage

lite_agent-0.2.0/CHANGELOG.md

@@ -0,0 +1,85 @@
+## v0.2.0
+
+[v0.1.0...v0.2.0](https://github.com/Jannchie/lite-agent/compare/v0.1.0...v0.2.0)
+
+### :rocket: Breaking Changes
+
+- **agent**: rename stream_async to completion - By [Jannchie](mailto:jannchie@gmail.com) in [dae25cc](https://github.com/Jannchie/lite-agent/commit/dae25cc)
+- **messages**: add support for function call message types and responses format conversion - By [Jannchie](mailto:jannchie@gmail.com) in [9e71ccc](https://github.com/Jannchie/lite-agent/commit/9e71ccc)
+- **project-structure**: rename package to lite-agent && update processor imports - By [Jannchie](mailto:jannchie@gmail.com) in [6e0747a](https://github.com/Jannchie/lite-agent/commit/6e0747a)
+- **runner**: rename run_stream to run throughout codebase - By [Jannchie](mailto:jannchie@gmail.com) in [525865f](https://github.com/Jannchie/lite-agent/commit/525865f)
+- **stream-handler**: rename litellm_raw to completion_raw - By [Jannchie](mailto:jannchie@gmail.com) in [9001f19](https://github.com/Jannchie/lite-agent/commit/9001f19)
+- **types**: remove AgentToolCallMessage usage and definition - By [Jannchie](mailto:jannchie@gmail.com) in [fb3592e](https://github.com/Jannchie/lite-agent/commit/fb3592e)
+- migrate chunk and message types to pydantic models && update runner and processors for model usage - By [Jannchie](mailto:jannchie@gmail.com) in [da0d6fc](https://github.com/Jannchie/lite-agent/commit/da0d6fc)
+
+### :sparkles: Features
+
+- **agent**: support dynamic handoff and parent transfer - By [Jannchie](mailto:jannchie@gmail.com) in [5019018](https://github.com/Jannchie/lite-agent/commit/5019018)
+- **agent**: add async tool call handling - By [Jannchie](mailto:jannchie@gmail.com) in [7de9103](https://github.com/Jannchie/lite-agent/commit/7de9103)
+- **context**: add context support for agent and runner - By [Jannchie](mailto:jannchie@gmail.com) in [e4885ee](https://github.com/Jannchie/lite-agent/commit/e4885ee)
+- **handoffs**: support multiple agents and add weather tools - By [Jannchie](mailto:jannchie@gmail.com) in [b9440ea](https://github.com/Jannchie/lite-agent/commit/b9440ea)
+- **handoffs**: add agent handoff and transfer functionality - By [Jannchie](mailto:jannchie@gmail.com) in [7655029](https://github.com/Jannchie/lite-agent/commit/7655029)
+- **message-transfer**: add message_transfer callback support to agent and runner && provide consolidate_history_transfer utility && add tests and usage examples - By [Jannchie](mailto:jannchie@gmail.com) in [16d4788](https://github.com/Jannchie/lite-agent/commit/16d4788)
+- **runner**: add record_to param to run_until_complete - By [Jannchie](mailto:jannchie@gmail.com) in [597bff6](https://github.com/Jannchie/lite-agent/commit/597bff6)
+- **runner**: support chat stream recording and improve sequence typing - By [Jannchie](mailto:jannchie@gmail.com) in [5c1d609](https://github.com/Jannchie/lite-agent/commit/5c1d609)
+- **runner**: add run_continue method and require confirm tool handling - By [Jannchie](mailto:jannchie@gmail.com) in [d130ebc](https://github.com/Jannchie/lite-agent/commit/d130ebc)
+- **testing**: add integration and unit tests for agent and file recording - By [Jannchie](mailto:jannchie@gmail.com) in [e5a58ce](https://github.com/Jannchie/lite-agent/commit/e5a58ce)
+- **tool**: add require_confirmation support for tool calls - By [Jannchie](mailto:panjianqi@preferred.jp) in [69625c8](https://github.com/Jannchie/lite-agent/commit/69625c8)
+- **tool-calls**: add require_confirm chunk type and flow - By [Jannchie](mailto:panjianqi@preferred.jp) in [b2d1654](https://github.com/Jannchie/lite-agent/commit/b2d1654)
+- **types**: add agent message and tool call types - By [Jannchie](mailto:jannchie@gmail.com) in [489e370](https://github.com/Jannchie/lite-agent/commit/489e370)
+
+### :adhesive_bandage: Fixes
+
+- **agent**: update tool target to completion - By [Jannchie](mailto:jannchie@gmail.com) in [6c8978c](https://github.com/Jannchie/lite-agent/commit/6c8978c)
+- **examples**: simplify temper agent instructions - By [Jannchie](mailto:jannchie@gmail.com) in [6632092](https://github.com/Jannchie/lite-agent/commit/6632092)
+- **loggers**: update logger name to lite_agent - By [Jannchie](mailto:jannchie@gmail.com) in [b4c59aa](https://github.com/Jannchie/lite-agent/commit/b4c59aa)
+- **types**: change tool_calls from sequence to list - By [Jannchie](mailto:jannchie@gmail.com) in [334c9c0](https://github.com/Jannchie/lite-agent/commit/334c9c0)
+
+### :art: Refactors
+
+- **agent**: rename prepare_messages to prepare_completion_messages - By [Jannchie](mailto:jannchie@gmail.com) in [02eeed6](https://github.com/Jannchie/lite-agent/commit/02eeed6)
+- **agent**: remove unused completions to responses converter - By [Jannchie](mailto:jannchie@gmail.com) in [f6432c6](https://github.com/Jannchie/lite-agent/commit/f6432c6)
+- **chunk-handler**: update type annotations and chunk types - By [Jannchie](mailto:panjianqi@preferred.jp) in [f2b95c5](https://github.com/Jannchie/lite-agent/commit/f2b95c5)
+- **chunk-handler**: extract helper functions for chunk processing - By [Jannchie](mailto:jannchie@gmail.com) in [083a164](https://github.com/Jannchie/lite-agent/commit/083a164)
+- **litellm**: extract chunk processing logic to function - By [Jannchie](mailto:jannchie@gmail.com) in [f31e459](https://github.com/Jannchie/lite-agent/commit/f31e459)
+- **litellm-stream**: decouple funcall from stream handler && adjust runner tool call logic - By [Jannchie](mailto:jannchie@gmail.com) in [2747e7e](https://github.com/Jannchie/lite-agent/commit/2747e7e)
+- **rich-channel**: extract rich channel to separate module && clean main - By [Jannchie](mailto:jannchie@gmail.com) in [a8be74d](https://github.com/Jannchie/lite-agent/commit/a8be74d)
+- **runner**: modularize runner methods and simplify message handling - By [Jannchie](mailto:jannchie@gmail.com) in [628a6b5](https://github.com/Jannchie/lite-agent/commit/628a6b5)
+- **tests**: parametrize tool call type tests and clean up redundant cases - By [Jannchie](mailto:jannchie@gmail.com) in [0791582](https://github.com/Jannchie/lite-agent/commit/0791582)
+- **types**: move chunk types to types.py && update type imports - By [Jannchie](mailto:panjianqi@preferred.jp) in [d2ebeec](https://github.com/Jannchie/lite-agent/commit/d2ebeec)
+
+### :lipstick: Styles
+
+- **tests**: improve formatting and consistency - By [Jannchie](mailto:jannchie@gmail.com) in [4342aa4](https://github.com/Jannchie/lite-agent/commit/4342aa4)
+
+### :memo: Documentation
+
+- add initial project readme - By [Jannchie](mailto:panjianqi@preferred.jp) in [f04ec7a](https://github.com/Jannchie/lite-agent/commit/f04ec7a)
+
+### :wrench: Chores
+
+- **ci**: switch to uv for dependency management and update to python 3.12 - By [Jannchie](mailto:panjianqi@preferred.jp) in [64d3085](https://github.com/Jannchie/lite-agent/commit/64d3085)
+- **ci**: add github actions workflow - By [Jannchie](mailto:panjianqi@preferred.jp) in [051be1b](https://github.com/Jannchie/lite-agent/commit/051be1b)
+- **deps**: update lock file - By [Jannchie](mailto:jannchie@gmail.com) in [9aa1e77](https://github.com/Jannchie/lite-agent/commit/9aa1e77)
+- **metadata**: rename project and reformat keywords array - By [Jannchie](mailto:jannchie@gmail.com) in [2659c4d](https://github.com/Jannchie/lite-agent/commit/2659c4d)
+- **metadata**: update ai topic classifier - By [Jannchie](mailto:jannchie@gmail.com) in [15d46ea](https://github.com/Jannchie/lite-agent/commit/15d46ea)
+- **rename**: rename project name - By [Jannchie](mailto:jannchie@gmail.com) in [84077a8](https://github.com/Jannchie/lite-agent/commit/84077a8)
+
+## v0.1.0
+
+[5859f296f4aa3bf9156e560e5bbd98b3d8795b0d...v0.1.0](https://github.com/Jannchie/lite-agent/compare/5859f296f4aa3bf9156e560e5bbd98b3d8795b0d...v0.1.0)
+
+### :sparkles: Features
+
+- **chunk-handler**: add content-delta and tool-call-delta chunk types && update includes list in runner - By [Jannchie](mailto:jannchie@gmail.com) in [a3ba7d1](https://github.com/Jannchie/lite-agent/commit/a3ba7d1)
+- **easy_agent**: add prompt toolkit dependency && update agent to use prompt session && improve chunk handling - By [Jannchie](mailto:jannchie@gmail.com) in [b408c14](https://github.com/Jannchie/lite-agent/commit/b408c14)
+
+### :art: Refactors
+
+- **main**: remove datetime imports and usage && add input validator - By [Jannchie](mailto:jannchie@gmail.com) in [eab45b3](https://github.com/Jannchie/lite-agent/commit/eab45b3)
+
+### :wrench: Chores
+
+- **import**: remove unused json tool import - By [Jannchie](mailto:jannchie@gmail.com) in [8de5975](https://github.com/Jannchie/lite-agent/commit/8de5975)
+- **lock**: update lock file - By [Jannchie](mailto:jannchie@gmail.com) in [64880d3](https://github.com/Jannchie/lite-agent/commit/64880d3)
+- **pyproject**: update version && improve description && expand keywords && add classifiers - By [Jannchie](mailto:jannchie@gmail.com) in [c3e8ee6](https://github.com/Jannchie/lite-agent/commit/c3e8ee6)

lite_agent-0.2.0/PKG-INFO

@@ -0,0 +1,111 @@
+Metadata-Version: 2.4
+Name: lite-agent
+Version: 0.2.0
+Summary: A lightweight, extensible framework for building AI agent.
+Author-email: Jianqi Pan <jannchie@gmail.com>
+License: MIT
+Keywords: AI,agent framework,assistant,chatbot,function call,openai,pydantic,rich,tooling
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Communications :: Chat
+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: prompt-toolkit>=3.0.51
+Requires-Dist: rich>=14.0.0
+Description-Content-Type: text/markdown
+
+# LiteAgent
+
+[![codecov](https://codecov.io/gh/Jannchie/lite-agent/graph/badge.svg?token=SJW89Z1VAZ)](https://codecov.io/gh/Jannchie/lite-agent)
+
+## Introduction
+
+LiteAgent is an easy-to-learn, lightweight, and extensible AI agent framework built on top of [LiteLLM](https://github.com/BerriAI/litellm). It is designed as a minimal yet practical implementation for quickly building intelligent assistants and chatbots with robust tool-calling capabilities. The codebase is intentionally simple, making it ideal for learning, extension, and rapid prototyping.
+
+**Key Advantages:**
+
+- **Minimal and approachable:** The simplest agent implementation for fast learning and hacking.
+- **Accurate and complete type hints:** All function signatures are fully type-hinted and never faked, ensuring reliable developer experience and static analysis.
+- **Flexible parameter definition:** Supports defining tool function parameters using basic types, Pydantic models, or Python dataclasses—even in combination.
+- **Streaming responses:** Seamless support for LiteLLM streaming output.
+- **Custom tool functions:** Easily integrate your own Python functions (e.g., weather, temperature queries).
+- **Rich type annotations, Pydantic-based.**
+- **Easy to extend and test.**
+
+## Installation
+
+You can install LiteAgent directly from PyPI:
+
+```bash
+pip install lite-agent
+```
+
+Or use [uv](https://github.com/astral-sh/uv):
+
+```bash
+uv pip install lite-agent
+```
+
+If you want to install from source for development:
+
+```bash
+uv pip install -e .
+# or
+pip install -e .
+```
+
+## Quick Start
+
+### Code Example
+
+See `examples/basic.py`:
+
+```python
+import asyncio
+from lite_agent.agent import Agent
+from lite_agent.runner import Runner
+
+async def get_whether(city: str) -> str:
+    await asyncio.sleep(1)
+    return f"The weather in {city} is sunny with a few clouds."
+
+async def main():
+    agent = Agent(
+        model="gpt-4.1",
+        name="Weather Assistant",
+        instructions="You are a helpful weather assistant.",
+        tools=[get_whether],
+    )
+    runner = Runner(agent)
+    resp = await runner.run_until_complete("What's the weather in New York?")
+    for chunk in resp:
+        print(chunk)
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+See `pyproject.toml` for details.
+
+## Testing
+
+```bash
+pytest
+```
+
+## License
+
+MIT License
+
+## Author
+
+Jianqi Pan ([jannchie@gmail.com](mailto:jannchie@gmail.com))

lite_agent-0.2.0/README.md

@@ -0,0 +1,86 @@
+# LiteAgent
+
+[![codecov](https://codecov.io/gh/Jannchie/lite-agent/graph/badge.svg?token=SJW89Z1VAZ)](https://codecov.io/gh/Jannchie/lite-agent)
+
+## Introduction
+
+LiteAgent is an easy-to-learn, lightweight, and extensible AI agent framework built on top of [LiteLLM](https://github.com/BerriAI/litellm). It is designed as a minimal yet practical implementation for quickly building intelligent assistants and chatbots with robust tool-calling capabilities. The codebase is intentionally simple, making it ideal for learning, extension, and rapid prototyping.
+
+**Key Advantages:**
+
+- **Minimal and approachable:** The simplest agent implementation for fast learning and hacking.
+- **Accurate and complete type hints:** All function signatures are fully type-hinted and never faked, ensuring reliable developer experience and static analysis.
+- **Flexible parameter definition:** Supports defining tool function parameters using basic types, Pydantic models, or Python dataclasses—even in combination.
+- **Streaming responses:** Seamless support for LiteLLM streaming output.
+- **Custom tool functions:** Easily integrate your own Python functions (e.g., weather, temperature queries).
+- **Rich type annotations, Pydantic-based.**
+- **Easy to extend and test.**
+
+## Installation
+
+You can install LiteAgent directly from PyPI:
+
+```bash
+pip install lite-agent
+```
+
+Or use [uv](https://github.com/astral-sh/uv):
+
+```bash
+uv pip install lite-agent
+```
+
+If you want to install from source for development:
+
+```bash
+uv pip install -e .
+# or
+pip install -e .
+```
+
+## Quick Start
+
+### Code Example
+
+See `examples/basic.py`:
+
+```python
+import asyncio
+from lite_agent.agent import Agent
+from lite_agent.runner import Runner
+
+async def get_whether(city: str) -> str:
+    await asyncio.sleep(1)
+    return f"The weather in {city} is sunny with a few clouds."
+
+async def main():
+    agent = Agent(
+        model="gpt-4.1",
+        name="Weather Assistant",
+        instructions="You are a helpful weather assistant.",
+        tools=[get_whether],
+    )
+    runner = Runner(agent)
+    resp = await runner.run_until_complete("What's the weather in New York?")
+    for chunk in resp:
+        print(chunk)
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+See `pyproject.toml` for details.
+
+## Testing
+
+```bash
+pytest
+```
+
+## License
+
+MIT License
+
+## Author
+
+Jianqi Pan ([jannchie@gmail.com](mailto:jannchie@gmail.com))

lite_agent-0.2.0/examples/basic.py

@@ -0,0 +1,46 @@
+import asyncio
+import logging
+
+from rich.logging import RichHandler
+
+from lite_agent.agent import Agent
+from lite_agent.runner import Runner
+
+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."
+
+
+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],
+)
+
+
+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",
+    )
+    async for chunk in resp:
+        logger.info(chunk)
+    print(runner.messages)
+
+if __name__ == "__main__":
+    asyncio.run(main())

lite_agent-0.2.0/examples/channels/rich_channel.py

@@ -0,0 +1,50 @@
+from rich.console import Console
+
+from lite_agent.types import AgentChunk, ContentDeltaChunk
+
+
+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,
+            "content_delta": self.handle_content_delta,
+            "usage": self.handle_usage,
+            "require_confirm": self.handle_require_confirm,
+        }
+        self.new_turn = True
+
+    async def handle(self, chunk: AgentChunk):
+        handler = self.map[chunk.type]
+        return await handler(chunk)
+
+    async def handle_final_message(self, _chunk: AgentChunk):
+        print()
+        self.new_turn = True
+
+    async def handle_tool_call(self, chunk: AgentChunk):
+        name = getattr(chunk, "name", "<unknown>")
+        arguments = getattr(chunk, "arguments", "")
+        self.console.print(f"🛠️  [green]{name}[/green]([yellow]{arguments}[/yellow])")
+
+    async def handle_tool_call_result(self, chunk: AgentChunk):
+        name = getattr(chunk, "name", "<unknown>")
+        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:
+            self.console.print("🤖 ", end="")
+            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.2.0/examples/confirm_and_continue.py

@@ -0,0 +1,60 @@
+import asyncio
+import logging
+
+from funcall.decorators import tool
+from rich.logging import RichHandler
+
+from lite_agent.agent import Agent
+from lite_agent.runner import Runner
+
+logging.basicConfig(
+    level=logging.WARNING,
+    format="%(message)s",
+    datefmt="[%X]",
+    handlers=[RichHandler(rich_tracebacks=True)],
+)
+
+logger = logging.getLogger("lite_agent")
+logger.setLevel(logging.DEBUG)
+
+
+@tool(require_confirmation=True)
+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."
+
+
+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."
+
+
+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_whether, get_temperature],
+)
+
+
+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"],
+        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"],
+        record_to="tests/mocks/confirm_and_continue/2.jsonl",
+    )
+    async for chunk in resp:
+        logger.info(chunk)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

lite_agent-0.2.0/examples/context.py

@@ -0,0 +1,60 @@
+import asyncio
+import logging
+
+from funcall import Context
+from rich.logging import RichHandler
+
+from lite_agent.agent import Agent, BaseModel
+from lite_agent.runner import Runner
+
+
+class WeatherContext(BaseModel):
+    city: str
+
+
+weather_context = Context(WeatherContext(city="New York"))
+
+
+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_current_city_temperature(context: Context[WeatherContext]) -> str:
+    """Get the temperature for the current city specified in the context."""
+    await asyncio.sleep(1)
+    if not context.value:
+        msg = "City must be specified in the context."
+        raise ValueError(msg)
+    return f"The temperature in {context.value.city} is 25°C."
+
+
+agent = Agent(
+    model="gpt-4.1-nano",
+    name="Weather Assistant",
+    instructions="You are a weather assistant. Use the tools provided to answer questions about the weather.",
+    tools=[get_current_city_temperature],
+)
+
+
+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"],
+        record_to="tests/mocks/context/1.jsonl",
+        context=weather_context,
+    )
+    async for chunk in resp:
+        logger.info(chunk)
+    print(runner.messages)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

lite_agent-0.2.0/examples/handoffs.py

@@ -0,0 +1,74 @@
+#!/usr/bin/env python3
+"""Example showing how to use handoffs between agents.
+
+This example demonstrates how agents can transfer conversations to each other
+using automatically generated transfer functions.
+"""
+
+import asyncio
+import logging
+
+from rich.logging import RichHandler
+
+from lite_agent import consolidate_history_transfer
+from lite_agent.agent import Agent
+from lite_agent.loggers import logger
+from lite_agent.runner import Runner
+
+logging.basicConfig(
+    format="%(message)s",
+    datefmt="[%X]",
+    handlers=[RichHandler(rich_tracebacks=True)],
+)
+logger.setLevel(logging.DEBUG)
+
+
+def get_temperature(city: str) -> str:
+    """Get the temperature of a city."""
+    # In a real application, this would call an API or database.
+    return f"The temperature in {city} is 25°C."
+
+
+def get_weather(city: str) -> str:
+    """Get the weather of a city."""
+    # In a real application, this would call an API or database.
+    return f"The weather in {city} is sunny."
+
+
+async def main():
+    """Demonstrate agent handoffs functionality."""
+    parent = Agent(
+        model="gpt-4.1",
+        name="ParentAgent",
+        instructions="You are a helpful agent.",
+    )
+
+    whether_agent = Agent(
+        model="gpt-4.1",
+        name="WhetherAgent",
+        instructions="You are a helpful agent to check weather.",
+        tools=[get_weather],
+        message_transfer=consolidate_history_transfer,
+    )
+
+    temper_agent = Agent(
+        model="gpt-4.1",
+        name="TemperatureAgent",
+        instructions="You are a helpful agent to check temperature.",
+        tools=[get_temperature],
+        message_transfer=consolidate_history_transfer,
+    )
+
+    parent.add_handoff(whether_agent)
+    parent.add_handoff(temper_agent)
+
+    print(temper_agent.fc.function_registry)
+    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"])
+    async for message in resp:
+        logger.info(message)
+    logger.info(runner.agent.name)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())