summaryrefslogtreecommitdiff
path: root/models/embed.go
diff options
context:
space:
mode:
Diffstat (limited to 'models/embed.go')
-rw-r--r--models/embed.go15
1 files changed, 15 insertions, 0 deletions
diff --git a/models/embed.go b/models/embed.go
new file mode 100644
index 0000000..078312c
--- /dev/null
+++ b/models/embed.go
@@ -0,0 +1,15 @@
+package models
+
+type LCPEmbedResp 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"`
+}