@maxim_mazurok/gapi.client.aiplatform-v1beta1 0.0.20240429 → 0.0.20240501

package/index.d.ts

@@ -9,7 +9,7 @@
 // This file was generated by https://github.com/Maxim-Mazurok/google-api-typings-generator. Please do not edit it manually.
 // In case of any problems please post issue to https://github.com/Maxim-Mazurok/google-api-typings-generator
 // Generated from: https://aiplatform.googleapis.com/$discovery/rest?version=v1beta1
-// Revision: 20240429
+// Revision: 20240501
 
 /// <reference types="gapi.client" />
 
@@ -1065,6 +1065,14 @@ declare namespace gapi.client {
       /** The operation generic information. */
       genericMetadata?: GoogleCloudAiplatformV1beta1GenericOperationMetadata;
     }
+    interface GoogleCloudAiplatformV1beta1CreateNotebookExecutionJobRequest {
+      /** Required. The NotebookExecutionJob to create. */
+      notebookExecutionJob?: GoogleCloudAiplatformV1beta1NotebookExecutionJob;
+      /** Optional. User specified ID for the NotebookExecutionJob. */
+      notebookExecutionJobId?: string;
+      /** Required. The resource name of the Location to create the NotebookExecutionJob. Format: `projects/{project}/locations/{location}` */
+      parent?: string;
+    }
     interface GoogleCloudAiplatformV1beta1CreateNotebookRuntimeTemplateOperationMetadata {
       /** The operation generic information. */
       genericMetadata?: GoogleCloudAiplatformV1beta1GenericOperationMetadata;
@@ -1491,6 +1499,7 @@ declare namespace gapi.client {
       /** The prediction output. */
       output?: string;
     }
+    interface GoogleCloudAiplatformV1beta1DirectUploadSource {}
     interface GoogleCloudAiplatformV1beta1DiskSpec {
       /** Size in GB of the boot disk (default is 100GB). */
       bootDiskSizeGb?: number;
@@ -2510,6 +2519,12 @@ declare namespace gapi.client {
       neighborCount?: number;
       /** Crowding is a constraint on a neighbor list produced by nearest neighbor search requiring that no more than some value k' of the k neighbors returned have the same value of crowding_attribute. It's used for improving result diversity. This field is the maximum number of matches with the same crowding tag. */
       perCrowdingAttributeNeighborCount?: number;
+      /** Optional. Represents RRF algorithm that combines search results. */
+      rrf?: GoogleCloudAiplatformV1beta1FindNeighborsRequestQueryRRF;
+    }
+    interface GoogleCloudAiplatformV1beta1FindNeighborsRequestQueryRRF {
+      /** Required. Users can provide an alpha value to give more weight to dense vs sparse results. For example, if the alpha is 0, we only return sparse and if the alpha is 1, we only return dense. */
+      alpha?: number;
     }
     interface GoogleCloudAiplatformV1beta1FindNeighborsResponse {
       /** The nearest neighbors of the query datapoints. */
@@ -2526,6 +2541,8 @@ declare namespace gapi.client {
       datapoint?: GoogleCloudAiplatformV1beta1IndexDatapoint;
       /** The distance between the neighbor and the dense embedding query. */
       distance?: number;
+      /** The distance between the neighbor and the query sparse_embedding. */
+      sparseDistance?: number;
     }
     interface GoogleCloudAiplatformV1beta1FluencyInput {
       /** Required. Fluency instance. */
@@ -2700,6 +2717,16 @@ declare namespace gapi.client {
       /** Required. The public base model URI. */
       baseModelUri?: string;
     }
+    interface GoogleCloudAiplatformV1beta1GoogleDriveSource {
+      /** Required. Google Drive resource IDs. */
+      resourceIds?: GoogleCloudAiplatformV1beta1GoogleDriveSourceResourceId[];
+    }
+    interface GoogleCloudAiplatformV1beta1GoogleDriveSourceResourceId {
+      /** Required. The ID of the Google Drive resource. */
+      resourceId?: string;
+      /** Required. The type of the Google Drive resource. */
+      resourceType?: string;
+    }
     interface GoogleCloudAiplatformV1beta1GroundednessInput {
       /** Required. Groundedness instance. */
       instance?: GoogleCloudAiplatformV1beta1GroundednessInstance;
@@ -2848,6 +2875,18 @@ declare namespace gapi.client {
       /** Required. Model evaluation resource to be imported. */
       modelEvaluation?: GoogleCloudAiplatformV1beta1ModelEvaluation;
     }
+    interface GoogleCloudAiplatformV1beta1ImportRagFilesConfig {
+      /** Google Cloud Storage location. Supports importing individual files as well as entire Google Cloud Storage directories. Sample formats: - `gs://bucket_name/my_directory/object_name/my_file.txt` - `gs://bucket_name/my_directory` */
+      gcsSource?: GoogleCloudAiplatformV1beta1GcsSource;
+      /** Google Drive location. Supports importing individual files as well as Google Drive folders. */
+      googleDriveSource?: GoogleCloudAiplatformV1beta1GoogleDriveSource;
+      /** Specifies the size and overlap of chunks after importing RagFiles. */
+      ragFileChunkingConfig?: GoogleCloudAiplatformV1beta1RagFileChunkingConfig;
+    }
+    interface GoogleCloudAiplatformV1beta1ImportRagFilesRequest {
+      /** Required. The config for the RagFiles to be synced and imported into the RagCorpus. VertexRagDataService.ImportRagFiles. */
+      importRagFilesConfig?: GoogleCloudAiplatformV1beta1ImportRagFilesConfig;
+    }
     interface GoogleCloudAiplatformV1beta1Index {
       /** Output only. Timestamp when this Index was created. */
       createTime?: string;
@@ -2887,6 +2926,8 @@ declare namespace gapi.client {
       numericRestricts?: GoogleCloudAiplatformV1beta1IndexDatapointNumericRestriction[];
       /** Optional. List of Restrict of the datapoint, used to perform "restricted searches" where boolean rule are used to filter the subset of the database eligible for matching. This uses categorical tokens. See: https://cloud.google.com/vertex-ai/docs/matching-engine/filtering */
       restricts?: GoogleCloudAiplatformV1beta1IndexDatapointRestriction[];
+      /** Optional. Feature embedding vector for sparse index. */
+      sparseEmbedding?: GoogleCloudAiplatformV1beta1IndexDatapointSparseEmbedding;
     }
     interface GoogleCloudAiplatformV1beta1IndexDatapointCrowdingTag {
       /** The attribute value used for crowding. The maximum number of neighbors to return per crowding attribute value (per_crowding_attribute_num_neighbors) is configured per-query. This field is ignored if per_crowding_attribute_num_neighbors is larger than the total number of neighbors to return for a given query. */
@@ -2912,6 +2953,12 @@ declare namespace gapi.client {
       /** The namespace of this restriction. e.g.: color. */
       namespace?: string;
     }
+    interface GoogleCloudAiplatformV1beta1IndexDatapointSparseEmbedding {
+      /** Optional. The list of indexes for the embedding values of the sparse vector. */
+      dimensions?: string[];
+      /** Optional. The list of embedding values of the sparse vector. */
+      values?: number[];
+    }
     interface GoogleCloudAiplatformV1beta1IndexEndpoint {
       /** Output only. Timestamp when this IndexEndpoint was created. */
       createTime?: string;
@@ -2953,6 +3000,8 @@ declare namespace gapi.client {
     interface GoogleCloudAiplatformV1beta1IndexStats {
       /** Output only. The number of shards in the Index. */
       shardsCount?: number;
+      /** Output only. The number of sparse vectors in the Index. */
+      sparseVectorsCount?: string;
       /** Output only. The number of dense vectors in the Index. */
       vectorsCount?: string;
     }
@@ -3256,6 +3305,18 @@ declare namespace gapi.client {
       /** List of PublisherModels in the requested page. */
       publisherModels?: GoogleCloudAiplatformV1beta1PublisherModel[];
     }
+    interface GoogleCloudAiplatformV1beta1ListRagCorporaResponse {
+      /** A token to retrieve the next page of results. Pass to ListRagCorporaRequest.page_token to obtain that page. */
+      nextPageToken?: string;
+      /** List of RagCorpora in the requested page. */
+      ragCorpora?: GoogleCloudAiplatformV1beta1RagCorpus[];
+    }
+    interface GoogleCloudAiplatformV1beta1ListRagFilesResponse {
+      /** A token to retrieve the next page of results. Pass to ListRagFilesRequest.page_token to obtain that page. */
+      nextPageToken?: string;
+      /** List of RagFiles in the requested page. */
+      ragFiles?: GoogleCloudAiplatformV1beta1RagFile[];
+    }
     interface GoogleCloudAiplatformV1beta1ListReasoningEnginesResponse {
       /** A token to retrieve the next page of results. Pass to ListReasoningEnginesRequest.page_token to obtain that page. */
       nextPageToken?: string;
@@ -4505,7 +4566,7 @@ declare namespace gapi.client {
       labels?: {[P in string]: string};
       /** Optional. Immutable. The specification of a single machine for the template. */
       machineSpec?: GoogleCloudAiplatformV1beta1MachineSpec;
-      /** Output only. The resource name of the NotebookRuntimeTemplate. */
+      /** The resource name of the NotebookRuntimeTemplate. */
       name?: string;
       /** Optional. Network spec. */
       networkSpec?: GoogleCloudAiplatformV1beta1NetworkSpec;
@@ -5204,6 +5265,64 @@ declare namespace gapi.client {
       /** Optional. Which version to use for evaluation. */
       version?: number;
     }
+    interface GoogleCloudAiplatformV1beta1RagContexts {
+      /** All its contexts. */
+      contexts?: GoogleCloudAiplatformV1beta1RagContextsContext[];
+    }
+    interface GoogleCloudAiplatformV1beta1RagContextsContext {
+      /** The distance between the query vector and the context text vector. */
+      distance?: number;
+      /** For vertex RagStore, if the file is imported from Cloud Storage or Google Drive, source_uri will be original file URI in Cloud Storage or Google Drive; if file is uploaded, source_uri will be file display name. */
+      sourceUri?: string;
+      /** The text chunk. */
+      text?: string;
+    }
+    interface GoogleCloudAiplatformV1beta1RagCorpus {
+      /** Output only. Timestamp when this RagCorpus was created. */
+      createTime?: string;
+      /** Optional. The description of the RagCorpus. */
+      description?: string;
+      /** Required. The display name of the RagCorpus. The name can be up to 128 characters long and can consist of any UTF-8 characters. */
+      displayName?: string;
+      /** Output only. The resource name of the RagCorpus. */
+      name?: string;
+      /** Output only. Timestamp when this RagCorpus was last updated. */
+      updateTime?: string;
+    }
+    interface GoogleCloudAiplatformV1beta1RagFile {
+      /** Output only. Timestamp when this RagFile was created. */
+      createTime?: string;
+      /** Optional. The description of the RagFile. */
+      description?: string;
+      /** Output only. The RagFile is encapsulated and uploaded in the UploadRagFile request. */
+      directUploadSource?: any;
+      /** Required. The display name of the RagFile. The name can be up to 128 characters long and can consist of any UTF-8 characters. */
+      displayName?: string;
+      /** Output only. Google Cloud Storage location of the RagFile. It does not support wildcards in the Cloud Storage uri for now. */
+      gcsSource?: GoogleCloudAiplatformV1beta1GcsSource;
+      /** Output only. Google Drive location. Supports importing individual files as well as Google Drive folders. */
+      googleDriveSource?: GoogleCloudAiplatformV1beta1GoogleDriveSource;
+      /** Output only. The resource name of the RagFile. */
+      name?: string;
+      /** Output only. The type of the RagFile. */
+      ragFileType?: string;
+      /** Output only. The size of the RagFile in bytes. */
+      sizeBytes?: string;
+      /** Output only. Timestamp when this RagFile was last updated. */
+      updateTime?: string;
+    }
+    interface GoogleCloudAiplatformV1beta1RagFileChunkingConfig {
+      /** The overlap between chunks. */
+      chunkOverlap?: number;
+      /** The size of the chunks. */
+      chunkSize?: number;
+    }
+    interface GoogleCloudAiplatformV1beta1RagQuery {
+      /** Optional. The number of contexts to retrieve. */
+      similarityTopK?: number;
+      /** Optional. The query in text format to get relevant contexts. */
+      text?: string;
+    }
     interface GoogleCloudAiplatformV1beta1RawPredictRequest {
       /** The prediction input. Supports HTTP headers and arbitrary data payload. A DeployedModel may have an upper limit on the number of instances it supports per request. When this limit it is exceeded for an AutoML model, the RawPredict method returns an error. When this limit is exceeded for a custom-trained model, the behavior varies depending on the model. You can specify the schema for each instance in the predict_schemata.instance_schema_uri field when you create a Model. This schema applies when you deploy the `Model` as a `DeployedModel` to an Endpoint and use the `RawPredict` method. */
       httpBody?: GoogleApiHttpBody;
@@ -5421,6 +5540,30 @@ declare namespace gapi.client {
       /** Set to use data source powered by Vertex RAG store. User data is uploaded via the VertexRagDataService. */
       vertexRagStore?: GoogleCloudAiplatformV1beta1VertexRagStore;
     }
+    interface GoogleCloudAiplatformV1beta1RetrieveContextsRequest {
+      /** Required. Single RAG retrieve query. */
+      query?: GoogleCloudAiplatformV1beta1RagQuery;
+      /** The data source for Vertex RagStore. */
+      vertexRagStore?: GoogleCloudAiplatformV1beta1RetrieveContextsRequestVertexRagStore;
+    }
+    interface GoogleCloudAiplatformV1beta1RetrieveContextsRequestVertexRagStore {
+      /** Optional. Deprecated. Please use rag_resources to specify the data source. */
+      ragCorpora?: string[];
+      /** 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. */
+      ragResources?: GoogleCloudAiplatformV1beta1RetrieveContextsRequestVertexRagStoreRagResource[];
+      /** Optional. Only return contexts with vector distance smaller than the threshold. */
+      vectorDistanceThreshold?: number;
+    }
+    interface GoogleCloudAiplatformV1beta1RetrieveContextsRequestVertexRagStoreRagResource {
+      /** Optional. RagCorpora resource name. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}` */
+      ragCorpus?: string;
+      /** Optional. rag_file_id. The files should be in the same rag_corpus set in rag_corpus field. */
+      ragFileIds?: string[];
+    }
+    interface GoogleCloudAiplatformV1beta1RetrieveContextsResponse {
+      /** The contexts of the query. */
+      contexts?: GoogleCloudAiplatformV1beta1RagContexts;
+    }
     interface GoogleCloudAiplatformV1beta1RougeInput {
       /** Required. Repeated rouge instances. */
       instances?: GoogleCloudAiplatformV1beta1RougeInstance[];
@@ -5464,7 +5607,9 @@ declare namespace gapi.client {
       fileOutputGcsBucket?: string;
     }
     interface GoogleCloudAiplatformV1beta1RuntimeConfigVertexAISearchRuntimeConfig {
-      /** Required. Vertext AI Search serving config name. Format: `projects/{project}/locations/{location}/collections/{collection}/engines/{engine}/servingConfigs/{serving_config}` */
+      /** Vertex AI Search App ID. This is used to construct the search request. By setting this app_id, API will construct the serving config which is required to call search API for the user. The app_id and serving_config_name cannot both be empty at the same time. */
+      appId?: string;
+      /** [Deprecated] Please use app_id instead. Vertex AI Search serving config name. Format: `projects/{project}/locations/{location}/collections/{collection}/engines/{engine}/servingConfigs/{serving_config}` */
       servingConfigName?: string;
     }
     interface GoogleCloudAiplatformV1beta1SafetyInput {
@@ -5564,6 +5709,8 @@ declare namespace gapi.client {
       catchUp?: boolean;
       /** Request for ModelMonitoringService.CreateModelMonitoringJob. */
       createModelMonitoringJobRequest?: GoogleCloudAiplatformV1beta1CreateModelMonitoringJobRequest;
+      /** Request for NotebookService.CreateNotebookExecutionJob. */
+      createNotebookExecutionJobRequest?: GoogleCloudAiplatformV1beta1CreateNotebookExecutionJobRequest;
       /** Request for PipelineService.CreatePipelineJob. CreatePipelineJobRequest.parent field is required (format: projects/{project}/locations/{location}). */
       createPipelineJobRequest?: GoogleCloudAiplatformV1beta1CreatePipelineJobRequest;
       /** Output only. Timestamp when this Schedule was created. */
@@ -7920,6 +8067,22 @@ declare namespace gapi.client {
       /** Output only. The version ID of the model that is uploaded. */
       modelVersionId?: string;
     }
+    interface GoogleCloudAiplatformV1beta1UploadRagFileConfig {
+      /** Specifies the size and overlap of chunks after uploading RagFile. */
+      ragFileChunkingConfig?: GoogleCloudAiplatformV1beta1RagFileChunkingConfig;
+    }
+    interface GoogleCloudAiplatformV1beta1UploadRagFileRequest {
+      /** Required. The RagFile to upload. */
+      ragFile?: GoogleCloudAiplatformV1beta1RagFile;
+      /** Required. The config for the RagFiles to be uploaded into the RagCorpus. VertexRagDataService.UploadRagFile. */
+      uploadRagFileConfig?: GoogleCloudAiplatformV1beta1UploadRagFileConfig;
+    }
+    interface GoogleCloudAiplatformV1beta1UploadRagFileResponse {
+      /** The error that occurred while processing the RagFile. */
+      error?: GoogleRpcStatus;
+      /** The RagFile that had been uploaded into the RagCorpus. */
+      ragFile?: GoogleCloudAiplatformV1beta1RagFile;
+    }
     interface GoogleCloudAiplatformV1beta1UpsertDatapointsRequest {
       /** A list of datapoints to be created/updated. */
       datapoints?: GoogleCloudAiplatformV1beta1IndexDatapoint[];
@@ -7948,13 +8111,21 @@ declare namespace gapi.client {
       datastore?: string;
     }
     interface GoogleCloudAiplatformV1beta1VertexRagStore {
-      /** 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. */
+      /** Optional. Deprecated. Please use rag_resources instead. */
       ragCorpora?: string[];
+      /** 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. */
+      ragResources?: GoogleCloudAiplatformV1beta1VertexRagStoreRagResource[];
       /** Optional. Number of top k results to return from the selected corpora. */
       similarityTopK?: number;
       /** Optional. Only return results with vector distance smaller than the threshold. */
       vectorDistanceThreshold?: number;
     }
+    interface GoogleCloudAiplatformV1beta1VertexRagStoreRagResource {
+      /** Optional. RagCorpora resource name. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}` */
+      ragCorpus?: string;
+      /** Optional. rag_file_id. The files should be in the same rag_corpus set in rag_corpus field. */
+      ragFileIds?: string[];
+    }
     interface GoogleCloudAiplatformV1beta1VideoMetadata {
       /** Optional. The end offset of the video. */
       endOffset?: string;
@@ -8620,7 +8791,12 @@ declare namespace gapi.client {
       /** The sum of the size of all the contents in the request. */
       totalContentSize?: string;
     }
+    interface LearningServingLlmAtlasOutputMetadata {
+      requestTopic?: string;
+      source?: string;
+    }
     interface LearningServingLlmMessageMetadata {
+      atlasMetadata?: LearningServingLlmAtlasOutputMetadata;
       /** Summary of classifier output. We attach this to all messages regardless of whether classification rules triggered or not. */
       classifierSummary?: LearningGenaiRootClassifierOutputSummary;
       /** Contains metadata related to Codey Processors. */
@@ -8647,13 +8823,15 @@ declare namespace gapi.client {
       lmPrefix?: string;
       /** The original text generated by LLM. This is the raw output for debugging purposes. */
       originalText?: string;
-      /** NOT YET IMPLEMENTED. Applies to streaming only. Number of tokens decoded / emitted by the model as part of this stream. This may be different from token_count, which contains number of tokens returned in this response after any response rewriting / truncation. */
+      /** Number of tokens decoded by the model as part of a stream. This count may be different from `per_stream_returned_token_count` which, is counted after any response rewriting or truncation. Applies to streaming response only. */
       perStreamDecodedTokenCount?: number;
+      /** Number of tokens returned per stream in a response candidate after any response rewriting or truncation. Applies to streaming response only. Applies to Gemini models only. */
+      perStreamReturnedTokenCount?: number;
       /** Results of running RAI on the query or this response candidate. One output per rai_config. It will be populated regardless of whether the threshold is exceeded or not. */
       raiOutputs?: LearningGenaiRootRAIOutput[];
       /** Recitation Results. It will be populated as long as Recitation processing is enabled, regardless of recitation outcome. */
       recitationResult?: LearningGenaiRecitationRecitationResult;
-      /** NOT YET IMPLEMENTED. Number of tokens returned as part of this candidate. */
+      /** NOT IMPLEMENTED TODO (b/334187574) Remove this field after Labs migrates to per_stream_returned_token_count and total_returned_token_count. */
       returnTokenCount?: number;
       /** All the different scores for a message are logged here. */
       scores?: LearningGenaiRootScore[];
@@ -8661,6 +8839,8 @@ declare namespace gapi.client {
       streamTerminated?: boolean;
       /** Total tokens decoded so far per response_candidate. For streaming: Count of all the tokens decoded so far (aggregated count). For unary: Count of all the tokens decoded per response_candidate. */
       totalDecodedTokenCount?: number;
+      /** Total number of tokens returned in a response candidate. For streaming, it is the aggregated count (i.e. total so far) Applies to Gemini models only. */
+      totalReturnedTokenCount?: number;
       /** Translated user-prompt used for RAI post processing. This is for internal processing only. We will translate in pre-processor and pass the translated text to the post processor using this field. It will be empty if non of the signals requested need translation. */
       translatedUserPrompts?: string[];
       /** The metadata from Vertex SafetyCat processors */
@@ -8712,6 +8892,66 @@ declare namespace gapi.client {
       /** The following are usually only present when code != 0 Space to which this status belongs */
       space?: string;
     }
+    interface MediaResource {
+      /** Upload a file into a RagCorpus. */
+      upload(request: {
+        /** V1 error format. */
+        '$.xgafv'?: string;
+        /** OAuth access token. */
+        access_token?: string;
+        /** Data format for response. */
+        alt?: string;
+        /** JSONP */
+        callback?: string;
+        /** Selector specifying which fields to include in a partial response. */
+        fields?: string;
+        /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+        key?: string;
+        /** OAuth 2.0 token for the current user. */
+        oauth_token?: string;
+        /** Required. The name of the RagCorpus resource into which to upload the file. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}` */
+        parent: string;
+        /** Returns response with indentations and line breaks. */
+        prettyPrint?: boolean;
+        /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+        quotaUser?: string;
+        /** Upload protocol for media (e.g. "raw", "multipart"). */
+        upload_protocol?: string;
+        /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+        uploadType?: string;
+        /** Request body */
+        resource: GoogleCloudAiplatformV1beta1UploadRagFileRequest;
+      }): Request<GoogleCloudAiplatformV1beta1UploadRagFileResponse>;
+      upload(
+        request: {
+          /** V1 error format. */
+          '$.xgafv'?: string;
+          /** OAuth access token. */
+          access_token?: string;
+          /** Data format for response. */
+          alt?: string;
+          /** JSONP */
+          callback?: string;
+          /** Selector specifying which fields to include in a partial response. */
+          fields?: string;
+          /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+          key?: string;
+          /** OAuth 2.0 token for the current user. */
+          oauth_token?: string;
+          /** Required. The name of the RagCorpus resource into which to upload the file. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}` */
+          parent: string;
+          /** Returns response with indentations and line breaks. */
+          prettyPrint?: boolean;
+          /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+          quotaUser?: string;
+          /** Upload protocol for media (e.g. "raw", "multipart"). */
+          upload_protocol?: string;
+          /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+          uploadType?: string;
+        },
+        body: GoogleCloudAiplatformV1beta1UploadRagFileRequest
+      ): Request<GoogleCloudAiplatformV1beta1UploadRagFileResponse>;
+    }
     interface OperationsResource {
       /** Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of 1, corresponding to `Code.CANCELLED`. */
       cancel(request?: {
@@ -26477,9 +26717,297 @@ declare namespace gapi.client {
       }): Request<GoogleLongrunningOperation>;
     }
     interface RagFilesResource {
+      /** Deletes a RagFile. */
+      delete(request?: {
+        /** V1 error format. */
+        '$.xgafv'?: string;
+        /** OAuth access token. */
+        access_token?: string;
+        /** Data format for response. */
+        alt?: string;
+        /** JSONP */
+        callback?: string;
+        /** Selector specifying which fields to include in a partial response. */
+        fields?: string;
+        /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+        key?: string;
+        /** Required. The name of the RagFile resource to be deleted. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}/ragFiles/{rag_file}` */
+        name: string;
+        /** OAuth 2.0 token for the current user. */
+        oauth_token?: string;
+        /** Returns response with indentations and line breaks. */
+        prettyPrint?: boolean;
+        /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+        quotaUser?: string;
+        /** Upload protocol for media (e.g. "raw", "multipart"). */
+        upload_protocol?: string;
+        /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+        uploadType?: string;
+      }): Request<GoogleLongrunningOperation>;
+      /** Gets a RagFile. */
+      get(request?: {
+        /** V1 error format. */
+        '$.xgafv'?: string;
+        /** OAuth access token. */
+        access_token?: string;
+        /** Data format for response. */
+        alt?: string;
+        /** JSONP */
+        callback?: string;
+        /** Selector specifying which fields to include in a partial response. */
+        fields?: string;
+        /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+        key?: string;
+        /** Required. The name of the RagFile resource. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}/ragFiles/{rag_file}` */
+        name: string;
+        /** OAuth 2.0 token for the current user. */
+        oauth_token?: string;
+        /** Returns response with indentations and line breaks. */
+        prettyPrint?: boolean;
+        /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+        quotaUser?: string;
+        /** Upload protocol for media (e.g. "raw", "multipart"). */
+        upload_protocol?: string;
+        /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+        uploadType?: string;
+      }): Request<GoogleCloudAiplatformV1beta1RagFile>;
+      /** Import files from Google Cloud Storage or Google Drive into a RagCorpus. */
+      import(request: {
+        /** V1 error format. */
+        '$.xgafv'?: string;
+        /** OAuth access token. */
+        access_token?: string;
+        /** Data format for response. */
+        alt?: string;
+        /** JSONP */
+        callback?: string;
+        /** Selector specifying which fields to include in a partial response. */
+        fields?: string;
+        /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+        key?: string;
+        /** OAuth 2.0 token for the current user. */
+        oauth_token?: string;
+        /** Required. The name of the RagCorpus resource into which to import files. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}` */
+        parent: string;
+        /** Returns response with indentations and line breaks. */
+        prettyPrint?: boolean;
+        /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+        quotaUser?: string;
+        /** Upload protocol for media (e.g. "raw", "multipart"). */
+        upload_protocol?: string;
+        /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+        uploadType?: string;
+        /** Request body */
+        resource: GoogleCloudAiplatformV1beta1ImportRagFilesRequest;
+      }): Request<GoogleLongrunningOperation>;
+      import(
+        request: {
+          /** V1 error format. */
+          '$.xgafv'?: string;
+          /** OAuth access token. */
+          access_token?: string;
+          /** Data format for response. */
+          alt?: string;
+          /** JSONP */
+          callback?: string;
+          /** Selector specifying which fields to include in a partial response. */
+          fields?: string;
+          /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+          key?: string;
+          /** OAuth 2.0 token for the current user. */
+          oauth_token?: string;
+          /** Required. The name of the RagCorpus resource into which to import files. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}` */
+          parent: string;
+          /** Returns response with indentations and line breaks. */
+          prettyPrint?: boolean;
+          /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+          quotaUser?: string;
+          /** Upload protocol for media (e.g. "raw", "multipart"). */
+          upload_protocol?: string;
+          /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+          uploadType?: string;
+        },
+        body: GoogleCloudAiplatformV1beta1ImportRagFilesRequest
+      ): Request<GoogleLongrunningOperation>;
+      /** Lists RagFiles in a RagCorpus. */
+      list(request?: {
+        /** V1 error format. */
+        '$.xgafv'?: string;
+        /** OAuth access token. */
+        access_token?: string;
+        /** Data format for response. */
+        alt?: string;
+        /** JSONP */
+        callback?: string;
+        /** Selector specifying which fields to include in a partial response. */
+        fields?: string;
+        /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+        key?: string;
+        /** OAuth 2.0 token for the current user. */
+        oauth_token?: string;
+        /** Optional. The standard list page size. */
+        pageSize?: number;
+        /** Optional. The standard list page token. Typically obtained via ListRagFilesResponse.next_page_token of the previous VertexRagDataService.ListRagFiles call. */
+        pageToken?: string;
+        /** Required. The resource name of the RagCorpus from which to list the RagFiles. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}` */
+        parent: string;
+        /** Returns response with indentations and line breaks. */
+        prettyPrint?: boolean;
+        /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+        quotaUser?: string;
+        /** Upload protocol for media (e.g. "raw", "multipart"). */
+        upload_protocol?: string;
+        /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+        uploadType?: string;
+      }): Request<GoogleCloudAiplatformV1beta1ListRagFilesResponse>;
       operations: OperationsResource;
     }
     interface RagCorporaResource {
+      /** Creates a RagCorpus. */
+      create(request: {
+        /** V1 error format. */
+        '$.xgafv'?: string;
+        /** OAuth access token. */
+        access_token?: string;
+        /** Data format for response. */
+        alt?: string;
+        /** JSONP */
+        callback?: string;
+        /** Selector specifying which fields to include in a partial response. */
+        fields?: string;
+        /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+        key?: string;
+        /** OAuth 2.0 token for the current user. */
+        oauth_token?: string;
+        /** Required. The resource name of the Location to create the RagCorpus in. Format: `projects/{project}/locations/{location}` */
+        parent: string;
+        /** Returns response with indentations and line breaks. */
+        prettyPrint?: boolean;
+        /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+        quotaUser?: string;
+        /** Upload protocol for media (e.g. "raw", "multipart"). */
+        upload_protocol?: string;
+        /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+        uploadType?: string;
+        /** Request body */
+        resource: GoogleCloudAiplatformV1beta1RagCorpus;
+      }): Request<GoogleLongrunningOperation>;
+      create(
+        request: {
+          /** V1 error format. */
+          '$.xgafv'?: string;
+          /** OAuth access token. */
+          access_token?: string;
+          /** Data format for response. */
+          alt?: string;
+          /** JSONP */
+          callback?: string;
+          /** Selector specifying which fields to include in a partial response. */
+          fields?: string;
+          /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+          key?: string;
+          /** OAuth 2.0 token for the current user. */
+          oauth_token?: string;
+          /** Required. The resource name of the Location to create the RagCorpus in. Format: `projects/{project}/locations/{location}` */
+          parent: string;
+          /** Returns response with indentations and line breaks. */
+          prettyPrint?: boolean;
+          /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+          quotaUser?: string;
+          /** Upload protocol for media (e.g. "raw", "multipart"). */
+          upload_protocol?: string;
+          /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+          uploadType?: string;
+        },
+        body: GoogleCloudAiplatformV1beta1RagCorpus
+      ): Request<GoogleLongrunningOperation>;
+      /** Deletes a RagCorpus. */
+      delete(request?: {
+        /** V1 error format. */
+        '$.xgafv'?: string;
+        /** OAuth access token. */
+        access_token?: string;
+        /** Data format for response. */
+        alt?: string;
+        /** JSONP */
+        callback?: string;
+        /** Selector specifying which fields to include in a partial response. */
+        fields?: string;
+        /** Optional. If set to true, any RagFiles in this RagCorpus will also be deleted. Otherwise, the request will only work if the RagCorpus has no RagFiles. */
+        force?: boolean;
+        /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+        key?: string;
+        /** Required. The name of the RagCorpus resource to be deleted. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}` */
+        name: string;
+        /** OAuth 2.0 token for the current user. */
+        oauth_token?: string;
+        /** Returns response with indentations and line breaks. */
+        prettyPrint?: boolean;
+        /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+        quotaUser?: string;
+        /** Upload protocol for media (e.g. "raw", "multipart"). */
+        upload_protocol?: string;
+        /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+        uploadType?: string;
+      }): Request<GoogleLongrunningOperation>;
+      /** Gets a RagCorpus. */
+      get(request?: {
+        /** V1 error format. */
+        '$.xgafv'?: string;
+        /** OAuth access token. */
+        access_token?: string;
+        /** Data format for response. */
+        alt?: string;
+        /** JSONP */
+        callback?: string;
+        /** Selector specifying which fields to include in a partial response. */
+        fields?: string;
+        /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+        key?: string;
+        /** Required. The name of the RagCorpus resource. Format: `projects/{project}/locations/{location}/ragCorpora/{rag_corpus}` */
+        name: string;
+        /** OAuth 2.0 token for the current user. */
+        oauth_token?: string;
+        /** Returns response with indentations and line breaks. */
+        prettyPrint?: boolean;
+        /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+        quotaUser?: string;
+        /** Upload protocol for media (e.g. "raw", "multipart"). */
+        upload_protocol?: string;
+        /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+        uploadType?: string;
+      }): Request<GoogleCloudAiplatformV1beta1RagCorpus>;
+      /** Lists RagCorpora in a Location. */
+      list(request?: {
+        /** V1 error format. */
+        '$.xgafv'?: string;
+        /** OAuth access token. */
+        access_token?: string;
+        /** Data format for response. */
+        alt?: string;
+        /** JSONP */
+        callback?: string;
+        /** Selector specifying which fields to include in a partial response. */
+        fields?: string;
+        /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+        key?: string;
+        /** OAuth 2.0 token for the current user. */
+        oauth_token?: string;
+        /** Optional. The standard list page size. */
+        pageSize?: number;
+        /** Optional. The standard list page token. Typically obtained via ListRagCorporaResponse.next_page_token of the previous VertexRagDataService.ListRagCorpora call. */
+        pageToken?: string;
+        /** Required. The resource name of the Location from which to list the RagCorpora. Format: `projects/{project}/locations/{location}` */
+        parent: string;
+        /** Returns response with indentations and line breaks. */
+        prettyPrint?: boolean;
+        /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+        quotaUser?: string;
+        /** Upload protocol for media (e.g. "raw", "multipart"). */
+        upload_protocol?: string;
+        /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+        uploadType?: string;
+      }): Request<GoogleCloudAiplatformV1beta1ListRagCorporaResponse>;
       operations: OperationsResource;
       ragFiles: RagFilesResource;
     }
@@ -31101,6 +31629,64 @@ declare namespace gapi.client {
         /** Legacy upload protocol for media (e.g. "media", "multipart"). */
         uploadType?: string;
       }): Request<GoogleCloudLocationListLocationsResponse>;
+      /** Retrieves relevant contexts for a query. */
+      retrieveContexts(request: {
+        /** V1 error format. */
+        '$.xgafv'?: string;
+        /** OAuth access token. */
+        access_token?: string;
+        /** Data format for response. */
+        alt?: string;
+        /** JSONP */
+        callback?: string;
+        /** Selector specifying which fields to include in a partial response. */
+        fields?: string;
+        /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+        key?: string;
+        /** OAuth 2.0 token for the current user. */
+        oauth_token?: string;
+        /** Required. The resource name of the Location from which to retrieve RagContexts. The users must have permission to make a call in the project. Format: `projects/{project}/locations/{location}`. */
+        parent: string;
+        /** Returns response with indentations and line breaks. */
+        prettyPrint?: boolean;
+        /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+        quotaUser?: string;
+        /** Upload protocol for media (e.g. "raw", "multipart"). */
+        upload_protocol?: string;
+        /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+        uploadType?: string;
+        /** Request body */
+        resource: GoogleCloudAiplatformV1beta1RetrieveContextsRequest;
+      }): Request<GoogleCloudAiplatformV1beta1RetrieveContextsResponse>;
+      retrieveContexts(
+        request: {
+          /** V1 error format. */
+          '$.xgafv'?: string;
+          /** OAuth access token. */
+          access_token?: string;
+          /** Data format for response. */
+          alt?: string;
+          /** JSONP */
+          callback?: string;
+          /** Selector specifying which fields to include in a partial response. */
+          fields?: string;
+          /** API key. Your API key identifies your project and provides you with API access, quota, and reports. Required unless you provide an OAuth 2.0 token. */
+          key?: string;
+          /** OAuth 2.0 token for the current user. */
+          oauth_token?: string;
+          /** Required. The resource name of the Location from which to retrieve RagContexts. The users must have permission to make a call in the project. Format: `projects/{project}/locations/{location}`. */
+          parent: string;
+          /** Returns response with indentations and line breaks. */
+          prettyPrint?: boolean;
+          /** Available to use for quota purposes for server-side applications. Can be any arbitrary string assigned to a user, but should not exceed 40 characters. */
+          quotaUser?: string;
+          /** Upload protocol for media (e.g. "raw", "multipart"). */
+          upload_protocol?: string;
+          /** Legacy upload protocol for media (e.g. "media", "multipart"). */
+          uploadType?: string;
+        },
+        body: GoogleCloudAiplatformV1beta1RetrieveContextsRequest
+      ): Request<GoogleCloudAiplatformV1beta1RetrieveContextsResponse>;
       agents: AgentsResource;
       apps: AppsResource;
       batchPredictionJobs: BatchPredictionJobsResource;
@@ -31306,6 +31892,8 @@ declare namespace gapi.client {
       models: ModelsResource;
     }
 
+    const media: MediaResource;
+
     const projects: ProjectsResource;
 
     const publishers: PublishersResource;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@maxim_mazurok/gapi.client.aiplatform-v1beta1",
-  "version": "0.0.20240429",
+  "version": "0.0.20240501",
   "description": "TypeScript typings for Vertex AI API v1beta1",
   "repository": {
     "type": "git",

package/readme.md

@@ -72,6 +72,11 @@ gapi.auth.authorize(
 After that you can use Vertex AI API resources: <!-- TODO: make this work for multiple namespaces -->
 
 ```typescript
+/*
+Upload a file into a RagCorpus.
+*/
+await gapi.client.aiplatform.media.upload({parent: 'parent'});
+
 /*
 Gets a GenAI cache config.
 */