Consolidated a bunch of reused code to agents

2025-03-26 00:21:19 -04:00
parent 5407c1a7cc
commit 5d2c350acf
33 changed files with 2866 additions and 803 deletions
--- a/pkg/agents/tools/wolfram.go
+++ b/pkg/agents/tools/wolfram.go
@@ -19,7 +19,7 @@ func CreateWolframFunctions(appId string) WolframFunctions {
 			"Query the Wolfram Alpha API",
 			func(ctx *gollm.Context, args struct {
 				Question string `description:"The question to ask Wolfram|Alpha"`
-			}) (string, error) {
+			}) (any, error) {
 				return client.GetShortAnswerQuery(args.Question, wolfram.Imperial, 10)
 			}),
 		Metric: gollm.NewFunction(
@@ -27,7 +27,7 @@ func CreateWolframFunctions(appId string) WolframFunctions {
 			"Query the Wolfram Alpha API",
 			func(ctx *gollm.Context, args struct {
 				Question string `description:"The question to ask Wolfram|Alpha"`
-			}) (string, error) {
+			}) (any, error) {
 				return client.GetShortAnswerQuery(args.Question, wolfram.Metric, 10)
 			}),
 	}