Add PromptFilterResult (#702)

This commit is contained in:
Mike Chaykowsky
2024-04-16 12:28:06 -07:00
committed by GitHub
parent a42f51967f
commit c6a63ed19a

View File

@@ -19,6 +19,11 @@ type ChatCompletionStreamChoice struct {
ContentFilterResults ContentFilterResults `json:"content_filter_results,omitempty"`
}
type PromptFilterResult struct {
Index int `json:"index"`
ContentFilterResults ContentFilterResults `json:"content_filter_results,omitempty"`
}
type ChatCompletionStreamResponse struct {
ID string `json:"id"`
Object string `json:"object"`
@@ -26,6 +31,7 @@ type ChatCompletionStreamResponse struct {
Model string `json:"model"`
Choices []ChatCompletionStreamChoice `json:"choices"`
PromptAnnotations []PromptAnnotation `json:"prompt_annotations,omitempty"`
PromptFilterResults []PromptFilterResult `json:"prompt_filter_results,omitempty"`
}
// ChatCompletionStream