Updated embedding call

Author: djthorpe

pkg/mistral/chat_test.go

@@ -16,7 +16,7 @@ func Test_chat_001(t *testing.T) {
 	client, err := mistral.New(GetApiKey(t), opts.OptTrace(os.Stderr, true))
 	assert.NoError(err)
 	assert.NotNil(client)
-	err = client.Chat([]schema.Message{
+	_, err = client.Chat([]schema.Message{
 		{Role: "user", Content: "What is the weather"},
 	})
 	assert.NoError(err)

pkg/mistral/embedding.go

@@ -2,11 +2,11 @@ package mistral
 
 import (
 	// Packages
-	"github.com/mutablelogic/go-client/pkg/client"
+	client "github.com/mutablelogic/go-client/pkg/client"
+	schema "github.com/mutablelogic/go-client/pkg/openai/schema"
 
 	// Namespace imports
 	. "github.com/djthorpe/go-errors"
-	. "github.com/mutablelogic/go-client/pkg/openai/schema"
 )
 
 ///////////////////////////////////////////////////////////////////////////////
@@ -19,12 +19,23 @@ type reqCreateEmbedding struct {
 	EncodingFormat string   `json:"encoding_format,omitempty"`
 }
 
+///////////////////////////////////////////////////////////////////////////////
+// GLOBALS
+
+const (
+	defaultEmbeddingModel = "mistral-embed"
+)
+
 ///////////////////////////////////////////////////////////////////////////////
 // API CALLS
 
 // CreateEmbedding creates an embedding from a string or array of strings
-func (c *Client) CreateEmbedding(content any) (Embeddings, error) {
+func (c *Client) CreateEmbedding(content any) (schema.Embeddings, error) {
 	var request reqCreateEmbedding
+	var response schema.Embeddings
+
+	// Set default model
+	request.Model = defaultEmbeddingModel
 
 	// Set the input, which is either a string or array of strings
 	switch v := content.(type) {
@@ -33,15 +44,14 @@ func (c *Client) CreateEmbedding(content any) (Embeddings, error) {
 	case []string:
 		request.Input = v
 	default:
-		return Embeddings{}, ErrBadParameter
+		return response, ErrBadParameter
 	}
 
-	// Return the response
-	var response Embeddings
+	// Request->Response
 	if payload, err := client.NewJSONRequest(request, client.ContentTypeJson); err != nil {
-		return Embeddings{}, err
+		return response, err
 	} else if err := c.Do(payload.Post(), &response, client.OptPath("embeddings")); err != nil {
-		return Embeddings{}, err
+		return response, err
 	}
 
 	// Return success

pkg/mistral/embedding_test.go

@@ -0,0 +1,23 @@
+package mistral_test
+
+import (
+	"encoding/json"
+	"os"
+	"testing"
+
+	// Packages
+	opts "github.com/mutablelogic/go-client/pkg/client"
+	mistral "github.com/mutablelogic/go-client/pkg/mistral"
+	assert "github.com/stretchr/testify/assert"
+)
+
+func Test_embedding_001(t *testing.T) {
+	assert := assert.New(t)
+	client, err := mistral.New(GetApiKey(t), opts.OptTrace(os.Stderr, true))
+	assert.NoError(err)
+	assert.NotNil(client)
+	response, err := client.CreateEmbedding("test")
+	assert.NoError(err)
+	data, _ := json.MarshalIndent(response, "", "  ")
+	t.Log(string(data))
+}