Extensions

package main

import (
	"context"
	"encoding/json"

	"github.com/pulseaiclub/phi/ext"
)

// Extension registers a minimal greet tool and /hello command.
func Extension(phi *ext.API) {
	phi.RegisterTool(ext.ToolDef{
		Name:        "greet",
		Label:       "Greet",
		Description: "Greet someone by name",
		Parameters: map[string]any{
			"type": "object",
			"properties": map[string]any{
				"name": map[string]any{"type": "string", "description": "Name to greet"},
			},
			"required": []any{"name"},
		},
		Execute: func(ctx context.Context, args json.RawMessage) (ext.ToolResult, error) {
			var in struct {
				Name string `json:"name"`
			}
			_ = json.Unmarshal(args, &in)
			msg := "Hello, " + in.Name + "!"
			return ext.ToolResult{Content: msg, Output: msg}, nil
		},
	})

	phi.On(ext.EventToolCall, func(ev ext.ToolCallEvent, ctx *ext.Context) *ext.ToolCallResult {
		// return &ext.ToolCallResult{Block: true, Reason: "..."} to deny
		return nil
	})

	phi.RegisterCommand("hello", ext.CommandDef{
		Description: "Say hello via toast",
		Handler: func(args string, ctx *ext.Context) error {
			name := args
			if name == "" {
				name = "world"
			}
			if ctx.UI != nil {
				ctx.UI.Notify("Hello "+name+"!", "info")
			}
			return nil
		},
	})
}