Restore the time service so it stops writing bad lua.

Signed-off-by: Katharine Berry <[email protected]>

Author: Katharine

service/assistant/functions/time.go

@@ -0,0 +1,75 @@
+package functions
+
+import (
+	"context"
+	"fmt"
+	"google.golang.org/genai"
+	"strings"
+	"time"
+
+	"github.com/honeycombio/beeline-go"
+	"github.com/pebble-dev/bobby-assistant/service/assistant/quota"
+)
+
+type TimeResponse struct {
+	Time string `json:"time"`
+}
+
+type GetTimeInput struct {
+	// The timezone, e.g. 'America/Los_Angeles'.
+	Timezone string `json:"timezone" jsonschema:"required"`
+	// The number of seconds to add to the current time.
+	Offset float64 `json:"offset"`
+}
+
+func init() {
+	registerFunction(Registration{
+		Definition: genai.FunctionDeclaration{
+			Name:        "get_time_elsewhere",
+			Description: "Get the current time in a given valid tzdb timezone. Not all cities have a tzdb entry - be sure to use one that exists. Call multiple times to find the time in multiple timezones.",
+			Parameters: &genai.Schema{
+				Type:     genai.TypeObject,
+				Nullable: false,
+				Properties: map[string]*genai.Schema{
+					"timezone": {
+						Type:        genai.TypeString,
+						Description: "The timezone, e.g. 'America/Los_Angeles'.",
+						Nullable:    false,
+					},
+					"offset": {
+						Type:        genai.TypeNumber,
+						Description: "The number of seconds to add to the current time, if checking a different time. Omit or set to zero for current time.",
+						Format:      "double",
+					},
+				},
+				Required: []string{"timezone"},
+			},
+		},
+		Fn:        getTimeElsewhere,
+		Thought:   getTimeThought,
+		InputType: GetTimeInput{},
+	})
+}
+
+func getTimeThought(args interface{}) string {
+	arg := args.(*GetTimeInput)
+	if arg.Timezone != "" {
+		s := strings.Split(arg.Timezone, "/")
+		place := strings.Replace(s[len(s)-1], "_", " ", -1)
+		return "Checking the time in " + place
+	}
+	return "Checking the time"
+}
+
+func getTimeElsewhere(ctx context.Context, quotaTracker *quota.Tracker, args interface{}) interface{} {
+	ctx, span := beeline.StartSpan(ctx, "get_time_elsewhere")
+	defer span.Send()
+	arg := args.(*GetTimeInput)
+	utc := time.Now().UTC().Add(time.Duration(arg.Offset) * time.Second)
+	loc, err := time.LoadLocation(arg.Timezone)
+	if err != nil {
+		return Error{fmt.Sprintf("The timezone %q is not valid", arg.Timezone)}
+	}
+	utc.In(loc)
+	return TimeResponse{utc.In(loc).Format(time.RFC1123)}
+}