arctl

package
v0.1.0 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Dec 29, 2023 License: Apache-2.0 Imports: 33 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func Chat

func Chat(ctx context.Context, similarDocs []schema.Document) error

func DatasetCreateCmd

func DatasetCreateCmd(homePath string) *cobra.Command

DatasetCreateCmd returns a new instance of the cobra.Command that is used to create a dataset.

func DatasetDeleteCmd

func DatasetDeleteCmd(homePath string) *cobra.Command

func DatasetExecuteCmd

func DatasetExecuteCmd(homePath string) *cobra.Command

func DatasetListCmd

func DatasetListCmd(homePath string) *cobra.Command

DatasetListCmd returns a Cobra command for listing datasets.

func DatasetShowCmd

func DatasetShowCmd(homePath string) *cobra.Command

func DatasourceCreateCmd

func DatasourceCreateCmd(kubeClient dynamic.Interface, namespace string) *cobra.Command

func DatasourceDeleteCmd

func DatasourceDeleteCmd(kubeClient dynamic.Interface, namespace string) *cobra.Command

func DatasourceGetCmd

func DatasourceGetCmd(kubeClient dynamic.Interface, namespace string) *cobra.Command

func DatasourceListCmd

func DatasourceListCmd(kubeClient dynamic.Interface, namespace string) *cobra.Command

func NewChatCmd

func NewChatCmd(homePath string) *cobra.Command

func NewDatasetCmd

func NewDatasetCmd(homePath string) *cobra.Command

func NewDatasourceCmd

func NewDatasourceCmd(kubeClient dynamic.Interface, namespace string) *cobra.Command

func SimilaritySearch

func SimilaritySearch(ctx context.Context, homePath string) ([]schema.Document, error)

Types

type Dataset

type Dataset struct {
	Name       string `json:"name"`
	CreateTime string `json:"create_time"`

	// Parameters for embedding service
	LLMType   string `json:"llm_type"`
	LLMApiKey string `json:"llm_api_key"`

	// Parameters for vectorization
	VectorStore      string `json:"vector_store"`
	DocumentLanguage string `json:"document_language"`
	TextSplitter     string `json:"text_splitter"`
	ChunkSize        int    `json:"chunk_size"`
	ChunkOverlap     int    `json:"chunk_overlap"`

	Files map[string]File `json:"files"`
}

type File

type File struct {
	// basic info
	Path string `json:"path"`
	Size int64  `json:"size"`

	// embedding status
	// Chunks is the number of split chunks
	Chunks int `json:"chunks"`
	// ChunksLoaded is the number of chunks loaded
	ChunksLoaded int `json:"chunks_loaded"`

	TimeCost float64 `json:"time_cost"`
}

Directories

Path Synopsis

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL