api

Documentation

Index

• Constants
• func GenerateImage(ctx context.Context, prompt string, size string) ([]byte, error)
• func GetApiModelList() ([]string, error)
• func GetApiTypeList() []string
• func GetGenerateFunction() (...)
• func GetLlmModel() (llm llms.Model, err error)
• func GetOllamaModelList() ([]string, error)
• func GetOpenAiImageModelList() ([]string, error)
• func GetOpenAiModelList() ([]string, error)
• func Mask(prompt string) (string, error)
• func SpeechToText(ctx context.Context, filename string, lang string) (string, error)
• func TextToSpeech(ctx context.Context, content string) (io.ReadCloser, error)
• type API_TYPE

Constants

const (
	API_OPENAI      = "OPENAI"
	API_HUGGINGFACE = "HUGGINGFACE"
	API_OLLAMA      = "OLLAMA"
)

Functions

func GenerateImage

func GenerateImage(ctx context.Context, prompt string, size string) ([]byte, error)

func GetApiModelList

func GetApiModelList() ([]string, error)

func GetApiTypeList

func GetApiTypeList() []string

func GetGenerateFunction

func GetGenerateFunction() (func(ctx context.Context, messages []llms.MessageContent, options ...llms.CallOption) (*llms.ContentResponse, error), error)

func GetLlmModel

func GetLlmModel() (llm llms.Model, err error)

func GetOllamaModelList

func GetOllamaModelList() ([]string, error)

func GetOpenAiImageModelList

func GetOpenAiImageModelList() ([]string, error)

func GetOpenAiModelList

func GetOpenAiModelList() ([]string, error)

func Mask

func Mask(prompt string) (string, error)

func SpeechToText

func SpeechToText(ctx context.Context, filename string, lang string) (string, error)

func TextToSpeech

func TextToSpeech(ctx context.Context, content string) (io.ReadCloser, error)

Types

type API_TYPE

type API_TYPE string