Added home assistant

Author: djthorpe

cmd/llm/main.go

@@ -36,6 +36,8 @@ type Globals struct {
 	AnthropicAPIKey string `name:"anthropic-api-key" env:"ANTHROPIC_API_KEY" help:"Anthropic API key"`
 	NewsAPIKey      string `name:"news-api-key" env:"NEWS_API_KEY" help:"NewsAPI key"`
 	WeatherAPIKey   string `name:"weather-api-key" env:"WEATHER_API_KEY" help:"WeatherAPI key"`
+	HAEndpoint      string `name:"ha-endpoint" env:"HA_ENDPOINT" help:"Home Assistant endpoint URL"`
+	HAToken         string `name:"ha-token" env:"HA_TOKEN" help:"Home Assistant long-lived access token"`
 
 	// Tool options
 	FsDir string `name:"fs" env:"FS_DIR" help:"Root directory for filesystem tools" type:"existingdir"`

cmd/llm/message.go

@@ -20,9 +20,10 @@ import (
 // GLOBALS
 
 const (
-	ansiDim   = "\033[2m"  // Dim text for thinking
-	ansiReset = "\033[0m"  // Reset formatting
-	ansiCyan  = "\033[36m" // Cyan for thinking label
+	ansiDim    = "\033[2m"  // Dim text for thinking
+	ansiReset  = "\033[0m"  // Reset formatting
+	ansiCyan   = "\033[36m" // Cyan for thinking label
+	ansiYellow = "\033[33m" // Yellow for tool calls
 )
 
 ///////////////////////////////////////////////////////////////////////////////
@@ -277,10 +278,12 @@ func streamCallback() opt.StreamFn {
 			switch role {
 			case "thinking":
 				fmt.Print(ansiDim + ansiCyan)
+			case "tool":
+				fmt.Print(ansiReset + ansiDim + ansiYellow)
 			default:
 				fmt.Print(ansiReset)
-				if lastRole == "thinking" {
-					fmt.Println() // newline between thinking and assistant
+				if lastRole == "thinking" || lastRole == "tool" {
+					fmt.Println() // newline between thinking/tool and assistant
 				}
 			}
 			lastRole = role

cmd/llm/tools.go

@@ -6,6 +6,7 @@ import (
 
 	// Packages
 	fstool "github.com/mutablelogic/go-llm/pkg/fstool"
+	homeassistant "github.com/mutablelogic/go-llm/pkg/homeassistant"
 	newsapi "github.com/mutablelogic/go-llm/pkg/newsapi"
 	tool "github.com/mutablelogic/go-llm/pkg/tool"
 	weatherapi "github.com/mutablelogic/go-llm/pkg/weatherapi"
@@ -68,6 +69,15 @@ func (g *Globals) Toolkit() (*tool.Toolkit, error) {
 		tools = append(tools, fsTools...)
 	}
 
+	// Add Home Assistant tools if endpoint and token are set
+	if g.HAEndpoint != "" && g.HAToken != "" {
+		haTools, err := homeassistant.NewTools(g.HAEndpoint, g.HAToken)
+		if err != nil {
+			return nil, fmt.Errorf("failed to create Home Assistant tools: %w", err)
+		}
+		tools = append(tools, haTools...)
+	}
+
 	return tool.NewToolkit(tools...)
 }
 

pkg/agent/generator.go

@@ -98,6 +98,13 @@ func (a *agent) WithSession(ctx context.Context, model schema.Model, session *sc
 			break
 		}
 
+		// Report tool calls via the streaming callback
+		if streamfn := o.GetStream(); streamfn != nil {
+			for _, call := range calls {
+				streamfn("tool", tk.Feedback(call)+"\n")
+			}
+		}
+
 		// Execute each tool call and collect result blocks
 		results := make([]schema.ContentBlock, 0, len(calls))
 		for _, call := range calls {

pkg/homeassistant/README.md

@@ -0,0 +1,8 @@
+# Home Assistant API Client
+
+This package provides a client for the Home Assistant API, which is used to interact with the Home Assistant home automation system.
+
+References:
+
+- API <https://developers.home-assistant.io/docs/api/rest/>
+- Package <https://pkg.go.dev/github.com/mutablelogic/go-llm/pkg/homeassistant>

pkg/homeassistant/client.go

@@ -0,0 +1,39 @@
+/*
+homeassistant implements an API client for Home Assistant API
+https://developers.home-assistant.io/docs/api/rest/
+*/
+package homeassistant
+
+import (
+	// Packages
+	"github.com/mutablelogic/go-client"
+)
+
+///////////////////////////////////////////////////////////////////////////////
+// TYPES
+
+type Client struct {
+	*client.Client
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// LIFECYCLE
+
+func New(endPoint, apiKey string, opts ...client.ClientOpt) (*Client, error) {
+	// Add a final slash to the endpoint
+	if len(endPoint) > 0 && endPoint[len(endPoint)-1] != '/' {
+		endPoint += "/"
+	}
+
+	// Create client
+	client, err := client.New(append(opts, client.OptEndpoint(endPoint), client.OptReqToken(client.Token{
+		Scheme: client.Bearer,
+		Value:  apiKey,
+	}))...)
+	if err != nil {
+		return nil, err
+	}
+
+	// Return the client
+	return &Client{client}, nil
+}

pkg/homeassistant/client_test.go

@@ -0,0 +1,40 @@
+package homeassistant_test
+
+import (
+	"os"
+	"testing"
+
+	// Packages
+	opts "github.com/mutablelogic/go-client"
+	homeassistant "github.com/mutablelogic/go-llm/pkg/homeassistant"
+	assert "github.com/stretchr/testify/assert"
+)
+
+func Test_client_001(t *testing.T) {
+	assert := assert.New(t)
+	client, err := homeassistant.New(GetEndPoint(t), GetApiKey(t), opts.OptTrace(os.Stderr, true))
+	assert.NoError(err)
+	assert.NotNil(client)
+	t.Log(client)
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// ENVIRONMENT
+
+func GetApiKey(t *testing.T) string {
+	key := os.Getenv("HA_TOKEN")
+	if key == "" {
+		t.Skip("HA_TOKEN not set")
+		t.SkipNow()
+	}
+	return key
+}
+
+func GetEndPoint(t *testing.T) string {
+	key := os.Getenv("HA_ENDPOINT")
+	if key == "" {
+		t.Skip("HA_ENDPOINT not set")
+		t.SkipNow()
+	}
+	return key
+}

pkg/homeassistant/config.go

@@ -0,0 +1,56 @@
+package homeassistant
+
+import (
+	"context"
+	"encoding/json"
+
+	// Packages
+	"github.com/mutablelogic/go-client"
+)
+
+///////////////////////////////////////////////////////////////////////////////
+// TYPES
+
+// Config represents the Home Assistant server configuration.
+type Config struct {
+	Components    []string       `json:"components"`
+	ConfigDir     string         `json:"config_dir"`
+	Elevation     float64        `json:"elevation"`
+	Latitude      float64        `json:"latitude"`
+	Longitude     float64        `json:"longitude"`
+	LocationName  string         `json:"location_name"`
+	TimeZone      string         `json:"time_zone"`
+	UnitSystem    map[string]any `json:"unit_system"`
+	Version       string         `json:"version"`
+	ExternalDirs  []string       `json:"whitelist_external_dirs,omitempty"`
+	AllowlistDirs []string       `json:"allowlist_external_dirs,omitempty"`
+	AllowlistURLs []string       `json:"allowlist_external_urls,omitempty"`
+	Currency      string         `json:"currency,omitempty"`
+	Country       string         `json:"country,omitempty"`
+	Language      string         `json:"language,omitempty"`
+	SafeMode      bool           `json:"safe_mode,omitempty"`
+	State         string         `json:"state,omitempty"`
+	InternalURL   string         `json:"internal_url,omitempty"`
+	ExternalURL   string         `json:"external_url,omitempty"`
+	RecoveryMode  bool           `json:"recovery_mode,omitempty"`
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// API CALLS
+
+// Config returns the current server configuration.
+func (c *Client) Config(ctx context.Context) (*Config, error) {
+	var response Config
+	if err := c.DoWithContext(ctx, nil, &response, client.OptPath("config")); err != nil {
+		return nil, err
+	}
+	return &response, nil
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// STRINGIFY
+
+func (v Config) String() string {
+	data, _ := json.MarshalIndent(v, "", "  ")
+	return string(data)
+}

pkg/homeassistant/config_test.go

@@ -0,0 +1,27 @@
+package homeassistant_test
+
+import (
+	"context"
+	"os"
+	"testing"
+
+	// Packages
+	opts "github.com/mutablelogic/go-client"
+	homeassistant "github.com/mutablelogic/go-llm/pkg/homeassistant"
+	assert "github.com/stretchr/testify/assert"
+)
+
+func Test_config_001(t *testing.T) {
+	assert := assert.New(t)
+	client, err := homeassistant.New(GetEndPoint(t), GetApiKey(t), opts.OptTrace(os.Stderr, true))
+	assert.NoError(err)
+	assert.NotNil(client)
+
+	config, err := client.Config(context.Background())
+	assert.NoError(err)
+	assert.NotNil(config)
+
+	t.Log(config)
+	assert.NotEmpty(config.Version)
+	assert.NotEmpty(config.LocationName)
+}

pkg/homeassistant/events.go

@@ -0,0 +1,59 @@
+package homeassistant
+
+import (
+	// Packages
+	"context"
+
+	"github.com/mutablelogic/go-client"
+)
+
+///////////////////////////////////////////////////////////////////////////////
+// TYPES
+
+type Event struct {
+	Event     string `json:"event"`
+	Listeners uint   `json:"listener_count"`
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// API CALLS
+
+// Events returns all the events and number of listeners
+func (c *Client) Events(ctx context.Context) ([]Event, error) {
+	var response []Event
+	if err := c.DoWithContext(ctx, nil, &response, client.OptPath("events")); err != nil {
+		return nil, err
+	}
+
+	// Return success
+	return response, nil
+}
+
+// FireEvent fires an event with the given type. Optional event data can be
+// passed as a map. Returns a confirmation message.
+func (c *Client) FireEvent(ctx context.Context, eventType string, eventData map[string]any) (string, error) {
+	type responseMessage struct {
+		Message string `json:"message"`
+	}
+
+	var payload client.Payload
+	var err error
+	if len(eventData) > 0 {
+		payload, err = client.NewJSONRequest(eventData)
+		if err != nil {
+			return "", err
+		}
+	} else {
+		payload, err = client.NewJSONRequest(struct{}{})
+		if err != nil {
+			return "", err
+		}
+	}
+
+	var response responseMessage
+	if err := c.DoWithContext(ctx, payload, &response, client.OptPath("events", eventType)); err != nil {
+		return "", err
+	}
+
+	return response.Message, nil
+}