Revert "Initial Batch Embedding"

This reverts commit c22d54895a.
This commit is contained in:
Roy Han
2024-06-24 15:26:05 -07:00
parent c22d54895a
commit 0f87628b6d
4 changed files with 10 additions and 93 deletions

View File

@@ -210,8 +210,7 @@ type EmbeddingRequest struct {
Model string `json:"model"`
// Prompt is the textual prompt to embed.
// Prompt string `json:"prompt"`
Prompt interface{} `json:"prompt"`
Prompt string `json:"prompt"`
// KeepAlive controls how long the model will stay loaded in memory following
// this request.
@@ -223,7 +222,7 @@ type EmbeddingRequest struct {
// EmbeddingResponse is the response from [Client.Embeddings].
type EmbeddingResponse struct {
Embedding [][]float64 `json:"embedding"`
Embedding []float64 `json:"embedding"`
}
// CreateRequest is the request passed to [Client.Create].