Feat: new tool to read from url

1 files changed, 43 insertions, 0 deletions

diff --git a/tools.go b/tools.go
index 47d0e01..aaba86d 100644
--- a/tools.go
+++ b/tools.go
@@ -171,6 +171,30 @@ func websearch(args map[string]string) []byte {
 	return data
 }
 
+// retrieves url content (text)
+func readURL(args map[string]string) []byte {
+	// make http request return bytes
+	link, ok := args["url"]
+	if !ok || link == "" {
+		msg := "linknot provided to read_url tool"
+		logger.Error(msg)
+		return []byte(msg)
+	}
+	resp, err := extra.WebSearcher.RetrieveFromLink(context.Background(), link)
+	if err != nil {
+		msg := "search tool failed; error: " + err.Error()
+		logger.Error(msg)
+		return []byte(msg)
+	}
+	data, err := json.Marshal(resp)
+	if err != nil {
+		msg := "failed to marshal search result; error: " + err.Error()
+		logger.Error(msg)
+		return []byte(msg)
+	}
+	return data
+}
+
 /*
 consider cases:
 - append mode (treat it like a journal appendix)
@@ -811,6 +835,7 @@ var fnMap = map[string]fnSig{
 	"recall_topics":   recallTopics,
 	"memorise":        memorise,
 	"websearch":       websearch,
+	"read_url":        readURL,
 	"file_create":     fileCreate,
 	"file_read":       fileRead,
 	"file_write":      fileWrite,
@@ -849,6 +874,24 @@ var baseTools = []models.Tool{
 			},
 		},
 	},
+	// read_url
+	models.Tool{
+		Type: "function",
+		Function: models.ToolFunc{
+			Name:        "read_url",
+			Description: "Retrieves text content of given link.",
+			Parameters: models.ToolFuncParams{
+				Type:     "object",
+				Required: []string{"url"},
+				Properties: map[string]models.ToolArgProps{
+					"url": models.ToolArgProps{
+						Type:        "string",
+						Description: "link to the webpage to read text from",
+					},
+				},
+			},
+		},
+	},
 	// memorise
 	models.Tool{
 		Type: "function",