internal/ai¶
github.com/gamidoc/backend/internal/ai
Types¶
Assistant¶
type Assistant interface {
NoopAssistant¶
type NoopAssistant struct{}
OpenAIAssistant¶
type OpenAIAssistant struct {
Section¶
type Section struct {
SectionText¶
type SectionText struct {
Functions¶
NewNoopAssistant¶
func NewNoopAssistant() *NoopAssistant {
NewOpenAIAssistant¶
func NewOpenAIAssistant(baseURL string, apiKey string, model string, client *http.Client) *OpenAIAssistant {
NoopAssistant.Chat¶
func (a *NoopAssistant) Chat(ctx context.Context, section Section, message string) (string, error) {
NoopAssistant.Enhance¶
func (a *NoopAssistant) Enhance(ctx context.Context, sections []SectionText) ([]SectionText, error) {
NoopAssistant.Prefill¶
func (a *NoopAssistant) Prefill(ctx context.Context, spark string, sections []Section) (map[int]json.RawMessage, error) {
NoopAssistant.RecommendBranch¶
func (a *NoopAssistant) RecommendBranch(ctx context.Context, spark string) (string, error) {
NoopAssistant.Rewrite¶
func (a *NoopAssistant) Rewrite(ctx context.Context, text string) (string, error) {
OpenAIAssistant.Chat¶
func (a *OpenAIAssistant) Chat(ctx context.Context, section Section, message string) (string, error) {
OpenAIAssistant.Enhance¶
func (a *OpenAIAssistant) Enhance(ctx context.Context, sections []SectionText) ([]SectionText, error) {
OpenAIAssistant.Prefill¶
func (a *OpenAIAssistant) Prefill(ctx context.Context, spark string, sections []Section) (map[int]json.RawMessage, error) {
OpenAIAssistant.RecommendBranch¶
func (a *OpenAIAssistant) RecommendBranch(ctx context.Context, spark string) (string, error) {
OpenAIAssistant.Rewrite¶
func (a *OpenAIAssistant) Rewrite(ctx context.Context, text string) (string, error) {
Variables¶
ErrEmptyText¶
var ErrEmptyText = errors.New("empty text")