qdrant

type CollectionConfig struct {
	Vectors VectorsConfig `json:"vectors"`

	// Optional tuning
	HNSW       *HNSWConfig       `json:"hnsw_config,omitempty"`
	Optimizers *OptimizersConfig `json:"optimizers_config,omitempty"`

	// Distributed / storage knobs
	ShardNumber       int   `json:"shard_number,omitempty"`
	ReplicationFactor int   `json:"replication_factor,omitempty"`
	OnDiskPayload     *bool `json:"on_disk_payload,omitempty"`

	// Escape hatch for unsupported / future fields
	Extras map[string]any `json:"-"`
}

type CollectionInfo struct {
	Status string `json:"status"` // can extend this struct if needed
}

type ConnectionConfig struct {
	BaseURL string // e.g. "http://localhost:6333"
	APIKey  string // optional

	Timeout         time.Duration // per-request hard timeout (http.Client)
	DialTimeout     time.Duration // TCP dial timeout
	IdleConnTimeout time.Duration // idle conn timeout

	// Retry
	RetryMaxAttempts int           // total attempts incl. first (default 3)
	RetryBaseDelay   time.Duration // default 150ms
	RetryMaxDelay    time.Duration // default 2s
	RetryJitter      float64       // 0..1 (default 0.2)
}

type CreateCollectionRequest struct {
	Collection string
	Config     CollectionConfig
}

type Defaults struct {
	Distance Distance
	HNSW     HNSWConfig
	Optim    OptimizersConfig
}

type DeletePointsRequest struct {
	Collection string
	IDs        []PointID
	Wait       bool
}

type Distance string

type HNSWConfig struct {
	M           int `json:"m,omitempty"`
	EfConstruct int `json:"ef_construct,omitempty"`
}

type HTTPError struct {
	StatusCode int
	Body       string
}

type OptimizersConfig struct {
	DefaultSegmentNumber int `json:"default_segment_number,omitempty"`
}

type Point struct {
	ID      PointID        `json:"id"`
	Vector  []float32      `json:"vector,omitempty"`  // single-vector baseline
	Payload map[string]any `json:"payload,omitempty"` // Qdrant payload
}

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

type QdrantClientInterface interface {
	Health(ctx context.Context) error

	UpsertPoints(ctx context.Context, req UpsertPointsRequest) error
	DeletePoints(ctx context.Context, req DeletePointsRequest) error
	Search(ctx context.Context, req SearchRequest) ([]ScoredPoint, error)

	// Collection management (minimal baseline)
	CreateCollection(ctx context.Context, req CreateCollectionRequest) error
	DeleteCollection(ctx context.Context, collection string) error
	GetCollectionInfo(ctx context.Context, collection string) (CollectionInfo, error)
}

type ScoredPoint struct {
	ID      PointID        `json:"id"`
	Score   float32        `json:"score"`
	Payload map[string]any `json:"payload,omitempty"`
	Vector  []float32      `json:"vector,omitempty"`
}

type SearchRequest struct {
	Collection     string
	Vector         []float32
	Limit          int
	Offset         int
	WithPayload    bool
	WithVector     bool
	ScoreThreshold *float32

	// Filter is raw Qdrant filter JSON. Keep flexible initially.
	Filter json.RawMessage
}

type UpsertPointsRequest struct {
	Collection string
	Points     []Point
	Wait       bool
}

type VectorParams struct {
	Size     int      `json:"size"`
	Distance Distance `json:"distance"`
}

type VectorsConfig struct {
	Single *VectorParams           `json:"-"`
	Named  map[string]VectorParams `json:"-"`
}