VaalaCat/go-openai
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add refusal (#844)

Browse Source 
* add custom marshaller, documentation and isolate tests

* fix linter

* add missing field
This commit is contained in:
Quest Henkart
2024-09-11 22:24:49 +02:00
committed by GitHub
parent 643da8d650
commit 194a03e763
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
chat.go
View File

@@ -82,6 +82,7 @@ type ChatMessagePart struct {
type ChatCompletionMessage struct { type ChatCompletionMessage struct {
Role string `json:"role"` Role string `json:"role"`
Content string `json:"content"` Content string `json:"content"`
Refusal string `json:"refusal,omitempty"`
MultiContent []ChatMessagePart MultiContent []ChatMessagePart
// This property isn't in the official documentation, but it's in // This property isn't in the official documentation, but it's in
@@ -107,6 +108,7 @@ func (m ChatCompletionMessage) MarshalJSON() ([]byte, error) {
msg := struct { msg := struct {
Role string `json:"role"` Role string `json:"role"`
Content string `json:"-"` Content string `json:"-"`
Refusal string `json:"refusal,omitempty"`
MultiContent []ChatMessagePart `json:"content,omitempty"` MultiContent []ChatMessagePart `json:"content,omitempty"`
Name string `json:"name,omitempty"` Name string `json:"name,omitempty"`
FunctionCall *FunctionCall `json:"function_call,omitempty"` FunctionCall *FunctionCall `json:"function_call,omitempty"`
@@ -115,9 +117,11 @@ func (m ChatCompletionMessage) MarshalJSON() ([]byte, error) {
}(m) }(m)
return json.Marshal(msg) return json.Marshal(msg)
} }
msg := struct { msg := struct {
Role string `json:"role"` Role string `json:"role"`
Content string `json:"content"` Content string `json:"content"`
Refusal string `json:"refusal,omitempty"`
MultiContent []ChatMessagePart `json:"-"` MultiContent []ChatMessagePart `json:"-"`
Name string `json:"name,omitempty"` Name string `json:"name,omitempty"`
FunctionCall *FunctionCall `json:"function_call,omitempty"` FunctionCall *FunctionCall `json:"function_call,omitempty"`
@@ -131,12 +135,14 @@ func (m *ChatCompletionMessage) UnmarshalJSON(bs []byte) error {
msg := struct { msg := struct {
Role string `json:"role"` Role string `json:"role"`
Content string `json:"content"` Content string `json:"content"`
Refusal string `json:"refusal,omitempty"`
MultiContent []ChatMessagePart MultiContent []ChatMessagePart
Name string `json:"name,omitempty"` Name string `json:"name,omitempty"`
FunctionCall *FunctionCall `json:"function_call,omitempty"` FunctionCall *FunctionCall `json:"function_call,omitempty"`
ToolCalls []ToolCall `json:"tool_calls,omitempty"` ToolCalls []ToolCall `json:"tool_calls,omitempty"`
ToolCallID string `json:"tool_call_id,omitempty"` ToolCallID string `json:"tool_call_id,omitempty"`
}{} }{}
if err := json.Unmarshal(bs, &msg); err == nil { if err := json.Unmarshal(bs, &msg); err == nil {
*m = ChatCompletionMessage(msg) *m = ChatCompletionMessage(msg)
return nil return nil
@@ -144,6 +150,7 @@ func (m *ChatCompletionMessage) UnmarshalJSON(bs []byte) error {
multiMsg := struct { multiMsg := struct {
Role string `json:"role"` Role string `json:"role"`
Content string Content string
Refusal string `json:"refusal,omitempty"`
MultiContent []ChatMessagePart `json:"content"` MultiContent []ChatMessagePart `json:"content"`
Name string `json:"name,omitempty"` Name string `json:"name,omitempty"`
FunctionCall *FunctionCall `json:"function_call,omitempty"` FunctionCall *FunctionCall `json:"function_call,omitempty"`