refactor: added hello world tools

Author: TobiasGleiter

agents/agents.go

@@ -8,6 +8,7 @@ package agents
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
 
 	"github.com/bit8bytes/beago/inputs/roles"
@@ -20,18 +21,17 @@ import (
 // FinalAnswer being non-empty signals the agent is done; otherwise Action and
 // ActionInput describe the next tool to call.
 type response struct {
-	Thought     string `json:"thought"`
-	Action      string `json:"action"`
-	ActionInput string `json:"action_input"`
-	FinalAnswer string `json:"final_answer"`
+	Thought     string          `json:"thought"`
+	Action      string          `json:"action"`
+	ActionInput json.RawMessage `json:"action_input"`
+	FinalAnswer string          `json:"final_answer"`
 }
 
 // Action is the domain representation of a tool call, extracted from a
-// response. Name matches the tool's Name() and Input is passed verbatim to
-// Tool.Execute.
+// response. Name matches the tool's Name() and Input is passed to Tool.Execute.
 type Action struct {
 	Name  string
-	Input string
+	Input json.RawMessage
 }
 
 type llm interface {

agents/react.go

@@ -35,6 +35,13 @@ func buildReActPrompt(tls map[string]tools.Tool, jsonInstructions string) []llms
 	var toolDescriptions strings.Builder
 	for _, t := range tls {
 		fmt.Fprintf(&toolDescriptions, "- %s: %s\n", t.Name(), t.Description())
+		for _, p := range t.Parameters() {
+			req := "optional"
+			if p.Required {
+				req = "required"
+			}
+			fmt.Fprintf(&toolDescriptions, "    - %s (%s): %s\n", p.Name, req, p.Description)
+		}
 	}
 
 	wd, _ := os.Getwd()
@@ -55,10 +62,10 @@ Available tools:
 Respond with a JSON object on each turn with these fields:
 - "thought": your reasoning about what to do next
 - "action": the exact tool name to call (empty string when giving final answer)
-- "action_input": the input to pass to the tool (empty string when giving final answer)
+- "action_input": a JSON object whose keys are the tool's parameter names (empty object {} when giving final answer)
 - "final_answer": your final answer to the user — MUST be non-empty when you are done; empty string ONLY when calling a tool
 
-When you have enough information to answer, set "action" and "action_input" to "" and put a detailed answer based on your observations — MUST be non-empty when done; be thorough and include all relevant findings.
+When you have enough information to answer, set "action" to "" and "action_input" to {} and put a detailed answer based on your observations — MUST be non-empty when done; be thorough and include all relevant findings.
 
 Think step by step. Do not hallucinate.`, wd, toolDescriptions.String(), jsonInstructions),
 		},

examples/agents/hello/main.go

@@ -27,10 +27,7 @@ func main() {
 		Format:  ollama.JSON,
 	})
 
-	// These tools are specifically designed for Golang.
-	tools := []tools.Tool{
-		&tools.HelloWorldTool{},
-	}
+	tools := []tools.Tool{&tools.HelloWorldTool{}}
 
 	agent, err := agents.NewReAct(ctx, model, tools, storage)
 	if err != nil {

tools/helloworld.go

@@ -1,6 +1,10 @@
 package tools
 
-import "context"
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+)
 
 type HelloWorldTool struct{}
 
@@ -9,9 +13,21 @@ func (t *HelloWorldTool) Name() string {
 }
 
 func (t *HelloWorldTool) Description() string {
-	return "Returns a hello greeting for the given name. Input: a name string."
+	return "Returns a hello greeting for the given name."
 }
 
-func (t *HelloWorldTool) Execute(_ context.Context, input string) (string, error) {
-	return "Hello, " + input + "!", nil
+func (t *HelloWorldTool) Parameters() []Parameter {
+	return []Parameter{
+		{Name: "name", Description: "The name to greet", Required: true},
+	}
+}
+
+func (t *HelloWorldTool) Execute(_ context.Context, params json.RawMessage) (string, error) {
+	var input struct {
+		Name string `json:"name"`
+	}
+	if err := json.Unmarshal(params, &input); err != nil {
+		return "", fmt.Errorf("helloWorld: invalid params: %w", err)
+	}
+	return "Hello, " + input.Name + "!", nil
 }

tools/tools.go

@@ -2,12 +2,23 @@
 // implementations that agents can use to interact with the outside world.
 package tools
 
-import "context"
+import (
+	"context"
+	"encoding/json"
+)
+
+// Parameter describes a single input field a tool accepts.
+type Parameter struct {
+	Name        string
+	Description string
+	Required    bool
+}
 
 // Tool represents an action the agent can perform.
-// Each tool must provide a name, description, and execution logic.
+// Each tool must provide a name, description, parameter schema, and execution logic.
 type Tool interface {
 	Name() string
 	Description() string
-	Execute(ctx context.Context, input string) (string, error)
+	Parameters() []Parameter
+	Execute(ctx context.Context, params json.RawMessage) (string, error)
 }