aiengine

Documentation

Index

• Variables
• type AiEngine
  • func NewAiEngine(apiBaseURL, apiKey string, log lib.ILogger) *AiEngine
  • func (a *AiEngine) Prompt(ctx context.Context, request *api.ChatCompletionRequest) (*api.ChatCompletionResponse, error)
  • func (a *AiEngine) PromptCb(ctx *gin.Context, body *openai.ChatCompletionRequest)
  • func (a *AiEngine) PromptStream(ctx context.Context, request *api.ChatCompletionRequest, ...) (*api.ChatCompletionStreamResponse, error)
• type CompletionCallback
• type ResponderFlusher

Variables

var (
	ErrChatCompletion = errors.New("chat completion error")
)

Types

type AiEngine

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

func NewAiEngine

func NewAiEngine(apiBaseURL, apiKey string, log lib.ILogger) *AiEngine

func (*AiEngine) Prompt

func (a *AiEngine) Prompt(ctx context.Context, request *api.ChatCompletionRequest) (*api.ChatCompletionResponse, error)

func (*AiEngine) PromptCb

func (a *AiEngine) PromptCb(ctx *gin.Context, body *openai.ChatCompletionRequest)

func (*AiEngine) PromptStream

func (a *AiEngine) PromptStream(ctx context.Context, request *api.ChatCompletionRequest, chunkCallback CompletionCallback) (*api.ChatCompletionStreamResponse, error)

type CompletionCallback

type CompletionCallback func(completion *api.ChatCompletionStreamResponse) error

type ResponderFlusher

type ResponderFlusher interface {
	http.ResponseWriter
	http.Flusher
}