search

type ArticleWithBody struct {
	*citygraph.Article
	Body string
}

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

type EmbeddingsRequest struct {
	Input []string `json:"input"`
	Model string   `json:"model"`
}

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

type QueryParams struct {
	IncludeMetadata bool      `json:"includeMetadata"`
	Vector          []float32 `json:"vector"`
	TopK            int64     `json:"topK"`
}

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

type QueryVector struct {
	Vector
	Score float32 `json:"score"`
}

type SearchResult struct {
	Name    string
	ID      string
	Path    string
	Slug    string
	Score   float32
	PubDate string
}

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

type UpsertVectorsResponse struct {
	UpsertedCount int `json:"upsertedCount"`
}

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