""" Editor: atgehrhardt Editor's Note: I recommend tempering expectations when using this tool Suggestions: - Add Valves or UserValves for configurable variables - Add logic to resolve device names from ID and allow controlling via device name. Device IDs are VERY specific and not very useful in a number or HA instances. - Add additional logic that allows for specific control based on device type. It's unlikely any models short of the best available could reliably use this implementation to control devices other than lights. """ """ title: Home Assistant Controls author: Crixle funding_url: https://github.com/open-webui version: 0.1 """ import os import requests from datetime import datetime import json HA_API_KEY = "YOUR API KEY" HA_URL = "IP OR FQDN OF HA INSTANCE" """ Recommend that these are replaced with env variables ^^ """ class Tools: def __init__(self): pass # Add your custom tools using pure Python code here, make sure to add type hints # Use Sphinx-style docstrings to document your tools, they will be used for generating tools specifications # Please refer to function_calling_filter_pipeline.py file from pipelines project for an example def controlDevice(self, entityID: str, domain: str, service: str) -> str: """ Controls a device in Home Assistant for lights, switches, and media players. :param domain: The domain of the device being controlled. This could be light, switch, media_player, or climate. :param service: Either "turn_on" or "turn_off based on context. :param entityID: The entityID that the request is commanding. :return: A simple confirmation or an error. Do not restate or appreciate what the user says. """ try: # Endpoint to control the light entity endpoint = f"{HA_URL}/api/services/{domain}/{service}" # Prepare the payload payload = json.dumps({"entity_id": entityID}) # Make the HTTP POST request response = requests.post( endpoint, data=payload, headers={"Authorization": f"Bearer {HA_API_KEY}"}, ) # Check if the request was successful return "Successfully changed the device" except Exception as e: return f"An error occured: {e}"