<?php
# Generated by the protocol buffer compiler.  DO NOT EDIT!
# source: google/cloud/aiplatform/v1/vertex_rag_service.proto

namespace Google\Cloud\AIPlatform\V1;

use Google\Protobuf\Internal\GPBType;
use Google\Protobuf\Internal\RepeatedField;
use Google\Protobuf\Internal\GPBUtil;

/**
 * Response message for AugmentPrompt.
 *
 * Generated from protobuf message <code>google.cloud.aiplatform.v1.AugmentPromptResponse</code>
 */
class AugmentPromptResponse extends \Google\Protobuf\Internal\Message
{
    /**
     * Augmented prompt, only text format is supported for now.
     *
     * Generated from protobuf field <code>repeated .google.cloud.aiplatform.v1.Content augmented_prompt = 1;</code>
     */
    private $augmented_prompt;
    /**
     * Retrieved facts from RAG data sources.
     *
     * Generated from protobuf field <code>repeated .google.cloud.aiplatform.v1.Fact facts = 2;</code>
     */
    private $facts;

    /**
     * Constructor.
     *
     * @param array $data {
     *     Optional. Data for populating the Message object.
     *
     *     @type array<\Google\Cloud\AIPlatform\V1\Content>|\Google\Protobuf\Internal\RepeatedField $augmented_prompt
     *           Augmented prompt, only text format is supported for now.
     *     @type array<\Google\Cloud\AIPlatform\V1\Fact>|\Google\Protobuf\Internal\RepeatedField $facts
     *           Retrieved facts from RAG data sources.
     * }
     */
    public function __construct($data = NULL) {
        \GPBMetadata\Google\Cloud\Aiplatform\V1\VertexRagService::initOnce();
        parent::__construct($data);
    }

    /**
     * Augmented prompt, only text format is supported for now.
     *
     * Generated from protobuf field <code>repeated .google.cloud.aiplatform.v1.Content augmented_prompt = 1;</code>
     * @return \Google\Protobuf\Internal\RepeatedField
     */
    public function getAugmentedPrompt()
    {
        return $this->augmented_prompt;
    }

    /**
     * Augmented prompt, only text format is supported for now.
     *
     * Generated from protobuf field <code>repeated .google.cloud.aiplatform.v1.Content augmented_prompt = 1;</code>
     * @param array<\Google\Cloud\AIPlatform\V1\Content>|\Google\Protobuf\Internal\RepeatedField $var
     * @return $this
     */
    public function setAugmentedPrompt($var)
    {
        $arr = GPBUtil::checkRepeatedField($var, \Google\Protobuf\Internal\GPBType::MESSAGE, \Google\Cloud\AIPlatform\V1\Content::class);
        $this->augmented_prompt = $arr;

        return $this;
    }

    /**
     * Retrieved facts from RAG data sources.
     *
     * Generated from protobuf field <code>repeated .google.cloud.aiplatform.v1.Fact facts = 2;</code>
     * @return \Google\Protobuf\Internal\RepeatedField
     */
    public function getFacts()
    {
        return $this->facts;
    }

    /**
     * Retrieved facts from RAG data sources.
     *
     * Generated from protobuf field <code>repeated .google.cloud.aiplatform.v1.Fact facts = 2;</code>
     * @param array<\Google\Cloud\AIPlatform\V1\Fact>|\Google\Protobuf\Internal\RepeatedField $var
     * @return $this
     */
    public function setFacts($var)
    {
        $arr = GPBUtil::checkRepeatedField($var, \Google\Protobuf\Internal\GPBType::MESSAGE, \Google\Cloud\AIPlatform\V1\Fact::class);
        $this->facts = $arr;

        return $this;
    }

}

