zai-adk-python-preview 0.1.0__py3-none-any.whl

zai_adk/__init__.py

@@ -0,0 +1,18 @@
+from zai_adk.agent import (
+    Agent,
+    AgentTemplate,
+    SubagentResult,
+    SubagentStatus,
+)
+from zai_adk.skills import SkillsManager, Skill, SkillMetadata, SkillContent
+
+__all__ = [
+    "Agent",
+    "AgentTemplate",
+    "SubagentResult",
+    "SubagentStatus",
+    "SkillsManager",
+    "Skill",
+    "SkillMetadata",
+    "SkillContent",
+]

zai_adk/agent/__init__.py

@@ -0,0 +1,26 @@
+from zai_adk.agent.events import (
+    AgentEvent,
+    FinalResponseEvent,
+    TextEvent,
+    ThinkingEvent,
+    ToolCallEvent,
+    ToolResultEvent,
+)
+from zai_adk.agent.service import Agent, TaskComplete
+from zai_adk.agent.types import AgentTemplate, SubagentResult, SubagentStatus
+
+__all__ = [
+    "Agent",
+    "TaskComplete",
+    # Events
+    "AgentEvent",
+    "FinalResponseEvent",
+    "TextEvent",
+    "ThinkingEvent",
+    "ToolCallEvent",
+    "ToolResultEvent",
+    # Subagent types
+    "AgentTemplate",
+    "SubagentResult",
+    "SubagentStatus",
+]

zai_adk/agent/events.py

@@ -0,0 +1,127 @@
+"""
+Event types for agent streaming.
+
+These events are yielded by `agent.query_stream()` to provide visibility
+into the agent's execution.
+
+Usage:
+    async for event in agent.query_stream("do something"):
+        match event:
+            case ToolCallEvent(tool=name, args=args):
+                print(f"Calling {name} with {args}")
+            case ToolResultEvent(tool=name, result=result):
+                print(f"{name} returned: {result}")
+            case TextEvent(content=text):
+                print(f"Assistant: {text}")
+"""
+
+import json
+from dataclasses import dataclass
+from typing import Any
+
+
+@dataclass
+class TextEvent:
+	"""Emitted when the assistant produces text content."""
+
+	content: str
+	"""The text content from the assistant."""
+
+	def __str__(self) -> str:
+		preview = self.content[:100] + '...' if len(self.content) > 100 else self.content
+		return f'💬 {preview}'
+
+
+@dataclass
+class ThinkingEvent:
+	"""Emitted when the model produces thinking/reasoning content."""
+
+	content: str
+	"""The thinking content."""
+
+	def __str__(self) -> str:
+		preview = self.content[:80] + '...' if len(self.content) > 80 else self.content
+		return f'🧠 {preview}'
+
+
+@dataclass
+class ToolCallEvent:
+	"""Emitted when the assistant calls a tool."""
+
+	tool: str
+	"""The name of the tool being called."""
+
+	args: dict[str, Any]
+	"""The arguments passed to the tool."""
+
+	tool_call_id: str
+	"""The unique ID of this tool call."""
+
+	display_name: str = ''
+	"""Human-readable description of the tool call (e.g., 'Browsing https://...')."""
+
+	def __str__(self) -> str:
+		if self.display_name:
+			return f'🔧 {self.display_name}'
+		args_str = json.dumps(self.args, default=str)
+		if len(args_str) > 80:
+			args_str = args_str[:77] + '...'
+		return f'🔧 {self.tool}({args_str})'
+
+
+@dataclass
+class ToolResultEvent:
+	"""Emitted when a tool returns a result."""
+
+	tool: str
+	"""The name of the tool that was called."""
+
+	result: str
+	"""The result returned by the tool."""
+
+	tool_call_id: str
+	"""The unique ID of the tool call this result corresponds to."""
+
+	is_error: bool = False
+	"""Whether the tool execution resulted in an error."""
+
+	def __str__(self) -> str:
+		prefix = '❌' if self.is_error else '✓'
+		preview = self.result[:80] + '...' if len(self.result) > 80 else self.result
+		return f'   {prefix} {self.tool}: {preview}'
+
+
+@dataclass
+class FinalResponseEvent:
+	"""Emitted when the agent produces its final response."""
+
+	content: str
+	"""The final response content."""
+
+	def __str__(self) -> str:
+		return f'✅ Final: {self.content[:100]}...' if len(self.content) > 100 else f'✅ Final: {self.content}'
+
+
+@dataclass
+class HiddenUserMessageEvent:
+	"""Emitted when the agent injects a hidden user message (ex: incomplete todos prompt).
+	Hidden messages are saved to history and sent to the LLM but not displayed in the UI.
+	"""
+
+	content: str
+	"""The content of the hidden user message."""
+
+	def __str__(self) -> str:
+		preview = self.content[:80] + '...' if len(self.content) > 80 else self.content
+		return f'👻 Hidden: {preview}'
+
+
+# Union type for all events
+AgentEvent = (
+	TextEvent
+	| ThinkingEvent
+	| ToolCallEvent
+	| ToolResultEvent
+	| FinalResponseEvent
+	| HiddenUserMessageEvent
+)