augmentPrompt method

Given an input prompt, it returns augmented prompt from vertex rag store to guide LLM towards generating grounded responses.

Throws a http.ClientException if there were problems communicating with the API service. Throws a StatusException if the API failed with a Status message. Throws a ServiceException for any other failure.

Implementation

Future<AugmentPromptResponse> augmentPrompt(
  AugmentPromptRequest request,
) async {
  final url = Uri.https(_host, '/v1beta1/${request.parent}:augmentPrompt');
  final response = await _client.post(url, body: request);
  return AugmentPromptResponse.fromJson(response);
}