query method

  1. @override
Future<VectorSpaceResult> query(
  1. String query, {
  2. int maxResults = 10,
})
override

Implementation

@override
Future<VectorSpaceResult> query(String query, {int maxResults = 10}) => client
    .searchText(
      namespace: namespace,
      search: SearchText.basic(
        rerank: rerank,
        query: query,
        textKey: "text",
        topK: maxResults,
        rerankModel: rerankModel,
        rerankTopK: rerankTopK,
      ),
    )
    .then(
      (v) => Future.wait(
        v.hits.map(
          (i) => _otk
              .estimateTokens(i.fields["text"] ?? "")
              .then(
                (tk) => VectorResult(
                  id: i.id,
                  score: i.score,
                  content: i.fields["text"] ?? "",
                  contentTokenCount: tk,
                ),
              ),
        ),
      ).then((r) => VectorSpaceResult(results: r)),
    );