diff options
Diffstat (limited to 'models')
| -rw-r--r-- | models/db.go | 10 | ||||
| -rw-r--r-- | models/models.go | 19 |
2 files changed, 29 insertions, 0 deletions
diff --git a/models/db.go b/models/db.go index 1f1c8b4..db4ae75 100644 --- a/models/db.go +++ b/models/db.go @@ -35,3 +35,13 @@ type Memory struct { CreatedAt time.Time `db:"created_at" json:"created_at"` UpdatedAt time.Time `db:"updated_at" json:"updated_at"` } + +// vector models + +type VectorRow struct { + ID uint32 `db:"id" json:"id"` + Embeddings []float32 `db:"embeddings" json:"embeddings"` + Slug string `db:"slug" json:"slug"` + RawText string `db:"raw_text" json:"raw_text"` + Distance float32 `db:"distance" json:"distance"` +} 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"` +// } |