Skip to content

Commit

Permalink
Feat/81 support new openai features (#88)
Browse files Browse the repository at this point in the history
  • Loading branch information
henomis committed Jun 15, 2023
1 parent 1cabb13 commit 36bb01b
Show file tree
Hide file tree
Showing 4 changed files with 29 additions and 15 deletions.
3 changes: 3 additions & 0 deletions .github/FUNDING.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Supported funding model platforms

github: [henomis]
2 changes: 1 addition & 1 deletion go.mod
Original file line number Diff line number Diff line change
Expand Up @@ -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 (
Expand Down
4 changes: 2 additions & 2 deletions go.sum
Original file line number Diff line number Diff line change
Expand Up @@ -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=
35 changes: 23 additions & 12 deletions llm/openai/openai.go
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 36bb01b

Please sign in to comment.