callbacks: Include context

tmc · Aug 26, 2023 · c5b84fa · c5b84fa
1 parent b2dee18
commit c5b84fa
Show file tree

Hide file tree

Showing 58 changed files with 114 additions and 111 deletions.
diff --git a/agents/executor.go b/agents/executor.go
@@ -86,7 +86,7 @@ func (e Executor) doAction(
 	action schema.AgentAction,
 ) ([]schema.AgentStep, error) {
 	if e.CallbacksHandler != nil {
-		e.CallbacksHandler.HandleAgentAction(action)
+		e.CallbacksHandler.HandleAgentAction(ctx, action)
 	}
 
 	tool, ok := nameToTool[strings.ToUpper(action.Tool)]

diff --git a/callbacks/callbacks.go b/callbacks/callbacks.go
@@ -1,23 +1,25 @@
 package callbacks
 
 import (
+	"context"
+
 	"github.com/tmc/langchaingo/llms"
 	"github.com/tmc/langchaingo/schema"
 )
 
 // Handler is the interface that allows for hooking into specific parts of an
 // LLM application.
 type Handler interface {
-	HandleText(text string)
-	HandleLLMStart(prompts []string)
-	HandleLLMEnd(output llms.LLMResult)
-	HandleChainStart(inputs map[string]any)
-	HandleChainEnd(outputs map[string]any)
-	HandleToolStart(input string)
-	HandleToolEnd(output string)
-	HandleAgentAction(action schema.AgentAction)
-	HandleRetrieverStart(query string)
-	HandleRetrieverEnd(documents []schema.Document)
+	HandleText(ctx context.Context, text string)
+	HandleLLMStart(ctx context.Context, prompts []string)
+	HandleLLMEnd(ctx context.Context, output llms.LLMResult)
+	HandleChainStart(ctx context.Context, inputs map[string]any)
+	HandleChainEnd(ctx context.Context, outputs map[string]any)
+	HandleToolStart(ctx context.Context, input string)
+	HandleToolEnd(ctx context.Context, output string)
+	HandleAgentAction(ctx context.Context, action schema.AgentAction)
+	HandleRetrieverStart(ctx context.Context, query string)
+	HandleRetrieverEnd(ctx context.Context, documents []schema.Document)
 }
 
 // HandlerHaver is an interface used to get callbacks handler.

diff --git a/callbacks/log.go b/callbacks/log.go
@@ -2,6 +2,7 @@
 package callbacks
 
 import (
+	"context"
 	"fmt"
 	"strings"
 
@@ -14,43 +15,43 @@ type LogHandler struct{}
 
 var _ Handler = LogHandler{}
 
-func (l LogHandler) HandleText(text string) {
+func (l LogHandler) HandleText(ctx context.Context, text string) {
 	fmt.Println(text)
 }
 
-func (l LogHandler) HandleLLMStart(prompts []string) {
+func (l LogHandler) HandleLLMStart(ctx context.Context, prompts []string) {
 	fmt.Println("Entering LLM with prompts:", prompts)
 }
 
-func (l LogHandler) HandleLLMEnd(output llms.LLMResult) {
+func (l LogHandler) HandleLLMEnd(ctx context.Context, output llms.LLMResult) {
 	fmt.Println("Exiting LLM with results:", formatLLMResult(output))
 }
 
-func (l LogHandler) HandleChainStart(inputs map[string]any) {
+func (l LogHandler) HandleChainStart(ctx context.Context, inputs map[string]any) {
 	fmt.Println("Entering chain with inputs:", formatChainValues(inputs))
 }
 
-func (l LogHandler) HandleChainEnd(outputs map[string]any) {
+func (l LogHandler) HandleChainEnd(ctx context.Context, outputs map[string]any) {
 	fmt.Println("Exiting chain with outputs:", formatChainValues(outputs))
 }
 
-func (l LogHandler) HandleToolStart(input string) {
+func (l LogHandler) HandleToolStart(ctx context.Context, input string) {
 	fmt.Println("Entering tool with input:", removeNewLines(input))
 }
 
-func (l LogHandler) HandleToolEnd(output string) {
+func (l LogHandler) HandleToolEnd(ctx context.Context, output string) {
 	fmt.Println("Exiting tool with output:", removeNewLines(output))
 }
 
-func (l LogHandler) HandleAgentAction(action schema.AgentAction) {
+func (l LogHandler) HandleAgentAction(ctx context.Context, action schema.AgentAction) {
 	fmt.Println("Agent selected action:", formatAgentAction(action))
 }
 
-func (l LogHandler) HandleRetrieverStart(query string) {
+func (l LogHandler) HandleRetrieverStart(ctx context.Context, query string) {
 	fmt.Println("Entering retriever with query:", removeNewLines(query))
 }
 
-func (l LogHandler) HandleRetrieverEnd(documents []schema.Document) {
+func (l LogHandler) HandleRetrieverEnd(ctx context.Context, documents []schema.Document) {
 	fmt.Println("Exiting retirer with documents:", documents)
 }
 

diff --git a/chains/chains.go b/chains/chains.go
@@ -44,7 +44,7 @@ func Call(ctx context.Context, c Chain, inputValues map[string]any, options ...C
 
 	callbacksHandler := getChainCallbackHandler(c)
 	if callbacksHandler != nil {
-		callbacksHandler.HandleChainStart(inputValues)
+		callbacksHandler.HandleChainStart(ctx, inputValues)
 	}
 
 	if err := validateInputs(c, fullValues); err != nil {
@@ -60,7 +60,7 @@ func Call(ctx context.Context, c Chain, inputValues map[string]any, options ...C
 	}
 
 	if callbacksHandler != nil {
-		callbacksHandler.HandleChainEnd(outputValues)
+		callbacksHandler.HandleChainEnd(ctx, outputValues)
 	}
 
 	err = c.GetMemory().SaveContext(ctx, inputValues, outputValues)

diff --git a/examples/anthropic-completion-example/go.mod b/examples/anthropic-completion-example/go.mod
@@ -2,7 +2,7 @@ module anthropic-completion-example
 
 go 1.19
 
-require github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+require github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 
 require (
 	github.com/dlclark/regexp2 v1.8.1 // indirect

diff --git a/examples/anthropic-completion-example/go.sum b/examples/anthropic-completion-example/go.sum
@@ -7,6 +7,6 @@ github.com/pkoukk/tiktoken-go v0.1.2 h1:u7PCSBiWJ3nJYoTGShyM9iHXz4dNyYkurwwp+GHt
 github.com/pkoukk/tiktoken-go v0.1.2/go.mod h1:boMWvk9pQCOTx11pgu0DrIdrAKgQzzJKUP6vLXaz7Rw=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849 h1:y4TnpS57FeE5QzBzV2wKysVxWvkuMsha12yGHPVVAFo=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849/go.mod h1:8T+nNIGBr3nYQEYFmF/YaT8t8YTKLvFYZBuVZOAYn5E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0 h1:7pmW0coaYnLm4evqJ+QR10EgDU1ku+xGOpcHBMuvE7E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0/go.mod h1:fd7jP67Fwvcr+i7J+oAZbrh2aiekUuBLVqW/vgDslnw=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
diff --git a/examples/cohere-llm-example/go.mod b/examples/cohere-llm-example/go.mod
@@ -2,7 +2,7 @@ module basic-llm-example
 
 go 1.19
 
-require github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+require github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 
 require (
 	github.com/cohere-ai/tokenizer v1.1.2 // indirect

diff --git a/examples/cohere-llm-example/go.sum b/examples/cohere-llm-example/go.sum
@@ -11,6 +11,6 @@ github.com/pkoukk/tiktoken-go v0.1.2 h1:u7PCSBiWJ3nJYoTGShyM9iHXz4dNyYkurwwp+GHt
 github.com/pkoukk/tiktoken-go v0.1.2/go.mod h1:boMWvk9pQCOTx11pgu0DrIdrAKgQzzJKUP6vLXaz7Rw=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849 h1:y4TnpS57FeE5QzBzV2wKysVxWvkuMsha12yGHPVVAFo=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849/go.mod h1:8T+nNIGBr3nYQEYFmF/YaT8t8YTKLvFYZBuVZOAYn5E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0 h1:7pmW0coaYnLm4evqJ+QR10EgDU1ku+xGOpcHBMuvE7E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0/go.mod h1:fd7jP67Fwvcr+i7J+oAZbrh2aiekUuBLVqW/vgDslnw=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
diff --git a/examples/document-qa-example/go.mod b/examples/document-qa-example/go.mod
@@ -2,7 +2,7 @@ module document-qa-example
 
 go 1.19
 
-require github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+require github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 
 require (
 	github.com/Masterminds/goutils v1.1.1 // indirect

diff --git a/examples/document-qa-example/go.sum b/examples/document-qa-example/go.sum
@@ -66,8 +66,8 @@ github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+
 github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
 github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
 github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849 h1:y4TnpS57FeE5QzBzV2wKysVxWvkuMsha12yGHPVVAFo=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849/go.mod h1:8T+nNIGBr3nYQEYFmF/YaT8t8YTKLvFYZBuVZOAYn5E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0 h1:7pmW0coaYnLm4evqJ+QR10EgDU1ku+xGOpcHBMuvE7E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0/go.mod h1:fd7jP67Fwvcr+i7J+oAZbrh2aiekUuBLVqW/vgDslnw=
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 go.starlark.net v0.0.0-20230302034142-4b1e35fe2254 h1:Ss6D3hLXTM0KobyBYEAygXzFfGcjnmfEJOBgSbemCtg=
 go.starlark.net v0.0.0-20230302034142-4b1e35fe2254/go.mod h1:jxU+3+j+71eXOW14274+SmmuW82qJzl6iZSeqEtTGds=

diff --git a/examples/huggingface-llm-example/go.mod b/examples/huggingface-llm-example/go.mod
@@ -2,7 +2,7 @@ module huggingface-llm-example
 
 go 1.19
 
-require github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+require github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 
 require (
 	github.com/dlclark/regexp2 v1.8.1 // indirect

diff --git a/examples/huggingface-llm-example/go.sum b/examples/huggingface-llm-example/go.sum
@@ -7,6 +7,6 @@ github.com/pkoukk/tiktoken-go v0.1.2 h1:u7PCSBiWJ3nJYoTGShyM9iHXz4dNyYkurwwp+GHt
 github.com/pkoukk/tiktoken-go v0.1.2/go.mod h1:boMWvk9pQCOTx11pgu0DrIdrAKgQzzJKUP6vLXaz7Rw=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849 h1:y4TnpS57FeE5QzBzV2wKysVxWvkuMsha12yGHPVVAFo=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849/go.mod h1:8T+nNIGBr3nYQEYFmF/YaT8t8YTKLvFYZBuVZOAYn5E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0 h1:7pmW0coaYnLm4evqJ+QR10EgDU1ku+xGOpcHBMuvE7E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0/go.mod h1:fd7jP67Fwvcr+i7J+oAZbrh2aiekUuBLVqW/vgDslnw=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
diff --git a/examples/llm-chain-example/go.mod b/examples/llm-chain-example/go.mod
@@ -2,7 +2,7 @@ module llm-chain-example
 
 go 1.19
 
-require github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+require github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 
 require (
 	github.com/Masterminds/goutils v1.1.1 // indirect

diff --git a/examples/llm-chain-example/go.sum b/examples/llm-chain-example/go.sum
@@ -66,8 +66,8 @@ github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+
 github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
 github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
 github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849 h1:y4TnpS57FeE5QzBzV2wKysVxWvkuMsha12yGHPVVAFo=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849/go.mod h1:8T+nNIGBr3nYQEYFmF/YaT8t8YTKLvFYZBuVZOAYn5E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0 h1:7pmW0coaYnLm4evqJ+QR10EgDU1ku+xGOpcHBMuvE7E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0/go.mod h1:fd7jP67Fwvcr+i7J+oAZbrh2aiekUuBLVqW/vgDslnw=
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 go.starlark.net v0.0.0-20230302034142-4b1e35fe2254 h1:Ss6D3hLXTM0KobyBYEAygXzFfGcjnmfEJOBgSbemCtg=
 go.starlark.net v0.0.0-20230302034142-4b1e35fe2254/go.mod h1:jxU+3+j+71eXOW14274+SmmuW82qJzl6iZSeqEtTGds=

diff --git a/examples/llmmath-chain-example/go.mod b/examples/llmmath-chain-example/go.mod
@@ -2,7 +2,7 @@ module llmmath-chain-example
 
 go 1.19
 
-require github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+require github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 
 require (
 	github.com/Masterminds/goutils v1.1.1 // indirect

diff --git a/examples/llmmath-chain-example/go.sum b/examples/llmmath-chain-example/go.sum
@@ -66,8 +66,8 @@ github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+
 github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
 github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
 github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849 h1:y4TnpS57FeE5QzBzV2wKysVxWvkuMsha12yGHPVVAFo=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849/go.mod h1:8T+nNIGBr3nYQEYFmF/YaT8t8YTKLvFYZBuVZOAYn5E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0 h1:7pmW0coaYnLm4evqJ+QR10EgDU1ku+xGOpcHBMuvE7E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0/go.mod h1:fd7jP67Fwvcr+i7J+oAZbrh2aiekUuBLVqW/vgDslnw=
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 go.starlark.net v0.0.0-20230302034142-4b1e35fe2254 h1:Ss6D3hLXTM0KobyBYEAygXzFfGcjnmfEJOBgSbemCtg=
 go.starlark.net v0.0.0-20230302034142-4b1e35fe2254/go.mod h1:jxU+3+j+71eXOW14274+SmmuW82qJzl6iZSeqEtTGds=

diff --git a/examples/llmsummarization-chain-example/go.mod b/examples/llmsummarization-chain-example/go.mod
@@ -2,7 +2,7 @@ module llmsummarization-chain-example
 
 go 1.19
 
-require github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+require github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 
 require (
 	cloud.google.com/go v0.110.0 // indirect

diff --git a/examples/llmsummarization-chain-example/go.sum b/examples/llmsummarization-chain-example/go.sum
@@ -127,8 +127,8 @@ github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/
 github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
 github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
 github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849 h1:y4TnpS57FeE5QzBzV2wKysVxWvkuMsha12yGHPVVAFo=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849/go.mod h1:8T+nNIGBr3nYQEYFmF/YaT8t8YTKLvFYZBuVZOAYn5E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0 h1:7pmW0coaYnLm4evqJ+QR10EgDU1ku+xGOpcHBMuvE7E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0/go.mod h1:fd7jP67Fwvcr+i7J+oAZbrh2aiekUuBLVqW/vgDslnw=
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 go.opencensus.io v0.24.0 h1:y73uSU6J157QMP2kn2r30vwW1A2W2WFwSCGnAVxeaD0=
 go.opencensus.io v0.24.0/go.mod h1:vNK8G9p7aAivkbmorf4v+7Hgx+Zs0yY+0fOtgBfjQKo=

diff --git a/examples/local-llm-example/go.mod b/examples/local-llm-example/go.mod
@@ -2,7 +2,7 @@ module local-llm-example
 
 go 1.19
 
-require github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+require github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 
 require (
 	github.com/dlclark/regexp2 v1.8.1 // indirect

diff --git a/examples/local-llm-example/go.sum b/examples/local-llm-example/go.sum
@@ -7,6 +7,6 @@ github.com/pkoukk/tiktoken-go v0.1.2 h1:u7PCSBiWJ3nJYoTGShyM9iHXz4dNyYkurwwp+GHt
 github.com/pkoukk/tiktoken-go v0.1.2/go.mod h1:boMWvk9pQCOTx11pgu0DrIdrAKgQzzJKUP6vLXaz7Rw=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849 h1:y4TnpS57FeE5QzBzV2wKysVxWvkuMsha12yGHPVVAFo=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849/go.mod h1:8T+nNIGBr3nYQEYFmF/YaT8t8YTKLvFYZBuVZOAYn5E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0 h1:7pmW0coaYnLm4evqJ+QR10EgDU1ku+xGOpcHBMuvE7E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0/go.mod h1:fd7jP67Fwvcr+i7J+oAZbrh2aiekUuBLVqW/vgDslnw=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
diff --git a/examples/mrkl-agent-example/go.mod b/examples/mrkl-agent-example/go.mod
@@ -2,7 +2,7 @@ module mrkl-agent-example
 
 go 1.19
 
-require github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+require github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 
 require (
 	github.com/Masterminds/goutils v1.1.1 // indirect

diff --git a/examples/mrkl-agent-example/go.sum b/examples/mrkl-agent-example/go.sum
@@ -66,8 +66,8 @@ github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+
 github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
 github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
 github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849 h1:y4TnpS57FeE5QzBzV2wKysVxWvkuMsha12yGHPVVAFo=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849/go.mod h1:8T+nNIGBr3nYQEYFmF/YaT8t8YTKLvFYZBuVZOAYn5E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0 h1:7pmW0coaYnLm4evqJ+QR10EgDU1ku+xGOpcHBMuvE7E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0/go.mod h1:fd7jP67Fwvcr+i7J+oAZbrh2aiekUuBLVqW/vgDslnw=
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 go.starlark.net v0.0.0-20230302034142-4b1e35fe2254 h1:Ss6D3hLXTM0KobyBYEAygXzFfGcjnmfEJOBgSbemCtg=
 go.starlark.net v0.0.0-20230302034142-4b1e35fe2254/go.mod h1:jxU+3+j+71eXOW14274+SmmuW82qJzl6iZSeqEtTGds=

diff --git a/examples/openai-chat-example/go.mod b/examples/openai-chat-example/go.mod
@@ -2,7 +2,7 @@ module openai-chat-example
 
 go 1.19
 
-require github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+require github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 
 require (
 	github.com/dlclark/regexp2 v1.8.1 // indirect

diff --git a/examples/openai-chat-example/go.sum b/examples/openai-chat-example/go.sum
@@ -7,6 +7,6 @@ github.com/pkoukk/tiktoken-go v0.1.2 h1:u7PCSBiWJ3nJYoTGShyM9iHXz4dNyYkurwwp+GHt
 github.com/pkoukk/tiktoken-go v0.1.2/go.mod h1:boMWvk9pQCOTx11pgu0DrIdrAKgQzzJKUP6vLXaz7Rw=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849 h1:y4TnpS57FeE5QzBzV2wKysVxWvkuMsha12yGHPVVAFo=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849/go.mod h1:8T+nNIGBr3nYQEYFmF/YaT8t8YTKLvFYZBuVZOAYn5E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0 h1:7pmW0coaYnLm4evqJ+QR10EgDU1ku+xGOpcHBMuvE7E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0/go.mod h1:fd7jP67Fwvcr+i7J+oAZbrh2aiekUuBLVqW/vgDslnw=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
diff --git a/examples/openai-completion-example/go.mod b/examples/openai-completion-example/go.mod
@@ -2,7 +2,7 @@ module openai-completion-example
 
 go 1.19
 
-require github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+require github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 
 require (
 	github.com/dlclark/regexp2 v1.8.1 // indirect

diff --git a/examples/openai-completion-example/go.sum b/examples/openai-completion-example/go.sum
@@ -7,6 +7,6 @@ github.com/pkoukk/tiktoken-go v0.1.2 h1:u7PCSBiWJ3nJYoTGShyM9iHXz4dNyYkurwwp+GHt
 github.com/pkoukk/tiktoken-go v0.1.2/go.mod h1:boMWvk9pQCOTx11pgu0DrIdrAKgQzzJKUP6vLXaz7Rw=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849 h1:y4TnpS57FeE5QzBzV2wKysVxWvkuMsha12yGHPVVAFo=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849/go.mod h1:8T+nNIGBr3nYQEYFmF/YaT8t8YTKLvFYZBuVZOAYn5E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0 h1:7pmW0coaYnLm4evqJ+QR10EgDU1ku+xGOpcHBMuvE7E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0/go.mod h1:fd7jP67Fwvcr+i7J+oAZbrh2aiekUuBLVqW/vgDslnw=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
diff --git a/examples/openai-function-call-example/go.mod b/examples/openai-function-call-example/go.mod
@@ -2,7 +2,7 @@ module openai-function-call-example
 
 go 1.19
 
-require github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+require github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 
 require (
 	github.com/dlclark/regexp2 v1.8.1 // indirect

diff --git a/examples/openai-function-call-example/go.sum b/examples/openai-function-call-example/go.sum
@@ -7,6 +7,6 @@ github.com/pkoukk/tiktoken-go v0.1.2 h1:u7PCSBiWJ3nJYoTGShyM9iHXz4dNyYkurwwp+GHt
 github.com/pkoukk/tiktoken-go v0.1.2/go.mod h1:boMWvk9pQCOTx11pgu0DrIdrAKgQzzJKUP6vLXaz7Rw=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849 h1:y4TnpS57FeE5QzBzV2wKysVxWvkuMsha12yGHPVVAFo=
-github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849/go.mod h1:8T+nNIGBr3nYQEYFmF/YaT8t8YTKLvFYZBuVZOAYn5E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0 h1:7pmW0coaYnLm4evqJ+QR10EgDU1ku+xGOpcHBMuvE7E=
+github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0/go.mod h1:fd7jP67Fwvcr+i7J+oAZbrh2aiekUuBLVqW/vgDslnw=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
diff --git a/examples/pinecone-vectorstore-example/go.mod b/examples/pinecone-vectorstore-example/go.mod
@@ -4,7 +4,7 @@ go 1.20
 
 require (
 	github.com/google/uuid v1.3.0
-	github.com/tmc/langchaingo v0.0.0-20230729231952-1f3948210849
+	github.com/tmc/langchaingo v0.0.0-20230826015154-aa97aec400c0
 )
 
 require (