fix: RAG - Fix the uri used for upload_file

PiperOrigin-RevId: 704443973

Author: vertex-sdk-bot

tests/unit/vertex_rag/test_rag_constants.py

@@ -163,10 +163,8 @@
 TEST_FILE_DISPLAY_NAME = "my-file.txt"
 TEST_FILE_DESCRIPTION = "my file."
 TEST_HEADERS = {"X-Goog-Upload-Protocol": "multipart"}
-TEST_UPLOAD_REQUEST_URI = (
- "https://{}/v1/projects/{}/locations/{}/ragCorpora/{}/ragFiles:upload".format(
- TEST_API_ENDPOINT, TEST_PROJECT_NUMBER, TEST_REGION, TEST_RAG_CORPUS_ID
- )
+TEST_UPLOAD_REQUEST_URI = "https://{}/upload/v1/projects/{}/locations/{}/ragCorpora/{}/ragFiles:upload".format(
+ TEST_API_ENDPOINT, TEST_PROJECT_NUMBER, TEST_REGION, TEST_RAG_CORPUS_ID
 )
 TEST_RAG_FILE_ID = "generate-456"
 TEST_RAG_FILE_RESOURCE_NAME = (

tests/unit/vertex_rag/test_rag_constants_preview.py

@@ -308,10 +308,8 @@
 TEST_FILE_DISPLAY_NAME = "my-file.txt"
 TEST_FILE_DESCRIPTION = "my file."
 TEST_HEADERS = {"X-Goog-Upload-Protocol": "multipart"}
-TEST_UPLOAD_REQUEST_URI = (
- "https://{}/v1beta1/projects/{}/locations/{}/ragCorpora/{}/ragFiles:upload".format(
- TEST_API_ENDPOINT, TEST_PROJECT_NUMBER, TEST_REGION, TEST_RAG_CORPUS_ID
- )
+TEST_UPLOAD_REQUEST_URI = "https://{}/upload/v1beta1/projects/{}/locations/{}/ragCorpora/{}/ragFiles:upload".format(
+ TEST_API_ENDPOINT, TEST_PROJECT_NUMBER, TEST_REGION, TEST_RAG_CORPUS_ID
 )
 TEST_RAG_FILE_ID = "generate-456"
 TEST_RAG_FILE_RESOURCE_NAME = (

vertexai/preview/rag/rag_data.py

@@ -415,7 +415,7 @@ def upload_file(
  )
  else:
  request_endpoint = initializer.global_config.api_endpoint
- upload_request_uri = "https://{}/v1beta1/{}/ragFiles:upload".format(
+ upload_request_uri = "https://{}/upload/v1beta1/{}/ragFiles:upload".format(
  request_endpoint,
  corpus_name,
  )

vertexai/rag/rag_data.py

@@ -338,7 +338,7 @@ def upload_file(
  )
  else:
  request_endpoint = initializer.global_config.api_endpoint
- upload_request_uri = "https://{}/v1/{}/ragFiles:upload".format(
+ upload_request_uri = "https://{}/upload/v1/{}/ragFiles:upload".format(
  request_endpoint,
  corpus_name,
  )