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,
     )