store

type ChunkingStrategyType string

type Collection struct {
	gorm.Model

	TenantID       string
	OrganizationID string
	ProjectID      string `gorm:"uniqueIndex:idx_collection_project_id_name"`

	// VectorStoreID is the ID of the vector store that is externally visible in the API.
	// This is also used as the name of the Milvus collection.
	VectorStoreID string `gorm:"uniqueIndex"`

	// CollectionID is the ID of the Milvus collection.
	CollectionID int64 `gorm:"uniqueIndex"`

	Name string `gorm:"uniqueIndex:idx_collection_project_id_name"`

	// UsageBytes is the total number of bytes used by the files in the vector store.
	UsageBytes int64

	FileCountsInProgress int64
	FileCountsCompleted  int64
	FileCountsFailed     int64
	FileCountsCancelled  int64
	FileCountsTotal      int64

	// TODO(guangrui): Update status.
	Status CollectionStatus

	Anchor ExpiresAfterAnchor
	// ExpiresAfterDays is the number of days the anchor time for when the vector store will expire.
	ExpiresAfterDays int64
	// ExpiresAt is the Unix timestamp (in seconds) for when the vector store will expire.
	ExpiresAt int64

	// LastActiveAt is the Unix timestamp (in seconds) for when the vector store was last active.
	LastActiveAt int64

	EmbeddingModel      string
	EmbeddingDimensions int

	Version int
}

type CollectionMetadata struct {
	gorm.Model

	// VectorStoreID is the ID of the vector store.
	VectorStoreID string `gorm:"uniqueIndex:idx_collectionmeta_vsid_key"`

	Key   string `gorm:"uniqueIndex:idx_collectionmeta_vsid_key"`
	Value string

	Version int
}

type CollectionStatus string

type ExpiresAfterAnchor string

type File struct {
	gorm.Model

	VectorStoreID string `gorm:"uniqueIndex:idx_file_vector_store_id_file_id"`

	// FileID is the file ID.
	FileID string `gorm:"uniqueIndex:idx_file_vector_store_id_file_id"`

	// UsageBytes is the total vector store usage in bytes. Note that this may be different from the original file size.
	UsageBytes int64

	// TODO(guangrui): handle status update.
	Status FileStatus

	LastErrorCode    LastErrorCode
	LastErrorMessage string

	ChunkingStrategyType ChunkingStrategyType
	MaxChunkSizeTokens   int64
	ChunkOverlapTokens   int64

	Version int
}

type FileStatus string

type LastErrorCode string

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