gapic

app.py

@@ -5,7 +5,7 @@ from langchain_core.prompts import PromptTemplate
 from google.cloud import aiplatform_v1
 import streamlit as st
 import json
-
+from google.cloud.aiplatform_v1 import EndpointServiceClient
 
 # Function to pad base64 strings
 def pad_base64(base64_string):
@@ -20,8 +20,8 @@ if credentials_json:
     os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = credentials_path
 
 # Initialize the Google AI Platform
-client_options = {"api_endpoint": "us-central1-aiplatform_v1.googleapis.com"}
-client = aiplatform_v1.gapic.PipelineServiceClient(client_options=client_options)
+client_options = {"api_endpoint": "us-central1-aiplatform.googleapis.com"}
+client = aiplatform_v1.PipelineServiceClient(client_options=client_options)
 
 # Define the documents as dictionaries, ensuring correct padding
 documents = [