diff options
Diffstat (limited to 'models/models.go')
| -rw-r--r-- | models/models.go | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/models/models.go b/models/models.go index 2c5c99c..bbf1bd0 100644 --- a/models/models.go +++ b/models/models.go @@ -106,3 +106,22 @@ type ChatToolsBody struct { } `json:"tools"` ToolChoice string `json:"tool_choice"` } + +type EmbeddingResp struct { + Embedding []float32 `json:"embedding"` + Index uint32 `json:"index"` +} + +// type EmbeddingsResp struct { +// Model string `json:"model"` +// Object string `json:"object"` +// Usage struct { +// PromptTokens int `json:"prompt_tokens"` +// TotalTokens int `json:"total_tokens"` +// } `json:"usage"` +// Data []struct { +// Embedding []float32 `json:"embedding"` +// Index int `json:"index"` +// Object string `json:"object"` +// } `json:"data"` +// } |