genai

Documentation

Index

• Constants
• Variables
• type AIGenerationError
  • func (e *AIGenerationError) Error() string
• type ClientOption
  • func WithApiKey(key string) ClientOption
  • func WithModelName(name string) ClientOption
• type ClientOptions
• type GenAiClient
  • func NewGenAiClient(ctx context.Context, options ...ClientOption) (*GenAiClient, error)
  • func (c *GenAiClient) Send(ctx context.Context, prompt string, mod *nexus.SchemaMod, ...) (string, error)

Constants

const DefaultModelName = "gemini-1.5-flash"

Variables

var ErrNoApiKey = errors.New("no generative ai api key provided")

Types

type AIGenerationError

type AIGenerationError struct {
	Reason genai.FinishReason
}

func (*AIGenerationError) Error

func (e *AIGenerationError) Error() string

type ClientOption

type ClientOption func(*ClientOptions)

func WithApiKey

func WithApiKey(key string) ClientOption

func WithModelName

func WithModelName(name string) ClientOption

type ClientOptions

type ClientOptions struct {
	ApiKey    string
	ModelName string
}

type GenAiClient

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

func NewGenAiClient

func NewGenAiClient(ctx context.Context, options ...ClientOption) (*GenAiClient, error)

func (*GenAiClient) Send

func (c *GenAiClient) Send(
	ctx context.Context,
	prompt string,
	mod *nexus.SchemaMod,
	hook markxus.LlmStreamConsumeHook,
) (string, error)