Feat: add rag [wip; skip-ci]

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"`
+}

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"`
+// }