googai

Documentation

Index

• Constants
• type Candidate
• type Client
  • func NewClient(key string) *Client
  • func (c *Client) CreateChatCompletion(contents []Content) (*ResponseBody, error)
  • func (c *Client) CreateImageCompletion(contents []Content) (*ResponseBody, error)
• type Content
• type Error
• type InlineData
• type Part
• type RequestBody
• type ResponseBody
• type SafetyRating
• type SafetySetting

Constants

const ApiBaseUrl = "https://generativelanguage.googleapis.com"

const ApiVersion = "v1beta"

const ChatMessageRoleAssistant = "model"

const ChatMessageRoleSystem = "user"

const ChatMessageRoleUser = "user"

Types

type Candidate

type Candidate struct {
	Content       Content        `json:"content"`
	FinishReason  string         `json:"finishReason"`
	Index         int            `json:"index"`
	SafetyRatings []SafetyRating `json:"safetyRatings"`
}

type Client

type Client struct {
	ApiBaseUrl     string
	ApiVersion     string
	ApiKey         string
	Model          string
	SafetySettings []SafetySetting
}

func NewClient

func NewClient(key string) *Client

func (*Client) CreateChatCompletion

func (c *Client) CreateChatCompletion(contents []Content) (*ResponseBody, error)

func (*Client) CreateImageCompletion

func (c *Client) CreateImageCompletion(contents []Content) (*ResponseBody, error)

type Content

type Content struct {
	Parts []Part `json:"parts"`
	Role  string `json:"role,omitempty"`
}

type Error

type Error struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
	Status  string `json:"status"`
}

type InlineData

type InlineData struct {
	MimeType string `json:"mime_type"`
	Data     string `json:"data"`
}

type Part

type Part struct {
	Text       string     `json:"text,omitempty"`
	InlineData InlineData `json:"inline_data,omitempty"`
}

type RequestBody

type RequestBody struct {
	Contents       []Content       `json:"contents"`
	SafetySettings []SafetySetting `json:"safetySettings"`
}

type ResponseBody

type ResponseBody struct {
	Candidates     []Candidate `json:"candidates"`
	Error          Error       `json:"error"`
	PromptFeedback struct {
		BlockReason   string         `json:"blockReason"`
		SafetyRatings []SafetyRating `json:"safetyRatings"`
	} `json:"promptFeedback"`
}

type SafetyRating

type SafetyRating struct {
	Category    string `json:"category"`
	Probability string `json:"probability"`
}

type SafetySetting

type SafetySetting struct {
	Category  string `json:"category"`
	Threshold string `json:"threshold"`
}