Merge pull request #39 from thedadams/add-tool-context

feat: add support for tools context

Author: thedadams

examples/example-file-stream.js

@@ -729,6 +729,7 @@ export interface ToolDef {
 	arguments: ArgumentSchema
 	tools: string[]
 	globalTools: string[]
+	context: string[]
 	export: string[]
 	blocking: boolean
 	instructions: string
@@ -960,6 +961,12 @@ function toolDefToString(tool: ToolDef) {
 	if (tool.tools?.length > 0) {
 		toolInfo.push(`Tools: ${tool.tools.join(", ")}`)
 	}
+	if (tool.context?.length > 0) {
+		toolInfo.push(`Context: ${tool.context.join(", ")}`)
+	}
+	if (tool.export?.length > 0) {
+		toolInfo.push(`Export: ${tool.export.join(", ")}`)
+	}
 	if (tool.maxTokens !== undefined) {
 		toolInfo.push(`Max Tokens: ${tool.maxTokens}`)
 	}

examples/example-models.js

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

examples/example-parse-file.js

@@ -63,6 +63,26 @@ describe("gptscript module", () => {
 		expect(err).toEqual("")
 	})
 
+	test("evaluate executes a prompt correctly with context", async () => {
+		let out = ""
+		let err = undefined
+		const t = {
+			instructions: "who was the president of the united states in 1928?",
+			context: [path.join(__dirname, "fixtures", "acorn-labs-context.gpt")]
+		}
+
+		try {
+			const run = client.evaluate(t as any, {disableCache: true})
+			out = await run.text()
+			err = run.err
+		} catch (e) {
+			console.error(e)
+		}
+
+		expect(out).toContain("Acorn Labs")
+		expect(err).toEqual("")
+	})
+
 	describe("run with test.gpt fixture", () => {
 		test("should execute test.gpt correctly", async () => {
 			const testGptPath = path.join(__dirname, "fixtures", "test.gpt")