43 lines
841 B
Go
43 lines
841 B
Go
|
// Package tools collects tools
|
||
|
|
package tools
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/ollama/ollama/api"
|
||
|
|
"tipsy.codes/charles/mc-god/v2/internal/pkg/rcon"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Tool interface {
|
||
|
|
Do(ctx context.Context, toolCall api.ToolCall, client *rcon.Client) error
|
||
|
|
Desc() api.Tool
|
||
|
|
Name() string
|
||
|
|
}
|
||
|
|
|
||
|
|
type Tools map[string]Tool
|
||
|
|
|
||
|
|
func New(tools ...Tool) Tools {
|
||
|
|
ret := make(map[string]Tool)
|
||
|
|
for _, tool := range tools {
|
||
|
|
ret[tool.Name()] = tool
|
||
|
|
}
|
||
|
|
return ret
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t Tools) AsAPI() api.Tools {
|
||
|
|
var ret api.Tools
|
||
|
|
for _, tool := range t {
|
||
|
|
ret = append(ret, tool.Desc())
|
||
|
|
}
|
||
|
|
return ret
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t Tools) Do(ctx context.Context, toolCall api.ToolCall, client *rcon.Client) error {
|
||
|
|
tool, found := t[toolCall.Function.Name]
|
||
|
|
if !found {
|
||
|
|
return fmt.Errorf("unknown tool %q", toolCall.Function.Name)
|
||
|
|
}
|
||
|
|
return tool.Do(ctx, toolCall, client)
|
||
|
|
}
|