langchain-dns-aid

DNS-AID agent discovery middleware for LangChain. Enables agents to discover, publish, and unpublish themselves via DNS using the DNS-AID protocol.

Why middleware instead of tools?

Unlike tool-only integrations (CrewAI, LlamaIndex, etc.), the middleware approach uses LangChain's lifecycle hooks to auto-publish the agent on startup and auto-unpublish on shutdown. This prevents stale DNS records when agents crash or forget to clean up.

Install

pip install langchain-dns-aid

Quick start

from langchain import create_agent, init_chat_model
from langchain_dns_aid import DnsAidMiddleware

agent = create_agent(
    model=init_chat_model("gpt-4o"),
    middleware=[DnsAidMiddleware(
        agent_name="my-assistant",
        domain="agents.example.com",
        endpoint="assistant.example.com",
        capabilities=["code-review", "documentation"],
        backend_name="route53",
    )],
)

# Agent is auto-published to DNS on startup.
# It can discover other agents during conversation.
# On shutdown, it is auto-unpublished.

What it does

Configuration

Discovery only (no auto-publish)

agent = create_agent(
    model=init_chat_model("gpt-4o"),
    middleware=[DnsAidMiddleware(
        auto_publish=False,
        auto_unpublish=False,
    )],
)
# Agent gets dns_aid_discover tool but doesn't publish itself.

License

Apache-2.0