Method Retrieve

internal/retrieval/opensearch_kb.go:45–99

(ctx context.Context, query string, opts map[string]any)

func (o *OpenSearchKB) Name() string { return o.name }

func (o *OpenSearchKB) Retrieve(ctx context.Context, query string, opts map[string]any) (types.RetrievalResult, error) {
	topK := effectiveTopK(o.cfg, opts, 8)
	q := map[string]any{
		"size": topK,
		"query": map[string]any{
			"multi_match": map[string]any{
				"query":  query,
				"fields": o.fields,
				"type":   "best_fields",
			},
		},
		"_source": true,
	}
	raw, err := json.Marshal(q)
	if err != nil {
		return types.RetrievalResult{}, err
	}
	base := strings.TrimRight(strings.TrimSpace(o.cfg.Endpoint), "/")
	index := strings.Trim(strings.TrimSpace(o.cfg.Index), "/")
	url := base + "/" + index + "/_search"
	req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(raw))
	if err != nil {
		return types.RetrievalResult{}, err
	}
	req.Header.Set("Content-Type", "application/json")
	for k, v := range o.cfg.Headers {
		if strings.TrimSpace(k) != "" && strings.TrimSpace(v) != "" {
			req.Header.Set(k, v)
		}
	}
	if req.Header.Get("Authorization") == "" {
		if k := strings.TrimSpace(o.cfg.APIKey); k != "" {
			switch {
			case strings.HasPrefix(k, "Basic "), strings.HasPrefix(k, "Bearer "), strings.HasPrefix(k, "ApiKey "):
				req.Header.Set("Authorization", k)
			default:
				req.Header.Set("Authorization", "ApiKey "+k)
			}
		}
	}

	resp, err := o.client.Do(req)
	if err != nil {
		return types.RetrievalResult{}, err
	}
	defer resp.Body.Close()
	b, err := io.ReadAll(resp.Body)
	if err != nil {
		return types.RetrievalResult{}, err
	}
	if resp.StatusCode < 200 || resp.StatusCode >= 300 {
		return types.RetrievalResult{}, fmt.Errorf("opensearch %s: %s: %s", o.name, resp.Status, truncateForErr(b, 512))
	}
	return parseOpenSearchHits(b)
}

func parseOpenSearchHits(b []byte) (types.RetrievalResult, error) {
	var root struct {