pinecone

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

type Cloud string

type Collection struct {
	Name        string
	Size        *int64
	Status      CollectionStatus
	Dimension   *int32
	VectorCount *int32
	Environment string
}

type CollectionStatus string

type CreateCollectionRequest struct {
	Name   string
	Source string
}

type CreatePodIndexRequest struct {
	Name             string
	Dimension        int32
	Metric           IndexMetric
	Environment      string
	PodType          string
	Shards           int32
	Replicas         int32
	SourceCollection *string
	MetadataConfig   *PodSpecMetadataConfig
}

type CreateServerlessIndexRequest struct {
	Name      string
	Dimension int32
	Metric    IndexMetric
	Cloud     Cloud
	Region    string
}

type DescribeIndexStatsResponse struct {
	Dimension        uint32
	IndexFullness    float32
	TotalVectorCount uint32
	Namespaces       map[string]*NamespaceSummary
}

type FetchVectorsResponse struct {
	Vectors map[string]*Vector
	Usage   *Usage
}

type Filter = structpb.Struct

type Index struct {
	Name      string
	Dimension int32
	Host      string
	Metric    IndexMetric
	Spec      *IndexSpec
	Status    *IndexStatus
}

type IndexConnection struct {
	Namespace string
	// contains filtered or unexported fields
}

type IndexMetric string

type IndexSpec struct {
	Pod        *PodSpec
	Serverless *ServerlessSpec
}

type IndexStatus struct {
	Ready bool
	State IndexStatusState
}

type IndexStatusState string

type ListVectorsRequest struct {
	Prefix          *string
	Limit           *uint32
	PaginationToken *string
}

type ListVectorsResponse struct {
	VectorIds           []*string
	Usage               *Usage
	NextPaginationToken *string
}

type Metadata = structpb.Struct

type NamespaceSummary struct {
	VectorCount uint32
}

type NewClientParams struct {
	ApiKey    string
	SourceTag string // optional
}

type PodSpec struct {
	Environment      string
	PodType          string
	PodCount         int32
	Replicas         int32
	ShardCount       int32
	SourceCollection *string
	MetadataConfig   *PodSpecMetadataConfig
}

type PodSpecMetadataConfig struct {
	Indexed *[]string
}

type QueryByVectorIdRequest struct {
	VectorId        string
	TopK            uint32
	Filter          *Filter
	IncludeValues   bool
	IncludeMetadata bool
	SparseValues    *SparseValues
}

type QueryByVectorValuesRequest struct {
	Vector          []float32
	TopK            uint32
	Filter          *Filter
	IncludeValues   bool
	IncludeMetadata bool
	SparseValues    *SparseValues
}

type QueryVectorsResponse struct {
	Matches []*ScoredVector
	Usage   *Usage
}

type ScoredVector struct {
	Vector *Vector
	Score  float32
}

type ServerlessSpec struct {
	Cloud  Cloud
	Region string
}

type SparseValues struct {
	Indices []uint32
	Values  []float32
}

type UpdateVectorRequest struct {
	Id           string
	Values       []float32
	SparseValues *SparseValues
	Metadata     *Metadata
}

type Usage struct {
	ReadUnits *uint32
}

type Vector struct {
	Id           string
	Values       []float32
	SparseValues *SparseValues
	Metadata     *Metadata
}