EscapeString

chat.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 	"net/http"
+	"strconv"
 )
 
 // Chat message role defined by the OpenAI API.
@@ -21,9 +22,15 @@ var (
 	ErrChatCompletionStreamNotSupported = errors.New("streaming is not supported with this method, please use CreateChatCompletionStream")              //nolint:lll
 )
 
+type EscapeString string
+
+func (esc EscapeString) MarshalJSON() ([]byte, error) {
+	return []byte(strconv.QuoteToASCII(string(esc))), nil
+}
+
 type ChatCompletionMessage struct {
-	Role    string `json:"role"`
-	Content string `json:"content"`
+	Role    string       `json:"role"`
+	Content EscapeString `json:"content"`
 
 	// This property isn't in the official documentation, but it's in
 	// the documentation for the official library for python:

chat_test.go

@@ -268,12 +268,12 @@ func handleChatCompletionEndpoint(w http.ResponseWriter, r *http.Request) {
 		res.Choices = append(res.Choices, ChatCompletionChoice{
 			Message: ChatCompletionMessage{
 				Role:    ChatMessageRoleAssistant,
-				Content: completionStr,
+				Content: EscapeString(completionStr),
 			},
 			Index: i,
 		})
 	}
-	inputTokens := numTokens(completionReq.Messages[0].Content) * n
+	inputTokens := numTokens(string(completionReq.Messages[0].Content)) * n
 	completionTokens := completionReq.MaxTokens * n
 	res.Usage = Usage{
 		PromptTokens:     inputTokens,

example_test.go

@@ -289,7 +289,7 @@ func Example_chatbot() {
 	for s.Scan() {
 		req.Messages = append(req.Messages, openai.ChatCompletionMessage{
 			Role:    openai.ChatMessageRoleUser,
-			Content: s.Text(),
+			Content: openai.EscapeString(s.Text()),
 		})
 		resp, err := client.CreateChatCompletion(context.Background(), req)
 		if err != nil {

examples/chatbot/main.go

@@ -26,9 +26,10 @@ func main() {
 	fmt.Print("> ")
 	s := bufio.NewScanner(os.Stdin)
 	for s.Scan() {
+		content := s.Text()
 		req.Messages = append(req.Messages, openai.ChatCompletionMessage{
 			Role:    openai.ChatMessageRoleUser,
-			Content: s.Text(),
+			Content: openai.EscapeString(content),
 		})
 		resp, err := client.CreateChatCompletion(context.Background(), req)
 		if err != nil {