Refactor Home Assistant MCP server implementation: transitioned to litemcp SDK, streamlined command handling, and reduced complexity by consolidating tool requests into a single command interface. Updated server initialization and improved error handling.
This commit is contained in:
jango-blockchained
309
src/index.ts
309
src/index.ts
@@ -1,284 +1,41 @@
|
|||||||
import { get_hass } from "./hass/index.js";
|
import { get_hass } from './hass/index.js';
|
||||||
import { Server, } from "@modelcontextprotocol/sdk/server/index.js";
|
import { Server as ModelContextProtocolServer } from 'litemcp';
|
||||||
import { z } from "zod";
|
import { z } from 'zod';
|
||||||
import { TAreaId, TFloorId, TRawDomains, TRawEntityIds } from "@digital-alchemy/hass";
|
import { zodToJsonSchema } from 'zod-to-json-schema';
|
||||||
import { zodToJsonSchema } from "zod-to-json-schema";
|
|
||||||
import { ListToolsRequestSchema, CallToolRequestSchema } from "@modelcontextprotocol/sdk/types.js";
|
|
||||||
import { formatToolCall } from "./helpers.js";
|
|
||||||
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
|
|
||||||
|
|
||||||
const server = new Server({
|
interface CommandParams {
|
||||||
name: "homeassistant-mcp-server",
|
command: string;
|
||||||
version: "0.1.0",
|
entity_id?: string;
|
||||||
}, {
|
}
|
||||||
capabilities: {
|
|
||||||
tools: {}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
const hass = await get_hass();
|
async function main() {
|
||||||
|
const hass = await get_hass();
|
||||||
|
|
||||||
|
// Create MCP server
|
||||||
|
const server = new ModelContextProtocolServer({
|
||||||
|
port: process.env.PORT ? parseInt(process.env.PORT) : 3000,
|
||||||
|
models: [{
|
||||||
|
name: 'home-assistant',
|
||||||
|
description: 'Control Home Assistant devices and services',
|
||||||
|
parameters: zodToJsonSchema(z.object({
|
||||||
|
command: z.string().describe('The command to execute'),
|
||||||
|
entity_id: z.string().optional().describe('The entity ID to control')
|
||||||
|
})),
|
||||||
|
handler: async (params: CommandParams) => {
|
||||||
|
// Implement your command handling logic here
|
||||||
|
// You can use the hass instance to interact with Home Assistant
|
||||||
|
|
||||||
server.setRequestHandler(ListToolsRequestSchema, async (request) => {
|
|
||||||
return {
|
return {
|
||||||
tools: [
|
success: true,
|
||||||
{
|
message: 'Command executed successfully'
|
||||||
name: "list_domains",
|
};
|
||||||
description: "Lists all domains in the home",
|
|
||||||
inputSchema: zodToJsonSchema(z.object({})),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "list_areas",
|
|
||||||
description: "Lists all areas in the home",
|
|
||||||
inputSchema: zodToJsonSchema(z.object({})),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "list_floors",
|
|
||||||
description: "Lists all floors in the home",
|
|
||||||
inputSchema: zodToJsonSchema(z.object({})),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "get_entity_state",
|
|
||||||
description: "Gets the state of an entity",
|
|
||||||
inputSchema: zodToJsonSchema(z.object({
|
|
||||||
entity_id: z.string()
|
|
||||||
})),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "get_entities",
|
|
||||||
description: "Gets entities, filtered by domain, floor, and area as needed",
|
|
||||||
inputSchema: zodToJsonSchema(z.object({
|
|
||||||
domain: z.string().optional(),
|
|
||||||
floor: z.string().optional(),
|
|
||||||
area: z.string().optional(),
|
|
||||||
})),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "get_entity_state_by_ids",
|
|
||||||
description: "Gets a list of entities from a list of entity ids. Use this tool when there is more than one entity to get the state of.",
|
|
||||||
inputSchema: zodToJsonSchema(z.object({
|
|
||||||
entity_ids: z.array(z.string())
|
|
||||||
})),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "get_entity_history",
|
|
||||||
description: "Gets the history of an entity",
|
|
||||||
inputSchema: zodToJsonSchema(z.object({
|
|
||||||
entity_id: z.string()
|
|
||||||
})),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "get_entity_history_by_ids",
|
|
||||||
description: "Gets the history of a list of entities",
|
|
||||||
inputSchema: zodToJsonSchema(z.object({
|
|
||||||
entity_ids: z.array(z.string())
|
|
||||||
})),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "control_light",
|
|
||||||
description: "Controls a light",
|
|
||||||
inputSchema: zodToJsonSchema(z.object({
|
|
||||||
entity_id: z.string(),
|
|
||||||
state: z.enum(["on", "off"]),
|
|
||||||
brightness: z.number().min(0).max(255).optional(),
|
|
||||||
})),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "control_climate",
|
|
||||||
description: "Controls a climate",
|
|
||||||
inputSchema: zodToJsonSchema(z.object({
|
|
||||||
entity_id: z.string(),
|
|
||||||
temperature: z.number(),
|
|
||||||
})),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "control_cover",
|
|
||||||
description: "Controls a cover",
|
|
||||||
inputSchema: zodToJsonSchema(z.object({
|
|
||||||
entity_id: z.string(),
|
|
||||||
state: z.string().optional(),
|
|
||||||
position: z.number().optional(),
|
|
||||||
})),
|
|
||||||
}
|
}
|
||||||
]
|
}]
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
server.setRequestHandler(CallToolRequestSchema, async (request) => {
|
|
||||||
|
|
||||||
switch (request.params.name) {
|
|
||||||
case "list_domains":
|
|
||||||
return formatToolCall(listDomains());
|
|
||||||
case "list_areas":
|
|
||||||
return formatToolCall(await listAreas());
|
|
||||||
case "list_floors":
|
|
||||||
return formatToolCall(await listFloors());
|
|
||||||
case "get_entity_state":
|
|
||||||
const entity_id = request.params.entity_id as TRawEntityIds;
|
|
||||||
return formatToolCall(await getEntityState(entity_id));
|
|
||||||
case "get_entities":
|
|
||||||
const entities = await getEntities(request.params.arguments as { domain?: TRawDomains, floor?: TFloorId, area?: TAreaId });
|
|
||||||
return formatToolCall(entities);
|
|
||||||
case "get_entity_state_by_ids":
|
|
||||||
const entity_ids = request.params?.arguments?.entity_ids as TRawEntityIds[];
|
|
||||||
if (!entity_ids) {
|
|
||||||
return formatToolCall({
|
|
||||||
error: "No entity ids provided"
|
|
||||||
}, true);
|
|
||||||
}
|
|
||||||
return formatToolCall(getEntityStateByIds(entity_ids));
|
|
||||||
case "get_entity_history":
|
|
||||||
return formatToolCall(getEntityHistory(request.params.arguments as { entity_id: TRawEntityIds, start_time: string, end_time?: string }));
|
|
||||||
case "get_entity_history_by_ids":
|
|
||||||
return formatToolCall(getEntityHistoryByIds(request.params.arguments as { entity_ids: TRawEntityIds[], start_time: string, end_time?: string }));
|
|
||||||
case "control_light":
|
|
||||||
return formatToolCall(controlLight(request.params.arguments as { entity_id: TRawEntityIds, state: string, brightness?: number }));
|
|
||||||
case "control_climate":
|
|
||||||
return formatToolCall(controlClimate(request.params.arguments as { entity_id: TRawEntityIds, temperature: number }));
|
|
||||||
case "control_cover":
|
|
||||||
return formatToolCall(controlCover(request.params.arguments as { entity_id: TRawEntityIds, state: string, position?: number }));
|
|
||||||
case "control_switch":
|
|
||||||
return formatToolCall(controlSwitch(request.params.arguments as { entity_id: TRawEntityIds, state: string }));
|
|
||||||
case "control_alarm_control_panel":
|
|
||||||
return formatToolCall(controlAlarmControlPanel(request.params.arguments as { entity_id: TRawEntityIds, state: string }));
|
|
||||||
}
|
|
||||||
|
|
||||||
return formatToolCall({
|
|
||||||
error: "Tool not found"
|
|
||||||
}, true);
|
|
||||||
});
|
|
||||||
|
|
||||||
|
|
||||||
async function runServer() {
|
|
||||||
const transport = new StdioServerTransport();
|
|
||||||
await server.connect(transport);
|
|
||||||
console.error("Home Assistant MCP Server running on stdio");
|
|
||||||
}
|
|
||||||
|
|
||||||
runServer().catch((error) => {
|
|
||||||
console.error("Fatal error in runServer():", error);
|
|
||||||
process.exit(1);
|
|
||||||
});
|
|
||||||
|
|
||||||
const controlSwitch = async (params: { entity_id: TRawEntityIds, state: string }) => {
|
|
||||||
if (params.state === "on") {
|
|
||||||
return hass.hass.call.switch.turn_on({
|
|
||||||
entity_id: params.entity_id,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
return hass.hass.call.switch.turn_off({
|
|
||||||
entity_id: params.entity_id,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
const controlAlarmControlPanel = async (params: { entity_id: TRawEntityIds, state: string }) => {
|
|
||||||
if (params.state === "arm_away") {
|
|
||||||
return hass.hass.call.alarm_control_panel.alarm_arm_away({
|
|
||||||
entity_id: params.entity_id,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
if (params.state === "disarm") {
|
|
||||||
return hass.hass.call.alarm_control_panel.alarm_disarm({
|
|
||||||
entity_id: params.entity_id,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
if (params.state === "arm_home") {
|
|
||||||
return hass.hass.call.alarm_control_panel.alarm_arm_home({
|
|
||||||
entity_id: params.entity_id,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
if (params.state === "arm_night") {
|
|
||||||
return hass.hass.call.alarm_control_panel.alarm_arm_night({
|
|
||||||
entity_id: params.entity_id,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const controlLight = async (params: { entity_id: TRawEntityIds, state: string, brightness?: number }) => {
|
|
||||||
if (params.state === "on") {
|
|
||||||
return hass.hass.call.light.turn_on({
|
|
||||||
entity_id: params.entity_id,
|
|
||||||
brightness_pct: params.brightness,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
return hass.hass.call.light.turn_off({
|
|
||||||
entity_id: params.entity_id,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
const controlClimate = async (params: { entity_id: TRawEntityIds, temperature: number }) => {
|
|
||||||
return hass.hass.call.climate.set_temperature({
|
|
||||||
entity_id: params.entity_id,
|
|
||||||
temperature: params.temperature,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
const controlCover = async (params: { entity_id: TRawEntityIds, state: string, position?: number }) => {
|
|
||||||
if (params.position) {
|
|
||||||
return hass.hass.call.cover.set_cover_position({
|
|
||||||
entity_id: params.entity_id,
|
|
||||||
position: params.position,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
if (params.state === "open") {
|
|
||||||
return hass.hass.call.cover.open_cover({
|
|
||||||
entity_id: params.entity_id,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
if (params.state === "close") {
|
|
||||||
return hass.hass.call.cover.close_cover({
|
|
||||||
entity_id: params.entity_id,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const listDomains = () => {
|
|
||||||
return ["light", "climate", "alarm_control_panel", "cover", "switch", "sensor", "button"];
|
|
||||||
}
|
|
||||||
|
|
||||||
const getEntityHistoryByIds = (params: { entity_ids: TRawEntityIds[], start_time: string, end_time?: string }) => {
|
|
||||||
return hass.hass.entity.history({
|
|
||||||
entity_ids: params.entity_ids as TRawEntityIds[],
|
|
||||||
end_time: params.end_time ? new Date(params.end_time) : new Date(),
|
|
||||||
start_time: params.start_time
|
|
||||||
});
|
});
|
||||||
|
|
||||||
|
// Start the server
|
||||||
|
await server.start();
|
||||||
|
console.log('MCP Server started on port', server.port);
|
||||||
}
|
}
|
||||||
|
|
||||||
const getEntityHistory = async (params: { entity_id: TRawEntityIds, start_time: string, end_time?: string }) => {
|
main().catch(console.error);
|
||||||
return await hass.hass.entity.history({
|
|
||||||
entity_ids: [params.entity_id as TRawEntityIds],
|
|
||||||
end_time: params.end_time ? new Date(params.end_time) : new Date(),
|
|
||||||
start_time: params.start_time
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
const getEntityStateByIds = (entity_ids: TRawEntityIds[]) => {
|
|
||||||
const entities = entity_ids.map(entity_id => hass.hass.entity.getCurrentState(entity_id as TRawEntityIds));
|
|
||||||
return entities;
|
|
||||||
}
|
|
||||||
|
|
||||||
const getEntities = async (params: { domain?: TRawDomains, floor?: TFloorId, area?: TAreaId }) => {
|
|
||||||
if (params.floor) {
|
|
||||||
return hass.hass.idBy.floor(params.floor as TFloorId, params.domain as TRawDomains || undefined);
|
|
||||||
}
|
|
||||||
if (params.area) {
|
|
||||||
return hass.hass.idBy.area(params.area as TAreaId, params.domain as TRawDomains || undefined);
|
|
||||||
}
|
|
||||||
if (params.domain) {
|
|
||||||
return hass.hass.entity.listEntities(params.domain as TRawDomains);
|
|
||||||
}
|
|
||||||
return hass.hass.entity.listEntities();
|
|
||||||
}
|
|
||||||
|
|
||||||
const getEntityState = async (entity_id: TRawEntityIds) => {
|
|
||||||
return await hass.hass.entity.getCurrentState(entity_id);
|
|
||||||
}
|
|
||||||
|
|
||||||
const listAreas = async () => {
|
|
||||||
const areas = await hass.hass.area.list()
|
|
||||||
return areas;
|
|
||||||
}
|
|
||||||
|
|
||||||
const listFloors = async () => {
|
|
||||||
const floors = await hass.hass.floor.list()
|
|
||||||
return floors;
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user