feat(firebaseml): update the api

#### firebaseml:v2beta The following keys were added: - schemas.RagResource (Total Keys: 5) - schemas.VertexRagStore.properties.ragResources (Total Keys: 2)
googleapis · Apr 30, 2024 · a406db6 · a406db6
1 parent dff33af
commit a406db6
Show file tree

Hide file tree

Showing 4 changed files with 47 additions and 6 deletions.
diff --git a/docs/dyn/firebaseml_v2beta.projects.locations.publishers.models.html b/docs/dyn/firebaseml_v2beta.projects.locations.publishers.models.html
@@ -330,9 +330,17 @@ <h3>Method Details</h3>
           &quot;datastore&quot;: &quot;A String&quot;, # Required. Fully-qualified Vertex AI Search&#x27;s datastore resource ID. Format: `projects/{project}/locations/{location}/collections/{collection}/dataStores/{dataStore}`
         },
         &quot;vertexRagStore&quot;: { # Retrieve from Vertex RAG Store for grounding. # Set to use data source powered by Vertex RAG store. User data is uploaded via the VertexRagDataService.
-          &quot;ragCorpora&quot;: [ # Required. Vertex RAG Store corpus resource name: `projects/{project}/locations/{location}/ragCorpora/{ragCorpus}` Currently only one corpus is allowed. In the future we may open up multiple corpora support. However, they should be from the same project and location.
+          &quot;ragCorpora&quot;: [ # Optional. Deprecated. Please use rag_resources instead.
             &quot;A String&quot;,
           ],
+          &quot;ragResources&quot;: [ # Optional. The representation of the rag source. It can be used to specify corpus only or ragfiles. Currently only support one corpus or multiple files from one corpus. In the future we may open up multiple corpora support.
+            { # The definition of the Rag resource.
+              &quot;ragCorpus&quot;: &quot;A String&quot;, # Optional. RagCorpora resource name. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}`
+              &quot;ragFileIds&quot;: [ # Optional. rag_file_id. The files should be in the same rag_corpus set in rag_corpus field.
+                &quot;A String&quot;,
+              ],
+            },
+          ],
           &quot;similarityTopK&quot;: 42, # Optional. Number of top k results to return from the selected corpora.
           &quot;vectorDistanceThreshold&quot;: 3.14, # Optional. Only return results with vector distance smaller than the threshold.
         },
@@ -623,9 +631,17 @@ <h3>Method Details</h3>
           &quot;datastore&quot;: &quot;A String&quot;, # Required. Fully-qualified Vertex AI Search&#x27;s datastore resource ID. Format: `projects/{project}/locations/{location}/collections/{collection}/dataStores/{dataStore}`
         },
         &quot;vertexRagStore&quot;: { # Retrieve from Vertex RAG Store for grounding. # Set to use data source powered by Vertex RAG store. User data is uploaded via the VertexRagDataService.
-          &quot;ragCorpora&quot;: [ # Required. Vertex RAG Store corpus resource name: `projects/{project}/locations/{location}/ragCorpora/{ragCorpus}` Currently only one corpus is allowed. In the future we may open up multiple corpora support. However, they should be from the same project and location.
+          &quot;ragCorpora&quot;: [ # Optional. Deprecated. Please use rag_resources instead.
             &quot;A String&quot;,
           ],
+          &quot;ragResources&quot;: [ # Optional. The representation of the rag source. It can be used to specify corpus only or ragfiles. Currently only support one corpus or multiple files from one corpus. In the future we may open up multiple corpora support.
+            { # The definition of the Rag resource.
+              &quot;ragCorpus&quot;: &quot;A String&quot;, # Optional. RagCorpora resource name. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}`
+              &quot;ragFileIds&quot;: [ # Optional. rag_file_id. The files should be in the same rag_corpus set in rag_corpus field.
+                &quot;A String&quot;,
+              ],
+            },
+          ],
           &quot;similarityTopK&quot;: 42, # Optional. Number of top k results to return from the selected corpora.
           &quot;vectorDistanceThreshold&quot;: 3.14, # Optional. Only return results with vector distance smaller than the threshold.
         },

diff --git a/googleapiclient/discovery_cache/documents/firebaseml.v1.json b/googleapiclient/discovery_cache/documents/firebaseml.v1.json
@@ -204,7 +204,7 @@
 }
 }
 },
-"revision": "20240419",
+"revision": "20240426",
 "rootUrl": "https://firebaseml.googleapis.com/",
 "schemas": {
 "CancelOperationRequest": {

diff --git a/googleapiclient/discovery_cache/documents/firebaseml.v1beta2.json b/googleapiclient/discovery_cache/documents/firebaseml.v1beta2.json
@@ -318,7 +318,7 @@
 }
 }
 },
-"revision": "20240419",
+"revision": "20240426",
 "rootUrl": "https://firebaseml.googleapis.com/",
 "schemas": {
 "DownloadModelResponse": {

diff --git a/googleapiclient/discovery_cache/documents/firebaseml.v2beta.json b/googleapiclient/discovery_cache/documents/firebaseml.v2beta.json
@@ -206,7 +206,7 @@
 }
 }
 },
-"revision": "20240419",
+"revision": "20240426",
 "rootUrl": "https://firebaseml.googleapis.com/",
 "schemas": {
 "Blob": {
@@ -771,6 +771,24 @@
 },
 "type": "object"
 },
+"RagResource": {
+"description": "The definition of the Rag resource.",
+"id": "RagResource",
+"properties": {
+"ragCorpus": {
+"description": "Optional. RagCorpora resource name. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}`",
+"type": "string"
+},
+"ragFileIds": {
+"description": "Optional. rag_file_id. The files should be in the same rag_corpus set in rag_corpus field.",
+"items": {
+"type": "string"
+},
+"type": "array"
+}
+},
+"type": "object"
+},
 "Retrieval": {
 "description": "Defines a retrieval tool that model can call to access external knowledge.",
 "id": "Retrieval",
@@ -1132,12 +1150,19 @@
 "id": "VertexRagStore",
 "properties": {
 "ragCorpora": {
-"description": "Required. Vertex RAG Store corpus resource name: `projects/{project}/locations/{location}/ragCorpora/{ragCorpus}` Currently only one corpus is allowed. In the future we may open up multiple corpora support. However, they should be from the same project and location.",
+"description": "Optional. Deprecated. Please use rag_resources instead.",
 "items": {
 "type": "string"
 },
 "type": "array"
 },
+"ragResources": {
+"description": "Optional. The representation of the rag source. It can be used to specify corpus only or ragfiles. Currently only support one corpus or multiple files from one corpus. In the future we may open up multiple corpora support.",
+"items": {
+"$ref": "RagResource"
+},
+"type": "array"
+},
 "similarityTopK": {
 "description": "Optional. Number of top k results to return from the selected corpora.",
 "format": "int32",