autofix

Documentation

Index

• Constants
• func GenerateSolution(aiApiProvider, aiApiKey, endpoint string, issues []*issue.Issue) error
• type GenAIClient
  • func NewGenAIClient(ctx context.Context, aiApiKey, endpoint string) (GenAIClient, error)
• type GenAIGenerativeModel

Constants

const (
	GeminiModel = "gemini-1.5-flash"
	AIPrompt    = `` /* 175-byte string literal not displayed */

	GeminiProvider = "gemini"
)

Functions

func GenerateSolution

func GenerateSolution(aiApiProvider, aiApiKey, endpoint string, issues []*issue.Issue) error

GenerateSolution generates a solution for the given issues using the specified AI provider

Types

type GenAIClient

type GenAIClient interface {
	// Close clean up and close the client.
	Close() error
	// GenerativeModel build the generative mode.
	GenerativeModel(name string) GenAIGenerativeModel
}

GenAIClient defines the interface for the GenAI client.

func NewGenAIClient

func NewGenAIClient(ctx context.Context, aiApiKey, endpoint string) (GenAIClient, error)

NewGenAIClient creates a new gemini API client.

type GenAIGenerativeModel

type GenAIGenerativeModel interface {
	// GenerateContent generates an response for given prompt.
	GenerateContent(ctx context.Context, prompt string) (string, error)
}

GenAIGenerativeModel defines the interface for the Generative Model.