vertexai

Documentation

Index

• func CreatePalmClient(projectId string) (*vertexai.LLM, error)
• func GenerateText(palmClient *vertexai.LLM, prompt string) (string, error)
• type Completion
• type Content
• type PalmRequest
• type PalmResponse
• type Parameters

Functions

func CreatePalmClient

func CreatePalmClient(projectId string) (*vertexai.LLM, error)

func GenerateText

func GenerateText(palmClient *vertexai.LLM, prompt string) (string, error)

Types

type Completion

type Completion struct {
	Text string `json:"text"`
}

type Content

type Content struct {
	Content string `json:"content"`
}

type PalmRequest

type PalmRequest struct {
	Instances  []Content   `json:"instances"`
	Parameters *Parameters `json:"parameters"`
}

type PalmResponse

type PalmResponse struct {
	Predictions []struct {
		SafetyAttributes struct {
			Categories []string  `json:"categories"`
			Blocked    bool      `json:"blocked"`
			Scores     []float64 `json:"scores"`
		} `json:"safetyAttributes"`
		CitationMetadata struct {
			Citations []any `json:"citations"`
		} `json:"citationMetadata"`
		Content string `json:"content"`
	} `json:"predictions"`
}

type Parameters

type Parameters struct {
	Temperature     float64 `json:"temperature"`
	MaxOutputTokens int     `json:"maxOutputTokens"`
	TopP            float64 `json:"topP"`
	TopK            int     `json:"topK"`
}