dapr-agents-ext-drasi

Drasi extension for Dapr Agents, enabling resilient, scalable, and business event-driven AI agents through Drasi's change notification capabilities.

Features

• Directly trigger agents from Drasi change events via Dapr pub/sub

Getting Started

Prerequisites

This extension is installed as an optional dependency on the core dapr-agents package; see the Getting Started section in the root README for a list of prerequisites.

Installation

uv add dapr-agents[drasi]

Public API

from dapr_agents.ext.drasi import (
    drasi_trigger,                  # Register Drasi query subscriptions for agents
    DrasiChangeEvent,               # Type for Drasi change events
    DrasiOperation,                 # Operation type for Drasi change events
)

Usage

Register a Drasi query subscription on an agent before hosting:

agent = DurableAgent(...)

drasi_trigger(
    agent,
    query_id="<YOUR_DRASI_QUERY_ID>",
    task_mapper=lambda event, ctx: TriggerAction(task="<AGENT_TASK_MESSAGE>")
)

runner = AgentRunner()
try:
    runner.subscribe(agent)
    await wait_for_shutdown()
finally:
    runner.shutdown(agent)

Examples

• Drasi Change-Driven Agents on Kubernetes — demonstrates how to subscribe an agent to Drasi queries in a Kubernetes environment.

Development

Install extension in editable mode

From the project root:

uv venv
source .venv/bin/activate
uv sync --active --group dev --group test --extra drasi

Run extension tests

uv run --group test pytest ext/dapr-agents-ext-drasi -m "not integration" -v

Extension code quality

See the Code Quality section in the development README for code quality commands.

Regenerate Drasi models

See the provenance file for context.

./scripts/regen-drasi-models.sh