in infra-as-code/modules/ingest-pipeline/cf-feedback-generator/lib.py [0:0]
def __init__(self, project_id, location_id, model_name, insights_endpoint,
insights_api_version, ccai_insights_location_id, conversation_id,
dataset_name, table_name, scorecard_id, ingest_record_bucket_id,
target_tags, target_values):
self.project_id = project_id
self.location_id = location_id
self.model_name = model_name
self.insights_endpoint = insights_endpoint
self.insights_api_version = insights_api_version
self.ccai_insights_location_id = ccai_insights_location_id
self.conversation_id = conversation_id
self.creds = self.get_credentials()
self.oauth_token = self.get_oauth_token()
vertexai.init(project=self.project_id, location=self.location_id, credentials=self.creds)
self.model = GenerativeModel(model_name)
self.dataset_name = dataset_name
self.table_name = table_name
self.scorecard_id = scorecard_id
self.ingest_record_bucket_id = ingest_record_bucket_id
self.target_tags = target_tags.split(",")
self.target_values = target_values.split(",")
self.storage_client = storage.Client(project=self.project_id, credentials=self.creds)