self_query_mcp_agent_v2

Self Query MCP Discovery Agent - Proper Haive Implementation.

This agent uses Haive’s configuration system properly by extending BaseRAGAgent and using SelfQueryRetrieverConfig instead of manually creating retrievers.

Classes

MCPDocumentLoader

Load and process MCP server documents.

MCPServerMetadata

Metadata schema for MCP servers.

SelfQueryMCPAgent

Enhanced MCP Discovery Agent using proper Haive configuration system.

Functions

test_mcp_agent()

Test the MCP agent.

Module Contents

class self_query_mcp_agent_v2.MCPDocumentLoader

Load and process MCP server documents.

static create_mcp_documents() list[langchain.schema.Document]

Create documents from MCP servers data.

class self_query_mcp_agent_v2.MCPServerMetadata(/, **data: Any)

Bases: pydantic.BaseModel

Metadata schema for MCP servers.

category: str = None
has_install: bool = None
language: str = None
prompts_count: int = None
resources_count: int = None
server_name: str = None
stars: int = None
tools_count: int = None
total_features: int = None
class self_query_mcp_agent_v2.SelfQueryMCPAgent(name: str = 'mcp_self_query_agent')

Bases: haive.agents.rag.base.agent.BaseRAGAgent

Enhanced MCP Discovery Agent using proper Haive configuration system.

analyze_query_intent(query: str) str

Analyze query to determine if it’s suitable for self-query.

async search_mcp_servers(query: str, k: int = 5) list[langchain.schema.Document]

Search for MCP servers using the configured retriever.

async self_query_mcp_agent_v2.test_mcp_agent()

Test the MCP agent.