openai

Documentation

Index

• func ConvertOpenAIRequestToGemini(openAIReq *EmbedRequest, model *genai.EmbeddingModel) (*genai.EmbeddingBatch, error)
• type EmbedRequest
• type EmbedResponse
  • func ConvertGeminiResponseToOpenAI(geminiBatchResp *genai.BatchEmbedContentsResponse, model string) *EmbedResponse
• type EmbedResponseData
• type ModelResponse
• type ModelResponseData
• type Usage

Functions

func ConvertOpenAIRequestToGemini

func ConvertOpenAIRequestToGemini(openAIReq *EmbedRequest, model *genai.EmbeddingModel) (*genai.EmbeddingBatch, error)

Types

type EmbedRequest

type EmbedRequest struct {
	Input          interface{} `json:"input"`
	Model          string      `json:"model"`
	EncodingFormat string      `json:"encoding_format,omitempty"`
	Dimensions     int         `json:"dimensions,omitempty"`
	User           string      `json:"user,omitempty"`
}

type EmbedResponse

type EmbedResponse struct {
	Object string               `json:"object"`
	Data   []*EmbedResponseData `json:"data"`
	Model  string               `json:"model"`
	Usage  *Usage               `json:"usage"`
}

func ConvertGeminiResponseToOpenAI

func ConvertGeminiResponseToOpenAI(geminiBatchResp *genai.BatchEmbedContentsResponse, model string) *EmbedResponse

type EmbedResponseData

type EmbedResponseData struct {
	Object    string    `json:"object"`
	Embedding []float32 `json:"embedding"`
	Index     int       `json:"index"`
}

type ModelResponse

type ModelResponse struct {
	Object string               `json:"object"`
	Data   []*ModelResponseData `json:"data"`
}

type ModelResponseData

type ModelResponseData struct {
	Object  string `json:"object"`
	ID      string `json:"id"`
	Created uint   `json:"created"`
	OwnedBy string `json:"owned_by"`
}

type Usage

type Usage struct {
	PromptTokens int `json:"prompt_tokens"`
	TotalTokens  int `json:"total_tokens"`
}