diff --git a/api/client.go b/api/client.go index d6d5c300b..c59fbc423 100644 --- a/api/client.go +++ b/api/client.go @@ -356,7 +356,7 @@ func (c *Client) Embed(ctx context.Context, req *EmbedRequest) (*EmbedResponse, return &resp, nil } -// Embeddings generates embeddings from a model. (Legacy) +// Embeddings generates an embedding from a model. 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 { diff --git a/server/routes.go b/server/routes.go index 89284163a..0e986d25b 100644 --- a/server/routes.go +++ b/server/routes.go @@ -384,7 +384,7 @@ func (s *Server) EmbeddingsHandler(c *gin.Context) { return } - embedding, err := r.Embed(c.Request.Context(), []string{req.Prompt}) + embeddings, err := r.Embed(c.Request.Context(), []string{req.Prompt}) if err != nil { slog.Info(fmt.Sprintf("embedding generation failed: %v", err)) @@ -392,14 +392,14 @@ func (s *Server) EmbeddingsHandler(c *gin.Context) { return } - embedding64 := make([]float64, len(embedding[0])) + embedding := make([]float64, len(embeddings[0])) - for i, v := range embedding[0] { - embedding64[i] = float64(v) + for i, v := range embeddings[0] { + embedding[i] = float64(v) } resp := api.EmbeddingResponse{ - Embedding: embedding64, + Embedding: embedding, } c.JSON(http.StatusOK, resp) }