pinecone

type Client interface {
	Upsert(ctx context.Context, req *UpsertRequest) (*UpsertResponse, error)
	Fetch(ctx context.Context, req *FetchRequest) (*FetchResponse, error)
	Query(ctx context.Context, req *QueryRequest) (*QueryResponse, error)
	Close() error
}

type Endpoint struct {
	IndexName   string
	ProjectName string
	Environment string
}

type ErrorResponse struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
}

type FetchRequest struct {
	IDs       []string `json:"ids"`
	Namespace string   `json:"namespace"`
}

type FetchResponse struct {
	Vectors   map[string]*Vector `json:"vectors"`
	Namespace string             `json:"namespace"`
}

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

type Match struct {
	ID       string         `json:"id"`
	Values   []float32      `json:"values"`
	Metadata map[string]any `json:"metadata"`
	Score    float64        `json:"score"`
}

type Options struct {
	UseGRPC bool
}

type QueryRequest struct {
	Filter          map[string]any `json:"filter"`
	IncludeValues   bool           `json:"includeValues"`
	IncludeMetadata bool           `json:"includeMetadata"`
	Vector          []float32      `json:"vector"`
	Namespace       string         `json:"namespace"`
	TopK            int64          `json:"topK"`
	ID              string         `json:"id"`
}

type QueryResponse struct {
	Matches   []*Match `json:"matches"`
	Namespace string   `json:"namespace"`
}

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

type UpsertRequest struct {
	Vectors   []*Vector `json:"vectors"`
	Namespace string    `json:"namespace"`
}

type UpsertResponse struct {
	UpsertedCount uint32 `json:"upsertedCount"`
}

type Vector struct {
	ID       string         `json:"id"`
	Values   []float32      `json:"values"`
	Metadata map[string]any `json:"metadata"`
}