Added chat

Author: djthorpe

pkg/openai/chat.go

@@ -0,0 +1,36 @@
+package openai
+
+import (
+	// Packages
+	"github.com/mutablelogic/go-client/pkg/client"
+)
+
+const (
+	defaultChatCompletion = "gpt-3.5-turbo"
+)
+
+///////////////////////////////////////////////////////////////////////////////
+// API CALLS
+
+// Chat creates a model response for the given chat conversation.
+func (c *Client) Chat(opts ...Opt) (Chat, error) {
+	// Create the request
+	var request reqChat
+	request.Model = defaultChatCompletion
+	for _, opt := range opts {
+		if err := opt(&request); err != nil {
+			return Chat{}, err
+		}
+	}
+
+	// Return the response
+	var response Chat
+	if payload, err := client.NewJSONRequest(request, client.ContentTypeJson); err != nil {
+		return Chat{}, err
+	} else if err := c.Do(payload.Post(), &response, client.OptPath("chat/completions")); err != nil {
+		return Chat{}, err
+	}
+
+	// Return success
+	return response, nil
+}

pkg/openai/chat_test.go

@@ -0,0 +1,21 @@
+package openai_test
+
+import (
+	"os"
+	"testing"
+
+	opts "github.com/mutablelogic/go-client/pkg/client"
+	openai "github.com/mutablelogic/go-client/pkg/openai"
+	assert "github.com/stretchr/testify/assert"
+)
+
+func Test_chat_001(t *testing.T) {
+	assert := assert.New(t)
+	client, err := openai.New(GetApiKey(t), opts.OptTrace(os.Stderr, true))
+	assert.NoError(err)
+	assert.NotNil(client)
+
+	response, err := client.Chat()
+	assert.NoError(err)
+	assert.NotNil(response)
+}

pkg/openai/opts.go

@@ -15,7 +15,7 @@ import (
 type Opt func(Request) error
 
 ///////////////////////////////////////////////////////////////////////////////
-// CreateEmbedding request options
+// setModel
 
 // Set the model identifier
 func (req *reqCreateEmbedding) setModel(value string) error {
@@ -27,6 +27,16 @@ func (req *reqCreateEmbedding) setModel(value string) error {
 	}
 }
 
+// Set the model identifier
+func (req *reqChat) setModel(value string) error {
+	if value == "" {
+		return ErrBadParameter.With("Model")
+	} else {
+		req.Model = value
+		return nil
+	}
+}
+
 ///////////////////////////////////////////////////////////////////////////////
 // PUBLIC METHODS
 

pkg/openai/schema.go

@@ -31,6 +31,33 @@ type Embeddings struct {
 	} `json:"usage"`
 }
 
+// A chat completion object
+type Chat struct {
+	Id                string           `json:"id"`
+	Object            string           `json:"object"`
+	Created           int64            `json:"created"`
+	Model             string           `json:"model"`
+	SystemFingerprint string           `json:"system_fingerprint"`
+	Choices           []*MessageChoice `json:"choices"`
+	Usage             struct {
+		PromptTokens     int `json:"prompt_tokens"`
+		CompletionTokens int `json:"completion_tokens"`
+		TotalTokens      int `json:"total_tokens"`
+	} `json:"usage"`
+}
+
+// A message choice object
+type MessageChoice struct {
+	Index        int    `json:"index"`
+	FinishReason string `json:"finish_reason"`
+}
+
+// A message choice object
+type Message struct {
+	Role    string `json:"role"`
+	Content string `json:"content"`
+}
+
 ///////////////////////////////////////////////////////////////////////////////
 // REQUESTS
 
@@ -41,6 +68,15 @@ type reqCreateEmbedding struct {
 	User           string   `json:"user,omitempty"`
 }
 
+type reqChat struct {
+	Model            string    `json:"model"`
+	Messages         []Message `json:"messages"`
+	FrequencyPenalty float64   `json:"frequency_penalty,omitempty"`
+	PresencePenalty  float64   `json:"presence_penalty,omitempty"`
+	MaxTokens        int       `json:"max_tokens,omitempty"`
+	Count            int       `json:"n,omitempty"`
+}
+
 ///////////////////////////////////////////////////////////////////////////////
 // RESPONSES