diff --git a/go.mod b/go.mod
index e1eb4bbe..e5e61371 100644
--- a/go.mod
+++ b/go.mod
@@ -2,7 +2,7 @@ module github.com/gptscript-ai/gptscript
 
 go 1.22.0
 
-replace github.com/sashabaranov/go-openai => github.com/gptscript-ai/go-openai v0.0.0-20240329145331-9325362ed457
+replace github.com/sashabaranov/go-openai => github.com/gptscript-ai/go-openai v0.0.0-20240330064700-d33320ca826f
 
 require (
 	github.com/AlecAivazis/survey/v2 v2.3.7
diff --git a/go.sum b/go.sum
index 4fda2332..9d410683 100644
--- a/go.sum
+++ b/go.sum
@@ -124,8 +124,8 @@ github.com/googleapis/gax-go/v2 v2.0.4/go.mod h1:0Wqv26UfaUD9n4G6kQubkQ+KchISgw+
 github.com/googleapis/gax-go/v2 v2.0.5/go.mod h1:DWXyrwAJ9X0FpwwEdw+IPEYBICEFu5mhpdKc/us6bOk=
 github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc=
 github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
-github.com/gptscript-ai/go-openai v0.0.0-20240329145331-9325362ed457 h1:ziAtY3K9pxk4Ua/0dQdndAopXyLlzguX9ECRbi8E9t4=
-github.com/gptscript-ai/go-openai v0.0.0-20240329145331-9325362ed457/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg=
+github.com/gptscript-ai/go-openai v0.0.0-20240330064700-d33320ca826f h1:XSqwZIal7wCjmkV3Ncrvu5TAvXlaUnjGwygn21OTlh0=
+github.com/gptscript-ai/go-openai v0.0.0-20240330064700-d33320ca826f/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg=
 github.com/hashicorp/errwrap v1.0.0 h1:hLrqtEDnRye3+sgx6z4qVLNuviH3MR5aQ0ykNJa/UYA=
 github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
 github.com/hashicorp/go-multierror v1.1.1 h1:H5DkEtf6CXdFp0N0Em5UCwQpXMWke8IA0+lD48awMYo=
diff --git a/pkg/openai/client.go b/pkg/openai/client.go
index 90b7d76c..02cf44d6 100644
--- a/pkg/openai/client.go
+++ b/pkg/openai/client.go
@@ -263,6 +263,8 @@ func toMessages(request types.CompletionRequest) (result []openai.ChatCompletion
 
 		if message.ToolCall != nil {
 			chatMessage.ToolCallID = message.ToolCall.ID
+			// This field is not documented but specifically Azure thinks it should be set
+			chatMessage.Name = message.ToolCall.Function.Name
 		}
 
 		for _, content := range message.Content {