google

Documentation

Index

• type GeminiClient
  • func NewGeminiClient(ctx context.Context, apiKey string) (*GeminiClient, error)
  • func (c *GeminiClient) Chat(ctx context.Context) (*genai.ChatSession, error)
  • func (c *GeminiClient) Close()
  • func (c *GeminiClient) GenerateImage(ctx context.Context, prompt string) ([]byte, error)
  • func (c *GeminiClient) GenerateText(ctx context.Context, prompt string) (string, error)
  • func (c *GeminiClient) SendMessage(ctx context.Context, chat *genai.ChatSession, message string) (string, error)

Types

type GeminiClient

type GeminiClient struct {
	// contains filtered or unexported fields
}

GeminiClient represents a client for interacting with Google Cloud Gemini API

func NewGeminiClient

func NewGeminiClient(ctx context.Context, apiKey string) (*GeminiClient, error)

NewGeminiClient creates a new Gemini API client

func (*GeminiClient) Chat

func (c *GeminiClient) Chat(ctx context.Context) (*genai.ChatSession, error)

Chat initiates a chat session with Gemini

func (*GeminiClient) Close

func (c *GeminiClient) Close()

Close closes the Gemini client

func (*GeminiClient) GenerateImage

func (c *GeminiClient) GenerateImage(ctx context.Context, prompt string) ([]byte, error)

GenerateImage generates an image based on text prompt using Gemini Pro Vision

func (*GeminiClient) GenerateText

func (c *GeminiClient) GenerateText(ctx context.Context, prompt string) (string, error)

GenerateText generates text response based on prompt

func (*GeminiClient) SendMessage

func (c *GeminiClient) SendMessage(ctx context.Context, chat *genai.ChatSession, message string) (string, error)

SendMessage sends a message in a chat session and returns the response