First AI generated version

embed.go

@@ -0,0 +1,24 @@
+package gollama
+
+import (
+	"context"
+
+	"github.com/ollama/ollama/api"
+)
+
+// Embed generates vector embeddings for the given texts.
+func (c *Client) Embed(ctx context.Context, model string, texts []string) ([][]float32, error) {
+	name, err := c.resolveModel(model)
+	if err != nil {
+		return nil, err
+	}
+	req := &api.EmbedRequest{
+		Model: name,
+		Input: texts,
+	}
+	resp, err := c.api.Embed(ctx, req)
+	if err != nil {
+		return nil, wrapError("embed", name, err)
+	}
+	return resp.Embeddings, nil
+}