add: more tools, fix prompt

internal/pkg/tools/time/time.go

@@ -0,0 +1,59 @@
+/*
+ * Package time provides a Ollama tool to control the time.
+ */
+package time
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/ollama/ollama/api"
+	"tipsy.codes/charles/mc-god/v2/internal/pkg/rcon"
+)
+
+func Get() *Tool {
+	return &Tool{}
+}
+
+type Tool struct{}
+
+func (t *Tool) Name() string {
+	return "change_time"
+}
+
+func (t *Tool) Desc() api.Tool {
+	toolPropertiesMap := api.NewToolPropertiesMap()
+	toolPropertiesMap.Set("time", api.ToolProperty{
+		Type:        api.PropertyType{"string"},
+		Description: "What to set the weather too",
+		Enum:        []any{"day", "noon", "midnight"},
+	})
+	return api.Tool{
+		Type: "function",
+		Function: api.ToolFunction{
+			Name:        Get().Name(),
+			Description: "Changes the current time",
+			Parameters: api.ToolFunctionParameters{
+				Type:       "object",
+				Properties: &api.ToolPropertiesMap{},
+				Required:   []string{"time"},
+			},
+		},
+	}
+}
+
+func (t *Tool) Do(ctx context.Context, toolCall api.ToolCall, client *rcon.Client) error {
+
+	time, found := toolCall.Function.Arguments.Get("time")
+	if !found {
+		return fmt.Errorf("missing time argument")
+	}
+	timeSting, ok := time.(string)
+	if !ok {
+		return fmt.Errorf("incorrect data type %v; want string", time)
+	}
+	if _, err := client.Execute("/time set " + timeSting); err != nil {
+		return fmt.Errorf("failed to call tool")
+	}
+	return nil
+}