ai

package

v0.5.7 Latest Latest Go to latest Published: Dec 10, 2024 License: Apache-2.0 Imports: 7 Imported by: 0

Details

This section is empty.

View Source

var (
	ErrMissingAuthToken = errors.New("auth token is required")
	ErrInvalidQuery     = errors.New("query is invalid")
)

func CheckAIManager(aiMgr *AIManager) error

CheckAIManager check if the AI manager is created

func ExtractSelectSQL(sql string) string

ExtractSelectSQL extracts SQL statements that start with "SELECT * FROM"

func IsInvalidQuery(sql string) bool

IsInvalidQuery check if the query is invalid

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

func NewAIManager(c registry.ExtraConfig) (*AIManager, error)

NewAIManager returns a new AIManager object

func (a *AIManager) ConvertTextToSQL(query string) (string, error)

ConvertTextToSQL converts natural language text to an SQL query

func (a *AIManager) FixSQL(sql string, query string, sqlErr string) (string, error)

FixSQL fix the error SQL