Refactor

hupe1980 · Jul 29, 2023 · 8f21dc5 · 8f21dc5
1 parent 34cad6d
commit 8f21dc5
Show file tree

Hide file tree

Showing 5 changed files with 45 additions and 2 deletions.
diff --git a/examples/openai_functions_agent/main.go b/examples/openai_functions_agent/main.go
@@ -8,8 +8,8 @@ import (
 
 	"github.com/hupe1980/golc"
 	"github.com/hupe1980/golc/agent"
-	"github.com/hupe1980/golc/agent/toolkit"
 	"github.com/hupe1980/golc/model/chatmodel"
+	"github.com/hupe1980/golc/toolkit"
 	"github.com/playwright-community/playwright-go"
 )
 

diff --git a/examples/react_description_agent/main.go b/examples/react_description_agent/main.go
@@ -8,8 +8,8 @@ import (
 
 	"github.com/hupe1980/golc"
 	"github.com/hupe1980/golc/agent"
-	"github.com/hupe1980/golc/agent/toolkit"
 	"github.com/hupe1980/golc/model/llm"
+	"github.com/hupe1980/golc/toolkit"
 	"github.com/playwright-community/playwright-go"
 )
 

diff --git a/agent/toolkit/browser.go → toolkit/browser.go b/agent/toolkit/browser.go → toolkit/browser.go
diff --git a/toolkit/browser_test.go b/toolkit/browser_test.go
@@ -0,0 +1,41 @@
+package toolkit
+
+import (
+	"testing"
+
+	"github.com/hupe1980/golc/schema"
+	"github.com/stretchr/testify/require"
+)
+
+// TestNewBrowser tests the creation of a new Browser object
+func TestNewBrowser(t *testing.T) {
+	browser, err := NewBrowser(nil)
+	require.NoError(t, err)
+	require.NotNil(t, browser)
+
+	// Ensure that the Browser has the expected tools
+	expectedToolNames := []string{
+		"CurrentPage",
+		"NavigateBrowser",
+		"ExtractText",
+	}
+	tools := browser.Tools()
+	require.Len(t, tools, len(expectedToolNames))
+
+	for _, name := range expectedToolNames {
+		assertToolExists(t, tools, name)
+	}
+}
+
+// assertToolExists checks if a tool with the specified name exists in the list of tools.
+func assertToolExists(t *testing.T, tools []schema.Tool, name string) {
+	t.Helper()
+
+	for _, tool := range tools {
+		if tool.Name() == name {
+			return // Found the tool
+		}
+	}
+
+	require.Fail(t, "tool not found: "+name)
+}
diff --git a/toolkit/toolkit.go b/toolkit/toolkit.go
@@ -0,0 +1,2 @@
+// package toolkit provides a collection of tools.
+package toolkit
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		// package toolkit provides a collection of tools.
		package toolkit