feat: support baidu's embedding model (close #324)

This commit is contained in:
JustSong
2023-07-29 12:15:07 +08:00
parent d1335ebc01
commit 130e6bfd83
5 changed files with 129 additions and 4 deletions

View File

@@ -99,6 +99,19 @@ type OpenAITextResponse struct {
Usage `json:"usage"`
}
type OpenAIEmbeddingResponseItem struct {
Object string `json:"object"`
Index int `json:"index"`
Embedding []float64 `json:"embedding"`
}
type OpenAIEmbeddingResponse struct {
Object string `json:"object"`
Data []OpenAIEmbeddingResponseItem `json:"data"`
Model string `json:"model"`
Usage `json:"usage"`
}
type ImageResponse struct {
Created int `json:"created"`
Data []struct {