Feat: add rag [wip; skip-ci]

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