diff --git a/go.mod b/go.mod index 2438264d..25c30a82 100644 --- a/go.mod +++ b/go.mod @@ -14,7 +14,7 @@ require ( github.com/henomis/qdrant-go v1.1.0 github.com/henomis/restclientgo v1.2.0 github.com/invopop/jsonschema v0.7.0 - github.com/sashabaranov/go-openai v1.19.2 + github.com/sashabaranov/go-openai v1.24.0 ) require ( diff --git a/go.sum b/go.sum index 3e878b8a..e2d3391c 100644 --- a/go.sum +++ b/go.sum @@ -27,8 +27,8 @@ github.com/mitchellh/mapstructure v1.5.0 h1:jeMsZIYE/09sWLaz43PL7Gy6RuMjD2eJVyua github.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo= github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= -github.com/sashabaranov/go-openai v1.19.2 h1:+dkuCADSnwXV02YVJkdphY8XD9AyHLUWwk6V7LB6EL8= -github.com/sashabaranov/go-openai v1.19.2/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg= +github.com/sashabaranov/go-openai v1.24.0 h1:4H4Pg8Bl2RH/YSnU8DYumZbuHnnkfioor/dtNlB20D4= +github.com/sashabaranov/go-openai v1.24.0/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg= github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= diff --git a/llm/openai/common.go b/llm/openai/common.go index 5d7b3e65..8b78f880 100644 --- a/llm/openai/common.go +++ b/llm/openai/common.go @@ -28,11 +28,16 @@ const ( GPT432K Model = openai.GPT432K GPT40613 Model = openai.GPT40613 GPT40314 Model = openai.GPT40314 + GPT4o Model = openai.GPT4o + GPT4o20240513 Model = openai.GPT4o20240513 + GPT4Turbo Model = openai.GPT4Turbo + GPT4Turbo20240409 Model = openai.GPT4Turbo20240409 GPT4Turbo0125 Model = openai.GPT4Turbo0125 GPT4Turbo1106 Model = openai.GPT4Turbo1106 GPT4TurboPreview Model = openai.GPT4TurboPreview GPT4VisionPreview Model = openai.GPT4VisionPreview GPT4 Model = openai.GPT4 + GPT3Dot5Turbo0125 Model = openai.GPT3Dot5Turbo0125 GPT3Dot5Turbo1106 Model = openai.GPT3Dot5Turbo1106 GPT3Dot5Turbo0613 Model = openai.GPT3Dot5Turbo0613 GPT3Dot5Turbo0301 Model = openai.GPT3Dot5Turbo0301 diff --git a/llm/openai/openai.go b/llm/openai/openai.go index 94005872..cffcfdf7 100644 --- a/llm/openai/openai.go +++ b/llm/openai/openai.go @@ -379,7 +379,7 @@ func (o *OpenAI) getChatCompletionRequestTools() []openai.Tool { for _, function := range o.functions { tools = append(tools, openai.Tool{ Type: "function", - Function: openai.FunctionDefinition{ + Function: &openai.FunctionDefinition{ Name: function.Name, Description: function.Description, Parameters: function.Parameters,