googleai

Documentation

Overview

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: MIT

Index

• type GeminiClient
  • func NewGeminiClient(apiKey, model string, logger *zap.Logger, maxAttempts int, ...) *GeminiClient
  • func (c *GeminiClient) GetModelName() string
  • func (c *GeminiClient) SendPrompt(ctx context.Context, prompt string, history []models.Message) (string, error)

Types

type GeminiClient

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

GeminiClient implementa o cliente para interagir com a API do Google Gemini

func NewGeminiClient

func NewGeminiClient(apiKey, model string, logger *zap.Logger, maxAttempts int, backoff time.Duration) *GeminiClient

NewGeminiClient cria uma nova instância de GeminiClient

func (*GeminiClient) GetModelName

func (c *GeminiClient) GetModelName() string

GetModelName retorna o nome amigável do modelo Gemini

func (*GeminiClient) SendPrompt

func (c *GeminiClient) SendPrompt(ctx context.Context, prompt string, history []models.Message) (string, error)

SendPrompt envia um prompt para o Gemini e retorna a resposta