vectorstores

type AttachOptions struct {
	Attributes map[string]any `json:"attributes,omitempty"`
	Chunking   map[string]any `json:"chunking_strategy,omitempty"`
}

type FilePage struct {
	Files []openai.VectorStoreFile
	Next  string
}

type IndexedContent struct {
	Text    string
	Partial bool
}

type Options struct {
	NewService   func(context.Context, *cli.Command) (Service, error)
	ErrorMessage func(error) string
}

type SearchPage struct {
	Results []openai.VectorStoreSearchResponse
	More    bool
}

type Service interface {
	ListStores(context.Context, string) (StorePage, error)
	GetStore(context.Context, string) (*openai.VectorStore, error)
	CreateStore(context.Context, string) (*openai.VectorStore, error)
	UpdateStore(context.Context, string, map[string]any) (*openai.VectorStore, error)
	DeleteStore(context.Context, string) error
	ListFiles(context.Context, string, string) (FilePage, error)
	ListSourceFiles(context.Context, string) (SourcePage, error)
	UploadFile(context.Context, string) (*openai.FileObject, error)
	AttachFile(context.Context, string, string, AttachOptions) (*openai.VectorStoreFile, error)
	GetFile(context.Context, string, string) (*openai.VectorStoreFile, error)
	UpdateFile(context.Context, string, string, map[string]any) (*openai.VectorStoreFile, error)
	RemoveFile(context.Context, string, string) error
	FileContent(context.Context, string, string) (IndexedContent, error)
	CreateBatch(context.Context, string, []string, AttachOptions) (*openai.VectorStoreFileBatch, error)
	GetBatch(context.Context, string, string) (*openai.VectorStoreFileBatch, error)
	CancelBatch(context.Context, string, string) (*openai.VectorStoreFileBatch, error)
	ListBatchFiles(context.Context, string, string, string) (FilePage, error)
	Search(context.Context, string, string) (SearchPage, error)
}

type SourcePage struct {
	Files []openai.FileObject
	Next  string
}

type StorePage struct {
	Stores []openai.VectorStore
	Next   string
}