From 0d6e3565aea328b83fa63b83d4be72da4d451109 Mon Sep 17 00:00:00 2001
From: Brian Murray <brian@bmurray.ca>
Date: Thu, 4 Jan 2024 13:00:52 -0700
Subject: [PATCH] Add embeddings to API (#1773)

---
 api/client.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/api/client.go b/api/client.go
index 72da30b8..4fe8668a 100644
--- a/api/client.go
+++ b/api/client.go
@@ -309,6 +309,13 @@ func (c *Client) Heartbeat(ctx context.Context) error {
 	}
 	return nil
 }
+func (c *Client) Embeddings(ctx context.Context, req *EmbeddingRequest) (*EmbeddingResponse, error) {
+	var resp EmbeddingResponse
+	if err := c.do(ctx, http.MethodPost, "/api/embeddings", req, &resp); err != nil {
+		return nil, err
+	}
+	return &resp, nil
+}
 
 func (c *Client) CreateBlob(ctx context.Context, digest string, r io.Reader) error {
 	if err := c.do(ctx, http.MethodHead, fmt.Sprintf("/api/blobs/%s", digest), nil, nil); err != nil {