pydantic-deepagents/pydantic_deep/mcp/__init__.py at 0585a72c9f05c6aa31f47625ab56a979f5b27d9b · vstorm-co/pydantic-deepagents · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
"""MCP (Model Context Protocol) client support for pydantic-deep.

Declarative server configs (:class:`MCPServerConfig` / :class:`MCPAuth`), a
secret-aware :class:`MCPRegistry` that builds connected pydantic-ai toolsets,
curated built-in servers, and a connection probe. The optional ``mcp`` extra
provides the underlying transport (``pip install 'pydantic-deep[mcp]'``).
"""

from __future__ import annotations

from pydantic_deep.mcp.builtins import BUILTIN_MCP_NAMES, builtin_mcp_servers
from pydantic_deep.mcp.config import (
    MCPAuth,
    MCPAuthKind,
    MCPConfigError,
    MCPServerConfig,
    MCPTransport,
)
from pydantic_deep.mcp.loader import expand_env_vars, parse_mcp_servers
from pydantic_deep.mcp.registry import (
    MCP_INSTALL_HINT,
    HttpClientFactory,
    MCPNotInstalledError,
    MCPProbeResult,
    MCPRegistry,
    SecretResolver,
    auth_satisfied,
    build_mcp_server,
    make_resilient,
    probe_mcp_server,
)
from pydantic_deep.mcp.resources import (
    SKILL_DOC_NAME,
    SKILL_URI_SCHEME,
    MCPResourceProvider,
    create_mcp_resources_toolset,
)

__all__ = [
    "MCPAuth",
    "MCPAuthKind",
    "MCPConfigError",
    "MCPServerConfig",
    "MCPTransport",
    "MCPRegistry",
    "MCPProbeResult",
    "MCPNotInstalledError",
    "MCP_INSTALL_HINT",
    "SecretResolver",
    "HttpClientFactory",
    "auth_satisfied",
    "build_mcp_server",
    "make_resilient",
    "probe_mcp_server",
    "create_mcp_resources_toolset",
    "MCPResourceProvider",
    "SKILL_URI_SCHEME",
    "SKILL_DOC_NAME",
    "builtin_mcp_servers",
    "BUILTIN_MCP_NAMES",
    "parse_mcp_servers",
    "expand_env_vars",
]