ai

package

v0.1.2 Latest Latest Go to latest Published: Jan 6, 2025 License: MIT Imports: 8 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type Ai interface {
	GetResponse(prompt *Prompt) (string, error)
}

type Image struct {
	Data      []byte
	Extension string
}

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

func NewOpenAIClient(apiKey string, model string) *OpenAI

func (ai *OpenAI) GetResponse(prompt *Prompt) (string, error)

type Prompt struct {
	UserPrompt   string
	SystemPrompt string
	Images       []Image
}

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

func NewVertexAIClient(projectID, localtion, modelName string) (*VertexAI, error)

func (ai *VertexAI) GetResponse(prompt *Prompt) (string, error)