lite-agent 0.8.0__tar.gz → 0.9.0__tar.gz

{lite_agent-0.8.0 → lite_agent-0.9.0}/CHANGELOG.md

@@ -1,3 +1,21 @@
+## v0.9.0
+
+[v0.8.0...v0.9.0](https://github.com/Jannchie/lite-agent/compare/v0.8.0...v0.9.0)
+
+### :sparkles: Features
+
+- **agent**: improve assistant message tool_call handling - By [Jannchie](mailto:jannchie@gmail.com) in [45e6de8](https://github.com/Jannchie/lite-agent/commit/45e6de8)
+- **chat-display**: add column-style message display and show model info - By [Jannchie](mailto:jannchie@gmail.com) in [3559137](https://github.com/Jannchie/lite-agent/commit/3559137)
+- **tests**: add tests for client message_builder and response handlers - By [Jannchie](mailto:jannchie@gmail.com) in [ceec7d7](https://github.com/Jannchie/lite-agent/commit/ceec7d7)
+
+### :art: Refactors
+
+- **examples**: update examples to use new message and content types - By [Jannchie](mailto:jannchie@gmail.com) in [59b4618](https://github.com/Jannchie/lite-agent/commit/59b4618)
+
+### :wrench: Chores
+
+- **deps**: bump funcall to 0.11.0 - By [Jannchie](mailto:jannchie@gmail.com) in [eb5e7ae](https://github.com/Jannchie/lite-agent/commit/eb5e7ae)
+
 ## v0.8.0
 
 [v0.7.0...v0.8.0](https://github.com/Jannchie/lite-agent/compare/v0.7.0...v0.8.0)

{lite_agent-0.8.0 → lite_agent-0.9.0}/PKG-INFO

@@ -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 → lite_agent-0.9.0}/examples/basic.py

@@ -31,7 +31,7 @@ agent = Agent(
 
 
 async def main():
-    runner = Runner(agent, streaming=False, api="completion")
+    runner = Runner(agent, streaming=True, api="responses")
     resp = await runner.run_until_complete(
         "What is the temperature in New York?",
         includes=["usage", "assistant_message", "function_call", "function_call_output", "timing"],

{lite_agent-0.8.0 → lite_agent-0.9.0}/examples/basic_agent.py

@@ -4,6 +4,7 @@ import logging
 from rich.logging import RichHandler
 
 from lite_agent.agent import Agent
+from lite_agent.types import NewUserMessage, UserTextContent
 
 logging.basicConfig(
     level=logging.WARNING,
@@ -39,10 +40,9 @@ agent = Agent(
 async def main():
     resp = await agent.completion(
         [
-            {
-                "role": "user",
-                "content": "What is the temperature and whether in New York?",
-            },
+            NewUserMessage(
+                content=[UserTextContent(text="What is the temperature and whether in New York?")],
+            ),
         ],
     )
     async for chunk in resp:

{lite_agent-0.8.0 → lite_agent-0.9.0}/examples/cancel_and_transfer_demo.py

@@ -6,6 +6,7 @@ Demo showing two new features:
 """
 
 import asyncio
+import json
 import logging
 
 from funcall.decorators import tool
@@ -13,7 +14,8 @@ from rich.logging import RichHandler
 
 from lite_agent.agent import Agent
 from lite_agent.runner import Runner
-from lite_agent.types import AssistantTextContent, AssistantToolCall, NewAssistantMessage
+from lite_agent.types import AssistantTextContent, AssistantToolCall, AssistantToolCallResult, NewAssistantMessage, ToolCall, ToolCallFunction
+from lite_agent.types.events import FunctionCallOutputEvent
 
 logging.basicConfig(
     level=logging.WARNING,
@@ -106,9 +108,9 @@ async def demo_1_cancel_pending_tools():
             chunks.append(chunk)
             if len(chunks) >= 3:  # Stop after getting cancellation events
                 break
-    except Exception:
+    except Exception as e:
         # Expected - no real LLM configured
-        pass
+        logger.debug("Expected exception in demo: %s", e)
 
     print(f"Events from run() method: {len(chunks)}")
     for i, chunk in enumerate(chunks):
@@ -156,7 +158,7 @@ async def demo_1_cancel_pending_tools():
         print(f"Content items in assistant message: {len(assistant_msg.content)}")
 
         # Show the cancellation results
-        cancellation_results = [item for item in assistant_msg.content if item.type == "tool_call_result"]
+        cancellation_results = [item for item in assistant_msg.content if isinstance(item, AssistantToolCallResult)]
         print(f"Cancellation results added: {len(cancellation_results)}")
         for result in cancellation_results:
             print(f"  - {result.call_id}: {result.output}")
@@ -194,9 +196,6 @@ async def demo_2_transfer_events():
     print(f"Current agent: {runner.agent.name}")
 
     # Simulate transfer_to_agent call
-    import json
-
-    from lite_agent.types import ToolCall, ToolCallFunction
 
     transfer_call = ToolCall(
         type="function",
@@ -209,13 +208,14 @@ async def demo_2_transfer_events():
     )
 
     # Handle transfer and collect events
-    chunks = []
-    async for chunk in runner._handle_tool_calls([transfer_call], ["function_call_output"]):
-        chunks.append(chunk)
+    chunks = [chunk async for chunk in runner._handle_tool_calls([transfer_call], ["function_call_output"])]
 
     print(f"Events generated: {len(chunks)}")
     for i, chunk in enumerate(chunks):
-        print(f"  Event {i + 1}: {chunk.type} - {chunk.content}")
+        if isinstance(chunk, FunctionCallOutputEvent):
+            print(f"  Event {i + 1}: {chunk.type} - {chunk.content}")
+        else:
+            print(f"  Event {i + 1}: {chunk.type}")
 
     print(f"Agent after transfer: {runner.agent.name}")
 
@@ -232,13 +232,14 @@ async def demo_2_transfer_events():
     )
 
     # Handle parent transfer and collect events
-    chunks = []
-    async for chunk in runner._handle_tool_calls([parent_transfer_call], ["function_call_output"]):
-        chunks.append(chunk)
+    chunks = [chunk async for chunk in runner._handle_tool_calls([parent_transfer_call], ["function_call_output"])]
 
     print(f"Events generated: {len(chunks)}")
     for i, chunk in enumerate(chunks):
-        print(f"  Event {i + 1}: {chunk.type} - {chunk.content}")
+        if isinstance(chunk, FunctionCallOutputEvent):
+            print(f"  Event {i + 1}: {chunk.type} - {chunk.content}")
+        else:
+            print(f"  Event {i + 1}: {chunk.type}")
 
     print(f"Agent after parent transfer: {runner.agent.name}")
 
@@ -265,13 +266,14 @@ async def demo_2_transfer_events():
         index=1,
     )
 
-    chunks = []
-    async for chunk in runner._handle_tool_calls([transfer_call_1, transfer_call_2], ["function_call_output"]):
-        chunks.append(chunk)
+    chunks = [chunk async for chunk in runner._handle_tool_calls([transfer_call_1, transfer_call_2], ["function_call_output"])]
 
     print(f"Events generated: {len(chunks)}")
     for i, chunk in enumerate(chunks):
-        print(f"  Event {i + 1}: {chunk.content}")
+        if isinstance(chunk, FunctionCallOutputEvent):
+            print(f"  Event {i + 1}: {chunk.content}")
+        else:
+            print(f"  Event {i + 1}: {chunk.type}")
 
     print(f"Final agent: {runner.agent.name}")
 

lite_agent-0.9.0/examples/knowledge/main.py

@@ -0,0 +1,67 @@
+import asyncio
+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(
+    level=logging.WARNING,
+    format="%(message)s",
+    datefmt="[%X]",
+    handlers=[RichHandler(rich_tracebacks=True)],
+)
+
+logger = logging.getLogger("lite_agent")
+logger.setLevel(logging.DEBUG)
+
+
+def list_knowledges() -> list[str]:
+    """
+    列出系统中存在的知识，返回的是知识的标题
+    """
+    return ["苹果的价格的计算方法", "香蕉的价格", "茄子的价格", "芒果的价格", "其他水果蔬菜的价格"]
+
+
+def read_knowledge(knowledge: str) -> str:
+    """
+    通过知识的标题获取其内容
+    """
+    match knowledge:
+        case "苹果的价格的计算方法":
+            return "苹果的价格是两倍的香蕉的价格减去茄子的价格"
+        case "香蕉的价格":
+            return "香蕉的价格为3元"
+        case "其他水果蔬菜的价格":
+            return "茄子的价格为5元"
+        case "茄子的价格":
+            return "茄子的价格请参阅其他知识库"
+        case "芒果的价格":
+            return "芒果的价格为7元"
+        case _:
+            return "不存在的知识"
+
+
+agent = Agent(
+    model="gpt-5-mini",
+    name="Assistant",
+    instructions="你是一个有帮助的助手。",
+    tools=[list_knowledges, read_knowledge],
+    reasoning={"effort": "minimal"},
+)
+
+
+async def main():
+    runner = Runner(agent, streaming=True)
+    await runner.run_until_complete(
+        "苹果的价格是多少？查询知识库告诉我答案。",
+        includes=["usage", "assistant_message", "function_call", "function_call_output", "timing"],
+    )
+
+    display_messages(runner.messages)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

{lite_agent-0.8.0 → lite_agent-0.9.0}/examples/non_streaming.py

@@ -3,6 +3,7 @@ Simple example demonstrating non-streaming mode in LiteAgent.
 """
 
 import asyncio
+from datetime import datetime, timezone
 
 from lite_agent import Agent, Runner
 
@@ -33,9 +34,7 @@ async def main():
     # Example with a simple tool
     def get_time() -> str:
         """Get the current time."""
-        from datetime import datetime
-
-        return f"Current time: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+        return f"Current time: {datetime.now(timezone.utc).strftime('%Y-%m-%d %H:%M:%S')}"
 
     agent_with_tools = Agent(
         name="TimeAgent",

{lite_agent-0.8.0 → lite_agent-0.9.0}/examples/response_api_example.py

@@ -86,7 +86,7 @@ async def main():
         )
 
         # This will raise an error when trying to convert
-        converted_messages = agent._convert_responses_to_completions_format(runner.messages)  # noqa: SLF001
+        converted_messages = agent._convert_responses_to_completions_format(runner.messages)
         print("❌ This should not be reached!")
 
     except ValueError as e:
@@ -108,7 +108,7 @@ async def main():
     print("\n=== Converted messages for LLM API ===")
     # Note: In a real application, this conversion happens automatically
     # We're accessing the private method here just for demonstration
-    converted_messages = agent._convert_responses_to_completions_format(runner.messages)  # noqa: SLF001
+    converted_messages = agent._convert_responses_to_completions_format(runner.messages)
     for i, message in enumerate(converted_messages):
         print(f"Converted Message {i + 1}:")
         print(f"  {message}")

{lite_agent-0.8.0 → lite_agent-0.9.0}/examples/simple_debug.py

@@ -3,6 +3,7 @@ Simple debug to check non-streaming response.
 """
 
 import asyncio
+import traceback
 
 from lite_agent import Agent, Runner
 
@@ -29,8 +30,6 @@ async def main():
             print(f"First chunk: {chunks[0]}")
     except Exception as e:
         print(f"Error: {e}")
-        import traceback
-
         traceback.print_exc()
 
 

lite_agent-0.9.0/htmlcov/.gitignore

@@ -0,0 +1,2 @@
+# Created by coverage.py
+*