Python SDK — ADK V2 Streaming

claims = client.auth.decode_token()
conversation_user_id = claims.sub or claims.user_id

result = await client.adk.converse(
    query="What were the main Q1 risks?",
    user_id=conversation_user_id,
    agent_id="agent-uuid",
    agent_version="1.0",
    channel="chat",
    reasoning_event=True,
)

print(result["session_id"])
print(result["response"]["content"])

from cosmos_sdk import ADKV2SendOptions, ADKV2TurnCallbacks


callbacks = ADKV2TurnCallbacks(
    on_chunk=lambda text: print(text, end="", flush=True),
    on_reasoning=lambda content: print(f"\nReasoning: {content}"),
    on_error=lambda code, message, fatal: print(f"\n{code}: {message}"),
)

result = await client.send_message(
    "Analyze this support case.",
    user_id=conversation_user_id,
    callbacks=callbacks,
    options=ADKV2SendOptions(
        agent_id="agent-uuid",
        agent_version="1.0",
        provider_id="azure_openai",
        model_id="gpt-4o",
        session_state={"project_id": "project-uuid"},
    ),
)

await client.send_message("What should we do next?", user_id=conversation_user_id)
client.close_adk_session()

callbacks = ADKV2TurnCallbacks(
    on_chunk=lambda text: ...,
    on_tool_start=lambda name: ...,
    on_tool_complete=lambda name, result: ...,
    on_tool_call_args_delta=lambda name, json_path, delta: ...,
    on_reasoning=lambda content: ...,
    on_error=lambda code, message, fatal: ...,
    on_custom_event=lambda event_type, payload: ...,
    on_plan_proposed=lambda plan_id, steps, summary: ...,
    on_plan_step_start=lambda plan_id, step_id: ...,
    on_plan_step_complete=lambda plan_id, step_id, status, preview: ...,
)

from cosmos_sdk import ToolDefinition, ToolResult


async def get_order(args: dict) -> ToolResult:
    order_id = args["order_id"]
    return ToolResult(ok=True, data={"order_id": order_id, "status": "shipped"})


client.tool_manager.register_tool(
    "get_order",
    handler=get_order,
    definition=ToolDefinition(
        name="get_order",
        description="Get an order by ID",
        input_schema={
            "type": "object",
            "properties": {"order_id": {"type": "string"}},
            "required": ["order_id"],
        },
    ),
)

await client.send_message(
    "Where is order 1001?",
    user_id=conversation_user_id,
    options=ADKV2SendOptions(agent_id="agent-uuid"),
)

async def get_order(args: dict) -> ToolResult:
    order = await orders.find(args["order_id"])
    if order is None:
        # The model receives: {"error": "Order 1001 not found"}
        return ToolResult(ok=False, error={"code": "TOOL_EXECUTION_FAILED", "message": f"Order {args['order_id']} not found"})
    return ToolResult(ok=True, data=order)

queued = await client.adk.submit_async(
    query="Generate a compliance report",
    user_id=conversation_user_id,
    agent_id="agent-uuid",
    agent_version="1.0",
    callback_url="https://app.example.com/webhooks/adk",
)

final = await client.adk.poll_async_until_done(queued["request_id"])
print(final["status"])

sessions = await client.adk.list_sessions(user_id=conversation_user_id, agent_id="agent-uuid")
events = await client.adk.get_session_events(sessions[0]["id"])

await client.adk.update_session_metadata(
    sessions[0]["id"],
    {"title": "Q1 risk review", "tags": ["risk", "q1"]},
)

result = await client.generate_skills(
    source="description",
    description="Help support agents resolve refund requests",
    provider_id="azure_openai",
    model_id="gpt-4o",
    max_skills=3,
)