60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
|
/*
|
||
|
|
* 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
|
||
|
|
}
|