adding Hue light control
Co-authored-by: Alex Dimmock <AtomicAlexD@users.noreply.github.com>
This commit is contained in:
@@ -10,6 +10,9 @@ build/
|
||||
dist/
|
||||
wheels/
|
||||
*.egg-info
|
||||
.cache*
|
||||
*.log
|
||||
|
||||
# Virtual environments
|
||||
.venv
|
||||
.env
|
||||
@@ -14,29 +14,90 @@ CRITICAL: When calling tools, use COMPACT JSON with NO SPACES:
|
||||
{
|
||||
"name": "get_weather",
|
||||
"description": "Get current weather for a location",
|
||||
"parameters"[{"city":"string"}],
|
||||
"examples": [
|
||||
{
|
||||
"input": {"tool":"get_weather","parameters":{"city":"New York"}},
|
||||
"output": {"temperature": 22, "condition": "partly cloudy", "humidity": 65}
|
||||
},
|
||||
{
|
||||
"input": {"tool":"get_weather","parameters":{"city":"London"}},
|
||||
"output": {"temperature": 18, "condition": "rainy", "humidity": 80}
|
||||
}
|
||||
{"tool":"get_weather","parameters":{"city":"London"}},
|
||||
{"tool":"get_weather","parameters":{"city":"Kettering"}},
|
||||
{"tool":"get_weather","parameters":{"city":"Peterborough"}},
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "find_folder",
|
||||
"description": "Find any folder that matches the name provided on your machine or an optional directory",
|
||||
"parameters"[{"folder_name":"string"}],
|
||||
"examples": [
|
||||
{
|
||||
"input": {"tool":"find_folder","parameters":{"folder_name":"devin"}},
|
||||
},
|
||||
{
|
||||
"input": {"tool":"find_folder","parameters":{"folder_name":"winutils"}},
|
||||
}
|
||||
{"tool":"find_folder","parameters":{"folder_name":"devin"}},
|
||||
{"tool":"find_folder","parameters":{"folder_name":"winutils"}},
|
||||
{"tool":"find_folder","parameters":{"folder_name":"Dygma"}},
|
||||
]
|
||||
}'''
|
||||
},
|
||||
{
|
||||
"name": "turn_on_light",
|
||||
"description": "Turn on any light in the house by name",
|
||||
"parameters"[{"light_name":"string"}],
|
||||
"examples": [
|
||||
{"tool":"turn_on_light","parameters":{"light_name":"Monkey"}},
|
||||
{"tool":"turn_on_light","parameters":{"light_name":"Bedside"}},
|
||||
{"tool":"turn_on_light","parameters":{"light_name":"Bookshelf"}},
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "turn_off_light",
|
||||
"description": "Turn off any light in the house by name",
|
||||
"parameters"[{"light_name":"string"}],
|
||||
"examples": [
|
||||
{"tool":"turn_off_light","parameters":{"light_name":"Monkey"}},
|
||||
{"tool":"turn_off_light","parameters":{"light_name":"Bedside"}},
|
||||
{"tool":"turn_off_light","parameters":{"light_name":"Bookshelf"}},
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "set_light_brightness",
|
||||
"description": "Set the brightness level of any light in the house by name",
|
||||
"parameters": [
|
||||
{"light_name": "string"},
|
||||
{"brightness": "integer (1-100)"}
|
||||
],
|
||||
"examples": [
|
||||
{"tool": "set_light_brightness", "parameters": {"light_name": "Monkey", "brightness": 25}},
|
||||
{"tool": "set_light_brightness", "parameters": {"light_name": "Bedside", "brightness": 50}},
|
||||
{"tool": "set_light_brightness", "parameters": {"light_name": "Bookshelf", "brightness": 75}},
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "turn_on_room",
|
||||
"description": "Turn on any room of lights in the house by name",
|
||||
"parameters"[{"room_name":"string"}],
|
||||
"examples": [
|
||||
{"tool":"turn_on_room","parameters":{"room_name":"Office"}},
|
||||
{"tool":"turn_on_room","parameters":{"room_name":"Bedroom"}},
|
||||
{"tool":"turn_on_room","parameters":{"room_name":"Lounge"}},
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "turn_off_room",
|
||||
"description": "Turn off any room of lights in the house by name",
|
||||
"parameters"[{"room_name":"string"}],
|
||||
"examples": [
|
||||
{"tool":"turn_off_room","parameters":{"room_name":"Kitchen"}},
|
||||
{"tool":"turn_off_room","parameters":{"room_name":"Lounge"}},
|
||||
{"tool":"turn_off_room","parameters":{"room_name":"Office"}},
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "set_room_brightness",
|
||||
"description": "Set the brightness level of any room of lights in the house by name",
|
||||
"parameters": [
|
||||
{"room_name": "string"},
|
||||
{"brightness": "integer (1-100)"}
|
||||
],
|
||||
"examples": [
|
||||
{"tool": "set_room_brightness", "parameters": {"room_name": "Office", "brightness": 25}},
|
||||
{"tool": "set_room_brightness", "parameters": {"room_name": "Bedroom", "brightness": 50}},
|
||||
{"tool": "set_room_brightness", "parameters": {"room_name": "Kitchen", "brightness": 75}},
|
||||
]
|
||||
}
|
||||
'''
|
||||
|
||||
# Tool config
|
||||
# WMO Weather interpretation codes mapping
|
||||
@@ -70,3 +131,26 @@ WEATHER_CODE_MAP = {
|
||||
96: "Thunderstorm with slight hail",
|
||||
99: "Thunderstorm with heavy hail"
|
||||
}
|
||||
|
||||
LIGHT_NAMES = {
|
||||
'monkey': '9b128205-0a98-46b5-9234-2bd0be9fd009',
|
||||
'bench': 'b0fae364-78ed-4f99-86d0-6c65cbf792cd',
|
||||
"bedside": '945298e2-96b6-4c61-a506-1691bf7d7989',
|
||||
'bookshelf': '96d805e6-f39a-4e5f-9bde-3a42cbadfc6c',
|
||||
'lounge 1': '3ffcd59a-a19d-4066-af35-b5f45a2cf946',
|
||||
'lounge 2': 'facf3d02-f88d-482c-acb1-f9a4ed519356',
|
||||
'kitchen 1': 'badef93d-10ab-437d-9fc2-a09181a08fae',
|
||||
'kitchen 2': 'f15db9c5-0b67-49a6-8687-f20d768048b7',
|
||||
'kitchen 3': '47be084d-9e35-4108-aa4d-8da8f13e7d42',
|
||||
'kitchen 4': 'be408308-b14e-4bc3-9065-68055fd74b68',
|
||||
'shelf': '296ff923-da19-43a5-b1b2-7de25b227469',
|
||||
'cupboards': 'fe27cf93-68e2-47f2-a39e-d9bc4650263b',
|
||||
'office white': 'a1e0f26b-90b8-4044-98a3-58ed6a0f84b0',
|
||||
}
|
||||
ROOM_NAMES = {
|
||||
'office': 'bb0856ac-81d9-439a-83dc-8703c90574ba',
|
||||
'bedroom': '621fea30-f8b6-4de9-a347-1b4436321398',
|
||||
'lounge': '3dc9aab6-6379-4fa4-8e96-aae94fa692cf',
|
||||
'Kitchen': 'eaa524bc-edb6-4fd8-89b7-7cfc563ed7f1',
|
||||
'graveyard': 'aec3c969-581f-45c4-8f8f-b9407ee8caa3',
|
||||
}
|
||||
@@ -11,8 +11,6 @@ from llm_interface import llm
|
||||
|
||||
import logging
|
||||
|
||||
#TODO: add context for llm calls so we have history and can chain messages
|
||||
|
||||
console = Console()
|
||||
|
||||
class RichConsoleHandler(logging.StreamHandler):
|
||||
@@ -45,9 +43,10 @@ def main(logger):
|
||||
logger.info(f"Response equals original: {handled_response == llm_reply}")
|
||||
|
||||
if handled_response != llm_reply:
|
||||
output = language_model.tool_response(prompt=f'Make this lovely markdown, use fun emojis {handled_response}')
|
||||
#TODO: Make sure to pass the history into the this so that if we ask additional questions it has context
|
||||
# For example get weather AND suggest clothing based on weather.
|
||||
output = language_model.tool_response(prompt=f'''Your original Request was: {llm_prompt},
|
||||
This is the tool reply: {handled_response}
|
||||
Make a markdown format reply, using fun emojis.
|
||||
''')
|
||||
elif handled_response == llm_reply:
|
||||
output = handled_response
|
||||
|
||||
|
||||
@@ -9,6 +9,7 @@ dependencies = [
|
||||
"openmeteo-requests>=1.7.2",
|
||||
"pydantic>=2.11.7",
|
||||
"pytest>=8.4.1",
|
||||
"python-dotenv>=1.1.1",
|
||||
"requests>=2.32.5",
|
||||
"requests-cache>=1.2.1",
|
||||
"retry-requests>=2.0.0",
|
||||
|
||||
@@ -7,14 +7,33 @@ import config
|
||||
import logging
|
||||
import os
|
||||
import subprocess
|
||||
import dotenv
|
||||
import urllib3
|
||||
|
||||
dotenv.load_dotenv()
|
||||
|
||||
# Suppress SSL warnings (this is fine for local network)
|
||||
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
||||
|
||||
class ToolExecutor:
|
||||
def __init__(self, logger):
|
||||
self.tools = {
|
||||
self.available_tools = {
|
||||
"get_weather": self.get_weather,
|
||||
"find_folder": self.find_folder
|
||||
"find_folder": self.find_folder,
|
||||
"turn_on_light": self.turn_on_light,
|
||||
"turn_off_light": self.turn_off_light,
|
||||
"set_light_brightness": self.set_light_brightness,
|
||||
"turn_on_room": self.turn_on_room,
|
||||
"turn_off_room": self.turn_off_room,
|
||||
"set_room_brightness": self.set_room_brightness
|
||||
}
|
||||
self.logger = logger
|
||||
self.bridge_ip = '192.168.0.152'
|
||||
self.app_key = os.getenv("USERNAME")
|
||||
self.base_url = f'https://{self.bridge_ip}/clip/v2/resource'
|
||||
self.session = requests.Session()
|
||||
self.session.verify = False # For local network devices
|
||||
|
||||
|
||||
def process_llm_response(self, llm_output):
|
||||
if llm_output.startswith('{"tool":'):
|
||||
@@ -27,7 +46,7 @@ class ToolExecutor:
|
||||
self.logger.info(f'parsed tool: {tool_name}, parsed parameters: {parameters}')
|
||||
|
||||
# Execute the actual function
|
||||
result = self.tools[tool_name](**parameters)
|
||||
result = self.available_tools[tool_name](**parameters)
|
||||
|
||||
# Return result for LLM to use
|
||||
return result
|
||||
@@ -137,7 +156,7 @@ class ToolExecutor:
|
||||
# Use find command to search for directories
|
||||
cmd = ['find', search_path, '-type', 'd', '-name', folder_name]
|
||||
result = subprocess.run(cmd, capture_output=True, text=True, timeout=30)
|
||||
self.logger.warning(result)
|
||||
self.logger.info(result)
|
||||
if result.stdout:
|
||||
found_folders = [path.strip() for path in result.stdout.split('\n') if path.strip()]
|
||||
|
||||
@@ -158,3 +177,46 @@ class ToolExecutor:
|
||||
raise Exception("Folder search timed out after 30 seconds")
|
||||
except Exception as e:
|
||||
raise Exception(f"Folder search failed: {str(e)}")
|
||||
|
||||
def tell_hue(self, endpoint, payload):
|
||||
headers = {"hue-application-key": self.app_key}
|
||||
api_path = self.base_url+endpoint
|
||||
return self.session.put(api_path, json=payload, headers=headers, timeout=10)
|
||||
|
||||
def control_light(self, light_name, on_state=True, brightness=None):
|
||||
"""Control a light"""
|
||||
endpoint = f'/light/{config.LIGHT_NAMES[light_name.lower()]}'
|
||||
data = {"on": {"on": on_state}}
|
||||
if brightness is not None:
|
||||
data["dimming"] = {
|
||||
"brightness": brightness
|
||||
}
|
||||
return self.tell_hue(endpoint,data)
|
||||
|
||||
def control_room(self, room_name, on_state=True, brightness=None):
|
||||
"""Control a light"""
|
||||
endpoint = f'/grouped_light/{config.ROOM_NAMES[room_name.lower()]}'
|
||||
data = {"on": {"on": on_state}}
|
||||
if brightness is not None:
|
||||
data["dimming"] = {
|
||||
"brightness": brightness
|
||||
}
|
||||
return self.tell_hue(endpoint,data)
|
||||
|
||||
def turn_on_light(self, light_name):
|
||||
return self.control_light(light_name)
|
||||
|
||||
def turn_off_light(self, light_name):
|
||||
return self.control_light(light_name, on_state=False)
|
||||
|
||||
def set_light_brightness(self, light_name, brightness):
|
||||
return self.control_light(light_name, on_state=True, brightness=brightness)
|
||||
|
||||
def turn_on_room(self, room_name):
|
||||
return self.control_room(room_name)
|
||||
|
||||
def turn_off_room(self, room_name):
|
||||
return self.control_room(room_name, on_state=False)
|
||||
|
||||
def set_room_brightness(self, room_name, brightness):
|
||||
return self.control_room(room_name, on_state=True, brightness=brightness)
|
||||
@@ -116,6 +116,7 @@ dependencies = [
|
||||
{ name = "openmeteo-requests" },
|
||||
{ name = "pydantic" },
|
||||
{ name = "pytest" },
|
||||
{ name = "python-dotenv" },
|
||||
{ name = "requests" },
|
||||
{ name = "requests-cache" },
|
||||
{ name = "retry-requests" },
|
||||
@@ -130,6 +131,7 @@ requires-dist = [
|
||||
{ name = "openmeteo-requests", specifier = ">=1.7.2" },
|
||||
{ name = "pydantic", specifier = ">=2.11.7" },
|
||||
{ name = "pytest", specifier = ">=8.4.1" },
|
||||
{ name = "python-dotenv", specifier = ">=1.1.1" },
|
||||
{ name = "requests", specifier = ">=2.32.5" },
|
||||
{ name = "requests-cache", specifier = ">=1.2.1" },
|
||||
{ name = "retry-requests", specifier = ">=2.0.0" },
|
||||
@@ -382,6 +384,15 @@ wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/29/16/c8a903f4c4dffe7a12843191437d7cd8e32751d5de349d45d3fe69544e87/pytest-8.4.1-py3-none-any.whl", hash = "sha256:539c70ba6fcead8e78eebbf1115e8b589e7565830d7d006a8723f19ac8a0afb7", size = 365474, upload-time = "2025-06-18T05:48:03.955Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "python-dotenv"
|
||||
version = "1.1.1"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/f6/b0/4bc07ccd3572a2f9df7e6782f52b0c6c90dcbb803ac4a167702d7d0dfe1e/python_dotenv-1.1.1.tar.gz", hash = "sha256:a8a6399716257f45be6a007360200409fce5cda2661e3dec71d23dc15f6189ab", size = 41978, upload-time = "2025-06-24T04:21:07.341Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/5f/ed/539768cf28c661b5b068d66d96a2f155c4971a5d55684a514c1a0e0dec2f/python_dotenv-1.1.1-py3-none-any.whl", hash = "sha256:31f23644fe2602f88ff55e1f5c79ba497e01224ee7737937930c448e4d0e24dc", size = 20556, upload-time = "2025-06-24T04:21:06.073Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "qh3"
|
||||
version = "1.5.4"
|
||||
|
||||
Reference in New Issue
Block a user