From 36bb01b179df19e1dac453054878b3fdcd7e8556 Mon Sep 17 00:00:00 2001 From: Simone Vellei Date: Thu, 15 Jun 2023 18:59:27 +0200 Subject: [PATCH] Feat/81 support new openai features (#88) --- .github/FUNDING.yml | 3 +++ go.mod | 2 +- go.sum | 4 ++-- llm/openai/openai.go | 35 +++++++++++++++++++++++------------ 4 files changed, 29 insertions(+), 15 deletions(-) create mode 100644 .github/FUNDING.yml diff --git a/.github/FUNDING.yml b/.github/FUNDING.yml new file mode 100644 index 00000000..c2bd53aa --- /dev/null +++ b/.github/FUNDING.yml @@ -0,0 +1,3 @@ +# Supported funding model platforms + +github: [henomis] \ No newline at end of file diff --git a/go.mod b/go.mod index 30170dc2..316617b8 100644 --- a/go.mod +++ b/go.mod @@ -8,7 +8,7 @@ require ( github.com/google/uuid v1.3.0 github.com/henomis/pinecone-go v1.1.1 github.com/pkoukk/tiktoken-go v0.1.1 - github.com/sashabaranov/go-openai v1.10.1 + github.com/sashabaranov/go-openai v1.11.1 ) require ( diff --git a/go.sum b/go.sum index ad5d4dd8..ed72eff6 100644 --- a/go.sum +++ b/go.sum @@ -12,7 +12,7 @@ github.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RR github.com/pkoukk/tiktoken-go v0.1.1 h1:jtkYlIECjyM9OW1w4rjPmTohK4arORP9V25y6TM6nXo= github.com/pkoukk/tiktoken-go v0.1.1/go.mod h1:boMWvk9pQCOTx11pgu0DrIdrAKgQzzJKUP6vLXaz7Rw= github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= -github.com/sashabaranov/go-openai v1.10.1 h1:6WyHJaNzF266VaEEuW6R4YW+Ei0wpMnqRYPGK7fhuhQ= -github.com/sashabaranov/go-openai v1.10.1/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg= +github.com/sashabaranov/go-openai v1.11.1 h1:QAyLx/MhiVrV4nsGTAZI+JPP/n0nlYZfxltzv9jTClw= +github.com/sashabaranov/go-openai v1.11.1/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg= github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8= gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= diff --git a/llm/openai/openai.go b/llm/openai/openai.go index c0280c3c..0b51e52d 100644 --- a/llm/openai/openai.go +++ b/llm/openai/openai.go @@ -30,18 +30,29 @@ const ( type Model string const ( - GPT4 Model = openai.GPT4 - GPT3Dot5Turbo Model = openai.GPT3Dot5Turbo - GPT3TextDavinci003 Model = openai.GPT3TextDavinci003 - GPT3TextDavinci002 Model = openai.GPT3TextDavinci002 - GPT3TextCurie001 Model = openai.GPT3TextCurie001 - GPT3TextBabbage001 Model = openai.GPT3TextBabbage001 - GPT3TextAda001 Model = openai.GPT3TextAda001 - GPT3TextDavinci001 Model = openai.GPT3TextDavinci001 - GPT3Davinci Model = openai.GPT3Davinci - GPT3Curie Model = openai.GPT3Curie - GPT3Ada Model = openai.GPT3Ada - GPT3Babbage Model = openai.GPT3Babbage + GPT432K0613 Model = openai.GPT432K0613 + GPT432K0314 Model = openai.GPT432K0314 + GPT432K Model = openai.GPT432K + GPT40613 Model = openai.GPT40613 + GPT40314 Model = openai.GPT40314 + GPT4 Model = openai.GPT4 + GPT3Dot5Turbo0613 Model = openai.GPT3Dot5Turbo0613 + GPT3Dot5Turbo0301 Model = openai.GPT3Dot5Turbo0301 + GPT3Dot5Turbo16K Model = openai.GPT3Dot5Turbo16K + GPT3Dot5Turbo16K0613 Model = openai.GPT3Dot5Turbo16K0613 + GPT3Dot5Turbo Model = openai.GPT3Dot5Turbo + GPT3TextDavinci003 Model = openai.GPT3TextDavinci003 + GPT3TextDavinci002 Model = openai.GPT3TextDavinci002 + GPT3TextCurie001 Model = openai.GPT3TextCurie001 + GPT3TextBabbage001 Model = openai.GPT3TextBabbage001 + GPT3TextAda001 Model = openai.GPT3TextAda001 + GPT3TextDavinci001 Model = openai.GPT3TextDavinci001 + GPT3DavinciInstructBeta Model = openai.GPT3DavinciInstructBeta + GPT3Davinci Model = openai.GPT3Davinci + GPT3CurieInstructBeta Model = openai.GPT3CurieInstructBeta + GPT3Curie Model = openai.GPT3Curie + GPT3Ada Model = openai.GPT3Ada + GPT3Babbage Model = openai.GPT3Babbage ) type OpenAIUsageCallback func(types.Meta)