clean up

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 {

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)
 }