base

Documentation

Index

• Variables
• type ChatCompletionProvider
• type Config
• type Info
• type Provider
• type StreamChatCompletionsProvider

Variables

var ProviderRegistry = map[string]func(provider.Provider) Provider{}

Types

type ChatCompletionProvider

type ChatCompletionProvider interface {
	ChatCompletions(context.Context, *chat.ChatCompletionRequest) (*chat.ChatCompletionResponse, error)
}

type Config

type Config struct {
	BaseURL         string
	ChatCompletions string
}

type Info

type Info struct {
	Title       string
	Name        string
	Description string
}

type Provider

type Provider interface {
	Info() Info
	Config() Config
	Models() []string
	Schema() []byte
}

type StreamChatCompletionsProvider

type StreamChatCompletionsProvider interface {
	StreamChatCompletions(context.Context, *chat.ChatCompletionRequest, chan []byte) error
}