llm

Documentation

Index

• Variables
• type ChatGPT
  • func NewChatGPT(apiKey string, opts ...ChatGPTOption) (*ChatGPT, error)
  • func (c *ChatGPT) CompleteChat(ctx context.Context, msgs []chat.Message) (chat.Message, error)
• type ChatGPTOption
  • func ChatGPTWithMaxTokens(maxTokens int) ChatGPTOption
  • func ChatGPTWithModel(model string) ChatGPTOption
  • func ChatGPTWithSocksProxy(socksProxy string) ChatGPTOption
  • func ChatGPTWithTemperature(temperature float32) ChatGPTOption
  • func ChatGPTWithTopP(topP float32) ChatGPTOption
• type GigaChat
  • func NewGigaChat(clientID, clientSecret string, opts ...GigaChatOption) (*GigaChat, error)
  • func (gc *GigaChat) CompleteChat(ctx context.Context, msgs []chat.Message) (chat.Message, error)
• type GigaChatOption
  • func GigaChatWithMaxTokens(maxTokens int64) GigaChatOption
  • func GigaChatWithModel(model string) GigaChatOption
  • func GigaChatWithRepetitionPenalty(penalty float64) GigaChatOption
  • func GigaChatWithTemperature(temperature float64) GigaChatOption
  • func GigaChatWithTopP(topP float64) GigaChatOption
• type Mock
  • func (m *Mock) CompleteChat(ctx context.Context, msgs []chat.Message) (chat.Message, error)

Variables

var (
	ErrChatGPTUnsupportedRole        = errors.New("unsupported role")
	ErrChatGPTUnsupportedContentType = errors.New("unsupported content type")
	ErrChatGPTRequestFailed          = errors.New("request failed")
)

var (
	ErrGigaChatUnsupportedRole        = errors.New("unsupported role")
	ErrGigaChatUnsupportedContentType = errors.New("unsupported content type")
	ErrGigaChatRequestFailed          = errors.New("request failed")
)

Types

type ChatGPT

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

ChatGPT реализует клиент для взаимодействия с ChatGPT.

func NewChatGPT

func NewChatGPT(apiKey string, opts ...ChatGPTOption) (*ChatGPT, error)

NewChatGPT создает новый экземпляр ChatGPT с заданными опциями.

func (*ChatGPT) CompleteChat

func (c *ChatGPT) CompleteChat(ctx context.Context, msgs []chat.Message) (chat.Message, error)

CompleteChat реализует интерфейс Completion.

type ChatGPTOption

type ChatGPTOption func(*ChatGPT)

ChatGPTOption определяет функцию-опцию для конфигурации клиента.

func ChatGPTWithMaxTokens

func ChatGPTWithMaxTokens(maxTokens int) ChatGPTOption

Установка максимального количества токенов.

func ChatGPTWithModel

func ChatGPTWithModel(model string) ChatGPTOption

Установка модели для использования.

func ChatGPTWithSocksProxy

func ChatGPTWithSocksProxy(socksProxy string) ChatGPTOption

func ChatGPTWithTemperature

func ChatGPTWithTemperature(temperature float32) ChatGPTOption

Установка температуры генерации.

func ChatGPTWithTopP

func ChatGPTWithTopP(topP float32) ChatGPTOption

Установка параметра top-p сэмплирования.

type GigaChat

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

Реализация интерфейса GigaChat.

func NewGigaChat

func NewGigaChat(
	clientID,
	clientSecret string,
	opts ...GigaChatOption,
) (*GigaChat, error)

Создание нового экземпляра Client с заданными опциями.

func (*GigaChat) CompleteChat

func (gc *GigaChat) CompleteChat(ctx context.Context, msgs []chat.Message) (chat.Message, error)

Реализация интерфейса Completion.

type GigaChatOption

type GigaChatOption func(*GigaChat)

Определение функции-опции для конфигурации клиента.

func GigaChatWithMaxTokens

func GigaChatWithMaxTokens(maxTokens int64) GigaChatOption

Установка максимального количества токенов.

func GigaChatWithModel

func GigaChatWithModel(model string) GigaChatOption

Установка модели для использования.

func GigaChatWithRepetitionPenalty

func GigaChatWithRepetitionPenalty(penalty float64) GigaChatOption

Установка штрафа за повторения.

func GigaChatWithTemperature

func GigaChatWithTemperature(temperature float64) GigaChatOption

Установка температуры генерации.

func GigaChatWithTopP

func GigaChatWithTopP(topP float64) GigaChatOption

Установка параметра top-p сэмплирования.

type Mock

type Mock struct {
	CompleteChatFn func(ctx context.Context, msgs []chat.Message) (chat.Message, error)
}

func (*Mock) CompleteChat

func (m *Mock) CompleteChat(ctx context.Context, msgs []chat.Message) (chat.Message, error)