mcp.integration.aug_llm_mcp_extension¶

Extension module to add MCP support to AugLLMConfig.

This module provides utilities and mixins to enhance AugLLMConfig with MCP integration capabilities, allowing seamless use of MCP tools, resources, and prompts within the Haive agent framework.

Attributes¶

logger

Classes¶

`MCPAugLLMConfig`	Extended AugLLMConfig with MCP integration support.
`MCPPromptTemplate`	Model representing an MCP prompt template.
`MCPResource`	Model representing an MCP resource.
`MCPToolWrapper`	Wrapper to convert MCP tools to Haive-compatible tools.

Functions¶

`create_mcp_enabled_aug_config`(→ MCPAugLLMConfig)	Factory function to create an MCP-enabled AugLLMConfig.
`extend_aug_llm_config_for_mcp`(→ MCPAugLLMConfig)	Utility function to extend an existing AugLLMConfig with MCP support.

Module Contents¶

class mcp.integration.aug_llm_mcp_extension.MCPAugLLMConfig¶

Bases: haive.core.engine.aug_llm.AugLLMConfig

Extended AugLLMConfig with MCP integration support.

This class extends the base AugLLMConfig to add MCP-specific fields and functionality, enabling agents to use MCP servers for tools, resources, and prompts.

enhance_system_prompt_with_mcp() → str¶

Enhance the system prompt with MCP information.

Returns:: Enhanced system prompt including MCP resources and capabilities

async get_mcp_resource_content(uri: str) → Any¶

Fetch content for an MCP resource.

Parameters:: uri – Resource URI
Returns:: Resource content

async setup_mcp() → None¶

Initialize MCP integration.

Sets up the MCP manager, discovers tools, loads resources, and configures prompts based on the MCP configuration.

auto_discover_mcp_tools: bool = None¶

inject_mcp_resources: bool = None¶

mcp_config: haive.mcp.config.MCPConfig | None = None¶

mcp_manager: haive.mcp.manager.MCPManager | None = None¶

mcp_prompts: dict[str, MCPPromptTemplate] | None = None¶

mcp_resources: list[MCPResource] | None = None¶

use_mcp_prompts: bool = None¶

class mcp.integration.aug_llm_mcp_extension.MCPPromptTemplate(/, **data: Any)¶

Bases: pydantic.BaseModel

Model representing an MCP prompt template.

arguments: list[dict[str, Any]] = None¶

description: str = None¶

name: str = None¶

template: str = None¶

class mcp.integration.aug_llm_mcp_extension.MCPResource(/, **data: Any)¶

Bases: pydantic.BaseModel

Model representing an MCP resource.

content: Any | None = None¶

description: str = None¶

mime_type: str = None¶

name: str = None¶

uri: str = None¶

class mcp.integration.aug_llm_mcp_extension.MCPToolWrapper(**kwargs: Any)¶

Bases: langchain_core.tools.BaseTool

Wrapper to convert MCP tools to Haive-compatible tools.

This wrapper allows MCP tools to be used seamlessly within the Haive framework by adapting their interface to match BaseTool expectations.

description: str¶

Used to tell the model how/when/why to use the tool.

You can provide few-shot examples as a part of the description.

mcp_client: Any¶

mcp_tool: dict[str, Any]¶

name: str¶: The unique name of the tool that clearly communicates its purpose.

async mcp.integration.aug_llm_mcp_extension.create_mcp_enabled_aug_config(name: str, model: str = 'gpt-4o-mini', mcp_servers: dict[str, haive.mcp.config.MCPServerConfig] | None = None, **kwargs) → MCPAugLLMConfig¶

Factory function to create an MCP-enabled AugLLMConfig.

Parameters:

name – Configuration name
model – LLM model to use
mcp_servers – Dictionary of MCP server configurations
**kwargs – Additional AugLLMConfig parameters

Returns:

Initialized MCPAugLLMConfig with MCP integration

mcp.integration.aug_llm_mcp_extension.extend_aug_llm_config_for_mcp(base_config: haive.core.engine.aug_llm.AugLLMConfig, mcp_config: haive.mcp.config.MCPConfig) → MCPAugLLMConfig¶

Utility function to extend an existing AugLLMConfig with MCP support.

Parameters:

base_config – Existing AugLLMConfig instance
mcp_config – MCP configuration to add

Returns:

MCPAugLLMConfig with MCP support added

mcp.integration.aug_llm_mcp_extension.logger¶