embeddings endpoint

Co-Authored-By: Jeffrey Morgan <jmorganca@gmail.com>
This commit is contained in:
Bruce MacDonald
2023-08-08 15:13:22 -04:00
parent 5ebce03c77
commit 4b3507f036
2 changed files with 85 additions and 31 deletions

View File

@@ -42,6 +42,17 @@ type GenerateRequest struct {
Options map[string]interface{} `json:"options"`
}
type EmbeddingRequest struct {
Model string `json:"model"`
Prompt string `json:"prompt"`
Options map[string]interface{} `json:"options"`
}
type EmbeddingResponse struct {
Embedding []float64 `json:"embedding"`
}
type CreateRequest struct {
Name string `json:"name"`
Path string `json:"path"`