Skip to content

Commit

Permalink
Add support for azure openai new version API (2023-07-01-preview) (#451)
Browse files Browse the repository at this point in the history
  • Loading branch information
ZeroDeng01 authored Jul 20, 2023
1 parent 1876e0c commit 1153eb2
Show file tree
Hide file tree
Showing 2 changed files with 39 additions and 8 deletions.
29 changes: 29 additions & 0 deletions chat.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,35 @@ var (
ErrChatCompletionStreamNotSupported = errors.New("streaming is not supported with this method, please use CreateChatCompletionStream") //nolint:lll
)

type Hate struct {
Filtered bool `json:"filtered"`
Severity string `json:"severity,omitempty"`
}
type SelfHarm struct {
Filtered bool `json:"filtered"`
Severity string `json:"severity,omitempty"`
}
type Sexual struct {
Filtered bool `json:"filtered"`
Severity string `json:"severity,omitempty"`
}
type Violence struct {
Filtered bool `json:"filtered"`
Severity string `json:"severity,omitempty"`
}

type ContentFilterResults struct {
Hate Hate `json:"hate,omitempty"`
SelfHarm SelfHarm `json:"self_harm,omitempty"`
Sexual Sexual `json:"sexual,omitempty"`
Violence Violence `json:"violence,omitempty"`
}

type PromptAnnotation struct {
PromptIndex int `json:"prompt_index,omitempty"`
ContentFilterResults ContentFilterResults `json:"content_filter_results,omitempty"`
}

type ChatCompletionMessage struct {
Role string `json:"role"`
Content string `json:"content"`
Expand Down
18 changes: 10 additions & 8 deletions chat_stream.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,17 +12,19 @@ type ChatCompletionStreamChoiceDelta struct {
}

type ChatCompletionStreamChoice struct {
Index int `json:"index"`
Delta ChatCompletionStreamChoiceDelta `json:"delta"`
FinishReason FinishReason `json:"finish_reason"`
Index int `json:"index"`
Delta ChatCompletionStreamChoiceDelta `json:"delta"`
FinishReason FinishReason `json:"finish_reason"`
ContentFilterResults ContentFilterResults `json:"content_filter_results,omitempty"`
}

type ChatCompletionStreamResponse struct {
ID string `json:"id"`
Object string `json:"object"`
Created int64 `json:"created"`
Model string `json:"model"`
Choices []ChatCompletionStreamChoice `json:"choices"`
ID string `json:"id"`
Object string `json:"object"`
Created int64 `json:"created"`
Model string `json:"model"`
Choices []ChatCompletionStreamChoice `json:"choices"`
PromptAnnotations []PromptAnnotation `json:"prompt_annotations,omitempty"`
}

// ChatCompletionStream
Expand Down

0 comments on commit 1153eb2

Please sign in to comment.