feat: add support for tools context

Signed-off-by: Donnie Adams <[email protected]>

Author: thedadams

client_test.go

@@ -85,6 +85,39 @@ func TestSimpleEvaluate(t *testing.T) {
 	}
 }
 
+func TestEvaluateWithContext(t *testing.T) {
+	wd, err := os.Getwd()
+	if err != nil {
+		t.Fatalf("Error getting current working directory: %v", err)
+	}
+	fmt.Println(wd)
+
+	tool := &ToolDef{
+		Instructions: "What is the capital of the united states?",
+		Context: []string{
+			wd + "/test/acorn-labs-context.gpt",
+		},
+	}
+
+	run, err := client.Evaluate(context.Background(), Opts{DisableCache: true, IncludeEvents: true}, tool)
+	if err != nil {
+		t.Errorf("Error executing tool: %v", err)
+	}
+
+	for event := range run.Events() {
+		fmt.Println(event)
+	}
+
+	out, err := run.Text()
+	if err != nil {
+		t.Errorf("Error reading output: %v", err)
+	}
+
+	if out != "Acorn Labs" {
+		t.Errorf("Unexpected output: %s", out)
+	}
+}
+
 func TestRunFileChdir(t *testing.T) {
 	wd, err := os.Getwd()
 	if err != nil {

test/acorn-labs-context.gpt

@@ -0,0 +1 @@
+Say "Answer every query with 'Acorn Labs'"

tool.go

@@ -52,6 +52,18 @@ func (t *ToolDef) String() string {
 	if t.Cache != nil && !*t.Cache {
 		sb.WriteString("Cache: false\n")
 	}
+	if len(t.Context) > 0 {
+		sb.WriteString(fmt.Sprintf("Context: %s\n", strings.Join(t.Context, ", ")))
+	}
+	if len(t.ExportContext) > 0 {
+		sb.WriteString(fmt.Sprintf("Export Context: %s\n", strings.Join(t.ExportContext, ", ")))
+	}
+	if len(t.Export) > 0 {
+		sb.WriteString(fmt.Sprintf("Export: %s\n", strings.Join(t.Export, ", ")))
+	}
+	if len(t.GlobalTools) > 0 {
+		sb.WriteString(fmt.Sprintf("Global Tools: %s\n", strings.Join(t.GlobalTools, ", ")))
+	}
 	if t.Temperature != nil {
 		sb.WriteString(fmt.Sprintf("Temperature: %f\n", *t.Temperature))
 	}