table

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

type AutofillRequest struct {
	Enable         bool     `json:"enable"`
	Offset         int      `json:"offset"`
	Columns        []string `json:"columns"`
	ContextColumns []string `json:"context_columns"`
}

type ColumnSchema struct {
	Id       string `json:"id"`
	Name     string `json:"name"`
	Type     string `json:"type"`
	LinkedTo string `json:"linked_to"`
	Mode     string `json:"mode"`
}

type GenerateRowsRequest struct {
	Table       string  `json:"table"`
	SaveTo      string  `json:"save_to"`
	Count       int     `json:"count"`
	Batch       int     `json:"batch"`
	Temperature float64 `json:"temperature"`
	Model       string  `json:"model"`
	// used `json:""` in API only to send streaming results
	Stream bool `json:"stream"`

	Autofill AutofillRequest `json:"autofill"`
}

type ListTablesResponse struct {
	Total  int         `json:"total"`
	Tables []TableInfo `json:"tables"`
}

type Rows struct {
	Columns []*ent.TableColumn `json:"columns"`
	Rows    []*ent.TableRow    `json:"rows"`
}

type RowsGenerator interface {
	Next(ctx context.Context) ([]map[string]*schema.CellValue, error)
	Table() *ent.TableMeta
}

type RowsGeneratorMock struct {
	// NextFunc mocks the Next method.
	NextFunc func(ctx context.Context) ([]map[string]*schema.CellValue, error)

	// TableFunc mocks the Table method.
	TableFunc func() *ent.TableMeta
	// contains filtered or unexported fields
}

type TableColumnInfo struct {
	ID          string `json:"id"`
	Name        string `json:"name"`
	Description string `json:"description"`
	Type        string `json:"type"`
	FillMode    string `json:"fill_mode"`
}

type TableGenColumn struct {
	Name          string `json:"name"`
	Description   string `json:"description"`
	Type          string `json:"type"`
	FillMode      string `json:"fill_mode"`
	Source        string `json:"source"`
	Repeat        int    `json:"repeat"`
	ContextLength int    `json:"context_length"`
}

type TableGenColumnSchema struct {
	Name        string `json:"name"`
	Description string `json:"description"`
	Type        string `json:"type"`
}

type TableGenRequest struct {
	Name        string            `json:"name"`
	Model       string            `json:"model"`
	Description string            `json:"description"`
	Columns     []TableGenColumn  `json:"columns"`
	Sources     []json.RawMessage `json:"sources"`
}

type TableInfo struct {
	ID          string            `json:"id"`
	Name        string            `json:"name"`
	Description string            `json:"description"`
	Model       string            `json:"model"`
	Columns     []TableColumnInfo `json:"columns"`
}

type TableService interface {
	CreateTable(ctx context.Context, req *TableGenRequest) (string, error)
	ListTables(ctx context.Context) (*ListTablesResponse, error)
	GetTableDetail(ctx context.Context, table string) (*TableInfo, error)
	Genetate(ctx context.Context, params GenerateRowsRequest) (RowsGenerator, error)
	Rows(ctx context.Context, table string) (*Rows, error)
	Truncate(ctx context.Context, table string) (int, error)
	Delete(ctx context.Context, table string) (int, error)
	Import(ctx context.Context, table string, reader io.Reader) (string, error)
}

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

type TableServiceMock struct {
	// CreateTableFunc mocks the CreateTable method.
	CreateTableFunc func(ctx context.Context, req *TableGenRequest) (string, error)

	// DeleteFunc mocks the Delete method.
	DeleteFunc func(ctx context.Context, table string) (int, error)

	// GenetateFunc mocks the Genetate method.
	GenetateFunc func(ctx context.Context, params GenerateRowsRequest) (RowsGenerator, error)

	// GetTableDetailFunc mocks the GetTableDetail method.
	GetTableDetailFunc func(ctx context.Context, table string) (*TableInfo, error)

	// ImportFunc mocks the Import method.
	ImportFunc func(ctx context.Context, table string, reader io.Reader) (string, error)

	// ListTablesFunc mocks the ListTables method.
	ListTablesFunc func(ctx context.Context) (*ListTablesResponse, error)

	// RowsFunc mocks the Rows method.
	RowsFunc func(ctx context.Context, table string) (*Rows, error)

	// TruncateFunc mocks the Truncate method.
	TruncateFunc func(ctx context.Context, table string) (int, error)
	// contains filtered or unexported fields
}