mcp_simple_tool_agent

MCP Simple Tool Agent - Using SimpleAgent with retrieval tool.

Attributes

Classes

QueryRequest

!!! abstract "Usage Documentation"

QueryResponse

!!! abstract "Usage Documentation"

Functions

ask_agent(request)

Ask the MCP agent a question.

create_mcp_documents(...)

Create documents from MCP server data.

create_mcp_tool_agent(...)

Create a SimpleAgent with MCP search tool.

initialize_vector_store()

Initialize the vector store with MCP documents.

root()

Web interface for the MCP agent.

search_mcp_servers(→ str)

Search for MCP servers based on query.

startup_event()

Initialize the MCP tool agent.

Module Contents

class mcp_simple_tool_agent.QueryRequest(/, **data: Any)

Bases: pydantic.BaseModel

!!! abstract “Usage Documentation”

[Models](../concepts/models.md)

A base class for creating Pydantic models.

__class_vars__

The names of the class variables defined on the model.

__private_attributes__

Metadata about the private attributes of the model.

__signature__

The synthesized __init__ [Signature][inspect.Signature] of the model.

__pydantic_complete__

Whether model building is completed, or if there are still undefined fields.

__pydantic_core_schema__

The core schema of the model.

__pydantic_custom_init__

Whether the model has a custom __init__ function.

__pydantic_decorators__

Metadata containing the decorators defined on the model. This replaces Model.__validators__ and Model.__root_validators__ from Pydantic V1.

__pydantic_generic_metadata__

Metadata for generic models; contains data used for a similar purpose to __args__, __origin__, __parameters__ in typing-module generics. May eventually be replaced by these.

__pydantic_parent_namespace__

Parent namespace of the model, used for automatic rebuilding of models.

__pydantic_post_init__

The name of the post-init method for the model, if defined.

__pydantic_root_model__

Whether the model is a [RootModel][pydantic.root_model.RootModel].

__pydantic_serializer__

The pydantic-core SchemaSerializer used to dump instances of the model.

__pydantic_validator__

The pydantic-core SchemaValidator used to validate instances of the model.

__pydantic_fields__

A dictionary of field names and their corresponding [FieldInfo][pydantic.fields.FieldInfo] objects.

__pydantic_computed_fields__

A dictionary of computed field names and their corresponding [ComputedFieldInfo][pydantic.fields.ComputedFieldInfo] objects.

__pydantic_extra__

A dictionary containing extra values, if [extra][pydantic.config.ConfigDict.extra] is set to ‘allow’.

__pydantic_fields_set__

The names of fields explicitly set during instantiation.

__pydantic_private__

Values of private attributes set on the model instance.

query: str
class mcp_simple_tool_agent.QueryResponse(/, **data: Any)

Bases: pydantic.BaseModel

!!! abstract “Usage Documentation”

[Models](../concepts/models.md)

A base class for creating Pydantic models.

__class_vars__

The names of the class variables defined on the model.

__private_attributes__

Metadata about the private attributes of the model.

__signature__

The synthesized __init__ [Signature][inspect.Signature] of the model.

__pydantic_complete__

Whether model building is completed, or if there are still undefined fields.

__pydantic_core_schema__

The core schema of the model.

__pydantic_custom_init__

Whether the model has a custom __init__ function.

__pydantic_decorators__

Metadata containing the decorators defined on the model. This replaces Model.__validators__ and Model.__root_validators__ from Pydantic V1.

__pydantic_generic_metadata__

Metadata for generic models; contains data used for a similar purpose to __args__, __origin__, __parameters__ in typing-module generics. May eventually be replaced by these.

__pydantic_parent_namespace__

Parent namespace of the model, used for automatic rebuilding of models.

__pydantic_post_init__

The name of the post-init method for the model, if defined.

__pydantic_root_model__

Whether the model is a [RootModel][pydantic.root_model.RootModel].

__pydantic_serializer__

The pydantic-core SchemaSerializer used to dump instances of the model.

__pydantic_validator__

The pydantic-core SchemaValidator used to validate instances of the model.

__pydantic_fields__

A dictionary of field names and their corresponding [FieldInfo][pydantic.fields.FieldInfo] objects.

__pydantic_computed_fields__

A dictionary of computed field names and their corresponding [ComputedFieldInfo][pydantic.fields.ComputedFieldInfo] objects.

__pydantic_extra__

A dictionary containing extra values, if [extra][pydantic.config.ConfigDict.extra] is set to ‘allow’.

__pydantic_fields_set__

The names of fields explicitly set during instantiation.

__pydantic_private__

Values of private attributes set on the model instance.

query: str
response: str
timestamp: str
async mcp_simple_tool_agent.ask_agent(request: QueryRequest)

Ask the MCP agent a question.

mcp_simple_tool_agent.create_mcp_documents() list[langchain_core.documents.Document]

Create documents from MCP server data.

mcp_simple_tool_agent.create_mcp_tool_agent(llm_config: haive.core.models.llm.base.LLMConfig | None = None) haive.agents.simple.agent.SimpleAgent

Create a SimpleAgent with MCP search tool.

mcp_simple_tool_agent.initialize_vector_store()

Initialize the vector store with MCP documents.

async mcp_simple_tool_agent.root()

Web interface for the MCP agent.

mcp_simple_tool_agent.search_mcp_servers(query: str, k: int = 5) str

Search for MCP servers based on query.

Parameters:
  • query – Search query (e.g., “python database”, “github integration”)

  • k – Number of results to return (default: 5)

Returns:

Formatted string with search results

async mcp_simple_tool_agent.startup_event()

Initialize the MCP tool agent.

mcp_simple_tool_agent.VECTOR_STORE = None
mcp_simple_tool_agent.app
mcp_simple_tool_agent.mcp_agent = None