From 461d19aa2512fea7ac07e50c3178609850ef07c3 Mon Sep 17 00:00:00 2001 From: Grail Finder Date: Tue, 31 Dec 2024 13:25:13 +0300 Subject: Feat: add rag [wip; skip-ci] --- models/models.go | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'models/models.go') 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"` +// } -- cgit v1.2.3